gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: task_executor bug fix (#7253)

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

The lock is not released correctly when task_exectuor is abnormal

### Type of change

- [ ] Bug Fix (non-breaking change which fixes an issue)
- [ ] New Feature (non-breaking change which adds functionality)
- [ ] Documentation Update
- [ ] Refactoring
- [ ] Performance Improvement
- [ ] Other (please describe):
This commit is contained in:
benni82 2025-04-24 11:44:34 +08:00 • committed by GitHub
parent 2c62652ea8
commit 216cd7474b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
rag/svr/task_executor.py
View file

@ -674,6 +674,8 @@ async def report_status():
REDIS_CONN.delete(consumer_name) REDIS_CONN.delete(consumer_name)
except Exception: except Exception:
logging.exception("report_status got exception") logging.exception("report_status got exception")
finally:
redis_lock.release()
await trio.sleep(30) await trio.sleep(30)
@ -702,6 +704,8 @@ def recover_pending_tasks():
stop_event.wait(60) stop_event.wait(60)
except Exception: except Exception:
logging.warning("recover_pending_tasks got exception") logging.warning("recover_pending_tasks got exception")
finally:
redis_lock.release()
async def main(): async def main():