ygxbnet/dypid
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(token): 修改参数验证和删除逻辑

Browse Source
This commit is contained in:
YGXB_net
2025-09-10 17:03:09 +08:00
parent 1e4c2d540e
commit 861ac5d365
2 changed files with 54 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
controller/tokenController.go
View File

@@ -3,7 +3,9 @@ package controller
import (
"dypid/db"
"dypid/global"
"fmt"
"net/http"
"strconv"
"github.com/gin-gonic/gin"
)
@@ -18,7 +20,7 @@ func CreateTokenHandler(c *gin.Context) {
Token string `form:"token" binding:"required"`
DedupObject string `form:"dedup_object" binding:"required"`
DataFormat string `form:"data_format" binding:"required"`
Notes string `form:"notes" binding:"required"`
Notes string `form:"notes"`
}{}
if err := c.ShouldBindQuery(&input); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "参数不能为空 " + err.Error()})
@@ -41,7 +43,7 @@ func UpdateTokenHandler(c *gin.Context) {
Token string `form:"token" binding:"required"`
DedupObject string `form:"dedup_object" binding:"required"`
DataFormat string `form:"data_format" binding:"required"`
Notes string `form:"notes" binding:"required"`
Notes string `form:"notes"`
}{}
if err := c.ShouldBindQuery(&input); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "参数不能为空 " + err.Error()})
@@ -115,8 +117,8 @@ func DeleteTokenInfoHandler(c *gin.Context) {
//解析输入数据
input := struct {
Token string `form:"token" binding:"required"`
DedupBF bool `form:"dedup_bf"`
CacheList bool `form:"cache_list"`
DedupBF string `form:"dedup_bf"`
CacheList string `form:"cache_list"`
}{}
if err := c.ShouldBindQuery(&input); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Token不能为空"})
@@ -131,12 +133,30 @@ func DeleteTokenInfoHandler(c *gin.Context) {
}
//删除去重对象
if input.DedupBF {
switch input.DedupBF {
case "":
case "all":
keys := global.RDB.Keys(global.RCtx, "dedup:"+input.Token+":*").Val()
global.RDB.Del(global.RCtx, keys...)
default:
_, err := strconv.Atoi(input.DedupBF)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "dedup_bf设置错误 " + err.Error()})
return
}
}
if input.CacheList {
switch input.CacheList {
case "":
case "all":
global.RDB.Del(global.RCtx, "list:"+input.Token)
default:
i, err := strconv.Atoi(input.CacheList)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "cache_list设置错误 " + err.Error()})
return
}
fmt.Println(-i)
global.RDB.LTrim(global.RCtx, "list:"+input.Token, 1, int64(-i))
}
//输出信息