feat(token): 添加Token存在性检查

This commit is contained in:
2025-09-18 20:32:19 +08:00
parent 2c8e25bdf8
commit 42cfe0dc0f
2 changed files with 35 additions and 0 deletions

View File

@@ -26,6 +26,11 @@ func CreateTokenHandler(c *gin.Context) {
c.JSON(http.StatusBadRequest, gin.H{"error": "参数不能为空 " + err.Error()})
return
}
//检查Token是否存在
if db.CheckToken(input.Token) {
c.JSON(http.StatusBadRequest, gin.H{"error": "创建Token失败Token已经存在"})
return
}
//创建Token
err := db.CreateToken(input.Token, input.DedupObject, input.DataFormat, input.Notes)
@@ -49,6 +54,11 @@ func UpdateTokenHandler(c *gin.Context) {
c.JSON(http.StatusBadRequest, gin.H{"error": "参数不能为空 " + err.Error()})
return
}
//检查Token是否存在
if !db.CheckToken(input.Token) {
c.JSON(http.StatusBadRequest, gin.H{"error": "更改失败Token不存在"})
return
}
err := db.UpdateToken(input.Token, input.DedupObject, input.DataFormat, input.Notes)
if err != nil {
@@ -67,6 +77,11 @@ func DeleteTokenHandler(c *gin.Context) {
c.JSON(http.StatusBadRequest, gin.H{"error": "参数不能为空 " + err.Error()})
return
}
//检查Token是否存在
if !db.CheckToken(input.Token) {
c.JSON(http.StatusBadRequest, gin.H{"error": "删除Token失败Token不存在"})
return
}
err := db.DeleteToken(input.Token)
if err != nil {
@@ -85,6 +100,11 @@ func GetTokenInfoHandler(c *gin.Context) {
c.JSON(http.StatusBadRequest, gin.H{"error": "Token不能为空"})
return
}
//检查Token是否存在
if !db.CheckToken(input.Token) {
c.JSON(http.StatusBadRequest, gin.H{"error": "获取信息失败Token不存在"})
return
}
dedupObject, err := db.GetDedupObject(input.Token)
if err != nil {
@@ -125,6 +145,11 @@ func DeleteTokenInfoHandler(c *gin.Context) {
c.JSON(http.StatusBadRequest, gin.H{"error": "Token不能为空"})
return
}
//检查Token是否存在
if !db.CheckToken(input.Token) {
c.JSON(http.StatusBadRequest, gin.H{"error": "删除Token失败Token不存在"})
return
}
//检查token是否存在
_, err := db.GetDedupObject(input.Token)