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

Fix: code review (#17442)

Browse Source
This commit is contained in:
Lynn
2026-07-28 09:47:22 +08:00
committed by GitHub
parent 8b793ee148
commit 15f8ef7409
4 changed files with 49 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
internal/handler/memory.go
View File

@@ -264,10 +264,17 @@ func (h *MemoryHandler) DeleteMemory(c *gin.Context) {
common.ResponseWithCodeData(c, common.CodeArgumentError, nil, "memory ID is required")
return
}
// Get current logged-in user for access control
user, errorCode, errorMessage := GetUser(c)
if errorCode != common.CodeSuccess {
common.ErrorWithCode(c, errorCode, errorMessage)
return
}
ctx := c.Request.Context()
// Call service layer to delete memory
err := h.memoryService.DeleteMemory(ctx, memoryID)
// Call service layer to delete memory (with access control)
err := h.memoryService.DeleteMemory(ctx, user.ID, memoryID)
if err != nil {
errMsg := err.Error()
// Check if it's a "not found" error
@@ -386,10 +393,17 @@ func (h *MemoryHandler) GetMemoryConfig(c *gin.Context) {
common.ResponseWithCodeData(c, common.CodeArgumentError, nil, "memory ID is required")
return
}
// Get current logged-in user for access control
user, errorCode, errorMessage := GetUser(c)
if errorCode != common.CodeSuccess {
common.ErrorWithCode(c, errorCode, errorMessage)
return
}
ctx := c.Request.Context()
// Call service layer to get memory configuration
result, err := h.memoryService.GetMemoryConfig(ctx, memoryID)
// Call service layer to get memory configuration (with access control)
result, err := h.memoryService.GetMemoryConfig(ctx, user.ID, memoryID)
if err != nil {
errMsg := err.Error()
// Check if it's a "not found" error