init: taotie-api 项目初始化

service/userservice.go

@@ -0,0 +1,125 @@
+package service
+
+import (
+	"context"
+	"fmt"
+	"taotie-api/common"
+	"taotie-api/core"
+	"taotie-api/model/do/userdo"
+	"taotie-api/model/po"
+	"taotie-api/repo"
+	"taotie-api/utils/sctx"
+	"taotie-api/utils/sjwt"
+	"time"
+
+	"github.com/gookit/goutil/errorx"
+)
+
+type UserService struct {
+	cfg        *core.Configuration
+	userRepo   *repo.UserRepo[*po.TUser]
+	tenantRepo *repo.TenantRepo[*po.TTenant]
+}
+
+func NewUserService(mdb *repo.MongoDb, cfg *core.Configuration, userRepo *repo.UserRepo[*po.TUser], tenantRepo *repo.TenantRepo[*po.TTenant]) *UserService {
+	return &UserService{
+		cfg,
+		userRepo,
+		tenantRepo,
+	}
+}
+
+// Login 登录
+func (s *UserService) Login(ctx context.Context, in *userdo.LoginIn) (*userdo.LoginOut, error) {
+	// 验证租户
+	ttenant, err := s.tenantRepo.FindOneByTenantNo(ctx, in.TenantNo)
+	if err != nil {
+		return nil, err
+	}
+	if ttenant == nil {
+		return nil, common.ErrUserTenantNotFound
+	}
+
+	// 验证用户
+	tuser, err := s.userRepo.FindOneByTenantIdAndUserName(ctx, ttenant.Id(), in.UserName)
+	if err != nil {
+		return nil, err
+	}
+	if tuser == nil {
+		return nil, common.ErrUserNotFound
+	}
+	if tuser.Password != in.Password {
+		return nil, common.ErrUserPasswordError
+	}
+
+	return &userdo.LoginOut{
+		TenantId: ttenant.Id(),
+		UserId:   tuser.Id(),
+		UserName: tuser.UserName,
+	}, nil
+}
+
+// GenToken 生成token
+func (s *UserService) GenToken(ctx context.Context, in *userdo.GenTokenIn) (*userdo.GenTokenOut, error) {
+	token, err := sjwt.GenerateToken(in.TenantNo, in.UserId, in.UserName, s.cfg.JWT.TimeOut, s.cfg.JWT.SignString)
+	if err != nil {
+		return nil, err
+	}
+	return &userdo.GenTokenOut{
+		Token: token,
+	}, nil
+}
+
+func (s *UserService) Create(ctx context.Context, in *userdo.CreateIn) (*userdo.CreateOut, error) {
+	// 查重：租户名称
+	ttenant, err := s.tenantRepo.FindOneByTenantName(ctx, in.TenantName)
+	if err != nil {
+		return nil, err
+	}
+	if ttenant != nil {
+		return nil, errorx.With(common.ErrSysDuplicate, "租户名称重复")
+	}
+
+	// 先设置临时用户到 context（避免 BaseRepo.Create 中 cuser 为 nil 导致 panic）
+	sctx.SetCurrentUser(ctx, &sctx.CurrentUser{
+		UserId:   "",
+		UserName: in.UserName,
+		TenantId: "",
+	})
+
+	// 生成 tenantNo（基于时间戳）
+	tenantNo := fmt.Sprintf("T%d", time.Now().UnixMilli())
+
+	// 创建租户
+	ttenant = &po.TTenant{
+		TenantNo:   tenantNo,
+		TenantName: in.TenantName,
+	}
+	tenantId, err := s.tenantRepo.Create(ctx, ttenant)
+	if err != nil {
+		return nil, err
+	}
+
+	// 更新 context 中的 TenantId
+	sctx.SetCurrentUser(ctx, &sctx.CurrentUser{
+		UserId:   "",
+		UserName: in.UserName,
+		TenantId: tenantId,
+	})
+
+	// 创建用户
+	tuser := &po.TUser{
+		UserName: in.UserName,
+		Password: in.Password,
+	}
+	userId, err := s.userRepo.Create(ctx, tuser)
+	if err != nil {
+		return nil, err
+	}
+
+	return &userdo.CreateOut{
+		TenantId: tenantId,
+		UserId:   userId,
+		UserName: in.UserName,
+	}, nil
+}