zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-12 20:03:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Go: fix context (#18076)

Browse Source 
Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2026-08-11 11:54:57 +08:00
committed by GitHub
parent 9f0663d4d0
commit fc44d2fe3f
4 changed files with 260 additions and 188 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
internal/admin/handler.go
View File

@@ -1016,9 +1016,8 @@ func (h *Handler) RemoveIngestionTasks(c *gin.Context) {
common.ErrorWithCode(c, common.CodeBadRequest, "Task ID is required")
return
}
ctx := c.Request.Context()
if req.Email == nil && req.Status == nil {
ctx := c.Request.Context()
tasks, err := h.service.RemoveIngestionTasks(ctx, req.Tasks)
if err != nil {
common.ErrorWithCode(c, handler.IngestionTaskErrorCode(err), err.Error())
@@ -1027,7 +1026,7 @@ func (h *Handler) RemoveIngestionTasks(c *gin.Context) {
common.SuccessWithData(c, tasks, "Remove tasks successfully")
} else {
tasks, err := h.service.RemoveIngestionTasksByCondition(req.Tasks, req.Email, req.Status)
tasks, err := h.service.RemoveIngestionTasksByCondition(ctx, req.Tasks, req.Email, req.Status)
if err != nil {
common.ErrorWithCode(c, handler.IngestionTaskErrorCode(err), err.Error())
return
@@ -1048,9 +1047,8 @@ func (h *Handler) StopIngestionTasks(c *gin.Context) {
common.ErrorWithCode(c, common.CodeBadRequest, "Task ID is required")
return
}
ctx := c.Request.Context()
if req.Email == nil && req.Status == nil {
ctx := c.Request.Context()
tasks, err := h.service.StopIngestionTasks(ctx, req.Tasks)
if err != nil {
common.ErrorWithCode(c, handler.IngestionTaskErrorCode(err), err.Error())
@@ -1066,7 +1064,7 @@ func (h *Handler) StopIngestionTasks(c *gin.Context) {
common.SuccessWithData(c, result, "Stop tasks successfully")
} else {
tasks, err := h.service.StopIngestionTasksByCondition(req.Tasks, req.Email, req.Status)
tasks, err := h.service.StopIngestionTasksByCondition(ctx, req.Tasks, req.Email, req.Status)
if err != nil {
common.ErrorWithCode(c, common.CodeBadRequest, err.Error())
return