CompareWithPopularPool.java

package org.opensource.clearpool;

import java.sql.DriverManager;
import java.util.concurrent.atomic.AtomicLong;

import org.apache.commons.dbcp.BasicDataSource;
import org.apache.log4j.PropertyConfigurator;
import org.opensource.clearpool.core.ClearPoolDataSource;
import org.opensource.clearpool.logging.PoolLoggerFactory;
import org.opensource.clearpool.util.MemoryUtil;
import org.opensource.clearpool.util.MockTestDriver;
import org.opensource.clearpool.util.ThreadProcessUtil;

import com.alibaba.druid.pool.DruidDataSource;
import com.jolbox.bonecp.BoneCPDataSource;
import com.mchange.v2.c3p0.ComboPooledDataSource;

import junit.framework.TestCase;

public class CompareWithPopularPool extends TestCase {
  private String jdbcUrl;
  private String user;
  private String password;
  private String driverClassName;
  private int corePoolSize = 20;
  private int maxPoolSize = 50;
  private static final int threadCount = 100;
  private int loop = 5;
  private int count = 100000 / threadCount;

  private static AtomicLong physicalCon = MockTestDriver.physicalCon;

  private static final String PATH = "log4j/special_log4j.properties";

  static {
    // threadCount = Runtime.getRuntime().availableProcessors();
    // System.out.println("available processors: " + threadCount);
    // System.out.println();
    ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
    String path = classLoader.getResource(PATH).getPath();
    PropertyConfigurator.configure(path);
  }

  @Override
  public void setUp() throws Exception {
    MemoryUtil.printMemoryInfo();
    System.setProperty(PoolLoggerFactory.LOG_UNABLE, "true");
    DriverManager.registerDriver(new MockTestDriver());
    this.driverClassName = MockTestDriver.CLASS;
    this.jdbcUrl = MockTestDriver.URL;
    this.user = "1";
    this.password = "1";
    physicalCon.set(0);
  }

  public void test_clearpool() throws Exception {
    ClearPoolDataSource dataSource = new ClearPoolDataSource();
    dataSource.setCorePoolSize(this.corePoolSize);
    dataSource.setMaxPoolSize(this.maxPoolSize);
    dataSource.setDriverClassName(this.driverClassName);
    dataSource.setJdbcUrl(this.jdbcUrl);
    dataSource.setJdbcUser(this.user);
    dataSource.setJdbcPassword(this.password);
    for (int i = 0; i < this.loop; ++i) {
      ThreadProcessUtil.process(dataSource, "clearpool", this.count, threadCount, physicalCon);
    }
    System.out.println();
  }

  public void test_druid() throws Exception {
    DruidDataSource dataSource = new DruidDataSource();
    dataSource.setInitialSize(this.corePoolSize);
    dataSource.setMaxActive(this.maxPoolSize);
    dataSource.setMinIdle(this.corePoolSize);
    dataSource.setPoolPreparedStatements(true);
    dataSource.setDriverClassName(this.driverClassName);
    dataSource.setUrl(this.jdbcUrl);
    dataSource.setPoolPreparedStatements(true);
    dataSource.setUsername(this.user);
    dataSource.setPassword(this.password);
    dataSource.setValidationQuery("select 1");
    dataSource.setTestOnBorrow(false);
    for (int i = 0; i < this.loop; ++i) {
      ThreadProcessUtil.process(dataSource, "druid", this.count, threadCount, physicalCon);
    }
    System.out.println();
  }

  public void test_dbcp() throws Exception {
    final BasicDataSource dataSource = new BasicDataSource();
    dataSource.setInitialSize(this.corePoolSize);
    dataSource.setMaxActive(this.maxPoolSize);
    dataSource.setMinIdle(this.corePoolSize);
    dataSource.setMaxIdle(this.maxPoolSize);
    dataSource.setPoolPreparedStatements(true);
    dataSource.setDriverClassName(this.driverClassName);
    dataSource.setUrl(this.jdbcUrl);
    dataSource.setPoolPreparedStatements(true);
    dataSource.setUsername(this.user);
    dataSource.setPassword(this.password);
    dataSource.setValidationQuery("SELECT 1");
    dataSource.setTestOnBorrow(false);
    for (int i = 0; i < this.loop; ++i) {
      ThreadProcessUtil.process(dataSource, "dbcp", this.count, threadCount, physicalCon);
    }
    System.out.println();
  }

  public void test_bonecp() throws Exception {
    BoneCPDataSource dataSource = new BoneCPDataSource();
    dataSource.setMinConnectionsPerPartition(this.corePoolSize);
    dataSource.setMaxConnectionsPerPartition(this.maxPoolSize);
    dataSource.setDriverClass(this.driverClassName);
    dataSource.setJdbcUrl(this.jdbcUrl);
    dataSource.setStatementsCacheSize(100);
    dataSource.setServiceOrder("LIFO");
    dataSource.setUsername(this.user);
    dataSource.setPassword(this.password);
    dataSource.setPartitionCount(1);
    dataSource.setAcquireIncrement(5);
    for (int i = 0; i < this.loop; ++i) {
      ThreadProcessUtil.process(dataSource, "boneCP", this.count, threadCount, physicalCon);
    }
    System.out.println();
  }

  public void test_c3p0() throws Exception {
    ComboPooledDataSource dataSource = new ComboPooledDataSource();
    dataSource.setMinPoolSize(this.corePoolSize);
    dataSource.setMaxPoolSize(this.maxPoolSize);
    dataSource.setDriverClass(this.driverClassName);
    dataSource.setJdbcUrl(this.jdbcUrl);
    dataSource.setUser(this.user);
    dataSource.setPassword(this.password);
    for (int i = 0; i < this.loop; ++i) {
      ThreadProcessUtil.process(dataSource, "c3p0", this.count, threadCount, physicalCon);
    }
    System.out.println();
  }

  public void test_tomcat_jdbc() throws Exception {
    org.apache.tomcat.jdbc.pool.DataSource dataSource =
        new org.apache.tomcat.jdbc.pool.DataSource();
    dataSource.setMaxIdle(this.maxPoolSize);
    dataSource.setMinIdle(this.corePoolSize);
    dataSource.setMaxActive(this.maxPoolSize);
    dataSource.setDriverClassName(this.driverClassName);
    dataSource.setUrl(this.jdbcUrl);
    dataSource.setUsername(this.user);
    dataSource.setPassword(this.password);
    for (int i = 0; i < this.loop; ++i) {
      ThreadProcessUtil.process(dataSource, "tomcat-jdbc", this.count, threadCount, physicalCon);
    }
    System.out.println();
  }
}