gf_demo_star/internal/logic/users/users_register.go

package users

import (
	"context"

	"star/internal/dao"
	"star/internal/model/do"

	"github.com/gogf/gf/v2/errors/gerror"
)

type RegisterInput struct {
	Username string
	Password string
	Email    string
}

// Register 注册用户
func (u *Users) Register(ctx context.Context, in RegisterInput) error {
	if err := u.checkUser(ctx, in.Username); err != nil {
		return err
	}

	_, err := dao.Users.Ctx(ctx).Data(do.Users{
		Username: in.Username,
		Password: u.encryptPassword(in.Password), // 加密密码
		Email:    in.Email,
	}).Insert()
	if err != nil {
		return err
	}
	return nil
}

// checkUser 检查用户是否存在
func (u *Users) checkUser(ctx context.Context, username string) error {
	count, err := dao.Users.Ctx(ctx).Where("username", username).Count()
	if err != nil {
		return err
	}
	if count > 0 {
		return gerror.New("用户名已存在")
	}
	return nil
}