perf: cache efm2 monitor key for better performance

Author: aaron-congo

wrapper/src/main/java/software/amazon/jdbc/plugin/efm2/HostMonitorService.java

@@ -30,10 +30,7 @@ public interface HostMonitorService {
   HostMonitorConnectionContext startMonitoring(
       Connection connectionToAbort,
       HostSpec hostSpec,
-      Properties properties,
-      int failureDetectionTimeMillis,
-      int failureDetectionIntervalMillis,
-      int failureDetectionCount) throws SQLException;
+      Properties properties) throws SQLException;
 
   /**
    * Stop monitoring for a connection represented by the given {@link HostMonitorConnectionContext}.

wrapper/src/main/java/software/amazon/jdbc/plugin/efm2/HostMonitorServiceImpl.java

@@ -16,9 +16,15 @@
 
 package software.amazon.jdbc.plugin.efm2;
 
+import static software.amazon.jdbc.plugin.efm2.HostMonitoringConnectionPlugin.FAILURE_DETECTION_COUNT;
+import static software.amazon.jdbc.plugin.efm2.HostMonitoringConnectionPlugin.FAILURE_DETECTION_INTERVAL;
+import static software.amazon.jdbc.plugin.efm2.HostMonitoringConnectionPlugin.FAILURE_DETECTION_TIME;
+
 import java.sql.Connection;
 import java.sql.SQLException;
 import java.util.Properties;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ConcurrentMap;
 import java.util.concurrent.Executor;
 import java.util.concurrent.TimeUnit;
 import java.util.logging.Logger;
@@ -49,11 +55,15 @@ public class HostMonitorServiceImpl implements HostMonitorService {
   protected static final Executor ABORT_EXECUTOR =
       ExecutorFactory.newSingleThreadExecutor("abort");
 
+  protected final ConcurrentMap<String, String> monitorKeysByUrl = new ConcurrentHashMap<>();
   protected final FullServicesContainer serviceContainer;
   protected final PluginService pluginService;
   protected final MonitorService coreMonitorService;
   protected final TelemetryFactory telemetryFactory;
   protected final TelemetryCounter abortedConnectionsCounter;
+  protected final int failureDetectionTimeMillis;
+  protected final int failureDetectionIntervalMillis;
+  protected final int failureDetectionCount;
 
   public HostMonitorServiceImpl(final @NonNull FullServicesContainer serviceContainer, Properties props) {
     this.serviceContainer = serviceContainer;
@@ -62,6 +72,10 @@ public HostMonitorServiceImpl(final @NonNull FullServicesContainer serviceContai
     this.telemetryFactory = serviceContainer.getTelemetryFactory();
     this.abortedConnectionsCounter = telemetryFactory.createCounter("efm2.connections.aborted");
 
+    this.failureDetectionTimeMillis = FAILURE_DETECTION_TIME.getInteger(props);
+    this.failureDetectionIntervalMillis = FAILURE_DETECTION_INTERVAL.getInteger(props);
+    this.failureDetectionCount = FAILURE_DETECTION_COUNT.getInteger(props);
+
     this.coreMonitorService.registerMonitorTypeIfAbsent(
         HostMonitorImpl.class,
         TimeUnit.MILLISECONDS.toNanos(MONITOR_DISPOSAL_TIME_MS.getLong(props)),
@@ -78,21 +92,10 @@ public static void closeAllMonitors() {
   public HostMonitorConnectionContext startMonitoring(
       final Connection connectionToAbort,
       final HostSpec hostSpec,
-      final Properties properties,
-      final int failureDetectionTimeMillis,
-      final int failureDetectionIntervalMillis,
-      final int failureDetectionCount) throws SQLException {
-
-    final HostMonitor monitor = this.getMonitor(
-        hostSpec,
-        properties,
-        failureDetectionTimeMillis,
-        failureDetectionIntervalMillis,
-        failureDetectionCount);
-
+      final Properties properties) throws SQLException {
+    final HostMonitor monitor = this.getMonitor(hostSpec, properties);
     final HostMonitorConnectionContext context = new HostMonitorConnectionContext(connectionToAbort);
     monitor.startMonitoring(context);
-
     return context;
   }
 
@@ -131,24 +134,18 @@ public void releaseResources() {
    *
    * @param hostSpec Information such as hostname of the server.
    * @param properties The user configuration for the current connection.
-   * @param failureDetectionTimeMillis A failure detection time in millis.
-   * @param failureDetectionIntervalMillis A failure detection interval in millis.
-   * @param failureDetectionCount A failure detection count.
    * @return A {@link HostMonitorImpl} object associated with a specific server.
    * @throws SQLException if there's errors getting or creating a monitor
    */
-  protected HostMonitor getMonitor(
-      final HostSpec hostSpec,
-      final Properties properties,
-      final int failureDetectionTimeMillis,
-      final int failureDetectionIntervalMillis,
-      final int failureDetectionCount) throws SQLException {
-
-    final String monitorKey = String.format("%d:%d:%d:%s",
-        failureDetectionTimeMillis,
-        failureDetectionIntervalMillis,
-        failureDetectionCount,
-        hostSpec.getUrl());
+  protected HostMonitor getMonitor(final HostSpec hostSpec, final Properties properties) throws SQLException {
+    final String monitorKey = monitorKeysByUrl.computeIfAbsent(
+        hostSpec.getUrl(),
+        (url) -> String.format(
+            "%d:%d:%d:%s",
+            failureDetectionTimeMillis,
+            failureDetectionIntervalMillis,
+            failureDetectionCount,
+            url));
 
     return this.coreMonitorService.runIfAbsent(
         HostMonitorImpl.class,

wrapper/src/main/java/software/amazon/jdbc/plugin/efm2/HostMonitoringConnectionPlugin.java

@@ -148,11 +148,6 @@ public <T, E extends Exception> T execute(
       return jdbcMethodFunc.call();
     }
 
-    final int failureDetectionTimeMillis = FAILURE_DETECTION_TIME.getInteger(this.properties);
-    final int failureDetectionIntervalMillis =
-        FAILURE_DETECTION_INTERVAL.getInteger(this.properties);
-    final int failureDetectionCount = FAILURE_DETECTION_COUNT.getInteger(this.properties);
-
     initMonitorService();
 
     T result;
@@ -167,14 +162,10 @@ public <T, E extends Exception> T execute(
       final HostSpec monitoringHostSpec = this.getMonitoringHostSpec();
 
       try {
-        monitorContext =
-            this.monitorService.startMonitoring(
-                this.pluginService.getCurrentConnection(), // abort this connection if needed
-                monitoringHostSpec,
-                this.properties,
-                failureDetectionTimeMillis,
-                failureDetectionIntervalMillis,
-                failureDetectionCount);
+        monitorContext = this.monitorService.startMonitoring(
+            this.pluginService.getCurrentConnection(), // abort this connection if needed
+            monitoringHostSpec,
+            this.properties);
       } catch (SQLException e) {
         throw WrapperUtils.wrapExceptionIfNeeded(exceptionClass, e);
       }