From 36e0fdb7974568ca85fc751d96d90f545a315f79 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Mon, 8 Jul 2024 10:17:20 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=B3=A8=E5=86=8C=E8=BF=94?=
 =?UTF-8?q?=E5=9B=9E=E6=95=B0=E6=8D=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 handler/user.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/handler/user.go b/handler/user.go
index d304dbf..35aa2c2 100644
--- a/handler/user.go
+++ b/handler/user.go
@@ -253,6 +253,7 @@ func loginHandler(c *gin.Context) {
 func registerHandler(c *gin.Context) {
 	var req_data RLReq
 	tokenString := ""
+	var id uint
 	if err := c.ShouldBindJSON(&req_data); err == nil {
 		if len(req_data.Password) != 32 {
 			hasher := md5.New()
@@ -263,7 +264,7 @@ func registerHandler(c *gin.Context) {
 			c.JSON(200, gin.H{"error": "user already exists", "code": proto.UsernameExists, "message": "error"})
 			return
 		}
-		id := service.CreateUser(req_data.User, req_data.Password, req_data.Email, req_data.Gender, req_data.Age)
+		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
@@ -290,6 +291,11 @@ func registerHandler(c *gin.Context) {
 		return
 	}
 	// 返回令牌
-	c.JSON(200, gin.H{"token": tokenString, "username": req_data.User, "code": proto.SuccessCode, "message": "success"})
+	data := make(map[string]interface{})
+	data["id"] = id
+	data["username"] = req_data.User
+	data["email"] = req_data.Email
+	data["token"] = tokenString
+	c.JSON(200, gin.H{"code": proto.SuccessCode, "message": "success", "date": data})
 	return
 }