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

Perf: make do_cancel quicker. (#8846)

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

### Type of change

- [x] Performance Improvement
This commit is contained in:
Kevin Hu
2025-07-15 14:35:00 +08:00
committed by GitHub
parent 5fa6f2f151
commit 24c41d2a61
3 changed files with 36 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rag/svr/task_executor.py
View File

@ -214,7 +214,7 @@ async def collect():
canceled = False
task = TaskService.get_task(msg["id"])
if task:
canceled = TaskService.do_cancel(task["id"])
canceled = DocumentService.do_cancel(task["doc_id"])
if not task or canceled:
state = "is unknown" if not task else "has been cancelled"
FAILED_TASKS += 1
@ -382,7 +382,7 @@ async def build_chunks(task, progress_callback):
docs_to_tag = []
for d in docs:
task_canceled = TaskService.do_cancel(task["id"])
task_canceled = DocumentService.do_cancel(task["doc_id"])
if task_canceled:
progress_callback(-1, msg="Task has been canceled.")
return
@ -531,7 +531,7 @@ async def do_handle_task(task):
progress_callback(-1, msg=error_message)
raise Exception(error_message)
task_canceled = TaskService.do_cancel(task_id)
task_canceled = DocumentService.do_cancel(task_doc_id)
if task_canceled:
progress_callback(-1, msg="Task has been canceled.")
return
@ -609,7 +609,7 @@ async def do_handle_task(task):
for b in range(0, len(chunks), DOC_BULK_SIZE):
doc_store_result = await trio.to_thread.run_sync(lambda: settings.docStoreConn.insert(chunks[b:b + DOC_BULK_SIZE], search.index_name(task_tenant_id), task_dataset_id))
task_canceled = TaskService.do_cancel(task_id)
task_canceled = DocumentService.do_cancel(task_doc_id)
if task_canceled:
progress_callback(-1, msg="Task has been canceled.")
return