test: rw performance tests

aws · Oct 25, 2024 · c332625 · c332625
1 parent cb12907
commit c332625
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/pg/lib/pg_pool_client.ts b/pg/lib/pg_pool_client.ts
@@ -40,7 +40,9 @@ export class AwsPgPoolClient implements AwsPoolClient {
   }
 
   async end(poolClient: any) {
-    await poolClient?.release(true);
+    if (poolClient) {
+      await poolClient.release(true);
+    }
   }
 
   getIdleCount(): number {

diff --git a/tests/integration/container/tests/read_write_splitting_performance.test.ts b/tests/integration/container/tests/read_write_splitting_performance.test.ts
@@ -80,8 +80,6 @@ describe("rwperformance", () => {
       ConnectionProviderManager.setConnectionProvider(new InternalPooledConnectionProvider());
       const rwPluginWithPoolConfig = initReadWritePluginConfig(env.databaseInfo.clusterEndpoint, env.databaseInfo.clusterEndpointPort);
       const rwPluginWithPoolData = await measurePerformance(rwPluginWithPoolConfig);
-      await ConnectionProviderManager.releaseResources();
-      ConnectionProviderManager.resetProvider();
 
       readerData = calculateReaderOverhead("Switch to reader", rwPluginWithPoolData, noPluginData);
       writerData = calculateWriterOverhead("Switch to writer (using cached connection)", rwPluginWithPoolData, noPluginData);
@@ -92,6 +90,9 @@ describe("rwperformance", () => {
         "InternalConnectionPool"
       );
 
+      await ConnectionProviderManager.releaseResources();
+      ConnectionProviderManager.resetProvider();
+
       setReadOnlyPerfDataList = [];
 
       // Internal connection pool results with warm up.
@@ -102,12 +103,12 @@ describe("rwperformance", () => {
         if (instance.host && instance.port) {
           const client = DriverHelper.getClient(driver)(initReadWritePluginConfig(instance.host, instance.port));
           await PerfTestUtility.connectWithRetry(client);
+          await client.setReadOnly(true);
+          await client.setReadOnly(false);
         }
       }
 
       const rwPluginWithPoolWithWarmUpData = await measurePerformance(rwPluginWithPoolConfig);
-      await ConnectionProviderManager.releaseResources();
-      ConnectionProviderManager.resetProvider();
 
       readerData = calculateReaderOverhead("Switch to reader", rwPluginWithPoolWithWarmUpData, noPluginData);
       writerData = calculateWriterOverhead("Switch to writer (using cached connection)", rwPluginWithPoolWithWarmUpData, noPluginData);
@@ -117,6 +118,9 @@ describe("rwperformance", () => {
         `ReadWriteSplittingPerformanceResults_${env.engine}_Instances_${env.instances.length}_SwitchReaderWriterConnection.xlsx`,
         "InternalConnectionPoolWithWarmUp"
       );
+
+      await ConnectionProviderManager.releaseResources();
+      ConnectionProviderManager.resetProvider();
     },
     13200000
   );