git-svn-id: http://code.alibabatech.com/svn/druid/trunk@633 b9813039-fb51-4c41-a8b9-e21c2acb5095

2023-03-04 19:17:16 +00:00 · 2011-09-05 02:40:05 +00:00
parent f3707be513
commit 7eb9d95a78
1 changed files with 71 additions and 0 deletions
@@ -0,0 +1,71 @@
+package com.alibaba.druid.bvt.pool;
+
+import java.sql.Connection;
+import java.util.Random;
+import java.util.concurrent.CountDownLatch;
+
+import com.alibaba.druid.mock.MockDriver;
+import com.alibaba.druid.pool.DruidDataSource;
+
+import junit.framework.TestCase;
+
+
+public class TestGraceShutdown extends TestCase {
+    private MockDriver      driver;
+    private DruidDataSource dataSource;
+
+    protected void setUp() throws Exception {
+        driver = new MockDriver();
+
+        dataSource = new DruidDataSource();
+        dataSource.setUrl("jdbc:mock:xxx");
+        dataSource.setDriver(driver);
+        dataSource.setInitialSize(1);
+        dataSource.setMaxActive(2);
+        dataSource.setMaxIdle(2);
+        dataSource.setMinIdle(1);
+        dataSource.setMinEvictableIdleTimeMillis(300 * 1000); // 300 / 10
+        dataSource.setTimeBetweenEvictionRunsMillis(180 * 1000); // 180 / 10
+        dataSource.setTestWhileIdle(true);
+        dataSource.setTestOnBorrow(false);
+        dataSource.setValidationQuery("SELECT 1");
+        dataSource.setFilters("stat");
+    }
+    
+    public void test_close() throws Exception {
+        final int threadCount = 100;
+        Thread[] threads = new Thread[threadCount];
+        final CountDownLatch startLatch = new CountDownLatch(1);
+        final CountDownLatch endLatch = new CountDownLatch(threadCount);
+
+        for (int i = 0; i < threadCount; ++i) {
+            threads[i] = new Thread("thread-" + i) {
+
+                public void run() {
+                    try {
+                        startLatch.await();
+                        Connection conn = dataSource.getConnection();
+                    } catch (Exception e) {
+                        e.printStackTrace();
+                    } finally {
+                        endLatch.countDown();
+                    }
+                }
+            };
+        }
+
+        startLatch.countDown();
+        for (int i = 0; i < threadCount; ++i) {
+            threads[i].start();
+        }
+        Thread.sleep(1000);
+        
+        new Thread("close thread") {
+            public void run() {
+                dataSource.close();
+            }
+        }.start();
+
+        endLatch.await();
+    }
+}