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: refactor (#17511)

Browse Source 
Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2026-07-29 18:00:59 +08:00
committed by GitHub
parent 317363e513
commit 1f90755c48
11 changed files with 116 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
internal/admin/handler_ee.go
View File

@@ -887,7 +887,9 @@ func (h *Handler) ShowUserSummary(c *gin.Context) {
return
}
userSummary, err := h.service.ShowUserSummary(username)
ctx := c.Request.Context()
userSummary, err := h.service.ShowUserSummary(ctx, username)
if err != nil {
if errors.Is(err, common.ErrUserNotFound) {
common.ErrorWithCode(c, common.CodeNotFound, "User not found")
@@ -927,7 +929,9 @@ func (h *Handler) ShowUserQuota(c *gin.Context) {
return
}
userQuota, err := h.service.ShowUserQuota(username)
ctx := c.Request.Context()
userQuota, err := h.service.ShowUserQuota(ctx, username)
if err != nil {
if errors.Is(err, common.ErrUserNotFound) {
common.ErrorWithCode(c, common.CodeNotFound, "User not found")
@@ -1923,7 +1927,9 @@ func (h *Handler) GetTokenStats(c *gin.Context) {
}
granularity = strings.ToLower(granularity)
stats, err := h.service.GetTokenStats(userName, fromDate, toDate, granularity)
ctx := c.Request.Context()
stats, err := h.service.GetTokenStats(ctx, userName, fromDate, toDate, granularity)
if err != nil {
common.ErrorWithCode(c, common.CodeDataError, err.Error())
return
@@ -1990,8 +1996,9 @@ func (h *Handler) ListLogs(c *gin.Context) {
common.ErrorWithCode(c, common.CodeBadRequest, "Invalid days")
return
}
ctx := c.Request.Context()
stats, err := h.service.ListLogs(userName, daysInt)
stats, err := h.service.ListLogs(ctx, userName, daysInt)
if err != nil {
common.ErrorWithCode(c, common.CodeDataError, err.Error())
return