From 8db7a7fa188977a25526373945867f405e6fa669 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Tue, 2 Jul 2024 15:55:35 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=B3=A8=E5=86=8C=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 handler/user.go | 7 +++++++
 1 file changed, 7 insertions(+)

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
 }