u

code/allapp/allapp-go-v3/internal/handle/auth.go

@@ -0,0 +1,201 @@
+package handle
+
+import (
+	"allapp-go/internal/errors"
+	"allapp-go/internal/httpx"
+	"allapp-go/internal/types"
+	"allapp-go/pkg/db"
+	"allapp-go/pkg/jwtx"
+	"allapp-go/pkg/uniqueid"
+	"allapp-go/pkg/wechat"
+	"fmt"
+
+	"github.com/gofiber/fiber/v3"
+	"github.com/gofiber/fiber/v3/log"
+)
+
+// LoginQq ======================== QQ 登录 ========================
+func LoginQq(c fiber.Ctx) error {
+	var req types.LoginQqDTO
+	vo := new(types.LoginVO)
+
+	if err := httpx.BindAndValidate(c, &req); err != nil {
+		return errors.WithStack(err)
+	}
+
+	return handleThirdLogin(c, req.Openid, req.Nickname, req.Avatar, 1, vo)
+}
+
+// LoginWechat ======================== 微信登录 ========================
+func LoginWechat(c fiber.Ctx) error {
+	var req types.LoginWechatDTO
+	vo := new(types.LoginVO)
+
+	if err := httpx.BindAndValidate(c, &req); err != nil {
+		return errors.WithStack(err)
+	}
+
+	openid, token, err := wechat.GetWechatAccess(req.Code)
+	if err != nil {
+		log.Errorw("wechat access failed", "code", req.Code, "error", err)
+		return httpx.Fail(c, "微信登录失败，请重试")
+	}
+
+	nickname, avatar, err := wechat.GetWechatUserInfo(token, openid)
+	if err != nil {
+		log.Errorw("wechat userinfo failed", "openid", openid, "error", err)
+		return httpx.Fail(c, "微信登录失败，请重试")
+	}
+
+	return handleThirdLogin(c, openid, nickname, avatar, 0, vo)
+}
+
+// ======================== 第三方登录主流程 ========================
+func handleThirdLogin(
+	c fiber.Ctx,
+	openid string,
+	nickname string,
+	avatar string,
+	loginType int16,
+	vo *types.LoginVO,
+) error {
+
+	dbClient := db.New()
+
+	user, err := getUserByOpenID(dbClient, c, openid, loginType)
+	if err != nil {
+		return errors.WithStack(err)
+	}
+
+	// 不存在 -> 注册
+	if user == nil {
+		return registerUser(dbClient, c, openid, nickname, avatar, loginType, vo)
+	}
+
+	// ========= 校验状态 =========
+	id := toInt64(user["id"])
+	status := toInt64(user["status"])
+
+	if status == 0 {
+		return httpx.Fail(c, "账号已被禁用")
+	}
+
+	// ========= 更新登录时间 =========
+	if err := dbClient.Update(
+		c.Context(),
+		"user",
+		"id",
+		map[string]any{
+			"id":              id,
+			"last_login_time": "NOW()",
+		},
+	); err != nil {
+		return errors.WithStack(err)
+	}
+
+	// ========= token =========
+	token, err := jwtx.CreateToken(c.Context(), map[string]any{
+		"id": id,
+	})
+	if err != nil {
+		return errors.WithStack(err)
+	}
+
+	// ========= 返回 =========
+	vo.Token = token
+	vo.UserId = id
+	vo.Nickname, _ = user["nickname"].(string)
+	vo.Avatar, _ = user["avatar"].(string)
+
+	return httpx.OK(c, vo)
+}
+
+// ======================== 注册 ========================
+func registerUser(
+	dbClient *db.Client,
+	c fiber.Ctx,
+	openid, nickname, avatar string,
+	loginType int16,
+	vo *types.LoginVO,
+) error {
+	userId := uniqueid.NextId()
+
+	if nickname == "" {
+		nickname = fmt.Sprintf("用户_%d", userId)
+	}
+	if avatar == "" {
+		avatar = "https://default-avatar-url.com/default.png"
+	}
+
+	err := dbClient.WithTx(c.Context(), func(tx *db.Client) error {
+
+		if err := tx.Insert(c.Context(), "user", "id", map[string]any{
+			"id":              userId,
+			"nickname":        nickname,
+			"avatar":          avatar,
+			"status":          1,
+			"create_time":     "NOW()",
+			"update_time":     "NOW()",
+			"last_login_time": "NOW()",
+		}); err != nil {
+			return err
+		}
+
+		if err := tx.Insert(c.Context(), "user_oauth", "id", map[string]any{
+			"id":          uniqueid.NextId(),
+			"user_id":     userId,
+			"type":        loginType,
+			"openid":      openid,
+			"create_time": "NOW()",
+			"update_time": "NOW()",
+		}); err != nil {
+			return err
+		}
+
+		return nil
+	})
+
+	if err != nil {
+		return errors.WithStack(err)
+	}
+
+	token, err := jwtx.CreateToken(c.Context(), map[string]any{
+		"id": userId,
+	})
+	if err != nil {
+		return errors.WithStack(err)
+	}
+
+	vo.Token = token
+	vo.UserId = userId
+	vo.Nickname = nickname
+	vo.Avatar = avatar
+
+	return httpx.OK(c, vo)
+}
+
+// ======================== DB 查询封装（去重复 SQL） ========================
+func getUserByOpenID(dbClient *db.Client, c fiber.Ctx, openid string, loginType int16) (map[string]any, error) {
+
+	users, err := dbClient.LoadDataBySQL(
+		c.Context(),
+		"SELECT u.* FROM user_oauth o JOIN user u ON u.id = o.user_id WHERE o.openid = ? AND o.type = ?",
+		[]any{openid, loginType},
+	)
+
+	if err != nil {
+		return nil, err
+	}
+
+	if len(users) == 0 {
+		return nil, nil
+	}
+
+	return users[0], nil
+}
+
+// ======================== 类型工具（极简版） ========================
+func toInt64(v any) int64 {
+	i, _ := v.(int64)
+	return i
+}

code/allapp/allapp-go-v3/internal/handle/data_save.go

@@ -1,6 +1,7 @@
 package handle
 
 import (
+	"allapp-go/internal/errors"
 	"allapp-go/internal/httpx"
 	"allapp-go/internal/types"
 	"allapp-go/pkg/db"
@@ -12,7 +13,7 @@ func SaveData(c fiber.Ctx) error {
 	var req []types.SaveDataReq
 
 	if err := httpx.BindAndValidate(c, &req); err != nil {
-		return err
+		return errors.WithStack(err)
 	}
 
 	if len(req) == 0 {
@@ -49,7 +50,7 @@ func SaveData(c fiber.Ctx) error {
 	})
 
 	if err != nil {
-		return err
+		return errors.WithStack(err)
 	}
 
 	return httpx.OK(c, nil)