From a03e6ae11458d787a0ecce29edd5fd25448297ff Mon Sep 17 00:00:00 2001
From: Lei OT <lyt@hainatravel.com>
Date: Wed, 25 Dec 2024 16:42:17 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9C=8D=E5=8A=A1=E5=90=AF=E5=8A=A8,=20?=
 =?UTF-8?q?=E7=99=BB=E5=87=BA=E6=89=80=E6=9C=89=E8=B4=A6=E5=8F=B7=E8=BF=9E?=
 =?UTF-8?q?=E6=8E=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 wai-server/core/services/whatsappHandler.js | 11 +++++++++--
 wai-server/index.js                         |  3 ++-
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/wai-server/core/services/whatsappHandler.js b/wai-server/core/services/whatsappHandler.js
index 763eac1..84c9686 100644
--- a/wai-server/core/services/whatsappHandler.js
+++ b/wai-server/core/services/whatsappHandler.js
@@ -2,7 +2,7 @@ const generateId = require('../../utils/generateId.util');
 const { domain } = require('../../config').server;
 const whatsappEvents = require('../emitter');
 const { callWebhook } = require('../webhook');
-const { addConnection, updateConnection, addCurrentConnection } = require('../../services/connections.service');
+const { addConnection, updateConnection, addCurrentConnection, resetConnection } = require('../../services/connections.service');
 const { objectMapper } = require('../../utils/commons.util');
 const { sessionService } = require('..');
 
@@ -100,4 +100,11 @@ function setupWhatsappHandler() {
   setupMessageHandler();
 }
 
-module.exports = { setupWhatsappHandler };
+/**
+ * 登出: 当前服务的所有连接
+ */
+async function resetCurrentConnection() {
+  await resetConnection();
+}
+
+module.exports = { setupWhatsappHandler, resetCurrentConnection };
diff --git a/wai-server/index.js b/wai-server/index.js
index 5ce34fb..de61271 100644
--- a/wai-server/index.js
+++ b/wai-server/index.js
@@ -5,7 +5,7 @@ const server = require('./server');
 
 const { port } = require('./config').server;
 
-const { setupWhatsappHandler } = require('./core/services/whatsappHandler');
+const { setupWhatsappHandler, resetCurrentConnection } = require('./core/services/whatsappHandler');
 
 async function bootstrap() {
   /**
@@ -15,6 +15,7 @@ async function bootstrap() {
    */
 
   setupWhatsappHandler();
+  await resetCurrentConnection();
 
   return http.createServer(server.callback()).listen(port, '0.0.0.0');
 }