Merge pull request #8 from bytemate/feat/add-heartbeat

✨ Add heartbeat
fff455 · Jan 21, 2023 · c2ca61f · c2ca61f
2 parents cf3c003 + 748fcc6
commit c2ca61f
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/chatgpt/chatgpt.service.ts b/src/chatgpt/chatgpt.service.ts
@@ -261,4 +261,19 @@ export class ChatgptService {
       await this.startChatgptInstance(account.email);
     }
   }
+  // Update account status to Running
+  @Cron('1 * * * * *')
+  async heartbeat() {
+    const emails = Array.from(this.chatgptPoolService.accounts);
+    // update db account status
+    this.logger.debug(`Update account status: ${emails}`);
+    this.prismaService.chatGPTAccount.updateMany({
+      where: { email: { in: emails } },
+      data: { status: 'Starting' },
+    });
+    this.prismaService.chatGPTAccount.updateMany({
+      where: { email: { notIn: emails } },
+      data: { status: 'Running' },
+    });
+  }
 }