Java ConnectionPool类代码示例

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
@Override
public void disconnected(ConnectionPool parent, PooledConnection con, boolean finalizing) {
    @SuppressWarnings("unchecked")
    ConcurrentHashMap<CacheKey,CachedStatement> statements =
        (ConcurrentHashMap<CacheKey,CachedStatement>)con.getAttributes().get(STATEMENT_CACHE_ATTR);

    if (statements!=null) {
        for (Map.Entry<CacheKey, CachedStatement> p : statements.entrySet()) {
            closeStatement(p.getValue());
        }
        statements.clear();
    }

    super.disconnected(parent, con, finalizing);
}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
/**
 * {@inheritDoc}
 */
@Override
public void poolStarted(ConnectionPool pool) {
    super.poolStarted(pool);
    //see if we already created a map for this pool
    queries = SlowQueryReport.perPoolStats.get(pool.getName());
    if (queries==null) {
        //create the map to hold our stats
        //however TODO we need to improve the eviction
        //selection
        queries = new ConcurrentHashMap<String,QueryStats>();
        if (perPoolStats.putIfAbsent(pool.getName(), queries)!=null) {
            //there already was one
            queries = SlowQueryReport.perPoolStats.get(pool.getName());
        }
    }
}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
@Test
public void testFastSql() throws Exception {
    int count = 3;
    this.init();
    this.datasource.setMaxActive(1);
    this.datasource.setJdbcInterceptors(SlowQueryReport.class.getName());
    Connection con = this.datasource.getConnection();
    String fastSql = this.datasource.getValidationQuery();
    for (int i=0; i<count; i++) {
        Statement st = con.createStatement();
        ResultSet rs = st.executeQuery(fastSql);
        rs.close();
        st.close();
    }
    Map<String,SlowQueryReport.QueryStats> map = SlowQueryReport.getPoolStats(datasource.getPool().getName());
    Assert.assertNotNull(map);
    Assert.assertEquals(0,map.size());
    ConnectionPool pool = datasource.getPool();
    con.close();
    tearDown();
    Assert.assertNull(SlowQueryReport.getPoolStats(pool.getName()));
}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
@Test
public void testPoolCleaner() throws Exception {
    datasource.getPoolProperties().setTimeBetweenEvictionRunsMillis(2000);
    datasource.getPoolProperties().setTestWhileIdle(true);
    Assert.assertEquals("Pool cleaner should not be started yet.",0,ConnectionPool.getPoolCleaners().size() );
    Assert.assertNull("Pool timer should be null", ConnectionPool.getPoolTimer());
    Assert.assertEquals("Pool cleaner threads should not be present.",0, countPoolCleanerThreads());

    datasource.getConnection().close();
    Assert.assertEquals("Pool cleaner should have 1 cleaner.",1,ConnectionPool.getPoolCleaners().size() );
    Assert.assertNotNull("Pool timer should not be null", ConnectionPool.getPoolTimer());
    Assert.assertEquals("Pool cleaner threads should be 1.",1, countPoolCleanerThreads());

    datasource.close();
    Assert.assertEquals("Pool shutdown, no cleaners should be present.",0,ConnectionPool.getPoolCleaners().size() );
    Assert.assertNull("Pool timer should be null after shutdown", ConnectionPool.getPoolTimer());
    Assert.assertEquals("Pool cleaner threads should not be present after close.",0, countPoolCleanerThreads());


}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
@Test
public void test2PoolCleaners() throws Exception {
    datasource.getPoolProperties().setTimeBetweenEvictionRunsMillis(2000);
    datasource.getPoolProperties().setTestWhileIdle(true);

    DataSource ds2 = new DataSource(datasource.getPoolProperties());

    Assert.assertEquals("Pool cleaner should not be started yet.",0,ConnectionPool.getPoolCleaners().size() );
    Assert.assertNull("Pool timer should be null", ConnectionPool.getPoolTimer());
    Assert.assertEquals("Pool cleaner threads should not be present.",0, countPoolCleanerThreads());

    datasource.getConnection().close();
    ds2.getConnection().close();
    Assert.assertEquals("Pool cleaner should have 2 cleaner.",2,ConnectionPool.getPoolCleaners().size() );
    Assert.assertNotNull("Pool timer should not be null", ConnectionPool.getPoolTimer());
    Assert.assertEquals("Pool cleaner threads should be 1.",1, countPoolCleanerThreads());

    datasource.close();
    Assert.assertEquals("Pool cleaner should have 1 cleaner.",1,ConnectionPool.getPoolCleaners().size() );
    Assert.assertNotNull("Pool timer should not be null", ConnectionPool.getPoolTimer());

    ds2.close();
    Assert.assertEquals("Pool shutdown, no cleaners should be present.",0,ConnectionPool.getPoolCleaners().size() );
    Assert.assertNull("Pool timer should be null after shutdown", ConnectionPool.getPoolTimer());
    Assert.assertEquals("Pool cleaner threads should not be present after close.",0, countPoolCleanerThreads());
}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
@Before
public void setUp() throws Exception {
    this.datasource.setDriverClassName(Driver.class.getName());
    this.datasource.setUrl("jdbc:tomcat:test");
    this.datasource.setPassword(password);
    this.datasource.setUsername(username);
    this.datasource.getConnection().close();
    MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();
    String domain = "tomcat.jdbc";
    Hashtable<String,String> properties = new Hashtable<String,String>();
    properties.put("type", "ConnectionPool");
    properties.put("class", this.getClass().getName());
    oname = new ObjectName(domain,properties);
    ConnectionPool pool = datasource.createPool();
    org.apache.tomcat.jdbc.pool.jmx.ConnectionPool jmxPool = new org.apache.tomcat.jdbc.pool.jmx.ConnectionPool(pool);
    mbs.registerMBean(jmxPool, oname);

}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
/** {@inheritDoc} */
@Override
public synchronized ConnectionPool createPool() throws SQLException {
    if (pool == null) { // is first pool initialization
        try {
            // super.start() will call createPool() method, so we must initialize pool first
            // to ensure single initialization and not run into recursive loop.
            ConnectionPool connectionPool = super.createPool();
            if (connectionPool == null) {
                // sanity check. must never happen, but taken into account all variety of jdbc providers...
                return null;
            }
            super.start();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }
    return super.createPool();
}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
@Override
public void reset(ConnectionPool parent, PooledConnection con)
{
    trackedConnPool = parent;
    trackedPooledCon = con;

    if (parent != null && con != null)
    {
        ConnPair cp = new ConnPair();
        cp.conn = con;
        cp.thread = Thread.currentThread();
        Set<ConnPair> pairs = conns.get(parent);
        if (pairs != null)
        {
            conns.get(parent).add(cp);
        }
    }
}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
/**
 * Called by the engine to trigger the cleanup at the end of a payload thread.
 */
public static int forceCleanup(Thread t)
{
    int i = 0;
    for (Map.Entry<ConnectionPool, Set<ConnPair>> e : conns.entrySet())
    {
        for (ConnPair c : e.getValue())
        {
            if (c.thread.equals(t))
            {
                try
                {
                    // This will in turn remove it from the static Map.
                    c.conn.getHandler().invoke(c.conn, Connection.class.getMethod("close"), null);
                }
                catch (Throwable e1)
                {
                    e1.printStackTrace();
                }
                i++;
            }
        }
    }
    return i;
}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
/**
 * {@inheritDoc}
 */
@Override
public void poolStarted(ConnectionPool pool) {
    super.poolStarted(pool);
    //see if we already created a map for this pool
    queries = SlowQueryReport.perPoolStats.get(pool.getName());
    if (queries==null) {
        //create the map to hold our stats
        //however TODO we need to improve the eviction
        //selection
        queries = new ConcurrentHashMap<String,QueryStats>() {
            
        };
        if (perPoolStats.putIfAbsent(pool.getName(), queries)!=null) {
            //there already was one
            queries = SlowQueryReport.perPoolStats.get(pool.getName());
        }
    }
}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
public void testFastSql() throws Exception {
    int count = 3;
    this.init();
    this.datasource.setMaxActive(1);
    this.datasource.setJdbcInterceptors(SlowQueryReport.class.getName());
    Connection con = this.datasource.getConnection();
    String fastSql = this.datasource.getValidationQuery();
    for (int i=0; i<count; i++) {
        Statement st = con.createStatement();
        ResultSet rs = st.executeQuery(fastSql);
        rs.close();
        st.close();
    }
    Map<String,SlowQueryReport.QueryStats> map = SlowQueryReport.getPoolStats(datasource.getPool().getName());
    assertNotNull(map);
    assertEquals(0,map.size());
    ConnectionPool pool = datasource.getPool();
    con.close();
    tearDown();
    assertNull(SlowQueryReport.getPoolStats(pool.getName()));
}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
@Override
public void setUp() throws Exception {
    super.setUp();
    this.datasource.setDriverClassName(Driver.class.getName());
    this.datasource.setUrl("jdbc:tomcat:test");
    this.datasource.setPassword(password);
    this.datasource.setUsername(username);
    this.datasource.getConnection().close();
    MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();
    String domain = "tomcat.jdbc";
    Hashtable<String,String> properties = new Hashtable<String,String>();
    properties.put("type", "ConnectionPool");
    properties.put("class", this.getClass().getName());
    oname = new ObjectName(domain,properties);
    ConnectionPool pool = datasource.createPool();
    org.apache.tomcat.jdbc.pool.jmx.ConnectionPool jmxPool = new org.apache.tomcat.jdbc.pool.jmx.ConnectionPool(pool);
    mbs.registerMBean(jmxPool, oname);
    
}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
@Override
public void reset(ConnectionPool parent, PooledConnection con) {
    super.reset(parent, con);
    if (parent!=null) {
        poolName = parent.getName();
        pool = parent;
        registerJmx();
    }
}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
@Override
public void reset(ConnectionPool parent, PooledConnection con) {
    super.reset(parent, con);
    if (parent==null) {
        cacheSize = null;
        this.pcon = null;
    } else {
        cacheSize = cacheSizeMap.get(parent);
        this.pcon = con;
        if (!pcon.getAttributes().containsKey(STATEMENT_CACHE_ATTR)) {
            ConcurrentHashMap<CacheKey,CachedStatement> cache = new ConcurrentHashMap<CacheKey, CachedStatement>();
            pcon.getAttributes().put(STATEMENT_CACHE_ATTR,cache);
        }
    }
}
 

import org.apache.tomcat.jdbc.pool.ConnectionPool; //导入依赖的package包/类
@Override
public void reset(ConnectionPool parent, PooledConnection con) {
    super.reset(parent, con);
    if (parent!=null)
        queries = SlowQueryReport.perPoolStats.get(parent.getName());
    else
        queries = null;
}