66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"tydata-server/app/user/model"
|
|||
|
|
|
|||
|
|
"github.com/google/uuid"
|
|||
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type UserService struct {
|
|||
|
|
userModel model.UserModel
|
|||
|
|
userAuthModel model.UserAuthModel
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewUserService 创建UserService实例
|
|||
|
|
func NewUserService(userModel model.UserModel, userAuthModel model.UserAuthModel) *UserService {
|
|||
|
|
return &UserService{
|
|||
|
|
userModel: userModel,
|
|||
|
|
userAuthModel: userAuthModel,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GenerateUUIDUserId 生成UUID用户ID
|
|||
|
|
func (s *UserService) GenerateUUIDUserId(ctx context.Context) (string, error) {
|
|||
|
|
id := uuid.NewString()
|
|||
|
|
return id, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RegisterUUIDUser 注册UUID用户，返回用户ID
|
|||
|
|
func (s *UserService) RegisterUUIDUser(ctx context.Context) (int64, error) {
|
|||
|
|
// 生成UUID
|
|||
|
|
uuidStr, err := s.GenerateUUIDUserId(ctx)
|
|||
|
|
if err != nil {
|
|||
|
|
return 0, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var userId int64
|
|||
|
|
err = s.userModel.Trans(ctx, func(ctx context.Context, session sqlx.Session) error {
|
|||
|
|
// 创建用户记录
|
|||
|
|
user := &model.User{}
|
|||
|
|
result, err := s.userModel.Insert(ctx, session, user)
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
userId, err = result.LastInsertId()
|
|||
|
|
if err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 创建用户认证记录
|
|||
|
|
userAuth := &model.UserAuth{
|
|||
|
|
UserId: userId,
|
|||
|
|
AuthType: model.UserAuthTypeUUID,
|
|||
|
|
AuthKey: uuidStr,
|
|||
|
|
}
|
|||
|
|
_, err = s.userAuthModel.Insert(ctx, session, userAuth)
|
|||
|
|
return err
|
|||
|
|
})
|
|||
|
|
if err != nil {
|
|||
|
|
return 0, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return userId, nil
|
|||
|
|
}
|