bug fixed pscache

2023-03-04 19:17:16 +00:00 · 2012-08-14 11:05:51 +08:00
parent 4941a2f98d
commit 66cb1de037
5 changed files with 245 additions and 4 deletions
@@ -0,0 +1,135 @@
+package com.alibaba.druid.bvt.pool;
+
+import junit.framework.Assert;
+import junit.framework.TestCase;
+
+import com.alibaba.druid.TestForZY.A;
+import com.alibaba.druid.pool.ConnectionHolder;
+import com.alibaba.druid.pool.DruidDataSource;
+import com.alibaba.druid.pool.DruidPooledConnection;
+import com.alibaba.druid.pool.DruidPooledPreparedStatement;
+import com.alibaba.druid.pool.DruidPooledResultSet;
+import com.alibaba.druid.pool.PreparedStatementHolder;
+import com.alibaba.druid.pool.PreparedStatementPool;
+import com.alibaba.druid.util.JdbcUtils;
+
+public class PSCacheTest4 extends TestCase {
+
+    private DruidDataSource dataSource;
+
+    protected void setUp() throws Exception {
+        dataSource = new DruidDataSource();
+        dataSource.setUrl("jdbc:mock:x1");
+        dataSource.setPoolPreparedStatements(true);
+        dataSource.setMaxOpenPreparedStatements(50);
+        dataSource.setFilters("log4j");
+    }
+
+    protected void tearDown() throws Exception {
+        JdbcUtils.close(dataSource);
+    }
+
+    public void test_pscache() throws Exception {
+        DruidPooledConnection conn = dataSource.getConnection();
+
+        ConnectionHolder holder = conn.getConnectionHolder();
+        PreparedStatementPool stmtPool = holder.getStatementPool();
+
+        final String sql_0 = "select 0";
+        final String sql_1 = "select 1";
+
+        Assert.assertEquals(0, stmtPool.size());
+
+        PreparedStatementHolder stmtHoler_0;
+        PreparedStatementHolder stmtHoler_1_A;
+        PreparedStatementHolder stmtHoler_1_B;
+        PreparedStatementHolder stmtHoler_1_C;
+
+        DruidPooledPreparedStatement stmt_0;
+        DruidPooledPreparedStatement stmt_1_A;
+        DruidPooledPreparedStatement stmt_1_B;
+        DruidPooledPreparedStatement stmt_1_C;
+        DruidPooledPreparedStatement stmt_1_D;
+        DruidPooledPreparedStatement stmt_1_E;
+        DruidPooledPreparedStatement stmt_1_F;
+        DruidPooledPreparedStatement stmt_1_G;
+        DruidPooledPreparedStatement stmt_1_H;
+        
+        DruidPooledResultSet rs_0;
+        DruidPooledResultSet rs_1_A;
+        DruidPooledResultSet rs_1_B;
+        DruidPooledResultSet rs_1_C;
+        DruidPooledResultSet rs_1_D;
+        DruidPooledResultSet rs_1_E;
+        DruidPooledResultSet rs_1_F;
+        DruidPooledResultSet rs_1_G;
+        DruidPooledResultSet rs_1_H;
+
+        stmt_0 = (DruidPooledPreparedStatement) conn.prepareStatement(sql_0);
+        rs_0 = (DruidPooledResultSet) stmt_0.executeQuery();
+        
+        Assert.assertTrue(stmt_0.getPreparedStatementHolder().isInUse());
+
+        stmt_1_A = (DruidPooledPreparedStatement) conn.prepareStatement(sql_1);
+        rs_1_A = (DruidPooledResultSet) stmt_1_A.executeQuery();
+        
+        Assert.assertTrue(stmt_0.getPreparedStatementHolder().isInUse());
+        Assert.assertTrue(stmt_1_A.getPreparedStatementHolder().isInUse());
+        
+        stmt_1_B = (DruidPooledPreparedStatement) conn.prepareStatement(sql_1);
+        rs_1_B = (DruidPooledResultSet) stmt_1_B.executeQuery();
+        rs_1_B.close();
+        stmt_1_B.close();
+        
+        Assert.assertTrue(stmt_0.getPreparedStatementHolder().isInUse());
+        Assert.assertTrue(stmt_1_A.getPreparedStatementHolder().isInUse());
+        Assert.assertFalse(stmt_1_B.getPreparedStatementHolder().isInUse());
+        
+        stmt_1_C = (DruidPooledPreparedStatement) conn.prepareStatement(sql_1);
+        rs_1_C = (DruidPooledResultSet) stmt_1_C.executeQuery();
+        rs_1_C.close();
+        stmt_1_C.close();
+        
+        Assert.assertTrue(stmt_0.getPreparedStatementHolder().isInUse());
+        Assert.assertTrue(stmt_1_A.getPreparedStatementHolder().isInUse());
+        Assert.assertFalse(stmt_1_B.getPreparedStatementHolder().isInUse());
+        Assert.assertFalse(stmt_1_C.getPreparedStatementHolder().isInUse());
+        
+        stmt_1_D = (DruidPooledPreparedStatement) conn.prepareStatement(sql_1);
+        rs_1_D = (DruidPooledResultSet) stmt_1_D.executeQuery();
+        rs_1_D.close();
+        stmt_1_D.close();
+        
+        Assert.assertTrue(stmt_0.getPreparedStatementHolder().isInUse());
+        Assert.assertTrue(stmt_1_A.getPreparedStatementHolder().isInUse());
+        Assert.assertFalse(stmt_1_B.getPreparedStatementHolder().isInUse());
+        Assert.assertFalse(stmt_1_C.getPreparedStatementHolder().isInUse());
+        Assert.assertFalse(stmt_1_D.getPreparedStatementHolder().isInUse());
+        
+        stmt_1_E = (DruidPooledPreparedStatement) conn.prepareStatement(sql_1);
+        rs_1_E = (DruidPooledResultSet) stmt_1_E.executeQuery();
+        rs_1_E.close();
+        stmt_1_E.close();
+        
+        rs_1_A.close();
+        stmt_1_A.close();
+        
+        stmt_1_F = (DruidPooledPreparedStatement) conn.prepareStatement(sql_1);
+        rs_1_F = (DruidPooledResultSet) stmt_1_F.executeQuery();
+        rs_1_F.close();
+        stmt_1_F.close();
+        
+        stmt_1_G = (DruidPooledPreparedStatement) conn.prepareStatement(sql_1);
+        rs_1_G = (DruidPooledResultSet) stmt_1_G.executeQuery();
+        
+        stmt_1_H = (DruidPooledPreparedStatement) conn.prepareStatement(sql_1);
+        rs_1_H = (DruidPooledResultSet) stmt_1_H.executeQuery();
+        rs_1_H.close();
+        stmt_1_H.close();
+        
+        rs_1_G.close();
+        stmt_1_G.close();
+        
+        conn.close();
+    }
+}