BasicCompareInOracle.java

package org.opensource.clearpool;

import java.sql.Connection;
import java.sql.Driver;
import java.sql.SQLException;
import java.util.Properties;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

import javax.sql.DataSource;

import org.apache.log4j.PropertyConfigurator;
import org.opensource.clearpool.core.ClearPoolDataSource;
import org.opensource.clearpool.datasource.JDBCDataSource;
import org.opensource.clearpool.logging.PoolLoggerFactory;
import org.opensource.clearpool.util.JdbcUtil;
import org.opensource.clearpool.util.MemoryUtil;
import org.opensource.clearpool.util.ThreadProcessUtil;

import com.alibaba.druid.pool.DruidDataSource;

import junit.framework.TestCase;

/**
 * Oracle Test.
 *
 * Note: <br />
 * 1.replace driverClass with your database's jdbc-class please; <br />
 * 2.replace jdbcUrl with your database's url please; <br />
 * 3.replace user with your database's user please; <br />
 * 4.replace password with your database's password please; <br />
 * 5.replace sql with your valid sql please.
 *
 * @author xionghui
 * @date 24.09.2014
 * @version 1.0
 */
public class BasicCompareInOracle extends TestCase {
  private String sql = "select count(1) from cm_stuff where dr=0";

  private String driverClassName = "oracle.jdbc.driver.OracleDriver";
  private String jdbcUrl = "jdbc:oracle:thin:@20.10.1.224:1521:ora10g";
  private String user = "cm65dev";
  private String password = "1";

  private int corePoolSize = 5;
  private int maxPoolSize = 10;
  private static final int threadCount = 10;
  private int loop = 5;
  private int count = 100 / threadCount;

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

  static {
    System.out.println("Oracle:");
    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");
  }

  public void test_Nopool() throws Exception {
    System.setProperty("jdbc.drivers", this.driverClassName);
    Driver driver = JdbcUtil.createDriver(this.driverClassName);
    Properties connectProperties = new Properties();
    if (this.user != null) {
      connectProperties.put("user", this.user);
    }
    if (this.password != null) {
      connectProperties.put("password", this.password);
    }
    DataSource dataSource =
        new NopoolDataSource(this.driverClassName, this.jdbcUrl, driver, connectProperties);
    for (int i = 0; i < this.loop; ++i) {
      ThreadProcessUtil.processSql(dataSource, "Nopool", this.count, threadCount, this.sql);
    }
    System.out.println();
  }

  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.processSql(dataSource, "clearpool", this.count, threadCount, this.sql);
    }
    dataSource.destory();
    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.processSql(dataSource, "druid", this.count, threadCount, this.sql);
    }
    dataSource.close();
    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.processSql(dataSource, "tomcat-jdbc", this.count, threadCount, this.sql);
    }
    dataSource.close();
    System.out.println();
  }

  /**
   * Nopool DataSource
   *
   * @author xionghui
   * @date 24.09.2014
   * @version 1.0
   */
  static class NopoolDataSource extends JDBCDataSource {
    private static Lock lock = new ReentrantLock();

    public NopoolDataSource(String clazz, String url, Driver driver, Properties connectProperties) {
      super(clazz, url, driver, connectProperties);
    }

    @Override
    public Connection getConnection() throws SQLException {
      lock.lock();
      try {
        Connection con = super.getConnection();
        return con;
      } finally {
        lock.unlock();
      }
    }
  }
}