summaryrefslogtreecommitdiff
path: root/src/test/java/com/zaxxer/hikari/pool
diff options
context:
space:
mode:
Diffstat (limited to 'src/test/java/com/zaxxer/hikari/pool')
-rwxr-xr-xsrc/test/java/com/zaxxer/hikari/pool/ConnectionPoolSizeVsThreadsTest.java4
-rwxr-xr-xsrc/test/java/com/zaxxer/hikari/pool/ConnectionRaceConditionTest.java2
-rw-r--r--src/test/java/com/zaxxer/hikari/pool/HouseKeeperCleanupTest.java2
-rw-r--r--src/test/java/com/zaxxer/hikari/pool/MetricsTrackerTest.java86
-rw-r--r--src/test/java/com/zaxxer/hikari/pool/MiscTest.java7
-rw-r--r--src/test/java/com/zaxxer/hikari/pool/SaturatedPoolTest830.java156
-rw-r--r--src/test/java/com/zaxxer/hikari/pool/TestConnectionTimeoutRetry.java5
-rw-r--r--src/test/java/com/zaxxer/hikari/pool/TestElf.java23
-rw-r--r--src/test/java/com/zaxxer/hikari/pool/TestJNDI.java27
-rw-r--r--src/test/java/com/zaxxer/hikari/pool/TestMBean.java93
-rw-r--r--src/test/java/com/zaxxer/hikari/pool/TestMetrics.java46
11 files changed, 388 insertions, 63 deletions
diff --git a/src/test/java/com/zaxxer/hikari/pool/ConnectionPoolSizeVsThreadsTest.java b/src/test/java/com/zaxxer/hikari/pool/ConnectionPoolSizeVsThreadsTest.java
index e66ba58..8585485 100755
--- a/src/test/java/com/zaxxer/hikari/pool/ConnectionPoolSizeVsThreadsTest.java
+++ b/src/test/java/com/zaxxer/hikari/pool/ConnectionPoolSizeVsThreadsTest.java
@@ -45,9 +45,9 @@ import com.zaxxer.hikari.mocks.StubDataSource;
*/
public class ConnectionPoolSizeVsThreadsTest {
- public static final Logger LOGGER = LoggerFactory.getLogger(ConnectionPoolSizeVsThreadsTest.class);
+ private static final Logger LOGGER = LoggerFactory.getLogger(ConnectionPoolSizeVsThreadsTest.class);
- public static final int ITERATIONS = 50_000;
+ private static final int ITERATIONS = 50_000;
@Test
public void testPoolSizeAboutSameSizeAsThreadCount() throws Exception {
diff --git a/src/test/java/com/zaxxer/hikari/pool/ConnectionRaceConditionTest.java b/src/test/java/com/zaxxer/hikari/pool/ConnectionRaceConditionTest.java
index 31f8363..c2bd000 100755
--- a/src/test/java/com/zaxxer/hikari/pool/ConnectionRaceConditionTest.java
+++ b/src/test/java/com/zaxxer/hikari/pool/ConnectionRaceConditionTest.java
@@ -51,7 +51,7 @@ public class ConnectionRaceConditionTest
config.setMinimumIdle(0);
config.setMaximumPoolSize(10);
config.setInitializationFailTimeout(Long.MAX_VALUE);
- config.setConnectionTimeout(2500);
+ config.setConnectionTimeout(5000);
config.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
setSlf4jLogLevel(ConcurrentBag.class, Level.INFO);
diff --git a/src/test/java/com/zaxxer/hikari/pool/HouseKeeperCleanupTest.java b/src/test/java/com/zaxxer/hikari/pool/HouseKeeperCleanupTest.java
index 1a7caa7..f62de43 100644
--- a/src/test/java/com/zaxxer/hikari/pool/HouseKeeperCleanupTest.java
+++ b/src/test/java/com/zaxxer/hikari/pool/HouseKeeperCleanupTest.java
@@ -59,7 +59,7 @@ public class HouseKeeperCleanupTest
config.setInitializationFailTimeout(Long.MAX_VALUE);
config.setConnectionTimeout(2500);
config.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
- config.setScheduledExecutorService(executor);
+ config.setScheduledExecutor(executor);
HikariConfig config2 = newHikariConfig();
config.copyState(config2);
diff --git a/src/test/java/com/zaxxer/hikari/pool/MetricsTrackerTest.java b/src/test/java/com/zaxxer/hikari/pool/MetricsTrackerTest.java
new file mode 100644
index 0000000..494fd4d
--- /dev/null
+++ b/src/test/java/com/zaxxer/hikari/pool/MetricsTrackerTest.java
@@ -0,0 +1,86 @@
+package com.zaxxer.hikari.pool;
+
+import com.zaxxer.hikari.HikariDataSource;
+import com.zaxxer.hikari.metrics.IMetricsTracker;
+import com.zaxxer.hikari.mocks.StubDataSource;
+import org.junit.Test;
+
+import java.sql.Connection;
+import java.sql.SQLTransientConnectionException;
+import java.util.concurrent.TimeUnit;
+
+import static com.zaxxer.hikari.pool.TestElf.newHikariDataSource;
+import static junit.framework.TestCase.fail;
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * @author wvuong@chariotsolutions.com on 2/16/17.
+ */
+public class MetricsTrackerTest
+{
+
+ @Test(expected = SQLTransientConnectionException.class)
+ public void connectionTimeoutIsRecorded() throws Exception
+ {
+ int timeoutMillis = 1000;
+ int timeToCreateNewConnectionMillis = timeoutMillis * 2;
+
+ StubDataSource stubDataSource = new StubDataSource();
+ stubDataSource.setConnectionAcquistionTime(timeToCreateNewConnectionMillis);
+
+ StubMetricsTracker metricsTracker = new StubMetricsTracker();
+
+ try (HikariDataSource ds = newHikariDataSource()) {
+ ds.setMinimumIdle(0);
+ ds.setMaximumPoolSize(1);
+ ds.setConnectionTimeout(timeoutMillis);
+ ds.setDataSource(stubDataSource);
+ ds.setMetricsTrackerFactory((poolName, poolStats) -> metricsTracker);
+
+ try (Connection c = ds.getConnection()) {
+ fail("Connection shouldn't have been successfully created due to configured connection timeout");
+
+ } finally {
+ // assert that connection timeout was measured
+ assertThat(metricsTracker.connectionTimeoutRecorded, is(true));
+ // assert that measured time to acquire connection should be roughly equal or greater than the configured connection timeout time
+ assertTrue(metricsTracker.connectionAcquiredNanos >= TimeUnit.NANOSECONDS.convert(timeoutMillis, TimeUnit.MILLISECONDS));
+ }
+ }
+ }
+
+ private static class StubMetricsTracker implements IMetricsTracker
+ {
+
+ private Long connectionCreatedMillis;
+ private Long connectionAcquiredNanos;
+ private Long connectionBorrowedMillis;
+ private boolean connectionTimeoutRecorded;
+
+ @Override
+ public void recordConnectionCreatedMillis(long connectionCreatedMillis)
+ {
+ this.connectionCreatedMillis = connectionCreatedMillis;
+ }
+
+ @Override
+ public void recordConnectionAcquiredNanos(long elapsedAcquiredNanos)
+ {
+ this.connectionAcquiredNanos = elapsedAcquiredNanos;
+ }
+
+ @Override
+ public void recordConnectionUsageMillis(long elapsedBorrowedMillis)
+ {
+ this.connectionBorrowedMillis = elapsedBorrowedMillis;
+ }
+
+ @Override
+ public void recordConnectionTimeout()
+ {
+ this.connectionTimeoutRecorded = true;
+ }
+ }
+}
diff --git a/src/test/java/com/zaxxer/hikari/pool/MiscTest.java b/src/test/java/com/zaxxer/hikari/pool/MiscTest.java
index f9698ef..2791f3c 100644
--- a/src/test/java/com/zaxxer/hikari/pool/MiscTest.java
+++ b/src/test/java/com/zaxxer/hikari/pool/MiscTest.java
@@ -102,7 +102,7 @@ public class MiscTest
try (PrintStream ps = new PrintStream(baos, true)) {
setSlf4jTargetStream(Class.forName("com.zaxxer.hikari.pool.ProxyLeakTask"), ps);
setConfigUnitTest(true);
-
+
HikariConfig config = newHikariConfig();
config.setMinimumIdle(0);
config.setMaximumPoolSize(4);
@@ -110,11 +110,11 @@ public class MiscTest
config.setMetricRegistry(null);
config.setLeakDetectionThreshold(TimeUnit.SECONDS.toMillis(1));
config.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
-
+
try (HikariDataSource ds = new HikariDataSource(config)) {
setSlf4jLogLevel(HikariPool.class, Level.DEBUG);
getPool(ds).logPoolState();
-
+
try (Connection connection = ds.getConnection()) {
quietlySleep(SECONDS.toMillis(4));
connection.close();
@@ -128,6 +128,7 @@ public class MiscTest
finally
{
setConfigUnitTest(false);
+ setSlf4jLogLevel(HikariPool.class, Level.INFO);
}
}
}
diff --git a/src/test/java/com/zaxxer/hikari/pool/SaturatedPoolTest830.java b/src/test/java/com/zaxxer/hikari/pool/SaturatedPoolTest830.java
new file mode 100644
index 0000000..db8f970
--- /dev/null
+++ b/src/test/java/com/zaxxer/hikari/pool/SaturatedPoolTest830.java
@@ -0,0 +1,156 @@
+/*
+ * Copyright (C) 2017 Brett Wooldridge
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.zaxxer.hikari.pool;
+
+import static com.zaxxer.hikari.pool.TestElf.getConcurrentBag;
+import static com.zaxxer.hikari.pool.TestElf.getPool;
+import static com.zaxxer.hikari.pool.TestElf.newHikariConfig;
+import static com.zaxxer.hikari.pool.TestElf.setSlf4jLogLevel;
+import static com.zaxxer.hikari.util.ClockSource.currentTime;
+import static com.zaxxer.hikari.util.ClockSource.elapsedMillis;
+import static com.zaxxer.hikari.util.UtilityElf.quietlySleep;
+import static java.lang.Math.round;
+import static java.util.concurrent.TimeUnit.SECONDS;
+import static org.junit.Assert.assertEquals;
+
+import java.sql.Connection;
+import java.sql.SQLException;
+import java.sql.Statement;
+import java.util.Arrays;
+import java.util.concurrent.LinkedBlockingQueue;
+import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicInteger;
+
+import org.apache.logging.log4j.Level;
+import org.junit.Test;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.zaxxer.hikari.HikariConfig;
+import com.zaxxer.hikari.HikariDataSource;
+import com.zaxxer.hikari.mocks.StubConnection;
+import com.zaxxer.hikari.mocks.StubStatement;
+
+/**
+ * @author Brett Wooldridge
+ */
+public class SaturatedPoolTest830
+{
+ private static final Logger LOGGER = LoggerFactory.getLogger(SaturatedPoolTest830.class);
+ private static final int MAX_POOL_SIZE = 10;
+
+ @Test
+ public void saturatedPoolTest() throws Exception {
+ HikariConfig config = newHikariConfig();
+ config.setMinimumIdle(5);
+ config.setMaximumPoolSize(MAX_POOL_SIZE);
+ config.setInitializationFailTimeout(Long.MAX_VALUE);
+ config.setConnectionTimeout(1000);
+ config.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
+
+ StubConnection.slowCreate = true;
+ StubStatement.setSimulatedQueryTime(1000);
+ setSlf4jLogLevel(HikariPool.class, Level.DEBUG);
+ System.setProperty("com.zaxxer.hikari.housekeeping.periodMs", "5000");
+
+ final long start = currentTime();
+
+ try (final HikariDataSource ds = new HikariDataSource(config)) {
+ LinkedBlockingQueue<Runnable> queue = new LinkedBlockingQueue<>();
+ ThreadPoolExecutor threadPool = new ThreadPoolExecutor( 50 /*core*/, 50 /*max*/, 2 /*keepalive*/, SECONDS, queue, new ThreadPoolExecutor.CallerRunsPolicy());
+ threadPool.allowCoreThreadTimeOut(true);
+
+ AtomicInteger windowIndex = new AtomicInteger();
+ boolean[] failureWindow = new boolean[100];
+ Arrays.fill(failureWindow, true);
+
+ // Initial saturation
+ for (int i = 0; i < 50; i++) {
+ threadPool.execute(() -> {
+ try (Connection conn = ds.getConnection();
+ Statement stmt = conn.createStatement()) {
+ stmt.execute("SELECT bogus FROM imaginary");
+ }
+ catch (SQLException e) {
+ LOGGER.info(e.getMessage());
+ }
+ });
+ }
+
+ long sleep = 80;
+outer: while (true) {
+ quietlySleep(sleep);
+
+ if (elapsedMillis(start) > SECONDS.toMillis(12) && sleep < 100) {
+ sleep = 100;
+ LOGGER.warn("Switching to 100ms sleep");
+ }
+ else if (elapsedMillis(start) > SECONDS.toMillis(6) && sleep < 90) {
+ sleep = 90;
+ LOGGER.warn("Switching to 90ms sleep");
+ }
+
+ threadPool.execute(() -> {
+ int ndx = windowIndex.incrementAndGet() % failureWindow.length;
+
+ try (Connection conn = ds.getConnection();
+ Statement stmt = conn.createStatement()) {
+ stmt.execute("SELECT bogus FROM imaginary");
+ failureWindow[ndx] = false;
+ }
+ catch (SQLException e) {
+ LOGGER.info(e.getMessage());
+ failureWindow[ndx] = true;
+ }
+ });
+
+ for (int i = 0; i < failureWindow.length; i++) {
+ if (failureWindow[i]) {
+ if (elapsedMillis(start) % (SECONDS.toMillis(1) - sleep) < sleep) {
+ LOGGER.info("Active threads {}, submissions per second {}, waiting threads {}",
+ threadPool.getActiveCount(),
+ SECONDS.toMillis(1) / sleep,
+ getPool(ds).getThreadsAwaitingConnection());
+ }
+ continue outer;
+ }
+ }
+
+ LOGGER.info("Timeouts have subsided.");
+ LOGGER.info("Active threads {}, submissions per second {}, waiting threads {}",
+ threadPool.getActiveCount(),
+ SECONDS.toMillis(1) / sleep,
+ getPool(ds).getThreadsAwaitingConnection());
+ break;
+ }
+
+ LOGGER.info("Waiting for completion of {} active tasks.", threadPool.getActiveCount());
+ while (getPool(ds).getActiveConnections() > 0) {
+ quietlySleep(50);
+ }
+
+ assertEquals("Rate not in balance at 10req/s", SECONDS.toMillis(1) / sleep, 10L);
+ }
+ finally {
+ StubStatement.setSimulatedQueryTime(0);
+ StubConnection.slowCreate = false;
+ System.clearProperty("com.zaxxer.hikari.housekeeping.periodMs");
+ setSlf4jLogLevel(HikariPool.class, Level.INFO);
+ }
+ }
+}
diff --git a/src/test/java/com/zaxxer/hikari/pool/TestConnectionTimeoutRetry.java b/src/test/java/com/zaxxer/hikari/pool/TestConnectionTimeoutRetry.java
index 15f8974..2760543 100644
--- a/src/test/java/com/zaxxer/hikari/pool/TestConnectionTimeoutRetry.java
+++ b/src/test/java/com/zaxxer/hikari/pool/TestConnectionTimeoutRetry.java
@@ -251,9 +251,9 @@ public class TestConnectionTimeoutRetry
Connection connection5 = ds.getConnection();
Connection connection6 = ds.getConnection();
Connection connection7 = ds.getConnection()) {
-
+
sleep(1300);
-
+
assertSame("Total connections not as expected", 10, pool.getTotalConnections());
assertSame("Idle connections not as expected", 3, pool.getIdleConnections());
}
@@ -273,5 +273,6 @@ public class TestConnectionTimeoutRetry
public void after()
{
System.getProperties().remove("com.zaxxer.hikari.housekeeping.periodMs");
+ setSlf4jLogLevel(HikariPool.class, Level.INFO);
}
}
diff --git a/src/test/java/com/zaxxer/hikari/pool/TestElf.java b/src/test/java/com/zaxxer/hikari/pool/TestElf.java
index 6438b11..7ec2925 100644
--- a/src/test/java/com/zaxxer/hikari/pool/TestElf.java
+++ b/src/test/java/com/zaxxer/hikari/pool/TestElf.java
@@ -19,7 +19,6 @@ package com.zaxxer.hikari.pool;
import java.io.PrintStream;
import java.lang.reflect.Field;
import java.sql.Connection;
-import java.util.HashMap;
import org.apache.logging.log4j.Level;
import org.apache.logging.log4j.core.Appender;
@@ -32,6 +31,7 @@ import org.slf4j.LoggerFactory;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
+import com.zaxxer.hikari.util.ConcurrentBag;
/**
* Utility methods for testing.
@@ -44,6 +44,10 @@ public final class TestElf
// default constructor
}
+ public static boolean isJava9() {
+ return System.getProperty("java.version").startsWith("9");
+ }
+
public static HikariPool getPool(HikariDataSource ds)
{
try {
@@ -56,20 +60,19 @@ public final class TestElf
}
}
- @SuppressWarnings("unchecked")
- public static HashMap<Object, HikariPool> getMultiPool(HikariDataSource ds)
+ static ConcurrentBag<?> getConcurrentBag(HikariDataSource ds)
{
try {
- Field field = ds.getClass().getDeclaredField("multiPool");
+ Field field = HikariPool.class.getDeclaredField("connectionBag");
field.setAccessible(true);
- return (HashMap<Object, HikariPool>) field.get(ds);
+ return (ConcurrentBag<?>) field.get(getPool(ds));
}
catch (Exception e) {
throw new RuntimeException(e);
}
}
- public static boolean getConnectionCommitDirtyState(Connection connection)
+ static boolean getConnectionCommitDirtyState(Connection connection)
{
try {
Field field = ProxyConnection.class.getDeclaredField("isCommitStateDirty");
@@ -81,7 +84,7 @@ public final class TestElf
}
}
- public static void setConfigUnitTest(boolean unitTest)
+ static void setConfigUnitTest(boolean unitTest)
{
try {
Field field = HikariConfig.class.getDeclaredField("unitTest");
@@ -93,7 +96,7 @@ public final class TestElf
}
}
- public static void setSlf4jTargetStream(Class<?> clazz, PrintStream stream)
+ static void setSlf4jTargetStream(Class<?> clazz, PrintStream stream)
{
try {
Log4jLogger log4Jlogger = (Log4jLogger) LoggerFactory.getLogger(clazz);
@@ -114,7 +117,7 @@ public final class TestElf
}
}
- public static void setSlf4jLogLevel(Class<?> clazz, Level logLevel)
+ static void setSlf4jLogLevel(Class<?> clazz, Level logLevel)
{
try {
Log4jLogger log4Jlogger = (Log4jLogger) LoggerFactory.getLogger(clazz);
@@ -144,7 +147,7 @@ public final class TestElf
return config;
}
- public static HikariDataSource newHikariDataSource()
+ static HikariDataSource newHikariDataSource()
{
final StackTraceElement callerStackTrace = Thread.currentThread().getStackTrace()[2];
diff --git a/src/test/java/com/zaxxer/hikari/pool/TestJNDI.java b/src/test/java/com/zaxxer/hikari/pool/TestJNDI.java
index b5cd40c..6e041cb 100644
--- a/src/test/java/com/zaxxer/hikari/pool/TestJNDI.java
+++ b/src/test/java/com/zaxxer/hikari/pool/TestJNDI.java
@@ -15,17 +15,21 @@
*/
package com.zaxxer.hikari.pool;
+import static com.zaxxer.hikari.pool.TestElf.newHikariConfig;
+import static com.zaxxer.hikari.pool.TestElf.newHikariDataSource;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import javax.naming.Context;
+import javax.naming.InitialContext;
import javax.naming.Name;
import javax.naming.NamingException;
import javax.naming.RefAddr;
import javax.naming.Reference;
+import com.zaxxer.hikari.HikariConfig;
import org.junit.Test;
import org.osjava.sj.jndi.AbstractContext;
@@ -33,6 +37,8 @@ import com.zaxxer.hikari.HikariDataSource;
import com.zaxxer.hikari.HikariJNDIFactory;
import com.zaxxer.hikari.mocks.StubDataSource;
+import java.sql.Connection;
+
public class TestJNDI
{
@Test
@@ -94,6 +100,27 @@ public class TestJNDI
}
}
+ @Test
+ public void testJndiLookup4() throws Exception
+ {
+ System.setProperty(Context.INITIAL_CONTEXT_FACTORY, "org.osjava.sj.memory.MemoryContextFactory");
+ System.setProperty("org.osjava.sj.jndi.shared", "true");
+ InitialContext ic = new InitialContext();
+
+ StubDataSource ds = new StubDataSource();
+
+ Context subcontext = ic.createSubcontext("java:/comp/env/jdbc");
+ subcontext.bind("java:/comp/env/jdbc/myDS", ds);
+
+ HikariConfig config = newHikariConfig();
+ config.setDataSourceJNDI("java:/comp/env/jdbc/myDS");
+
+ try (HikariDataSource hds = new HikariDataSource(config);
+ Connection conn = hds.getConnection()) {
+ assertNotNull(conn);
+ }
+ }
+
private class BogusContext extends AbstractContext
{
@Override
diff --git a/src/test/java/com/zaxxer/hikari/pool/TestMBean.java b/src/test/java/com/zaxxer/hikari/pool/TestMBean.java
index 55ba733..a77c960 100644
--- a/src/test/java/com/zaxxer/hikari/pool/TestMBean.java
+++ b/src/test/java/com/zaxxer/hikari/pool/TestMBean.java
@@ -15,30 +15,83 @@
*/
package com.zaxxer.hikari.pool;
-import static com.zaxxer.hikari.pool.TestElf.newHikariConfig;
+import com.zaxxer.hikari.HikariConfig;
+import com.zaxxer.hikari.HikariDataSource;
+import com.zaxxer.hikari.HikariPoolMXBean;
+import org.junit.Test;
+import javax.management.JMX;
+import javax.management.MBeanServer;
+import javax.management.MalformedObjectNameException;
+import javax.management.ObjectName;
+import java.lang.management.ManagementFactory;
+import java.sql.Connection;
import java.sql.SQLException;
+import java.util.concurrent.TimeUnit;
-import org.junit.Test;
-
-import com.zaxxer.hikari.HikariConfig;
-import com.zaxxer.hikari.HikariDataSource;
+import static com.zaxxer.hikari.pool.TestElf.newHikariConfig;
+import static org.junit.Assert.assertEquals;
public class TestMBean
{
- @Test
- public void testMBeanRegistration() throws SQLException
- {
- HikariConfig config = newHikariConfig();
- config.setMinimumIdle(0);
- config.setMaximumPoolSize(1);
- config.setRegisterMbeans(true);
- config.setConnectionTimeout(2800);
- config.setConnectionTestQuery("VALUES 1");
- config.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
-
- try (HikariDataSource ds = new HikariDataSource(config)) {
- // Close immediately
- }
- }
+ @Test
+ public void testMBeanRegistration() throws SQLException {
+ HikariConfig config = newHikariConfig();
+ config.setMinimumIdle(0);
+ config.setMaximumPoolSize(1);
+ config.setRegisterMbeans(true);
+ config.setConnectionTimeout(2800);
+ config.setConnectionTestQuery("VALUES 1");
+ config.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
+
+ try (HikariDataSource ds = new HikariDataSource(config)) {
+ // Close immediately
+ }
+ }
+
+ @Test
+ public void testMBeanReporting() throws SQLException, InterruptedException, MalformedObjectNameException {
+ HikariConfig config = newHikariConfig();
+ config.setMinimumIdle(3);
+ config.setMaximumPoolSize(5);
+ config.setRegisterMbeans(true);
+ config.setConnectionTimeout(2800);
+ config.setConnectionTestQuery("VALUES 1");
+ config.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
+
+ System.setProperty("com.zaxxer.hikari.housekeeping.periodMs", "100");
+
+ try (HikariDataSource ds = new HikariDataSource(config)) {
+
+ ds.setIdleTimeout(3000);
+
+ TimeUnit.SECONDS.sleep(1);
+
+ MBeanServer mBeanServer = ManagementFactory.getPlatformMBeanServer();
+ ObjectName poolName = new ObjectName("com.zaxxer.hikari:type=Pool (testMBeanReporting)");
+ HikariPoolMXBean hikariPoolMXBean = JMX.newMXBeanProxy(mBeanServer, poolName, HikariPoolMXBean.class);
+
+ assertEquals(0, hikariPoolMXBean.getActiveConnections());
+ assertEquals(3, hikariPoolMXBean.getIdleConnections());
+
+ try (Connection connection = ds.getConnection()) {
+ assertEquals(1, hikariPoolMXBean.getActiveConnections());
+
+ TimeUnit.SECONDS.sleep(1);
+
+ assertEquals(3, hikariPoolMXBean.getIdleConnections());
+ assertEquals(4, hikariPoolMXBean.getTotalConnections());
+ }
+
+ TimeUnit.SECONDS.sleep(2);
+
+ assertEquals(0, hikariPoolMXBean.getActiveConnections());
+ assertEquals(3, hikariPoolMXBean.getIdleConnections());
+ assertEquals(3, hikariPoolMXBean.getTotalConnections());
+
+ }
+ finally {
+ System.clearProperty("com.zaxxer.hikari.housekeeping.periodMs");
+ }
+ }
}
diff --git a/src/test/java/com/zaxxer/hikari/pool/TestMetrics.java b/src/test/java/com/zaxxer/hikari/pool/TestMetrics.java
index fcba58e..775bb49 100644
--- a/src/test/java/com/zaxxer/hikari/pool/TestMetrics.java
+++ b/src/test/java/com/zaxxer/hikari/pool/TestMetrics.java
@@ -44,8 +44,6 @@ import com.zaxxer.hikari.metrics.MetricsTrackerFactory;
import com.zaxxer.hikari.metrics.dropwizard.CodahaleMetricsTrackerFactory;
import com.zaxxer.hikari.util.UtilityElf;
-import shaded.org.codehaus.plexus.interpolation.os.Os;
-
/**
* Test HikariCP/CodaHale metrics integration.
*
@@ -85,7 +83,7 @@ public class TestMetrics
@Test
public void testMetricUsage() throws SQLException
{
- assumeFalse(Os.isFamily(Os.FAMILY_WINDOWS));
+ assumeFalse(System.getProperty("os.name").contains("Windows"));
MetricRegistry metricRegistry = new MetricRegistry();
HikariConfig config = newHikariConfig();
@@ -156,19 +154,19 @@ public class TestMetrics
try (HikariDataSource ds = newHikariDataSource()) {
ds.setMaximumPoolSize(1);
ds.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
-
+
MetricRegistry metricRegistry = new MetricRegistry();
HealthCheckRegistry healthRegistry = new HealthCheckRegistry();
-
+
try {
try (Connection connection = ds.getConnection()) {
// close immediately
}
-
+
// After the pool as started, we can only set them once...
ds.setMetricRegistry(metricRegistry);
ds.setHealthCheckRegistry(healthRegistry);
-
+
// and never again...
ds.setMetricRegistry(metricRegistry);
fail("Should not have been allowed to set registry after pool started");
@@ -192,19 +190,19 @@ public class TestMetrics
try (HikariDataSource ds = newHikariDataSource()) {
ds.setMaximumPoolSize(1);
ds.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
-
+
MetricRegistry metricRegistry = new MetricRegistry();
HealthCheckRegistry healthRegistry = new HealthCheckRegistry();
-
+
ds.setMetricRegistry(metricRegistry);
ds.setHealthCheckRegistry(healthRegistry);
-
+
// before the pool is started, we can set it any number of times...
ds.setMetricRegistry(metricRegistry);
ds.setHealthCheckRegistry(healthRegistry);
-
+
try (Connection connection = ds.getConnection()) {
-
+
// after the pool is started, we cannot set it any more
ds.setMetricRegistry(metricRegistry);
fail("Should not have been allowed to set registry after pool started");
@@ -221,15 +219,15 @@ public class TestMetrics
try (HikariDataSource ds = newHikariDataSource()) {
ds.setMaximumPoolSize(1);
ds.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
-
+
MetricRegistry metricRegistry = new MetricRegistry();
MetricsTrackerFactory metricsTrackerFactory = new CodahaleMetricsTrackerFactory(metricRegistry);
-
+
try (Connection connection = ds.getConnection()) {
-
+
// After the pool as started, we can only set them once...
ds.setMetricsTrackerFactory(metricsTrackerFactory);
-
+
// and never again...
ds.setMetricsTrackerFactory(metricsTrackerFactory);
fail("Should not have been allowed to set metricsTrackerFactory after pool started");
@@ -254,16 +252,16 @@ public class TestMetrics
try (HikariDataSource ds = newHikariDataSource()) {
ds.setMaximumPoolSize(1);
ds.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
-
+
MetricRegistry metricRegistry = new MetricRegistry();
-
+
// before the pool is started, we can set it any number of times using either setter
ds.setMetricRegistry(metricRegistry);
ds.setMetricRegistry(metricRegistry);
ds.setMetricRegistry(metricRegistry);
-
+
try (Connection connection = ds.getConnection()) {
-
+
// after the pool is started, we cannot set it any more
ds.setMetricRegistry(metricRegistry);
fail("Should not have been allowed to set registry after pool started");
@@ -280,17 +278,17 @@ public class TestMetrics
try (HikariDataSource ds = newHikariDataSource()) {
ds.setMaximumPoolSize(1);
ds.setDataSourceClassName("com.zaxxer.hikari.mocks.StubDataSource");
-
+
MetricRegistry metricRegistry = new MetricRegistry();
MetricsTrackerFactory metricsTrackerFactory = new CodahaleMetricsTrackerFactory(metricRegistry);
-
+
// before the pool is started, we can set it any number of times using either setter
ds.setMetricsTrackerFactory(metricsTrackerFactory);
ds.setMetricsTrackerFactory(metricsTrackerFactory);
ds.setMetricsTrackerFactory(metricsTrackerFactory);
-
+
try (Connection connection = ds.getConnection()) {
-
+
// after the pool is started, we cannot set it any more
ds.setMetricsTrackerFactory(metricsTrackerFactory);
fail("Should not have been allowed to set registry factory after pool started");