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

Go: context, part8 (#17405)

Browse Source 
Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2026-07-27 13:38:15 +08:00
committed by GitHub
parent c48eb70e67
commit f5aa5f7d94
5 changed files with 67 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
internal/handler/memory.go
View File

@@ -130,9 +130,10 @@ func (h *MemoryHandler) CreateMemory(c *gin.Context) {
// Record request parsing completion time (for timing)
tParsed := time.Now()
ctx := c.Request.Context()
// Call service layer to create memory
result, err := h.memoryService.CreateMemory(userID, &req)
result, err := h.memoryService.CreateMemory(ctx, userID, &req)
if err != nil {
// Log error if timing is enabled
if timingEnabled != "" {
@@ -218,8 +219,10 @@ func (h *MemoryHandler) UpdateMemory(c *gin.Context) {
return
}
ctx := c.Request.Context()
// Call service layer to update memory
result, err := h.memoryService.UpdateMemory(userID, memoryID, &req)
result, err := h.memoryService.UpdateMemory(ctx, userID, memoryID, &req)
if err != nil {
errMsg := err.Error()
// Check if it's a "not found" error
@@ -261,9 +264,10 @@ func (h *MemoryHandler) DeleteMemory(c *gin.Context) {
common.ResponseWithCodeData(c, common.CodeArgumentError, nil, "memory ID is required")
return
}
ctx := c.Request.Context()
// Call service layer to delete memory
err := h.memoryService.DeleteMemory(memoryID)
err := h.memoryService.DeleteMemory(ctx, memoryID)
if err != nil {
errMsg := err.Error()
// Check if it's a "not found" error
@@ -351,8 +355,10 @@ func (h *MemoryHandler) ListMemories(c *gin.Context) {
}
}
ctx := c.Request.Context()
// Call service layer to get memory list
result, err := h.memoryService.ListMemories(user.ID, tenantIDs, memoryTypes, storageType, keywords, page, pageSize)
result, err := h.memoryService.ListMemories(ctx, user.ID, tenantIDs, memoryTypes, storageType, keywords, page, pageSize)
if err != nil {
common.ResponseWithCodeData(c, common.CodeServerError, nil, err.Error())
return
@@ -380,9 +386,10 @@ func (h *MemoryHandler) GetMemoryConfig(c *gin.Context) {
common.ResponseWithCodeData(c, common.CodeArgumentError, nil, "memory ID is required")
return
}
ctx := c.Request.Context()
// Call service layer to get memory configuration
result, err := h.memoryService.GetMemoryConfig(memoryID)
result, err := h.memoryService.GetMemoryConfig(ctx, memoryID)
if err != nil {
errMsg := err.Error()
// Check if it's a "not found" error