chore: uncomment writer switch time calculations

aws · Oct 23, 2024 · 8de1c7a · 8de1c7a
1 parent 6bdbdca
commit 8de1c7a
Showing 1 changed file with 19 additions and 19 deletions.
diff --git a/tests/integration/container/tests/read_write_splitting_performance.test.ts b/tests/integration/container/tests/read_write_splitting_performance.test.ts
@@ -105,7 +105,7 @@ async function measurePerformance(config: any): Promise<Result> {
   const elapsedSwitchToReaderTimeNanos: bigint[] = [];
   const elapsedSwitchToWriterTimeNanos: bigint[] = [];
 
-  for (let i = 0; i < 1; i++) {
+  for (let i = 0; i < REPEAT_TIMES; i++) {
     logger.info(`Test iteration ${i}`);
     const client = DriverHelper.getClient(driver)(config);
     try {
@@ -127,19 +127,19 @@ async function measurePerformance(config: any): Promise<Result> {
       logger.info(`connect time ${connectTime}, execution time ${executionTime}, time1 ${time1}: ${time1 - connectTime - executionTime}`);
       elapsedSwitchToReaderTimeNanos.push(time1 - connectTime - executionTime);
 
-      // // Calculate time required to switch to an existing writer connection.
-      // ConnectTimePlugin.resetConnectTime();
-      // ExecuteTimePlugin.resetExecuteTime();
-      //
-      // switchToWriterStartTime = getTimeInNanos();
-      // await client.setReadOnly(false);
-      //
-      // connectTime = ConnectTimePlugin.getTotalConnectTime();
-      // executionTime = ExecuteTimePlugin.getTotalExecuteTime();
-      // elapsedSwitchToWriterTimeNanos.push(getTimeInNanos() - switchToWriterStartTime - connectTime - executionTime);
-      // const time2 = getTimeInNanos() - switchToWriterStartTime;
-      // logger.info(`connect time ${connectTime}, execution time ${executionTime}, time2 ${time1}: ${time2 - connectTime - executionTime}`);
-      // elapsedSwitchToReaderTimeNanos.push(time2 - connectTime - executionTime);
+      // Calculate time required to switch to an existing writer connection.
+      ConnectTimePlugin.resetConnectTime();
+      ExecuteTimePlugin.resetExecuteTime();
+
+      switchToWriterStartTime = getTimeInNanos();
+      await client.setReadOnly(false);
+
+      connectTime = ConnectTimePlugin.getTotalConnectTime();
+      executionTime = ExecuteTimePlugin.getTotalExecuteTime();
+      elapsedSwitchToWriterTimeNanos.push(getTimeInNanos() - switchToWriterStartTime - connectTime - executionTime);
+      const time2 = getTimeInNanos() - switchToWriterStartTime;
+      logger.info(`connect time ${connectTime}, execution time ${executionTime}, time2 ${time1}: ${time2 - connectTime - executionTime}`);
+      elapsedSwitchToReaderTimeNanos.push(time2 - connectTime - executionTime);
     } finally {
       await client.end();
     }
@@ -152,11 +152,11 @@ async function measurePerformance(config: any): Promise<Result> {
   data.switchToReaderMax = max;
   data.switchToReaderAvg = avg;
 
-  // [min, max, sum] = calculateStats(elapsedSwitchToWriterTimeNanos);
-  // avg = sum / BigInt(elapsedSwitchToWriterTimeNanos.length);
-  // data.switchToWriterMin = min;
-  // data.switchToWriterMax = max;
-  // data.switchToWriterAvg = avg;
+  [min, max, sum] = calculateStats(elapsedSwitchToWriterTimeNanos);
+  avg = sum / BigInt(elapsedSwitchToWriterTimeNanos.length);
+  data.switchToWriterMin = min;
+  data.switchToWriterMax = max;
+  data.switchToWriterAvg = avg;
 
   return data;
 }