initial commit

user/internal/logic/registerlogic.go

@@ -0,0 +1,90 @@
+package logic
+
+import (
+	"context"
+	"database/sql"
+	"godemo/user/internal/model"
+	"godemo/user/internal/svc"
+	"godemo/user/user"
+	"time"
+
+	"github.com/lib/pq"
+	"github.com/zeromicro/go-zero/core/logx"
+	"golang.org/x/crypto/bcrypt"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+type RegisterLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewRegisterLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RegisterLogic {
+	return &RegisterLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+// 用户注册
+func (l *RegisterLogic) Register(in *user.RegisterRequest) (*user.RegisterResponse, error) {
+	// 1. 检查用户名是否已存在
+	var existingUser model.User
+	err := l.svcCtx.DB.Get(&existingUser, "SELECT * FROM users WHERE username = $1", in.Username)
+	if err == nil {
+		// 用户名已存在
+		return nil, status.Error(codes.AlreadyExists, "用户名已存在")
+	}
+
+	// 2. 加密密码
+	hashedPassword, err := bcrypt.GenerateFromPassword([]byte(in.Password), bcrypt.DefaultCost)
+	if err != nil {
+		// 密码加密失败
+		return nil, status.Error(codes.Internal, err.Error())
+	}
+
+	// 3. 转换 email 为 sql.NullString
+	var email sql.NullString
+	if in.Email != "" {
+		email = sql.NullString{
+			String: in.Email,
+			Valid:  true,
+		}
+	} else {
+		email = sql.NullString{Valid: false}
+	}
+
+	// 4. 保存新用户到数据库
+	userModel := model.User{
+		Username:     in.Username,
+		PasswordHash: string(hashedPassword),
+		Email:        email,            // 使用 sql.NullString 类型存储 email
+		Roles:        []string{"user"}, // 默认角色为 "user"
+		CreatedAt:    time.Now(),       // 当前时间戳
+	}
+
+	// 执行插入数据库操作
+	_, err = l.svcCtx.DB.Exec(
+		"INSERT INTO users (username, password_hash, email, roles, created_at) VALUES ($1, $2, $3, $4, $5)",
+		userModel.Username,
+		userModel.PasswordHash,
+		userModel.Email,
+		pq.Array(userModel.Roles), // pq.Array 用于处理 PostgreSQL 数组类型
+		userModel.CreatedAt,
+	)
+	if err != nil {
+		// 数据库插入失败
+		return nil, status.Error(codes.Internal, err.Error())
+	}
+
+	// 获取插入的记录的 UserId
+	l.svcCtx.DB.Get(&existingUser, "SELECT * FROM users WHERE username = $1", in.Username)
+
+	// 5. 返回成功响应
+	return &user.RegisterResponse{
+		UserId: existingUser.UserId, // 将 UserId 转换为字符串
+	}, nil
+}