diff --git a/handler/user.go b/handler/user.go index 25e9bc9..d304dbf 100644 --- a/handler/user.go +++ b/handler/user.go @@ -264,6 +264,10 @@ func registerHandler(c *gin.Context) { return } id := service.CreateUser(req_data.User, req_data.Password, req_data.Email, req_data.Gender, req_data.Age) + if id == 0 { + c.JSON(200, gin.H{"error": "create user error", "code": proto.OperationFailed, "message": "error"}) + return + } // 生成 JWT 令牌 token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.MapClaims{ "username": req_data.User, @@ -273,9 +277,11 @@ func registerHandler(c *gin.Context) { tokenString, err = token.SignedString(signingKey) if err != nil { c.JSON(200, gin.H{"error": err.Error(), "code": proto.TokenGenerationError, "message": "error"}) + return } } else { c.JSON(200, gin.H{"error": err.Error(), "code": proto.DeviceRestartFailed, "message": "error"}) + return } fmt.Println(req_data) res := worker.SetRedisWithExpire(tokenString, tokenString, time.Hour*10) // 设置过期时间为10分钟 @@ -285,4 +291,5 @@ func registerHandler(c *gin.Context) { } // 返回令牌 c.JSON(200, gin.H{"token": tokenString, "username": req_data.User, "code": proto.SuccessCode, "message": "success"}) + return }