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

handle_task catch all exception (#3441)

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

handle_task catch all exception
Report heartbeats

### Type of change

- [x] Refactoring
This commit is contained in:
Zhichang Yu
2024-11-15 18:51:09 +08:00
committed by GitHub
parent 1e90a1bf36
commit 4ed5ca2666
4 changed files with 129 additions and 131 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
rag/utils/redis_conn.py
View File

@ -225,14 +225,16 @@ class RedisDB:
logging.exception("xpending_range: " + consumer_name + " got exception")
self.__open__()
def queue_length(self, queue) -> int:
def queue_info(self, queue, group_name) -> dict:
for _ in range(3):
try:
num = self.REDIS.xlen(queue)
return num
groups = self.REDIS.xinfo_groups(queue)
for group in groups:
if group["name"] == group_name:
return group
except Exception:
logging.exception("queue_length" + str(queue) + " got exception")
return 0
return None
def queue_head(self, queue) -> int:
for _ in range(3):