zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-05 07:10:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Go: add context (#17314)

Browse Source 
### Summary

As title.

---------

Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2026-07-24 16:47:12 +08:00
committed by GitHub
parent 008fa3e10e
commit bdfc3ada41
78 changed files with 1031 additions and 904 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
internal/handler/file.go
View File

@@ -306,7 +306,8 @@ func (h *FileHandler) UploadFile(c *gin.Context) {
}
}
result, err := h.fileService.UploadFile(userID, parentID, files)
ctx := c.Request.Context()
result, err := h.fileService.UploadFile(ctx, userID, parentID, files)
if err != nil {
common.ErrorWithCode(c, common.CodeBadRequest, err.Error())
return
@@ -427,7 +428,8 @@ func (h *FileHandler) MoveFiles(c *gin.Context) {
return
}
success, message := h.fileService.MoveFiles(user.ID, req.SrcFileIDs, req.DestFileID, req.NewName)
ctx := c.Request.Context()
success, message := h.fileService.MoveFiles(ctx, user.ID, req.SrcFileIDs, req.DestFileID, req.NewName)
if !success {
common.ResponseWithCodeData(c, common.CodeBadRequest, nil, message)
return
@@ -482,7 +484,8 @@ func (h *FileHandler) Download(c *gin.Context) {
// If blob is empty, try fallback via file2document
if len(blob) == 0 {
storageAddr, err := h.fileService.GetStorageAddress(fileID)
ctx := c.Request.Context()
storageAddr, err := h.fileService.GetStorageAddress(ctx, fileID)
if err != nil {
common.ResponseWithCodeData(c, common.CodeServerError, nil, "Failed to get file storage address: "+err.Error())
return
@@ -564,8 +567,8 @@ func (h *FileHandler) LinkToDatasets(c *gin.Context) {
common.ResponseWithCodeData(c, common.CodeArgumentError, nil, "mode must be 'add' or 'replace'")
return
}
if err := h.file2DocumentService.LinkToDatasets(user.ID, &req, mode); err != nil {
ctx := c.Request.Context()
if err := h.file2DocumentService.LinkToDatasets(ctx, user.ID, &req, mode); err != nil {
common.ResponseWithCodeData(c, linkToDatasetsErrorCode(err), nil, err.Error())
return
}