Added health checks

billoneil · billoneil · commit a621fa3954f9 · 2017-02-08T22:45:08.000-05:00
diff --git a/stubbornjava-common/src/main/java/com/stubbornjava/common/HealthChecks.java b/stubbornjava-common/src/main/java/com/stubbornjava/common/HealthChecks.java
@@ -0,0 +1,16 @@
+package com.stubbornjava.common;
+
+import com.codahale.metrics.health.HealthCheckRegistry;
+
+public class HealthChecks {
+    private HealthChecks() {}
+
+    private static final HealthCheckRegistry healthCheckRegistry;
+    static {
+        healthCheckRegistry = new HealthCheckRegistry();
+    }
+
+    public static HealthCheckRegistry getHealthCheckRegistry() {
+        return healthCheckRegistry;
+    }
+}
diff --git a/stubbornjava-examples/src/main/java/com/stubbornjava/examples/hikaricp/ConnectionPools.java b/stubbornjava-examples/src/main/java/com/stubbornjava/examples/hikaricp/ConnectionPools.java
@@ -6,6 +6,7 @@
 import org.slf4j.LoggerFactory;
 
 import com.stubbornjava.common.Configs;
+import com.stubbornjava.common.HealthChecks;
 import com.stubbornjava.common.Metrics;
 import com.stubbornjava.common.db.ConnectionPool;
 import com.typesafe.config.Config;
@@ -25,7 +26,7 @@ public class ConnectionPools {
      *  Using enum singleton pattern for lazy singletons
      */
     private enum Transactional {
-        INSTANCE(ConnectionPool.getDataSourceFromConfig(conf.getConfig("pools.transactional"), Metrics.registry(), null));
+        INSTANCE(ConnectionPool.getDataSourceFromConfig(conf.getConfig("pools.transactional"), Metrics.registry(), HealthChecks.getHealthCheckRegistry()));
         private final DataSource dataSource;
         private Transactional(DataSource dataSource) {
             this.dataSource = dataSource;
@@ -53,7 +54,7 @@ public static DataSource getTransactional() {
      *  run while the other pool is backed up.
      */
     private enum Processing {
-        INSTANCE(ConnectionPool.getDataSourceFromConfig(conf.getConfig("pools.processing"), Metrics.registry(), null));
+        INSTANCE(ConnectionPool.getDataSourceFromConfig(conf.getConfig("pools.processing"), Metrics.registry(), HealthChecks.getHealthCheckRegistry()));
         private final DataSource dataSource;
         private Processing(DataSource dataSource) {
             this.dataSource = dataSource;
@@ -69,9 +70,10 @@ public static DataSource getProcessing() {
 
     public static void main(String[] args) {
         logger.debug("starting");
+        DataSource processing = ConnectionPools.getProcessing();
+        logger.debug("processing started");
         DataSource transactional = ConnectionPools.getTransactional();
         logger.debug("transactional started");
-        DataSource processing = ConnectionPools.getProcessing();
         logger.debug("done");
     }
 }
