feat: connection time plugin

common/lib/connection_plugin_chain_builder.ts

@@ -22,6 +22,7 @@ import { WrapperProperties } from "./wrapper_property";
 import { AwsWrapperError } from "./utils/aws_wrapper_error";
 import { Messages } from "./utils/messages";
 import { DefaultPlugin } from "./plugins/default_plugin";
+import { ConnectTimePluginFactory } from './plugins/connect_time_plugin';
 
 export class PluginFactoryInfo {}
 
@@ -33,7 +34,8 @@ export class ConnectionPluginChainBuilder {
 
   static readonly PLUGIN_FACTORIES = new Map<string, FactoryClass>([
     ["iam", IamAuthenticationPluginFactory],
-    ["failover", FailoverPluginFactory]
+    ["failover", FailoverPluginFactory],
+    ["connectTime", ConnectTimePluginFactory]
   ]);
 
   getPlugins(pluginService: PluginService, props: Map<string, any>): ConnectionPlugin[] {

common/lib/plugins/connect_time_plugin.ts

@@ -0,0 +1,69 @@
+/*
+  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+
+  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.
+*/
+
+import { AbstractConnectionPlugin } from "../abstract_connection_plugin";
+import { logger } from '../../logutils';
+import { HostInfo } from "../host_info";
+import { getTimeInNanos } from '../utils/utils';
+import { Messages } from "../utils/messages";
+import { ConnectionPluginFactory } from "../plugin_factory";
+import { ConnectionPlugin } from "../connection_plugin";
+import { PluginService } from '../plugin_service';
+
+export class ConnectTimePlugin extends AbstractConnectionPlugin {
+    private static subscribedMethods: Set<string> = new Set<string>(["connect", "forceConnect"]);
+    private static connectTime: bigint = 0n;
+
+    public override getSubscribedMethods(): Set<string> {
+        return ConnectTimePlugin.subscribedMethods;
+    }
+
+    public override async connect<T>(hostInfo: HostInfo, props: Map<string, any>, isInitialConnection: boolean, connectFunc: () => Promise<T>): Promise<T> {
+        const startTime = getTimeInNanos();
+
+        const result = await connectFunc();
+
+        const elapsedTimeNanos = getTimeInNanos() - startTime;
+        ConnectTimePlugin.connectTime += elapsedTimeNanos;
+        logger.debug(Messages.get("ConnectTimePlugin.connectTime", elapsedTimeNanos.toString()));
+        return result;
+    }
+
+    public override async forceConnect<T>(hostInfo: HostInfo, props: Map<string, any>, isInitialConnection: boolean, forceConnectFunc: () => Promise<T>): Promise<T> {
+        const startTime = getTimeInNanos();
+
+        const result = await forceConnectFunc();
+
+        const elapsedTimeNanos = getTimeInNanos() - startTime;
+        ConnectTimePlugin.connectTime += elapsedTimeNanos;
+        logger.debug(Messages.get("ConnectTimePlugin.connectTime", elapsedTimeNanos.toString()));
+        return result;
+    }
+
+    public static resetConnectTime(): void {
+        this.connectTime = 0n;
+    }
+
+    public static getTotalConnectTime(): bigint {
+        return this.connectTime;
+    }
+}
+
+export class ConnectTimePluginFactory implements ConnectionPluginFactory {
+    getInstance(pluginService: PluginService, properties: Map<string, any>): ConnectionPlugin {
+        return new ConnectTimePlugin();
+    }
+}

common/lib/utils/locales/en.json

@@ -33,5 +33,6 @@
   "ClusterAwareWriterFailoverHandler.taskBFailedToConnectToAnyReader": "[TaskB] Failed to connect to any reader.",
   "ClusterAwareWriterFailoverHandler.standaloneNode": "[TaskB] Host %s is not yet connected to a cluster. The cluster is still being reconfigured.",
   "ClusterAwareWriterFailoverHandler.taskBAttemptConnectionToNewWriter": "[TaskB] Trying to connect to a new writer: '%s'",
-  "ClusterAwareWriterFailoverHandler.alreadyWriter": "Current reader connection is actually a new writer connection."
+  "ClusterAwareWriterFailoverHandler.alreadyWriter": "Current reader connection is actually a new writer connection.",
+  "ConnectTimePlugin.connectTime": "Connected in %s nanoseconds."
 }

common/lib/utils/utils.ts

@@ -38,7 +38,7 @@ export function logTopology(hosts: HostInfo[], msgPrefix: string) {
 }
 
 export function getTimeInNanos() {
-  return performance.now();
+  return process.hrtime.bigint();
 }
 
 export function maskProperties(props: Map<string, any>) {