v1.1

app/main/api/internal/logic/user/wxminiauthlogic.go

@@ -1,20 +1,21 @@
 package user
 
 import (
-	"context"
-	"encoding/json"
-	"fmt"
-	"io"
-	"net/http"
-	"time"
+    "context"
+    "encoding/json"
+    "fmt"
+    "io"
+    "net/http"
+    "time"
 
-	"ycc-server/app/main/api/internal/svc"
-	"ycc-server/app/main/api/internal/types"
-	"ycc-server/app/main/model"
-	"ycc-server/common/xerr"
+    "ycc-server/app/main/api/internal/svc"
+    "ycc-server/app/main/api/internal/types"
+    "ycc-server/app/main/model"
+    "ycc-server/common/xerr"
 
-	"github.com/pkg/errors"
-	"github.com/zeromicro/go-zero/core/logx"
+    "github.com/google/uuid"
+    "github.com/pkg/errors"
+    "github.com/zeromicro/go-zero/core/logx"
 )
 
 type WxMiniAuthLogic struct {
@@ -45,41 +46,29 @@ func (l *WxMiniAuthLogic) WxMiniAuth(req *types.WXMiniAuthReq) (resp *types.WXMi
 	}
 
 	// 3. 处理用户信息
-	var userID int64
-	var userType int64
-	if userAuth != nil {
-		// 已存在用户，直接登录
-		userID = userAuth.UserId
-		userType = model.UserTypeNormal
-	} else {
-		// 注册临时用户
-		userTemp, err := l.svcCtx.UserTempModel.FindOneByAuthTypeAuthKey(l.ctx, model.UserAuthTypeWxMiniOpenID, sessionKeyResp.Openid)
-		if err != nil && !errors.Is(err, model.ErrNotFound) {
-			return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户临时信息失败: %v", err)
-		}
-		if userTemp == nil {
-			// 创建新的临时用户
-			userTemp = &model.UserTemp{}
-			userTemp.AuthType = model.UserAuthTypeWxMiniOpenID
-			userTemp.AuthKey = sessionKeyResp.Openid
-			result, err := l.svcCtx.UserTempModel.Insert(l.ctx, nil, userTemp)
-			if err != nil {
-				return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建临时用户信息失败: %v", err)
-			}
-			// 获取新创建的临时用户ID
-			userID, err = result.LastInsertId()
-			if err != nil {
-				return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "获取新创建的临时用户ID失败: %v", err)
-			}
-		} else {
-			// 使用已存在的临时用户ID
-			userID = userTemp.Id
-		}
-		userType = model.UserTypeTemp
-	}
+    var userID string
+    var userType int64
+    if userAuth != nil {
+        // 已存在用户，直接登录
+        userID = userAuth.UserId
+        userType = model.UserTypeNormal
+    } else {
+        user := &model.User{Id: uuid.NewString()}
+        _, err := l.svcCtx.UserModel.Insert(l.ctx, nil, user)
+        if err != nil {
+            return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户失败: %v", err)
+        }
+        ua := &model.UserAuth{Id: uuid.NewString(), UserId: user.Id, AuthType: model.UserAuthTypeWxMiniOpenID, AuthKey: sessionKeyResp.Openid}
+        _, err = l.svcCtx.UserAuthModel.Insert(l.ctx, nil, ua)
+        if err != nil {
+            return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户授权失败: %v", err)
+        }
+        userID = user.Id
+        userType = model.UserTypeTemp
+    }
 
 	// 4. 生成JWT Token
-	token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID, userType)
+    token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID, userType)
 	if err != nil {
 		return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成JWT Token失败: %v", err)
 	}