更新用户登录功能

2025-08-10 02:10:01 +08:00
parent a2e5a0c36d
commit 19fea1ca91
2 changed files with 142 additions and 0 deletions
--- a/controllers/user/userLogin.go
+++ b/controllers/user/userLogin.go
@@ -0,0 +1,116 @@
+package user
+
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"net/url"
+	"time"
+	"toutoukan/config"
+	"toutoukan/init/databaseInit"
+	"toutoukan/model/usermodel"
+	"toutoukan/utill"
+)
+
+const wxLoginURL = "https://api.weixin.qq.com/sns/jscode2session"
+
+func UserLogin(c *gin.Context) {
+	fmt.Println("Request Body:", c.Request.Body)
+	var req usermodel.WxLoginM
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+	}
+	fmt.Println("req:123")
+	fmt.Println(req.EncryptedData)
+	fmt.Println(req.Code)
+	fmt.Println(req.Iv)
+	fmt.Println("使用的appid:", config.Conf.WxID)
+	fmt.Println("使用的secret:", config.Conf.Wxsecret)
+	//----------------发送验证请求
+	params := url.Values{}
+	params.Add("appid", config.Conf.WxID)
+	params.Add("secret", config.Conf.Wxsecret)
+	params.Add("js_code", req.Code)
+	params.Add("grant_type", "authorization_code")
+	requestURL := fmt.Sprintf("%s?%s", wxLoginURL, params.Encode())
+	fmt.Println("微信接口请求URL:", requestURL)
+
+	resp, err := http.Get(fmt.Sprintf("%s?%s", wxLoginURL, params.Encode()))
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "调用微信登录接口失败: " + err.Error()})
+		return
+	}
+	fmt.Println("微信登录返回结果：", resp.Body)
+	defer resp.Body.Close()
+
+	// 解析微信返回结果
+	var wxResp usermodel.WxLoginResponse
+	if err := json.NewDecoder(resp.Body).Decode(&wxResp); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "解析微信登录响应失败: " + err.Error(), "code": "10026"})
+		return
+	}
+
+	// 检查微信返回的错误
+	if wxResp.ErrCode != 0 {
+		c.JSON(http.StatusBadRequest, gin.H{ // http.StatusBadRequest 对应 400
+			"error": fmt.Sprintf("微信登录失败: %s (错误码: %d)", wxResp.ErrCode, wxResp.ErrMsg),
+			"code":  "10027",
+		})
+		return
+	}
+
+	// 到这里登录验证成功，获取到了openid和session_key
+	// 可以在这里进行后续处理，如：
+	// 1. 根据openid查询或创建用户
+	// 2. 生成自定义登录态（如token）返回给客户端
+	// 3. （可选）使用session_key、encryptedData和iv解密用户信息
+
+	openid := wxResp.OpenID
+	ctx := c.Request.Context() // 获取请求上下文，用于控制数据库操作超时
+
+	// 1. 查询用户是否存在
+	var exists bool
+	query := "SELECT EXISTS(SELECT 1 FROM user_info WHERE uid = ? LIMIT 1)"
+	err = databaseInit.UserDB.QueryRowContext(ctx, query, openid).Scan(&exists)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{
+			"error": "查询用户存在性失败: " + err.Error(),
+			"code":  "10029",
+		})
+		return
+	}
+
+	// 2. 如果用户不存在，插入新用户
+	if !exists {
+		now := time.Now()
+		insertSQL := `
+            INSERT INTO user_info (
+                uid, gender,createdtime, updatedtime
+            ) VALUES (?, ?, ?,?)
+        `
+		_, err := databaseInit.UserDB.ExecContext(
+			ctx,
+			insertSQL,
+			openid, // uid（使用微信 openid）
+			2,
+			now, // createdtime
+			now, // updatedtime
+		)
+		if err != nil {
+			c.JSON(http.StatusInternalServerError, gin.H{
+				"error": "插入新用户失败: " + err.Error(),
+				"code":  "10029",
+			})
+			return
+		}
+	}
+
+	token, err := utill.GenerateJWTAndStore(openid)
+	if err != nil {
+		c.JSON(500, gin.H{"error": "生成令牌失败", "code": "10036"})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{"result": "success", "error": nil, "code": "20001", "token": token})
+}