Merge branch 'dbpool-logging'

src/test/java/com/avaje/ebeaninternal/server/lib/sql/TestBusyBuffer.java

@@ -44,6 +44,7 @@ public class TestBusyBuffer extends BaseTestCase {
 
   }
 
+  @Test
   public void test_rotate() {
 
     BusyConnectionBuffer b = new BusyConnectionBuffer(2, 2);
@@ -54,13 +55,17 @@ public class TestBusyBuffer extends BaseTestCase {
     PooledConnection p3 = new PooledConnection("3");
 
     Assert.assertEquals(2, b.getCapacity());
+    Assert.assertEquals(0, b.size());
 
     b.add(p0);
     b.add(p1);
+    Assert.assertEquals(2, b.size());
     Assert.assertEquals(2, b.getCapacity());
     b.add(p2);
+    Assert.assertEquals(3, b.size());
     Assert.assertEquals(4, b.getCapacity());
     b.add(p3);
+    Assert.assertEquals(4, b.size());
     Assert.assertEquals(4, b.getCapacity());
 
     Assert.assertEquals(0, p0.getSlotId());
@@ -69,13 +74,20 @@ public class TestBusyBuffer extends BaseTestCase {
     Assert.assertEquals(3, p3.getSlotId());
 
     b.remove(p2);
+    Assert.assertEquals(3, b.size());
     b.remove(p0);
+    Assert.assertEquals(2, b.size());
     b.remove(p3);
+    Assert.assertEquals(1, b.size());
     b.add(p2);
+    Assert.assertEquals(2, b.size());
     Assert.assertEquals(0, p2.getSlotId());
 
     b.remove(p0);
+    Assert.assertEquals(2, b.size());
     b.add(p0);
+    Assert.assertEquals(3, b.size());
+
     // p1 is still in it's slot
     Assert.assertEquals(2, p0.getSlotId());
 

src/test/java/com/avaje/ebeaninternal/server/lib/sql/TestDataSourceMax.java

@@ -1,8 +1,9 @@
 package com.avaje.ebeaninternal.server.lib.sql;
 
 import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
 
-import org.junit.Assert;
 import org.junit.Test;
 
 import com.avaje.ebean.BaseTestCase;
@@ -15,44 +16,50 @@ public class TestDataSourceMax extends BaseTestCase {
   @Test
   public void test() {
 
-    boolean runThisManuallyNow = true;
+    boolean skipThisTest = true;
 
-    if (!runThisManuallyNow) {
+    if (skipThisTest) {
       return;
     }
-
-    String name = "h2";
+    
+    String name = "mysql";
 
     DataSourceConfig dsConfig = new DataSourceConfig();
     dsConfig.loadSettings(name);
-    dsConfig.setMinConnections(3);
-    dsConfig.setMaxConnections(3);
+    dsConfig.setMinConnections(2);
+    dsConfig.setMaxConnections(25);
     dsConfig.setWaitTimeoutMillis(30000);
+    dsConfig.setCaptureStackTrace(true);
 
     DataSourcePool pool = new DataSourcePool(null, name, dsConfig);
 
-    Assert.assertEquals(3, pool.getMaxSize());
-
-    DefaultBackgroundExecutor bg = new DefaultBackgroundExecutor(10, 2, 180, 30, "testDs");
+    
+    //pool.checkDataSource();
+    
+//    if (true) {
+//      pool.shutdown(false);
+//      return;
+//    }
+    
+    DefaultBackgroundExecutor bg = new DefaultBackgroundExecutor(1, 2, 180, 30, "testDs");
 
     try {
       for (int i = 0; i < 12; i++) {
         // Thread.sleep(10*i);
-        bg.execute(new ConnRunner(pool, 100));
+        bg.execute(new ConnRunner(pool, 4000, i));
       }
 
       System.out.println("main thread sleep ... " + pool.getStatus(false));
 
-      Thread.sleep(1000);
+      Thread.sleep(10000);
+      pool.getStatistics(true);
+
+      Thread.sleep(30000);
+
       Status status = pool.getStatus(false);
       System.out.println(status);
 
-      // this dumpOrder was for 3 vectors used in PooledConnectionQueue
-      // that logged the order of wait, notify and obtain events
-      // I have remove that code.
-
-      // String s = pool.dumpOrder();
-      // System.err.println(s);
+      pool.shutdown(false);
 
     } catch (Exception e) {
       e.printStackTrace();
@@ -64,21 +71,65 @@ public class TestDataSourceMax extends BaseTestCase {
 
     final DataSourcePool pool;
     final long sleepMillis;
+    final int position;
 
-    ConnRunner(DataSourcePool pool, long sleepMillis) {
+    ConnRunner(DataSourcePool pool, long sleepMillis, int position) {
       this.pool = pool;
       this.sleepMillis = sleepMillis;
+      this.position = position;
     }
 
-    public void run() {
+    private void waitSomeTime(long count) {
       try {
-        Connection connection = pool.getConnection();
-        Thread.sleep(sleepMillis);
-        connection.close();
-      } catch (Exception e) {
-        e.printStackTrace();
+      System.out.println(position+" sleep " + sleepMillis+" count:"+count);
+      Thread.sleep(sleepMillis);
+      System.out.println(position+" sleep done");
+      } catch (InterruptedException e){
+        throw new RuntimeException(e);
       }
     }
+    
+    public void run() {
+      Connection connection = null;
+      PreparedStatement pstmt = null;
+      ResultSet rset = null;
+      long count = -1;
+      try {
+        connection = pool.getConnection();
+        pstmt = connection.prepareStatement("select count(*) from o_customer");
+        rset = pstmt.executeQuery();
+        
+        while (rset.next()) {
+          // do nothing actually
+          count = rset.getLong(1);
+        }
 
+      } catch (Exception ex) {
+        ex.printStackTrace();
+      } finally {
+        if (rset != null) {
+          try {
+            rset.close();
+          } catch (Exception e) {
+            e.printStackTrace();
+          }
+          if (pstmt != null) {
+            try {
+              pstmt.close();
+            } catch (Exception e) {
+              e.printStackTrace();
+            }
+          }
+          if (connection != null) {
+            try {
+              connection.close();
+            } catch (Exception e) {
+              e.printStackTrace();
+            }
+          }
+        }
+        waitSomeTime(count);
+      }
+    }
   }
 }

src/test/java/com/avaje/ebeaninternal/server/lib/sql/TestDataSourceMaxWithEntity.java

@@ -0,0 +1,70 @@
+package com.avaje.ebeaninternal.server.lib.sql;
+
+import org.junit.Test;
+
+import com.avaje.ebean.BaseTestCase;
+import com.avaje.ebean.Ebean;
+import com.avaje.ebean.EbeanServer;
+import com.avaje.ebeaninternal.server.core.DefaultBackgroundExecutor;
+import com.avaje.tests.model.basic.Customer;
+
+public class TestDataSourceMaxWithEntity extends BaseTestCase {
+
+  @Test
+  public void test() {
+
+    boolean skipThisTest = true;
+
+    if (skipThisTest) {
+      return;
+    }
+
+    EbeanServer server = Ebean.getServer(null);
+    
+
+    DefaultBackgroundExecutor bg = new DefaultBackgroundExecutor(1, 2, 180, 30, "testDs");
+
+    try {
+      for (int i = 0; i < 12; i++) {
+        // Thread.sleep(10*i);
+        bg.execute(new ConnRunner(server, 4000, i));
+      }
+
+      System.out.println("main thread sleep ... ");
+
+   
+      Thread.sleep(30000);
+      
+      server.shutdown(true, false);
+
+    } catch (Exception e) {
+      e.printStackTrace();
+    }
+
+  }
+
+  private static class ConnRunner implements Runnable {
+
+    final EbeanServer server;
+    final long sleepMillis;
+    final int position;
+
+    ConnRunner(EbeanServer server, long sleepMillis, int position) {
+      this.server = server;
+      this.sleepMillis = sleepMillis;
+      this.position = position;
+    }
+
+    public void run() {
+      
+      server.find(Customer.class).findRowCount();
+      try {
+        System.out.println(position+" sleep " + sleepMillis);
+        Thread.sleep(sleepMillis);
+        System.out.println(position+" sleep done");
+      } catch (InterruptedException e) {
+        e.printStackTrace();
+      }
+    }
+  }
+}

src/test/java/com/avaje/ebeaninternal/server/lib/sql/TestFreeBuffer.java

@@ -10,14 +10,13 @@ public class TestFreeBuffer extends BaseTestCase {
   @Test
   public void test() {
 
-    FreeConnectionBuffer b = new FreeConnectionBuffer(3);
+    FreeConnectionBuffer b = new FreeConnectionBuffer();
 
     PooledConnection p0 = new PooledConnection("0");
     PooledConnection p1 = new PooledConnection("1");
     PooledConnection p2 = new PooledConnection("2");
     // PooledConnection p3 = new PooledConnection("3");
 
-    Assert.assertEquals(3, b.getCapacity());
     Assert.assertEquals(0, b.size());
     Assert.assertEquals(true, b.isEmpty());
 

src/test/java/com/avaje/ebeaninternal/server/lib/sql/TestFreeBufferTrim.java

@@ -0,0 +1,40 @@
+package com.avaje.ebeaninternal.server.lib.sql;
+
+import org.junit.Assert;
+import org.junit.Test;
+import org.mockito.Mockito;
+
+import com.avaje.ebean.BaseTestCase;
+
+public class TestFreeBufferTrim extends BaseTestCase {
+  
+  @Test
+  public void testWithTime() {
+
+    FreeConnectionBuffer b = new FreeConnectionBuffer();
+    Assert.assertEquals(0, b.size());
+    
+    PooledConnection p0 = Mockito.mock(PooledConnection.class);
+    Mockito.when(p0.shouldTrim(1500, 0)).thenReturn(true);
+    
+    PooledConnection p1 = Mockito.mock(PooledConnection.class);
+    Mockito.when(p1.shouldTrim(1500, 0)).thenReturn(true);
+    
+    PooledConnection p2 = Mockito.mock(PooledConnection.class);
+    Mockito.when(p2.shouldTrim(1500, 0)).thenReturn(false);
+    
+    b.add(p0);
+    b.add(p1);
+    b.add(p2);
+
+    Assert.assertEquals(3, b.size());
+
+    int trimCount = b.trim(1500, 0);
+
+    Assert.assertEquals(1, b.size());
+    Assert.assertEquals(2, trimCount);
+  }
+  
+  
+  
+}