From cae387704839a4769ccf400c7fb3043c28bc5422 Mon Sep 17 00:00:00 2001
From: heimoshuiyu <heimoshuiyu@gmail.com>
Date: Fri, 18 Oct 2024 16:59:43 +0800
Subject: [PATCH] =?UTF-8?q?risk-analyze=20main=20=E5=BE=AA=E7=8E=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cucyuqing/cmd/risk-analyze.py | 19 ++++++++++++++++---
 1 file changed, 16 insertions(+), 3 deletions(-)

diff --git a/cucyuqing/cmd/risk-analyze.py b/cucyuqing/cmd/risk-analyze.py
index 962e573..d431e01 100644
--- a/cucyuqing/cmd/risk-analyze.py
+++ b/cucyuqing/cmd/risk-analyze.py
@@ -1,7 +1,5 @@
 from dataclasses import dataclass
-from os import system
-from typing import Iterable, Required
-from typing_extensions import Doc
+from typing import Iterable
 import openai
 import asyncio
 from openai.types.chat import ChatCompletionMessageParam
@@ -16,6 +14,16 @@ from cucyuqing.dbscan import Document, run_dbscan
 
 
 async def main():
+    while True:
+        try:
+            await do_analyze()
+            await asyncio.sleep(60 * 30)
+        except Exception as e:
+            print(e)
+            await asyncio.sleep(60*60)
+
+
+async def do_analyze():
     await asyncio.gather(
         pool.open(),
         mysql.connect(),
@@ -54,6 +62,11 @@ async def main():
             },
         )
 
+    await asyncio.gather(
+        pool.close(),
+        mysql.disconnect(),
+    )
+
 
 @dataclass
 class RiskType: