luojunhui 2 tygodni temu
rodzic
commit
5376b8e70d

+ 3 - 1
applications/async_task/auto_rechunk_task.py

@@ -1,3 +1,5 @@
+from tqdm.asyncio import tqdm
+
 from applications.utils.http import AsyncHttpClient
 
 
@@ -23,7 +25,7 @@ class AutoRechunkTask:
 
     async def deal(self):
         tasks = await self.get_tasks()
-        for task in tasks:
+        for task in tqdm(tasks, desc="rechunk each doc"):
             await self.rechunk(task)
 
         return len(tasks)

+ 2 - 2
routes/buleprint.py

@@ -475,5 +475,5 @@ async def chunk_list():
 async def auto_rechunk():
     resource = get_resource_manager()
     auto_rechunk_task = AutoRechunkTask(mysql_client=resource.mysql_client)
-    res = await auto_rechunk_task.deal()
-    return jsonify({"status_code": 200, "detail": "success", "cnt": len(res)})
+    process_cnt = await auto_rechunk_task.deal()
+    return jsonify({"status_code": 200, "detail": "success", "cnt": process_cnt})