fix: 修复session维护过程的bug

This commit is contained in:
Rock Chin 2022-12-08 13:37:33 +08:00
parent b06754506a
commit c9547c32eb
2 changed files with 7 additions and 1 deletions

View File

@ -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

View File

@ -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,