diff --git a/handler/user.go b/handler/user.go index 7079247..d8ea2c2 100644 --- a/handler/user.go +++ b/handler/user.go @@ -58,7 +58,7 @@ func SetQRStatus(c *gin.Context) { var qrsetReq QRReq if err := c.ShouldBind(&qrsetReq); err == nil || qrsetReq.UUID != "" { if worker.IsContainKey(qrsetReq.UUID) == false { - c.JSON(200, gin.H{"code": 18, "message": "uuid not found", "data": "0"}) + c.JSON(200, gin.H{"code": 18, "message": "uuid not found in server", "data": "0"}) return } res := worker.SetHashWithField(qrsetReq.UUID, "status", "1") @@ -85,6 +85,10 @@ func ConfirmQRLogin(c *gin.Context) { if token == "" { c.JSON(200, gin.H{"code": 20, "message": "Token不存在", "data": "20"}) } + if worker.IsContainKey(qrsetReq.UUID) == false { + c.JSON(200, gin.H{"code": 18, "message": "uuid not found in server", "data": "0"}) + return + } if worker.SetHashWithField(qrsetReq.UUID, "status", token) { c.JSON(200, gin.H{"code": 0, "message": "success", "data": "0"}) } else {