Search in sources :

Example 6 with DruidPooledPreparedStatement

use of com.alibaba.druid.pool.DruidPooledPreparedStatement in project druid by alibaba.

the class PSCacheTest3 method test_pscache.

public void test_pscache() throws Exception {
    DruidPooledConnection conn = dataSource.getConnection();
    DruidConnectionHolder holder = conn.getConnectionHolder();
    PreparedStatementPool stmtPool = holder.getStatementPool();
    final String sql_0 = "select 0";
    final String sql_1 = "select 1";
    final String sql_2 = "select 2";
    final String sql_3 = "select 3";
    final String sql_4 = "select 4";
    Assert.assertEquals(0, stmtPool.size());
    PreparedStatementHolder stmtHoler_0;
    PreparedStatementHolder stmtHoler_1;
    PreparedStatementHolder stmtHoler_2;
    PreparedStatementHolder stmtHoler_3;
    PreparedStatementHolder stmtHoler_4;
    // sql_0连续执行两次
    {
        DruidPooledPreparedStatement stmt_0 = (DruidPooledPreparedStatement) conn.prepareStatement(sql_0);
        Assert.assertFalse(stmt_0.getPreparedStatementHolder().isPooling());
        stmt_0.close();
        Assert.assertEquals(1, stmtPool.size());
        Assert.assertTrue(stmt_0.getPreparedStatementHolder().isPooling());
    }
    {
        DruidPooledPreparedStatement stmt_0 = (DruidPooledPreparedStatement) conn.prepareStatement(sql_0);
        Assert.assertEquals(1, stmtPool.size());
        Assert.assertTrue(stmt_0.getPreparedStatementHolder().isInUse());
        Assert.assertTrue(stmt_0.getPreparedStatementHolder().isPooling());
        stmt_0.close();
        Assert.assertFalse(stmt_0.getPreparedStatementHolder().isInUse());
        Assert.assertTrue(stmt_0.getPreparedStatementHolder().isPooling());
        Assert.assertEquals(1, stmtPool.size());
    }
    DruidPooledPreparedStatement stmt_0 = (DruidPooledPreparedStatement) conn.prepareStatement(sql_0);
    stmtHoler_0 = stmt_0.getPreparedStatementHolder();
    Assert.assertTrue(stmtHoler_0.isInUse());
    Assert.assertTrue(stmtHoler_0.isPooling());
    stmt_0.execute();
    {
        DruidPooledPreparedStatement stmt_1 = (DruidPooledPreparedStatement) conn.prepareStatement(sql_1);
        Assert.assertEquals(1, stmtPool.size());
        stmtHoler_1 = stmt_1.getPreparedStatementHolder();
        Assert.assertTrue(stmt_1.getPreparedStatementHolder().isInUse());
        Assert.assertFalse(stmt_1.getPreparedStatementHolder().isPooling());
        stmt_1.close();
        Assert.assertFalse(stmt_1.getPreparedStatementHolder().isInUse());
        Assert.assertTrue(stmt_1.getPreparedStatementHolder().isPooling());
        Assert.assertTrue(stmt_1.getPreparedStatementHolder().isPooling());
        Assert.assertEquals(2, stmtPool.size());
    }
    Assert.assertTrue(stmtHoler_0.isPooling());
    Assert.assertTrue(stmtHoler_1.isPooling());
    {
        DruidPooledPreparedStatement stmt_2 = (DruidPooledPreparedStatement) conn.prepareStatement(sql_2);
        Assert.assertEquals(2, stmtPool.size());
        stmtHoler_2 = stmt_2.getPreparedStatementHolder();
        Assert.assertTrue(stmt_2.getPreparedStatementHolder().isInUse());
        Assert.assertFalse(stmt_2.getPreparedStatementHolder().isPooling());
        stmt_2.close();
        Assert.assertFalse(stmt_2.getPreparedStatementHolder().isInUse());
        Assert.assertTrue(stmt_2.getPreparedStatementHolder().isPooling());
        Assert.assertTrue(stmt_2.getPreparedStatementHolder().isPooling());
        Assert.assertEquals(3, stmtPool.size());
    }
    Assert.assertTrue(stmtHoler_0.isPooling());
    Assert.assertTrue(stmtHoler_1.isPooling());
    Assert.assertTrue(stmtHoler_2.isPooling());
    {
        DruidPooledPreparedStatement stmt_3 = (DruidPooledPreparedStatement) conn.prepareStatement(sql_3);
        Assert.assertEquals(3, stmtPool.size());
        stmtHoler_3 = stmt_3.getPreparedStatementHolder();
        Assert.assertTrue(stmt_3.getPreparedStatementHolder().isInUse());
        Assert.assertFalse(stmt_3.getPreparedStatementHolder().isPooling());
        stmt_3.close();
        Assert.assertFalse(stmt_3.getPreparedStatementHolder().isInUse());
        Assert.assertTrue(stmt_3.getPreparedStatementHolder().isPooling());
        Assert.assertTrue(stmt_3.getPreparedStatementHolder().isPooling());
        Assert.assertEquals(3, stmtPool.size());
    }
    Assert.assertFalse(stmtHoler_0.isPooling());
    Assert.assertTrue(stmtHoler_1.isPooling());
    Assert.assertTrue(stmtHoler_2.isPooling());
    Assert.assertTrue(stmtHoler_3.isPooling());
    {
        DruidPooledPreparedStatement stmt_4 = (DruidPooledPreparedStatement) conn.prepareStatement(sql_4);
        Assert.assertEquals(3, stmtPool.size());
        stmtHoler_4 = stmt_4.getPreparedStatementHolder();
        Assert.assertTrue(stmt_4.getPreparedStatementHolder().isInUse());
        Assert.assertFalse(stmt_4.getPreparedStatementHolder().isPooling());
        stmt_4.close();
        Assert.assertFalse(stmt_4.getPreparedStatementHolder().isInUse());
        Assert.assertTrue(stmt_4.getPreparedStatementHolder().isPooling());
        Assert.assertTrue(stmt_4.getPreparedStatementHolder().isPooling());
        Assert.assertEquals(3, stmtPool.size());
    }
    Assert.assertFalse(stmtHoler_0.isPooling());
    Assert.assertFalse(stmtHoler_1.isPooling());
    Assert.assertTrue(stmtHoler_2.isPooling());
    Assert.assertTrue(stmtHoler_3.isPooling());
    Assert.assertTrue(stmtHoler_4.isPooling());
    stmt_0.close();
    Assert.assertTrue(stmtHoler_0.isPooling());
    Assert.assertFalse(stmtHoler_1.isPooling());
    Assert.assertFalse(stmtHoler_2.isPooling());
    Assert.assertTrue(stmtHoler_3.isPooling());
    Assert.assertTrue(stmtHoler_4.isPooling());
    conn.close();
}
Also used : PreparedStatementHolder(com.alibaba.druid.pool.PreparedStatementHolder) PreparedStatementPool(com.alibaba.druid.pool.PreparedStatementPool) DruidPooledConnection(com.alibaba.druid.pool.DruidPooledConnection) DruidPooledPreparedStatement(com.alibaba.druid.pool.DruidPooledPreparedStatement) DruidConnectionHolder(com.alibaba.druid.pool.DruidConnectionHolder)

Example 7 with DruidPooledPreparedStatement

use of com.alibaba.druid.pool.DruidPooledPreparedStatement in project druid by alibaba.

the class DruidPooledConnectionTest1 method test_closePoolableStatement.

public void test_closePoolableStatement() throws Exception {
    DruidPooledConnection conn = (DruidPooledConnection) dataSource.getConnection();
    DruidPooledPreparedStatement stmt = (DruidPooledPreparedStatement) conn.prepareStatement("select 1");
    conn.close();
    conn.closePoolableStatement(stmt);
}
Also used : DruidPooledConnection(com.alibaba.druid.pool.DruidPooledConnection) DruidPooledPreparedStatement(com.alibaba.druid.pool.DruidPooledPreparedStatement)

Example 8 with DruidPooledPreparedStatement

use of com.alibaba.druid.pool.DruidPooledPreparedStatement in project druid by alibaba.

the class PoolablePreparedStatementTest method setUp.

protected void setUp() throws Exception {
    DruidDataSource dataSource = new DruidDataSource();
    MockConnection mockConn = new MockConnection();
    DruidConnectionHolder connHolder = new DruidConnectionHolder(dataSource, mockConn, 0);
    DruidPooledConnection conn = new DruidPooledConnection(connHolder);
    raw = new MockPreparedStatement(null, null);
    stmt = new DruidPooledPreparedStatement(conn, new PreparedStatementHolder(new PreparedStatementKey("", null, null, 0, 0, 0), raw)) {

        protected SQLException checkException(Throwable error) throws SQLException {
            if (error instanceof SQLException) {
                return (SQLException) error;
            }
            return new SQLException(error);
        }
    };
}
Also used : PreparedStatementHolder(com.alibaba.druid.pool.PreparedStatementHolder) SQLException(java.sql.SQLException) DruidPooledConnection(com.alibaba.druid.pool.DruidPooledConnection) DruidPooledPreparedStatement(com.alibaba.druid.pool.DruidPooledPreparedStatement) DruidConnectionHolder(com.alibaba.druid.pool.DruidConnectionHolder) MockPreparedStatement(com.alibaba.druid.mock.MockPreparedStatement) PreparedStatementKey(com.alibaba.druid.pool.DruidPooledPreparedStatement.PreparedStatementKey) DruidDataSource(com.alibaba.druid.pool.DruidDataSource) MockConnection(com.alibaba.druid.mock.MockConnection)

Aggregations

DruidPooledPreparedStatement (com.alibaba.druid.pool.DruidPooledPreparedStatement)8 DruidPooledConnection (com.alibaba.druid.pool.DruidPooledConnection)4 Connection (java.sql.Connection)4 PreparedStatement (java.sql.PreparedStatement)4 DruidConnectionHolder (com.alibaba.druid.pool.DruidConnectionHolder)3 PreparedStatementHolder (com.alibaba.druid.pool.PreparedStatementHolder)3 PreparedStatementPool (com.alibaba.druid.pool.PreparedStatementPool)2 MockConnection (com.alibaba.druid.mock.MockConnection)1 MockPreparedStatement (com.alibaba.druid.mock.MockPreparedStatement)1 DruidDataSource (com.alibaba.druid.pool.DruidDataSource)1 PreparedStatementKey (com.alibaba.druid.pool.DruidPooledPreparedStatement.PreparedStatementKey)1 DruidPooledResultSet (com.alibaba.druid.pool.DruidPooledResultSet)1 SQLException (java.sql.SQLException)1