From c9547c32eb00581340a80e1acf221543588cfaaa Mon Sep 17 00:00:00 2001 From: Rock Chin Date: Thu, 8 Dec 2022 13:37:33 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8Dsession=E7=BB=B4?= =?UTF-8?q?=E6=8A=A4=E8=BF=87=E7=A8=8B=E7=9A=84bug?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 6 ++++++ pkg/database/manager.py | 2 +- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 0b5acf64..3072748d 100644 --- a/main.py +++ b/main.py @@ -53,5 +53,11 @@ if __name__ == '__main__': try: time.sleep(86400) except KeyboardInterrupt: + try: + for session in pkg.openai.session.sessions: + pkg.openai.session.sessions[session].persistence() + except Exception as e: + if not isinstance(e, KeyboardInterrupt): + raise e print("程序退出") break diff --git a/pkg/database/manager.py b/pkg/database/manager.py index 1aca8d2b..2fde0fbb 100644 --- a/pkg/database/manager.py +++ b/pkg/database/manager.py @@ -35,8 +35,8 @@ class DatabaseManager: def heartbeat(self): while True: - self.conn.ping(reconnect=True) time.sleep(30) + self.conn.ping(reconnect=True) def reconnect(self): self.conn = pymysql.connect(host=self.host, port=self.port, user=self.user, password=self.password,