nyne/nysoure
1
0
Fork 0
mirror of https://github.com/wgh136/nysoure.git synced 2025-09-27 04:17:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

enhance file download handling to track real users

Browse Source
This commit is contained in:
nyne
2025-09-01 17:51:06 +08:00
parent 44b876ba0e
commit b804741e27
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
server/api/file.go
View File

@@ -205,7 +205,7 @@ func deleteFile(c fiber.Ctx) error {
func downloadFile(c fiber.Ctx) error { func downloadFile(c fiber.Ctx) error {
cfToken := c.Query("cf_token") cfToken := c.Query("cf_token")
s, filename, err := service.DownloadFile(c.Params("id"), cfToken) s, filename, err := service.DownloadFile(c.Params("id"), cfToken, c.Locals("real_user") == true)
if err != nil { if err != nil {
return err return err
} }

10
server/service/file.go
View File

@@ -389,7 +389,7 @@ func GetFile(fid string) (*model.FileView, error) {
} }
// DownloadFile handles the file download request. Return a presigned URL or a direct file path. // DownloadFile handles the file download request. Return a presigned URL or a direct file path.
func DownloadFile(fid, cfToken string) (string, string, error) { func DownloadFile(fid, cfToken string, isRealUser bool) (string, string, error) {
file, err := dao.GetFile(fid) file, err := dao.GetFile(fid)
if err != nil { if err != nil {
log.Error("failed to get file: ", err) log.Error("failed to get file: ", err)
@@ -436,9 +436,11 @@ func DownloadFile(fid, cfToken string) (string, string, error) {
return "", "", model.NewInternalServerError("failed to download file from storage") return "", "", model.NewInternalServerError("failed to download file from storage")
} }
err = dao.AddResourceDownloadCount(file.ResourceID) if isRealUser {
if err != nil { err = dao.AddResourceDownloadCount(file.ResourceID)
log.Errorf("failed to add resource download count: %v", err) if err != nil {
log.Errorf("failed to add resource download count: %v", err)
}
} }
return path, file.Filename, nil return path, file.Filename, nil