pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

support monitoring task executor (#2069)

Browse Source 
### What problem does this PR solve?
#1383

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
Kevin Hu
2024-08-23 15:55:58 +08:00
committed by GitHub
parent 19396998eb
commit cc6a48b128
2 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
rag/svr/task_executor.py
View File

@ -379,11 +379,12 @@ def report_status():
while True:
try:
obj = REDIS_CONN.get("TASKEXE")
obj = json.load(obj)
if not obj: obj = {}
else: obj = json.load(obj)
if id not in obj: obj[id] = []
obj[id].append(timer()*1000)
obj[id] = obj[id][:-60]
REDIS_CONN.set_obj("TASKEXE", obj)
obj[id] = obj[id][-60:]
REDIS_CONN.set_obj("TASKEXE", obj, 60*2)
except Exception as e:
print("[Exception]:", str(e))
time.sleep(60)