Browse Source

fix: task_executor bug fix (#7253)

### 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):
tags/v0.19.0
benni82 6 months ago
parent
commit
216cd7474b
No account linked to committer's email address
1 changed files with 4 additions and 0 deletions
  1. 4
    0
      rag/svr/task_executor.py

+ 4
- 0
rag/svr/task_executor.py View File

@@ -674,6 +674,8 @@ async def report_status():
REDIS_CONN.delete(consumer_name)
except Exception:
logging.exception("report_status got exception")
finally:
redis_lock.release()
await trio.sleep(30)


@@ -702,6 +704,8 @@ def recover_pending_tasks():
stop_event.wait(60)
except Exception:
logging.warning("recover_pending_tasks got exception")
finally:
redis_lock.release()


async def main():

Loading…
Cancel
Save