HiveBasicDataSourceProvider.java

/**
 *
 */
package org.hivedb.meta.persistence;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.hivedb.HiveRuntimeException;
import org.hivedb.meta.Node;
import org.springframework.jdbc.datasource.LazyConnectionDataSourceProxy;

import javax.sql.DataSource;
import java.util.Collection;
import java.util.HashSet;

public class HiveBasicDataSourceProvider implements HiveDataSourceProvider {
  private Log log = LogFactory.getLog(HiveBasicDataSource.class);

  private Collection<HiveBasicDataSource> dataSourcesToClose;

  public HiveBasicDataSourceProvider() {
    dataSourcesToClose = new HashSet<HiveBasicDataSource>();
  }

  public DataSource getDataSource(Node node) {
    return getDataSource(node.getUri());
  }

  /**
   * @see HiveDataSourceProvider#close()
   */
  public void close() {
    HiveRuntimeException exceptionWhileClosing = null;
    for (HiveBasicDataSource dataSource : dataSourcesToClose) {
      try {
        dataSource.close();
      } catch (Exception e) {
        exceptionWhileClosing = new HiveRuntimeException("Error closing datasources. Possibly more than one cause.", e);
      }
    }
    if (exceptionWhileClosing != null) {
      throw exceptionWhileClosing;
    }
  }

  public DataSource getDataSource(String uri) {
    HiveBasicDataSource ds = new HiveBasicDataSource(uri);
    LazyConnectionDataSourceProxy dataSourceProxy = new LazyConnectionDataSourceProxy(ds);
    dataSourcesToClose.add(ds);
    return dataSourceProxy;
  }
}