feat: read-write splitting plugin

aws · Jun 5, 2024 · e630ab2 · e630ab2
1 parent 85507b1
commit e630ab2
Show file tree

Hide file tree

Showing 9 changed files with 1,206 additions and 7 deletions.
diff --git a/common/lib/connection_plugin_chain_builder.ts b/common/lib/connection_plugin_chain_builder.ts
@@ -26,6 +26,7 @@ import { ExecuteTimePluginFactory } from "./plugins/execute_time_plugin";
 import { ConnectTimePluginFactory } from "./plugins/connect_time_plugin";
 import { AwsSecretsManagerPluginFactory } from "./authentication/aws_secrets_manager_plugin";
 import { ConnectionProvider } from "./connection_provider";
+import { ReadWriteSplittingPluginFactory } from "./plugins/read_write_splitting";
 import { StaleDnsPluginFactory } from "./plugins/stale_dns/stale_dns_plugin";
 import { FederatedAuthPluginFactory } from "./plugins/federated_auth/federated_auth_plugin";
 
@@ -43,6 +44,7 @@ export class ConnectionPluginChainBuilder {
     ["connectTime", ConnectTimePluginFactory],
     ["secretsManager", AwsSecretsManagerPluginFactory],
     ["failover", FailoverPluginFactory],
+    ["readWriteSplitting", ReadWriteSplittingPluginFactory],
     ["staleDns", StaleDnsPluginFactory],
     ["federatedAuth", FederatedAuthPluginFactory]
   ]);

diff --git a/common/lib/driver_connection_provider.ts b/common/lib/driver_connection_provider.ts
@@ -113,7 +113,7 @@ export class DriverConnectionProvider implements ConnectionProvider {
   getHostInfoByStrategy(hosts: HostInfo[], role: HostRole, strategy: string, props?: Map<string, any>): HostInfo {
     const acceptedStrategy = DriverConnectionProvider.acceptedStrategies.get(strategy);
     if (!acceptedStrategy) {
-      throw new AwsWrapperError(Messages.get("ConnectionProvider.unsupportedHostInfoSelectorStrategy", strategy, "DriverConnectionProvider"));
+      throw new AwsWrapperError(Messages.get("ConnectionProvider.unsupportedHostSelectorStrategy", strategy, "DriverConnectionProvider"));
     }
     return acceptedStrategy.getHost(hosts, role, props);
   }

diff --git a/common/lib/plugin_service.ts b/common/lib/plugin_service.ts
@@ -102,6 +102,11 @@ export class PluginService implements ErrorHandler, HostListProviderService {
     this._initialConnectionHostInfo = initialConnectionHostInfo;
   }
 
+  getHostInfoByStrategy(role: HostRole, strategy: string): HostInfo | undefined {
+    const pluginManager = this.pluginServiceManagerContainer.pluginManager;
+    return pluginManager?.getHostInfoByStrategy(role, strategy);
+  }
+
   getCurrentHostInfo(): HostInfo | null {
     return this._currentHostInfo ? this._currentHostInfo : null;
   }
@@ -131,6 +136,10 @@ export class PluginService implements ErrorHandler, HostListProviderService {
     return false;
   }
 
+  acceptsStrategy(role: HostRole, strategy: string): boolean {
+    return this.pluginServiceManagerContainer.pluginManager?.acceptsStrategy(role, strategy) ?? false;
+  }
+
   async forceRefreshHostList(): Promise<void>;
   async forceRefreshHostList(targetClient?: any): Promise<void>;
   async forceRefreshHostList(targetClient?: any): Promise<void> {