Stop the stress clients when the stress test is over (#652)

river-build · Aug 6, 2024 · 162f0d6 · 162f0d6
1 parent c4f0e70
commit 162f0d6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/packages/stress/src/mode/chat/root_chat.ts b/packages/stress/src/mode/chat/root_chat.ts
@@ -163,6 +163,12 @@ export async function startStressChat(opts: {
 
     logger.log('done', { summary })
 
+    for (let i = 0; i < clients.length; i += 1) {
+        const client = clients[i]
+        logger.log(`stopping ${client.logId}`)
+        await client.stop()
+    }
+
     return { summary, chatConfig, opts }
 }
 

diff --git a/packages/stress/src/utils/stressClient.ts b/packages/stress/src/utils/stressClient.ts
@@ -88,7 +88,9 @@ export class StressClient {
         public rpcClient: StreamRpcClient,
         public spaceDapp: SpaceDapp,
         public streamsClient: StreamsClient,
-    ) {}
+    ) {
+        logger.log('StressClient', { clientIndex, userId, logId: this.logId })
+    }
 
     get logId(): string {
         return `client${this.clientIndex}:${shortenHexString(this.userId)}`