org.cloudgraph.rocksdb.connect.RocksDBConnectionManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cloudgraph-rocksdb Show documentation
CloudGraph(tm) is a suite of Service Data Object (SDO) 2.1 services designed for relational and big-table style "cloud" databases, such as HBase and others.
The newest version!
/**
 * Copyright 2017 TerraMeta Software, Inc.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *     http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.cloudgraph.rocksdb.connect;

import java.io.IOException;
import java.util.Properties;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.commons.pool2.impl.GenericKeyedObjectPool;
import org.apache.commons.pool2.impl.GenericObjectPool;
import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
import org.cloudgraph.core.Connection;
import org.cloudgraph.core.ConnectionManager;
import org.cloudgraph.core.ServiceContext;
import org.cloudgraph.core.client.TableName;
import org.cloudgraph.rocksdb.service.CloudGraphContext;
import org.cloudgraph.rocksdb.service.RocksDBServiceContext;
import org.cloudgraph.store.mapping.StoreMapping;
import org.cloudgraph.store.mapping.StoreMappingContext;
import org.cloudgraph.store.mapping.TableMapping;
import org.cloudgraph.store.service.GraphServiceException;

/**
 * Manages HBase table pool and table interface access.
 * 
 * @see CloudGraphContext
 * @see TableMapping
 * @author Scott Cinnamond
 * @since 0.5
 */
public class RocksDBConnectionManager implements ConnectionManager {

  /**
   * Synonym for min idle. The property name for the {@code minIdle}
   * configuration attribute for connection pools.
   */
  public static final String CONNECTION_POOL_MIN_SIZE = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolMinSize";

  /**
   * Synonym for max total. The property name for the {@code maxTotal}
   * configuration attribute for connection pools.
   */
  public static final String CONNECTION_POOL_MAX_SIZE = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolMaxSize";

  /**
   * The property name for the {@code maxTotal} configuration attribute for
   * connection pools.
   *
   *
   * @see GenericObjectPool#setMaxTotal(int)
   */
  public static final String CONNECTION_POOL_MAX_TOTAL = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolMaxTotal";

  /**
   * The property name for the {@code maxIdle} configuration attribute for
   * connection pools.
   *
   * @see GenericObjectPool#setMaxIdle(int)
   */
  public static final String CONNECTION_POOL_MAX_IDLE = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolMaxIdle";

  /**
   * The property name for the {@code minIdle} configuration attribute for
   * connection pools.
   *
   *
   * @see GenericObjectPool#setMinIdle(int)
   */
  public static final String CONNECTION_POOL_MIN_IDLE = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolMinIdle";

  /**
   * The property name for the {@code lifo} configuration attribute for pools
   * created with this configuration instance.
   *
   * @see GenericObjectPool#getLifo()
   * @see GenericKeyedObjectPool#getLifo()
   */
  public static final String CONNECTION_POOL_LIFO = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolLifo";

  /**
   * The property name for the {@code maxWait} configuration attribute for pools
   * created with this configuration instance.
   *
   * @see GenericObjectPool#getMaxWaitMillis()
   * @see GenericKeyedObjectPool#getMaxWaitMillis()
   */
  public static final String CONNECTION_POOL_MAX_WAIT_MILLIS = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolMaxWaitMillis";

  /**
   * The property name for the {@code minEvictableIdleTimeMillis} configuration
   * attribute for connection pools.
   *
   * @see GenericObjectPool#getMinEvictableIdleTimeMillis()
   * @see GenericKeyedObjectPool#getMinEvictableIdleTimeMillis()
   */
  public static final String CONNECTION_POOL_MIN_EVICTABLE_IDLE_TIME_MILLIS = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolMinEvictableIdleTimeMillis";

  /**
   * The property name for the {@code softMinEvictableIdleTimeMillis}
   * configuration attribute for pools created with this configuration instance.
   *
   * @see GenericObjectPool#getSoftMinEvictableIdleTimeMillis()
   * @see GenericKeyedObjectPool#getSoftMinEvictableIdleTimeMillis()
   */
  public static final String CONNECTION_POOL_SOFT_MIN_EVICTABLE_IDLE_TIME_MILLIS = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolSoftMinEvictableIdleTimeMillis";

  /**
   * The property name for the {@code numTestsPerEvictionRun} configuration
   * attribute for connection pools.
   *
   *
   * @see GenericObjectPool#getNumTestsPerEvictionRun()
   * @see GenericKeyedObjectPool#getNumTestsPerEvictionRun()
   */
  public static final String CONNECTION_POOL_NUM_TESTS_PER_EVICTION_RUN = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolNumTestsPerEvictionRun";

  /**
   * The property name for the {@code testOnCreate} configuration attribute for
   * connection pools.
   *
   *
   * @see GenericObjectPool#getTestOnCreate()
   * @see GenericKeyedObjectPool#getTestOnCreate()
   */
  public static final String CONNECTION_POOL_TEST_ON_CREATE = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolTestOnCreate";

  /**
   * The property name for the {@code testOnBorrow} configuration attribute for
   * connection pools.
   *
   * @see GenericObjectPool#getTestOnBorrow()
   * @see GenericKeyedObjectPool#getTestOnBorrow()
   */
  public static final String CONNECTION_POOL_TEST_ON_BORROW = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolTestOnBorrow";

  /**
   * The property name for the {@code testOnReturn} configuration attribute for
   * connection pools.
   *
   * @see GenericObjectPool#getTestOnReturn()
   * @see GenericKeyedObjectPool#getTestOnReturn()
   */
  public static final String CONNECTION_POOL_TEST_ON_RETURN = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolTestOnReturn";

  /**
   * The property name for the {@code testWhileIdle} configuration attribute for
   * connection pools.
   */
  public static final String CONNECTION_POOL_TEST_WHILE_IDLE = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolTestWhileIdle";

  /**
   * The property name for the {@code timeBetweenEvictionRunsMillis}
   * configuration attribute for connection pools.
   */
  public static final String CONNECTION_POOL_TIME_BETWEEN_EVICTION_RUNS_MILLIS = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolTimeBetweenEvictionRunsMillis";

  /**
   * The property name for the {@code evictionPolicyClassName} configuration
   * attribute for connection pools.
   */
  public static final String CONNECTION_POOL_EVICTION_POLICY_CLASS_NAME = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolEvictionPolicyClassName";

  /**
   * The property name for the {@code blockWhenExhausted} configuration
   * attribute for connection pools.
   */
  public static final String CONNECTION_POOL_BLOCK_WHEN_EXHAUSTED = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolBlockWhenExhausted";

  /**
   * The property name of the flag that determines if JMX will be enabled for
   * connection pools.
   *
   */
  public static final String CONNECTION_POOL_JMX_ENABLED = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolJmxEnabled";

  /**
   * The property name of the JMX name base that will be used as part of the
   * name assigned to JMX enabled pools created with this configuration
   * instance. A value of null means that the pool will define the
   * JMX name base.
   */
  public static final String CONNECTION_POOL_JMX_NAME_BASE = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolJmxNameBase";

  /**
   * The property name of the JMX name prefix that will be used as part of the
   * name assigned to JMX enabled pools created with this configuration
   * instance.
   */
  public static final String CONNECTION_POOL_JMX_NAME_PREFIX = "org.plasma.sdo.access.provider.rocksdb.ConnectionPoolJmxNamePrefix";

  private GenericObjectPool pool;

  private static final Log log = LogFactory.getLog(RocksDBConnectionManager.class);

  private static volatile RocksDBConnectionManager instance;

  // private Configuration config;

  private RocksDBConnectionManager() {
    // this.config = CloudGraphContext.instance().getConfig();

    GenericObjectPoolConfig poolConfig = new GenericObjectPoolConfig();

    // poolConfig.setMaxTotal(this.config.getInt(CONNECTION_POOL_MAX_TOTAL,
    // GenericObjectPoolConfig.DEFAULT_MAX_TOTAL));
    // if (this.config.get(CONNECTION_POOL_MAX_SIZE) != null)
    // poolConfig.setMaxTotal(this.config.getInt(CONNECTION_POOL_MAX_SIZE,
    // GenericObjectPoolConfig.DEFAULT_MAX_TOTAL));
    //
    // poolConfig.setMaxIdle(this.config.getInt(CONNECTION_POOL_MAX_IDLE,
    // GenericObjectPoolConfig.DEFAULT_MAX_IDLE));
    // poolConfig.setMinIdle(this.config.getInt(CONNECTION_POOL_MIN_IDLE,
    // GenericObjectPoolConfig.DEFAULT_MIN_IDLE));
    // if (this.config.get(CONNECTION_POOL_MIN_SIZE) != null)
    // poolConfig.setMinIdle(this.config.getInt(CONNECTION_POOL_MIN_SIZE,
    // GenericObjectPoolConfig.DEFAULT_MIN_IDLE));
    //
    // poolConfig.setLifo(this.config.getBoolean(CONNECTION_POOL_LIFO,
    // GenericObjectPoolConfig.DEFAULT_LIFO));
    //
    // poolConfig.setMaxWaitMillis(this.config.getLong(CONNECTION_POOL_MAX_WAIT_MILLIS,
    // GenericObjectPoolConfig.DEFAULT_MAX_WAIT_MILLIS));

    // eviction
    // poolConfig.setTimeBetweenEvictionRunsMillis(this.config.getLong(
    // CONNECTION_POOL_TIME_BETWEEN_EVICTION_RUNS_MILLIS,
    // GenericObjectPoolConfig.DEFAULT_TIME_BETWEEN_EVICTION_RUNS_MILLIS));
    // poolConfig.setEvictionPolicyClassName(this.config.get(
    // CONNECTION_POOL_EVICTION_POLICY_CLASS_NAME,
    // GenericObjectPoolConfig.DEFAULT_EVICTION_POLICY_CLASS_NAME));
    // poolConfig.setMinEvictableIdleTimeMillis(this.config.getLong(
    // CONNECTION_POOL_MIN_EVICTABLE_IDLE_TIME_MILLIS,
    // GenericObjectPoolConfig.DEFAULT_MIN_EVICTABLE_IDLE_TIME_MILLIS));
    // poolConfig.setSoftMinEvictableIdleTimeMillis(this.config.getLong(
    // CONNECTION_POOL_SOFT_MIN_EVICTABLE_IDLE_TIME_MILLIS,
    // GenericObjectPoolConfig.DEFAULT_SOFT_MIN_EVICTABLE_IDLE_TIME_MILLIS));
    // poolConfig.setNumTestsPerEvictionRun(this.config.getInt(
    // CONNECTION_POOL_NUM_TESTS_PER_EVICTION_RUN,
    // GenericObjectPoolConfig.DEFAULT_NUM_TESTS_PER_EVICTION_RUN));

    // poolConfig.setTestOnCreate(this.config.getBoolean(CONNECTION_POOL_TEST_ON_CREATE,
    // GenericObjectPoolConfig.DEFAULT_TEST_ON_CREATE));
    // poolConfig.setTestOnBorrow(this.config.getBoolean(CONNECTION_POOL_TEST_ON_BORROW,
    // GenericObjectPoolConfig.DEFAULT_TEST_ON_BORROW));
    // poolConfig.setTestOnReturn(this.config.getBoolean(CONNECTION_POOL_TEST_ON_RETURN,
    // GenericObjectPoolConfig.DEFAULT_TEST_ON_RETURN));
    // poolConfig.setTestWhileIdle(this.config.getBoolean(CONNECTION_POOL_TEST_WHILE_IDLE,
    // GenericObjectPoolConfig.DEFAULT_TEST_WHILE_IDLE));
    // poolConfig.setBlockWhenExhausted(this.config.getBoolean(CONNECTION_POOL_BLOCK_WHEN_EXHAUSTED,
    // GenericObjectPoolConfig.DEFAULT_BLOCK_WHEN_EXHAUSTED));
    // poolConfig.setJmxEnabled(this.config.getBoolean(CONNECTION_POOL_JMX_ENABLED,
    // false));
    // poolConfig.setJmxNameBase(this.config.get(CONNECTION_POOL_JMX_NAME_BASE,
    // GenericObjectPoolConfig.DEFAULT_JMX_NAME_BASE));
    // poolConfig.setJmxNamePrefix(this.config.get(CONNECTION_POOL_JMX_NAME_PREFIX,
    // GenericObjectPoolConfig.DEFAULT_JMX_NAME_PREFIX));

    // FIXME: hack this is service specific
    Properties properties = new Properties();
    // StoreMappingContext mappingContext = new StoreMappingContext(properties);
    // ServiceContext serviceCOntext = new RocksDBServiceContext();

    PooledConnectionFactory factory = new PooledConnectionFactory();
    this.pool = new GenericObjectPool(factory, poolConfig);
    factory.setPool(pool);

    log.info("created connection pool[ " + "\n\tMaxTotal:\t\t" + poolConfig.getMaxTotal()
        + "\n\tMinIdle:\t\t" + poolConfig.getMinIdle() + "\n\tMaxIdle:\t\t"
        + poolConfig.getMaxIdle() + "\n\tLifo:\t\t" + poolConfig.getLifo()
        + "\n\tMaxWaitMillis:\t\t" + poolConfig.getMaxWaitMillis()
        + "\n\tTimeBetweenEvictionRunsMillis:\t\t" + poolConfig.getTimeBetweenEvictionRunsMillis()
        + "\n\tEvictionPolicyClassName:\t\t" + poolConfig.getEvictionPolicyClassName()
        + "\n\tMinEvictableIdleTimeMillis:\t\t" + poolConfig.getMinEvictableIdleTimeMillis()
        + "\n\tSoftMinEvictableIdleTimeMillis:\t\t"
        + poolConfig.getSoftMinEvictableIdleTimeMillis() + "\n\tNumTestsPerEvictionRun:\t\t"
        + poolConfig.getNumTestsPerEvictionRun() + "\n...]");
  }

  public static RocksDBConnectionManager instance() {
    if (instance == null)
      initInstance();
    return instance;
  }

  private static synchronized void initInstance() {
    if (instance == null)
      instance = new RocksDBConnectionManager();
  }

  protected void finalize() {
    log.debug("Finalizing ConnectionManager");
    try {
      super.finalize();
    } catch (Throwable ex) {
      log.error("ConnectionManager finalize failed to disconnect: ", ex);
    }
  }

  public Connection getConnection() {
    try {
      return this.pool.borrowObject();
    } catch (Exception e) {
      throw new GraphServiceException(e);
    } finally {
    }
  }

  public void createTable(Connection connection, TableName name, ServiceContext serviceContext) {
    log.warn("create table not supported");
  }

  public void deleteTable(Connection connection, TableName name) {
    log.warn("delete table not supported");
  }

}