tenant: add invites and join requests

This commit is contained in:
2025-12-18 18:27:23 +08:00
parent 462bde351d
commit ec4506fd2d
28 changed files with 5206 additions and 201 deletions

View File

@@ -0,0 +1,66 @@
package dto
import (
"strings"
"time"
"quyun/v2/app/requests"
"quyun/v2/pkg/consts"
)
// AdminTenantInviteCreateForm 租户管理员创建邀请码的请求参数。
type AdminTenantInviteCreateForm struct {
// Code 邀请码(可选):为空时由后端生成;建议只包含数字/字母,便于人工输入。
Code string `json:"code"`
// MaxUses 最大可使用次数可选0 表示不限次数;大于 0 时用尽后自动失效。
MaxUses *int `json:"max_uses"`
// ExpiresAt 过期时间(可选):为空表示不过期;到期后不可再使用。
ExpiresAt *time.Time `json:"expires_at"`
// Remark 备注(可选):用于审计记录生成目的/投放渠道等。
Remark string `json:"remark"`
}
// AdminTenantInviteDisableForm 租户管理员禁用邀请码的请求参数。
type AdminTenantInviteDisableForm struct {
// Reason 禁用原因(可选):用于审计与追溯。
Reason string `json:"reason"`
}
// AdminTenantInviteListFilter 租户管理员分页查询邀请码列表的过滤条件。
type AdminTenantInviteListFilter struct {
requests.Pagination
// Status 按状态过滤可选active/disabled/expired。
Status *consts.TenantInviteStatus `query:"status" json:"status"`
// Code 按邀请码模糊过滤可选支持部分匹配like
Code *string `query:"code" json:"code"`
}
// CodeTrimmed 对 code 进行空白与大小写处理,便于统一查询。
func (f *AdminTenantInviteListFilter) CodeTrimmed() string {
if f == nil || f.Code == nil {
return ""
}
return strings.ToLower(strings.TrimSpace(*f.Code))
}
// AdminTenantJoinRequestListFilter 租户管理员分页查询加入申请的过滤条件。
type AdminTenantJoinRequestListFilter struct {
requests.Pagination
// UserID 按申请人用户ID过滤可选
UserID *int64 `query:"user_id" json:"user_id"`
// Status 按申请状态过滤可选pending/approved/rejected。
Status *consts.TenantJoinRequestStatus `query:"status" json:"status"`
}
// AdminTenantJoinRequestDecideForm 租户管理员通过/拒绝加入申请的请求参数。
type AdminTenantJoinRequestDecideForm struct {
// Reason 审核说明(可选):用于审计记录通过/拒绝原因。
Reason string `json:"reason"`
}

View File

@@ -60,17 +60,21 @@ func Provide(opts ...opt.Option) error {
order *order,
orderAdmin *orderAdmin,
orderMe *orderMe,
tenantInviteAdmin *tenantInviteAdmin,
tenantJoinAdmin *tenantJoinAdmin,
tenantUserAdmin *tenantUserAdmin,
) (contracts.HttpRoute, error) {
obj := &Routes{
content: content,
contentAdmin: contentAdmin,
me: me,
middlewares: middlewares,
order: order,
orderAdmin: orderAdmin,
orderMe: orderMe,
tenantUserAdmin: tenantUserAdmin,
content: content,
contentAdmin: contentAdmin,
me: me,
middlewares: middlewares,
order: order,
orderAdmin: orderAdmin,
orderMe: orderMe,
tenantInviteAdmin: tenantInviteAdmin,
tenantJoinAdmin: tenantJoinAdmin,
tenantUserAdmin: tenantUserAdmin,
}
if err := obj.Prepare(); err != nil {
return nil, err
@@ -80,6 +84,20 @@ func Provide(opts ...opt.Option) error {
}, atom.GroupRoutes); err != nil {
return err
}
if err := container.Container.Provide(func() (*tenantInviteAdmin, error) {
obj := &tenantInviteAdmin{}
return obj, nil
}); err != nil {
return err
}
if err := container.Container.Provide(func() (*tenantJoinAdmin, error) {
obj := &tenantJoinAdmin{}
return obj, nil
}); err != nil {
return err
}
if err := container.Container.Provide(func() (*tenantUserAdmin, error) {
obj := &tenantUserAdmin{}

View File

@@ -24,13 +24,15 @@ type Routes struct {
log *log.Entry `inject:"false"`
middlewares *middlewares.Middlewares
// Controller instances
content *content
contentAdmin *contentAdmin
me *me
order *order
orderAdmin *orderAdmin
orderMe *orderMe
tenantUserAdmin *tenantUserAdmin
content *content
contentAdmin *contentAdmin
me *me
order *order
orderAdmin *orderAdmin
orderMe *orderMe
tenantInviteAdmin *tenantInviteAdmin
tenantJoinAdmin *tenantJoinAdmin
tenantUserAdmin *tenantUserAdmin
}
// Prepare initializes the routes provider with logging configuration.
@@ -185,6 +187,53 @@ func (r *Routes) Register(router fiber.Router) {
Local[*models.User]("user"),
PathParam[int64]("orderID"),
))
// Register routes for controller: tenantInviteAdmin
r.log.Debugf("Registering route: Get /t/:tenantCode/v1/admin/invites -> tenantInviteAdmin.adminInviteList")
router.Get("/t/:tenantCode/v1/admin/invites"[len(r.Path()):], DataFunc3(
r.tenantInviteAdmin.adminInviteList,
Local[*models.Tenant]("tenant"),
Local[*models.TenantUser]("tenant_user"),
Query[dto.AdminTenantInviteListFilter]("filter"),
))
r.log.Debugf("Registering route: Patch /t/:tenantCode/v1/admin/invites/:inviteID/disable -> tenantInviteAdmin.adminDisableInvite")
router.Patch("/t/:tenantCode/v1/admin/invites/:inviteID/disable"[len(r.Path()):], DataFunc4(
r.tenantInviteAdmin.adminDisableInvite,
Local[*models.Tenant]("tenant"),
Local[*models.TenantUser]("tenant_user"),
PathParam[int64]("inviteID"),
Body[dto.AdminTenantInviteDisableForm]("form"),
))
r.log.Debugf("Registering route: Post /t/:tenantCode/v1/admin/invites -> tenantInviteAdmin.adminCreateInvite")
router.Post("/t/:tenantCode/v1/admin/invites"[len(r.Path()):], DataFunc3(
r.tenantInviteAdmin.adminCreateInvite,
Local[*models.Tenant]("tenant"),
Local[*models.TenantUser]("tenant_user"),
Body[dto.AdminTenantInviteCreateForm]("form"),
))
// Register routes for controller: tenantJoinAdmin
r.log.Debugf("Registering route: Get /t/:tenantCode/v1/admin/join-requests -> tenantJoinAdmin.adminJoinRequests")
router.Get("/t/:tenantCode/v1/admin/join-requests"[len(r.Path()):], DataFunc3(
r.tenantJoinAdmin.adminJoinRequests,
Local[*models.Tenant]("tenant"),
Local[*models.TenantUser]("tenant_user"),
Query[dto.AdminTenantJoinRequestListFilter]("filter"),
))
r.log.Debugf("Registering route: Post /t/:tenantCode/v1/admin/join-requests/:requestID/approve -> tenantJoinAdmin.adminApproveJoinRequest")
router.Post("/t/:tenantCode/v1/admin/join-requests/:requestID/approve"[len(r.Path()):], DataFunc4(
r.tenantJoinAdmin.adminApproveJoinRequest,
Local[*models.Tenant]("tenant"),
Local[*models.TenantUser]("tenant_user"),
PathParam[int64]("requestID"),
Body[dto.AdminTenantJoinRequestDecideForm]("form"),
))
r.log.Debugf("Registering route: Post /t/:tenantCode/v1/admin/join-requests/:requestID/reject -> tenantJoinAdmin.adminRejectJoinRequest")
router.Post("/t/:tenantCode/v1/admin/join-requests/:requestID/reject"[len(r.Path()):], DataFunc4(
r.tenantJoinAdmin.adminRejectJoinRequest,
Local[*models.Tenant]("tenant"),
Local[*models.TenantUser]("tenant_user"),
PathParam[int64]("requestID"),
Body[dto.AdminTenantJoinRequestDecideForm]("form"),
))
// Register routes for controller: tenantUserAdmin
r.log.Debugf("Registering route: Delete /t/:tenantCode/v1/admin/users/:userID -> tenantUserAdmin.adminRemoveUser")
router.Delete("/t/:tenantCode/v1/admin/users/:userID"[len(r.Path()):], Func3(

View File

@@ -0,0 +1,134 @@
package tenant
import (
"time"
"quyun/v2/app/errorx"
"quyun/v2/app/http/tenant/dto"
"quyun/v2/app/requests"
"quyun/v2/app/services"
"quyun/v2/database/models"
"github.com/gofiber/fiber/v3"
log "github.com/sirupsen/logrus"
)
// tenantInviteAdmin 提供“租户管理员管理邀请码”的相关接口。
//
// @provider
type tenantInviteAdmin struct{}
// adminCreateInvite
//
// @Summary 创建邀请码(租户管理)
// @Tags Tenant
// @Accept json
// @Produce json
// @Param tenantCode path string true "Tenant Code"
// @Param form body dto.AdminTenantInviteCreateForm true "Form"
// @Success 200 {object} models.TenantInvite
//
// @Router /t/:tenantCode/v1/admin/invites [post]
// @Bind tenant local key(tenant)
// @Bind tenantUser local key(tenant_user)
// @Bind form body
func (*tenantInviteAdmin) adminCreateInvite(
ctx fiber.Ctx,
tenant *models.Tenant,
tenantUser *models.TenantUser,
form *dto.AdminTenantInviteCreateForm,
) (*models.TenantInvite, error) {
if err := requireTenantAdmin(tenantUser); err != nil {
return nil, err
}
if form == nil {
return nil, errorx.ErrInvalidParameter
}
log.WithFields(log.Fields{
"tenant_id": tenant.ID,
"user_id": tenantUser.UserID,
}).Info("tenant.admin.invites.create")
return services.Tenant.AdminCreateInvite(ctx.Context(), tenant.ID, tenantUser.UserID, form)
}
// adminInviteList
//
// @Summary 邀请码列表(租户管理)
// @Tags Tenant
// @Accept json
// @Produce json
// @Param tenantCode path string true "Tenant Code"
// @Param filter query dto.AdminTenantInviteListFilter true "Filter"
// @Success 200 {object} requests.Pager{items=models.TenantInvite}
//
// @Router /t/:tenantCode/v1/admin/invites [get]
// @Bind tenant local key(tenant)
// @Bind tenantUser local key(tenant_user)
// @Bind filter query
func (*tenantInviteAdmin) adminInviteList(
ctx fiber.Ctx,
tenant *models.Tenant,
tenantUser *models.TenantUser,
filter *dto.AdminTenantInviteListFilter,
) (*requests.Pager, error) {
if err := requireTenantAdmin(tenantUser); err != nil {
return nil, err
}
if filter == nil {
filter = &dto.AdminTenantInviteListFilter{}
}
log.WithFields(log.Fields{
"tenant_id": tenant.ID,
"user_id": tenantUser.UserID,
"status": filter.Status,
"code": filter.CodeTrimmed(),
}).Info("tenant.admin.invites.list")
return services.Tenant.AdminInvitePage(ctx.Context(), tenant.ID, filter)
}
// adminDisableInvite
//
// @Summary 禁用邀请码(租户管理)
// @Tags Tenant
// @Accept json
// @Produce json
// @Param tenantCode path string true "Tenant Code"
// @Param inviteID path int64 true "InviteID"
// @Param form body dto.AdminTenantInviteDisableForm true "Form"
// @Success 200 {object} models.TenantInvite
//
// @Router /t/:tenantCode/v1/admin/invites/:inviteID/disable [patch]
// @Bind tenant local key(tenant)
// @Bind tenantUser local key(tenant_user)
// @Bind inviteID path
// @Bind form body
func (*tenantInviteAdmin) adminDisableInvite(
ctx fiber.Ctx,
tenant *models.Tenant,
tenantUser *models.TenantUser,
inviteID int64,
form *dto.AdminTenantInviteDisableForm,
) (*models.TenantInvite, error) {
if err := requireTenantAdmin(tenantUser); err != nil {
return nil, err
}
if inviteID <= 0 {
return nil, errorx.ErrInvalidParameter.WithMsg("invite_id must be > 0")
}
if form == nil {
return nil, errorx.ErrInvalidParameter
}
log.WithFields(log.Fields{
"tenant_id": tenant.ID,
"user_id": tenantUser.UserID,
"invite_id": inviteID,
"disable_time": time.Now(),
}).Info("tenant.admin.invites.disable")
return services.Tenant.AdminDisableInvite(ctx.Context(), tenant.ID, tenantUser.UserID, inviteID, form.Reason)
}

View File

@@ -0,0 +1,142 @@
package tenant
import (
"time"
"quyun/v2/app/errorx"
"quyun/v2/app/http/tenant/dto"
"quyun/v2/app/requests"
"quyun/v2/app/services"
"quyun/v2/database/models"
"github.com/gofiber/fiber/v3"
log "github.com/sirupsen/logrus"
)
// tenantJoinAdmin 提供“租户管理员审核加入申请”的相关接口。
//
// @provider
type tenantJoinAdmin struct{}
// adminJoinRequests
//
// @Summary 加入申请列表(租户管理)
// @Tags Tenant
// @Accept json
// @Produce json
// @Param tenantCode path string true "Tenant Code"
// @Param filter query dto.AdminTenantJoinRequestListFilter true "Filter"
// @Success 200 {object} requests.Pager{items=models.TenantJoinRequest}
//
// @Router /t/:tenantCode/v1/admin/join-requests [get]
// @Bind tenant local key(tenant)
// @Bind tenantUser local key(tenant_user)
// @Bind filter query
func (*tenantJoinAdmin) adminJoinRequests(
ctx fiber.Ctx,
tenant *models.Tenant,
tenantUser *models.TenantUser,
filter *dto.AdminTenantJoinRequestListFilter,
) (*requests.Pager, error) {
if err := requireTenantAdmin(tenantUser); err != nil {
return nil, err
}
if filter == nil {
filter = &dto.AdminTenantJoinRequestListFilter{}
}
log.WithFields(log.Fields{
"tenant_id": tenant.ID,
"user_id": tenantUser.UserID,
"status": filter.Status,
"query_uid": filter.UserID,
}).Info("tenant.admin.join_requests.list")
return services.Tenant.AdminJoinRequestPage(ctx.Context(), tenant.ID, filter)
}
// adminApproveJoinRequest
//
// @Summary 通过加入申请(租户管理)
// @Tags Tenant
// @Accept json
// @Produce json
// @Param tenantCode path string true "Tenant Code"
// @Param requestID path int64 true "RequestID"
// @Param form body dto.AdminTenantJoinRequestDecideForm true "Form"
// @Success 200 {object} models.TenantJoinRequest
//
// @Router /t/:tenantCode/v1/admin/join-requests/:requestID/approve [post]
// @Bind tenant local key(tenant)
// @Bind tenantUser local key(tenant_user)
// @Bind requestID path
// @Bind form body
func (*tenantJoinAdmin) adminApproveJoinRequest(
ctx fiber.Ctx,
tenant *models.Tenant,
tenantUser *models.TenantUser,
requestID int64,
form *dto.AdminTenantJoinRequestDecideForm,
) (*models.TenantJoinRequest, error) {
if err := requireTenantAdmin(tenantUser); err != nil {
return nil, err
}
if requestID <= 0 {
return nil, errorx.ErrInvalidParameter.WithMsg("request_id must be > 0")
}
if form == nil {
return nil, errorx.ErrInvalidParameter
}
log.WithFields(log.Fields{
"tenant_id": tenant.ID,
"user_id": tenantUser.UserID,
"request_id": requestID,
"decide_time": time.Now(),
}).Info("tenant.admin.join_requests.approve")
return services.Tenant.AdminApproveJoinRequest(ctx.Context(), tenant.ID, tenantUser.UserID, requestID, form.Reason)
}
// adminRejectJoinRequest
//
// @Summary 拒绝加入申请(租户管理)
// @Tags Tenant
// @Accept json
// @Produce json
// @Param tenantCode path string true "Tenant Code"
// @Param requestID path int64 true "RequestID"
// @Param form body dto.AdminTenantJoinRequestDecideForm true "Form"
// @Success 200 {object} models.TenantJoinRequest
//
// @Router /t/:tenantCode/v1/admin/join-requests/:requestID/reject [post]
// @Bind tenant local key(tenant)
// @Bind tenantUser local key(tenant_user)
// @Bind requestID path
// @Bind form body
func (*tenantJoinAdmin) adminRejectJoinRequest(
ctx fiber.Ctx,
tenant *models.Tenant,
tenantUser *models.TenantUser,
requestID int64,
form *dto.AdminTenantJoinRequestDecideForm,
) (*models.TenantJoinRequest, error) {
if err := requireTenantAdmin(tenantUser); err != nil {
return nil, err
}
if requestID <= 0 {
return nil, errorx.ErrInvalidParameter.WithMsg("request_id must be > 0")
}
if form == nil {
return nil, errorx.ErrInvalidParameter
}
log.WithFields(log.Fields{
"tenant_id": tenant.ID,
"user_id": tenantUser.UserID,
"request_id": requestID,
"decide_time": time.Now(),
}).Info("tenant.admin.join_requests.reject")
return services.Tenant.AdminRejectJoinRequest(ctx.Context(), tenant.ID, tenantUser.UserID, requestID, form.Reason)
}

View File

@@ -0,0 +1,13 @@
package dto
// JoinByInviteForm 用户通过邀请码加入租户的请求参数。
type JoinByInviteForm struct {
// InviteCode 邀请码:由租户管理员生成;用户提交后加入对应租户。
InviteCode string `json:"invite_code"`
}
// JoinRequestCreateForm 用户提交加入租户申请的请求参数(无邀请码场景)。
type JoinRequestCreateForm struct {
// Reason 申请原因(可选):用于向租户管理员说明申请加入的目的。
Reason string `json:"reason"`
}

View File

@@ -0,0 +1,87 @@
package tenantjoin
import (
"quyun/v2/app/errorx"
"quyun/v2/app/http/tenantjoin/dto"
"quyun/v2/app/services"
"quyun/v2/database/models"
"quyun/v2/providers/jwt"
"github.com/gofiber/fiber/v3"
log "github.com/sirupsen/logrus"
)
// join 提供“非成员加入租户”的相关接口(邀请码加入 / 申请加入)。
//
// @provider
type join struct{}
// joinByInvite
//
// @Summary 通过邀请码加入租户
// @Tags TenantJoin
// @Accept json
// @Produce json
// @Param tenantCode path string true "Tenant Code"
// @Param form body dto.JoinByInviteForm true "Form"
// @Success 200 {object} models.TenantUser
//
// @Router /t/:tenantCode/v1/join/invite [post]
// @Bind tenant local key(tenant)
// @Bind claims local key(claims)
// @Bind form body
func (*join) joinByInvite(
ctx fiber.Ctx,
tenant *models.Tenant,
claims *jwt.Claims,
form *dto.JoinByInviteForm,
) (*models.TenantUser, error) {
if tenant == nil || claims == nil {
return nil, errorx.ErrInternalError.WithMsg("context missing")
}
if form == nil {
return nil, errorx.ErrInvalidParameter
}
log.WithFields(log.Fields{
"tenant_id": tenant.ID,
"user_id": claims.UserID,
}).Info("tenantjoin.join_by_invite")
return services.Tenant.JoinByInvite(ctx.Context(), tenant.ID, claims.UserID, form.InviteCode)
}
// createJoinRequest
//
// @Summary 提交加入租户申请
// @Tags TenantJoin
// @Accept json
// @Produce json
// @Param tenantCode path string true "Tenant Code"
// @Param form body dto.JoinRequestCreateForm true "Form"
// @Success 200 {object} models.TenantJoinRequest
//
// @Router /t/:tenantCode/v1/join/request [post]
// @Bind tenant local key(tenant)
// @Bind claims local key(claims)
// @Bind form body
func (*join) createJoinRequest(
ctx fiber.Ctx,
tenant *models.Tenant,
claims *jwt.Claims,
form *dto.JoinRequestCreateForm,
) (*models.TenantJoinRequest, error) {
if tenant == nil || claims == nil {
return nil, errorx.ErrInternalError.WithMsg("context missing")
}
if form == nil {
return nil, errorx.ErrInvalidParameter
}
log.WithFields(log.Fields{
"tenant_id": tenant.ID,
"user_id": claims.UserID,
}).Info("tenantjoin.create_join_request")
return services.Tenant.CreateJoinRequest(ctx.Context(), tenant.ID, claims.UserID, form)
}

View File

@@ -0,0 +1,37 @@
package tenantjoin
import (
"quyun/v2/app/middlewares"
"go.ipao.vip/atom"
"go.ipao.vip/atom/container"
"go.ipao.vip/atom/contracts"
"go.ipao.vip/atom/opt"
)
func Provide(opts ...opt.Option) error {
if err := container.Container.Provide(func() (*join, error) {
obj := &join{}
return obj, nil
}); err != nil {
return err
}
if err := container.Container.Provide(func(
join *join,
middlewares *middlewares.Middlewares,
) (contracts.HttpRoute, error) {
obj := &Routes{
join: join,
middlewares: middlewares,
}
if err := obj.Prepare(); err != nil {
return nil, err
}
return obj, nil
}, atom.GroupRoutes); err != nil {
return err
}
return nil
}

View File

@@ -0,0 +1,63 @@
// Code generated by atomctl. DO NOT EDIT.
// Package tenantjoin provides HTTP route definitions and registration
// for the quyun/v2 application.
package tenantjoin
import (
"quyun/v2/app/http/tenantjoin/dto"
"quyun/v2/app/middlewares"
"quyun/v2/database/models"
"quyun/v2/providers/jwt"
"github.com/gofiber/fiber/v3"
log "github.com/sirupsen/logrus"
_ "go.ipao.vip/atom"
_ "go.ipao.vip/atom/contracts"
. "go.ipao.vip/atom/fen"
)
// Routes implements the HttpRoute contract and provides route registration
// for all controllers in the tenantjoin module.
//
// @provider contracts.HttpRoute atom.GroupRoutes
type Routes struct {
log *log.Entry `inject:"false"`
middlewares *middlewares.Middlewares
// Controller instances
join *join
}
// Prepare initializes the routes provider with logging configuration.
func (r *Routes) Prepare() error {
r.log = log.WithField("module", "routes.tenantjoin")
r.log.Info("Initializing routes module")
return nil
}
// Name returns the unique identifier for this routes provider.
func (r *Routes) Name() string {
return "tenantjoin"
}
// Register registers all HTTP routes with the provided fiber router.
// Each route is registered with its corresponding controller action and parameter bindings.
func (r *Routes) Register(router fiber.Router) {
// Register routes for controller: join
r.log.Debugf("Registering route: Post /t/:tenantCode/v1/join/invite -> join.joinByInvite")
router.Post("/t/:tenantCode/v1/join/invite"[len(r.Path()):], DataFunc3(
r.join.joinByInvite,
Local[*models.Tenant]("tenant"),
Local[*jwt.Claims]("claims"),
Body[dto.JoinByInviteForm]("form"),
))
r.log.Debugf("Registering route: Post /t/:tenantCode/v1/join/request -> join.createJoinRequest")
router.Post("/t/:tenantCode/v1/join/request"[len(r.Path()):], DataFunc3(
r.join.createJoinRequest,
Local[*models.Tenant]("tenant"),
Local[*jwt.Claims]("claims"),
Body[dto.JoinRequestCreateForm]("form"),
))
r.log.Info("Successfully registered all routes")
}

View File

@@ -0,0 +1,12 @@
package tenantjoin
func (r *Routes) Path() string {
return "/t/:tenantCode/v1"
}
func (r *Routes) Middlewares() []any {
return []any{
r.middlewares.TenantResolve,
r.middlewares.TenantAuth,
}
}

View File

@@ -0,0 +1,511 @@
package services
import (
"context"
"crypto/rand"
"encoding/base32"
"strings"
"time"
"quyun/v2/app/errorx"
"quyun/v2/app/http/tenant/dto"
tenantjoindto "quyun/v2/app/http/tenantjoin/dto"
"quyun/v2/app/requests"
"quyun/v2/database/models"
"quyun/v2/pkg/consts"
"github.com/jackc/pgx/v5/pgconn"
"github.com/pkg/errors"
"github.com/sirupsen/logrus"
"go.ipao.vip/gen"
"go.ipao.vip/gen/types"
"gorm.io/gorm"
"gorm.io/gorm/clause"
)
func isUniqueViolation(err error) bool {
var pgErr *pgconn.PgError
if errors.As(err, &pgErr) {
return pgErr.Code == "23505"
}
return errors.Is(err, gorm.ErrDuplicatedKey)
}
func newInviteCode() (string, error) {
// 邀请码为安全敏感值:使用强随机数,避免可预测性导致被撞库加入租户。
buf := make([]byte, 10) // 80-bit
if _, err := rand.Read(buf); err != nil {
return "", err
}
// base32去掉 padding便于输入统一转小写存储与比较。
return strings.ToLower(base32.StdEncoding.WithPadding(base32.NoPadding).EncodeToString(buf)), nil
}
// AdminCreateInvite 租户管理员创建邀请(用于用户通过邀请码加入租户)。
func (t *tenant) AdminCreateInvite(ctx context.Context, tenantID, operatorUserID int64, form *dto.AdminTenantInviteCreateForm) (*models.TenantInvite, error) {
if tenantID <= 0 {
return nil, errorx.ErrInvalidParameter.WithMsg("tenant_id must be > 0")
}
if operatorUserID <= 0 {
return nil, errorx.ErrInvalidParameter.WithMsg("operator_user_id must be > 0")
}
if form == nil {
return nil, errorx.ErrInvalidParameter.WithMsg("form is nil")
}
now := time.Now().UTC()
code := strings.ToLower(strings.TrimSpace(form.Code))
if code == "" {
var err error
code, err = newInviteCode()
if err != nil {
return nil, err
}
}
if form.ExpiresAt != nil && !form.ExpiresAt.IsZero() && form.ExpiresAt.Before(now) {
return nil, errorx.ErrInvalidParameter.WithMsg("expires_at must be in future")
}
if form.MaxUses != nil && *form.MaxUses < 0 {
return nil, errorx.ErrInvalidParameter.WithMsg("max_uses must be >= 0")
}
logrus.WithFields(logrus.Fields{
"tenant_id": tenantID,
"operator_user_id": operatorUserID,
"code": code,
"max_uses": form.MaxUses,
"expires_at_present": form.ExpiresAt != nil,
}).Info("services.tenant.admin.create_invite")
invite := &models.TenantInvite{
TenantID: tenantID,
UserID: operatorUserID,
Code: code,
Status: consts.TenantInviteStatusActive,
MaxUses: 0,
UsedCount: 0,
Remark: strings.TrimSpace(form.Remark),
CreatedAt: now,
UpdatedAt: now,
}
if form.MaxUses != nil {
invite.MaxUses = int32(*form.MaxUses)
}
if form.ExpiresAt != nil && !form.ExpiresAt.IsZero() {
invite.ExpiresAt = form.ExpiresAt.UTC()
}
// 关键点expires_at/disabled_at 允许为空,避免写入 0001-01-01 造成误判。
db := models.Q.TenantInvite.WithContext(ctx).UnderlyingDB().Omit("disabled_at", "disabled_operator_user_id")
if invite.ExpiresAt.IsZero() {
db = db.Omit("expires_at")
}
if err := db.Create(invite).Error; err != nil {
if isUniqueViolation(err) {
return nil, errorx.ErrRecordDuplicated.WithMsg("邀请码已存在,请重试")
}
return nil, err
}
return invite, nil
}
// AdminDisableInvite 租户管理员禁用邀请(幂等)。
func (t *tenant) AdminDisableInvite(ctx context.Context, tenantID, operatorUserID, inviteID int64, reason string) (*models.TenantInvite, error) {
if tenantID <= 0 || operatorUserID <= 0 || inviteID <= 0 {
return nil, errorx.ErrInvalidParameter.WithMsg("invalid tenant_id/operator_user_id/invite_id")
}
now := time.Now().UTC()
reason = strings.TrimSpace(reason)
logrus.WithFields(logrus.Fields{
"tenant_id": tenantID,
"operator_user_id": operatorUserID,
"invite_id": inviteID,
}).Info("services.tenant.admin.disable_invite")
var out models.TenantInvite
err := _db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
var inv models.TenantInvite
if err := tx.
Clauses(clause.Locking{Strength: "UPDATE"}).
Where("id = ? AND tenant_id = ?", inviteID, tenantID).
First(&inv).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return errorx.ErrRecordNotFound.WithMsg("邀请码不存在")
}
return err
}
// 幂等:重复禁用直接返回当前状态。
if inv.Status == consts.TenantInviteStatusDisabled || inv.Status == consts.TenantInviteStatusExpired {
out = inv
return nil
}
inv.Status = consts.TenantInviteStatusDisabled
inv.DisabledOperatorUserID = operatorUserID
inv.DisabledAt = now
if reason != "" {
inv.Remark = reason
}
inv.UpdatedAt = now
// 关键点disabled_at/disabled_operator_user_id 允许为空,但禁用时必须落审计信息。
if err := tx.Save(&inv).Error; err != nil {
return err
}
out = inv
return nil
})
if err != nil {
return nil, err
}
return &out, nil
}
// AdminInvitePage 租户管理员分页查询邀请列表。
func (t *tenant) AdminInvitePage(ctx context.Context, tenantID int64, filter *dto.AdminTenantInviteListFilter) (*requests.Pager, error) {
if tenantID <= 0 {
return nil, errorx.ErrInvalidParameter.WithMsg("tenant_id must be > 0")
}
if filter == nil {
filter = &dto.AdminTenantInviteListFilter{}
}
filter.Pagination.Format()
tbl, query := models.TenantInviteQuery.QueryContext(ctx)
conds := []gen.Condition{tbl.TenantID.Eq(tenantID)}
if filter.Status != nil && *filter.Status != "" {
conds = append(conds, tbl.Status.Eq(*filter.Status))
}
if code := filter.CodeTrimmed(); code != "" {
conds = append(conds, tbl.Code.Like("%"+strings.ToLower(code)+"%"))
}
items, total, err := query.Where(conds...).Order(tbl.ID.Desc()).FindByPage(int(filter.Offset()), int(filter.Limit))
if err != nil {
return nil, err
}
return &requests.Pager{
Pagination: filter.Pagination,
Total: total,
Items: items,
}, nil
}
// JoinByInvite 用户通过邀请码加入租户(无须已是租户成员)。
func (t *tenant) JoinByInvite(ctx context.Context, tenantID, userID int64, inviteCode string) (*models.TenantUser, error) {
if tenantID <= 0 || userID <= 0 {
return nil, errorx.ErrInvalidParameter.WithMsg("invalid tenant_id/user_id")
}
inviteCode = strings.ToLower(strings.TrimSpace(inviteCode))
if inviteCode == "" {
return nil, errorx.ErrInvalidParameter.WithMsg("invite_code is empty")
}
now := time.Now().UTC()
logrus.WithFields(logrus.Fields{
"tenant_id": tenantID,
"user_id": userID,
"invite_code": inviteCode,
"invite_token": "[masked]",
}).Info("services.tenant.join_by_invite")
var out models.TenantUser
err := _db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
// 关键前置条件:已经是成员时直接成功返回,不消耗邀请码使用次数。
var existingTU models.TenantUser
if err := tx.Where("tenant_id = ? AND user_id = ?", tenantID, userID).First(&existingTU).Error; err == nil {
out = existingTU
return nil
} else if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
return err
}
// 邀请校验必须加行锁,避免并发超发 used_count。
var inv models.TenantInvite
if err := tx.
Clauses(clause.Locking{Strength: "UPDATE"}).
Where("tenant_id = ? AND code = ?", tenantID, inviteCode).
First(&inv).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return errorx.ErrRecordNotFound.WithMsg("邀请码不存在")
}
return err
}
// 关键规则:禁用/过期的邀请码不可使用。
if inv.Status != consts.TenantInviteStatusActive {
return errorx.ErrPreconditionFailed.WithMsg("邀请码不可用")
}
if !inv.ExpiresAt.IsZero() && inv.ExpiresAt.Before(now) {
// 业务侧保持状态一致:过期时顺手标记 expired避免后续误用。
_ = tx.Model(&inv).Updates(map[string]any{
"status": consts.TenantInviteStatusExpired,
"updated_at": now,
}).Error
return errorx.ErrPreconditionFailed.WithMsg("邀请码已过期")
}
if inv.MaxUses > 0 && inv.UsedCount >= inv.MaxUses {
_ = tx.Model(&inv).Updates(map[string]any{
"status": consts.TenantInviteStatusExpired,
"updated_at": now,
}).Error
return errorx.ErrPreconditionFailed.WithMsg("邀请码已用尽")
}
// 加入租户:默认 member + verified与 tenant.AddUser 保持一致。
tu := &models.TenantUser{
TenantID: tenantID,
UserID: userID,
Role: types.NewArray([]consts.TenantUserRole{consts.TenantUserRoleMember}),
Status: consts.UserStatusVerified,
Balance: 0,
BalanceFrozen: 0,
CreatedAt: now,
UpdatedAt: now,
}
if err := tx.Create(tu).Error; err != nil {
if isUniqueViolation(err) {
// 并发幂等:重复插入按已加入处理,不消耗邀请码次数。
if err := tx.Where("tenant_id = ? AND user_id = ?", tenantID, userID).First(&out).Error; err != nil {
return err
}
return nil
}
return err
}
out = *tu
// 只有在“新加入”成功时才消耗邀请码次数。
updates := map[string]any{
"used_count": inv.UsedCount + 1,
"updated_at": now,
}
if inv.MaxUses > 0 && inv.UsedCount+1 >= inv.MaxUses {
updates["status"] = consts.TenantInviteStatusExpired
}
return tx.Model(&inv).Updates(updates).Error
})
if err != nil {
return nil, err
}
return &out, nil
}
// CreateJoinRequest 用户提交加入租户申请(无邀请码场景)。
func (t *tenant) CreateJoinRequest(ctx context.Context, tenantID, userID int64, form *tenantjoindto.JoinRequestCreateForm) (*models.TenantJoinRequest, error) {
if tenantID <= 0 || userID <= 0 {
return nil, errorx.ErrInvalidParameter.WithMsg("invalid tenant_id/user_id")
}
if form == nil {
return nil, errorx.ErrInvalidParameter.WithMsg("form is nil")
}
now := time.Now().UTC()
reason := strings.TrimSpace(form.Reason)
logrus.WithFields(logrus.Fields{
"tenant_id": tenantID,
"user_id": userID,
}).Info("services.tenant.create_join_request")
// 关键前置条件:已是成员则不允许重复申请。
var existingTU models.TenantUser
if err := _db.WithContext(ctx).Where("tenant_id = ? AND user_id = ?", tenantID, userID).First(&existingTU).Error; err == nil {
return nil, errorx.ErrPreconditionFailed.WithMsg("已是该租户成员")
} else if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
return nil, err
}
req := &models.TenantJoinRequest{
TenantID: tenantID,
UserID: userID,
Status: consts.TenantJoinRequestStatusPending,
Reason: reason,
CreatedAt: now,
UpdatedAt: now,
DecidedAt: time.Time{},
DecidedReason: "",
}
// 关键点decided_at/decided_operator_user_id 允许为空,避免写入 0001-01-01 造成误判。
db := models.Q.TenantJoinRequest.WithContext(ctx).UnderlyingDB().Omit("decided_at", "decided_operator_user_id")
if err := db.Create(req).Error; err != nil {
if isUniqueViolation(err) {
// 幂等:重复提交时返回现有 pending 申请。
tbl, query := models.TenantJoinRequestQuery.QueryContext(ctx)
existing, qErr := query.Where(
tbl.TenantID.Eq(tenantID),
tbl.UserID.Eq(userID),
tbl.Status.Eq(consts.TenantJoinRequestStatusPending),
).First()
if qErr == nil {
return existing, nil
}
return nil, err
}
return nil, err
}
return req, nil
}
// AdminJoinRequestPage 租户管理员分页查询加入申请列表。
func (t *tenant) AdminJoinRequestPage(ctx context.Context, tenantID int64, filter *dto.AdminTenantJoinRequestListFilter) (*requests.Pager, error) {
if tenantID <= 0 {
return nil, errorx.ErrInvalidParameter.WithMsg("tenant_id must be > 0")
}
if filter == nil {
filter = &dto.AdminTenantJoinRequestListFilter{}
}
filter.Pagination.Format()
tbl, query := models.TenantJoinRequestQuery.QueryContext(ctx)
conds := []gen.Condition{tbl.TenantID.Eq(tenantID)}
if filter.UserID != nil && *filter.UserID > 0 {
conds = append(conds, tbl.UserID.Eq(*filter.UserID))
}
if filter.Status != nil && *filter.Status != "" {
conds = append(conds, tbl.Status.Eq(*filter.Status))
}
items, total, err := query.Where(conds...).Order(tbl.ID.Desc()).FindByPage(int(filter.Offset()), int(filter.Limit))
if err != nil {
return nil, err
}
return &requests.Pager{
Pagination: filter.Pagination,
Total: total,
Items: items,
}, nil
}
// AdminApproveJoinRequest 租户管理员通过加入申请(幂等)。
func (t *tenant) AdminApproveJoinRequest(ctx context.Context, tenantID, operatorUserID, requestID int64, reason string) (*models.TenantJoinRequest, error) {
if tenantID <= 0 || operatorUserID <= 0 || requestID <= 0 {
return nil, errorx.ErrInvalidParameter.WithMsg("invalid tenant_id/operator_user_id/request_id")
}
now := time.Now().UTC()
reason = strings.TrimSpace(reason)
logrus.WithFields(logrus.Fields{
"tenant_id": tenantID,
"operator_user_id": operatorUserID,
"request_id": requestID,
}).Info("services.tenant.admin.approve_join_request")
var out models.TenantJoinRequest
err := _db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
var req models.TenantJoinRequest
if err := tx.
Clauses(clause.Locking{Strength: "UPDATE"}).
Where("id = ? AND tenant_id = ?", requestID, tenantID).
First(&req).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return errorx.ErrRecordNotFound.WithMsg("申请不存在")
}
return err
}
// 幂等:已通过则直接返回。
if req.Status == consts.TenantJoinRequestStatusApproved {
out = req
return nil
}
if req.Status != consts.TenantJoinRequestStatusPending {
return errorx.ErrPreconditionFailed.WithMsg("申请状态不可通过")
}
// 先落成员关系,再更新申请状态,保证“通过后一定能成为成员”(至少幂等)。
tu := &models.TenantUser{
TenantID: tenantID,
UserID: req.UserID,
Role: types.NewArray([]consts.TenantUserRole{consts.TenantUserRoleMember}),
Status: consts.UserStatusVerified,
Balance: 0,
BalanceFrozen: 0,
CreatedAt: now,
UpdatedAt: now,
}
if err := tx.Create(tu).Error; err != nil && !isUniqueViolation(err) {
return err
}
req.Status = consts.TenantJoinRequestStatusApproved
req.DecidedAt = now
req.DecidedOperatorUserID = operatorUserID
req.DecidedReason = reason
req.UpdatedAt = now
if err := tx.Save(&req).Error; err != nil {
return err
}
out = req
return nil
})
if err != nil {
return nil, err
}
return &out, nil
}
// AdminRejectJoinRequest 租户管理员拒绝加入申请(幂等)。
func (t *tenant) AdminRejectJoinRequest(ctx context.Context, tenantID, operatorUserID, requestID int64, reason string) (*models.TenantJoinRequest, error) {
if tenantID <= 0 || operatorUserID <= 0 || requestID <= 0 {
return nil, errorx.ErrInvalidParameter.WithMsg("invalid tenant_id/operator_user_id/request_id")
}
now := time.Now().UTC()
reason = strings.TrimSpace(reason)
logrus.WithFields(logrus.Fields{
"tenant_id": tenantID,
"operator_user_id": operatorUserID,
"request_id": requestID,
}).Info("services.tenant.admin.reject_join_request")
var out models.TenantJoinRequest
err := _db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
var req models.TenantJoinRequest
if err := tx.
Clauses(clause.Locking{Strength: "UPDATE"}).
Where("id = ? AND tenant_id = ?", requestID, tenantID).
First(&req).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return errorx.ErrRecordNotFound.WithMsg("申请不存在")
}
return err
}
// 幂等:已拒绝则直接返回。
if req.Status == consts.TenantJoinRequestStatusRejected {
out = req
return nil
}
if req.Status != consts.TenantJoinRequestStatusPending {
return errorx.ErrPreconditionFailed.WithMsg("申请状态不可拒绝")
}
req.Status = consts.TenantJoinRequestStatusRejected
req.DecidedAt = now
req.DecidedOperatorUserID = operatorUserID
req.DecidedReason = reason
req.UpdatedAt = now
if err := tx.Save(&req).Error; err != nil {
return err
}
out = req
return nil
})
if err != nil {
return nil, err
}
return &out, nil
}

View File

@@ -0,0 +1,369 @@
package services
import (
"context"
"database/sql"
"errors"
"testing"
"time"
"quyun/v2/app/commands/testx"
"quyun/v2/app/errorx"
tenantdto "quyun/v2/app/http/tenant/dto"
tenantjoindto "quyun/v2/app/http/tenantjoin/dto"
"quyun/v2/database"
"quyun/v2/database/models"
"quyun/v2/pkg/consts"
. "github.com/smartystreets/goconvey/convey"
"github.com/stretchr/testify/suite"
_ "go.ipao.vip/atom"
"go.ipao.vip/atom/contracts"
"go.ipao.vip/gen/types"
"go.uber.org/dig"
)
type TenantJoinTestSuiteInjectParams struct {
dig.In
DB *sql.DB
Initials []contracts.Initial `group:"initials"` // nolint:structcheck
}
type TenantJoinTestSuite struct {
suite.Suite
TenantJoinTestSuiteInjectParams
}
func Test_TenantJoin(t *testing.T) {
providers := testx.Default().With(Provide)
testx.Serve(providers, t, func(p TenantJoinTestSuiteInjectParams) {
suite.Run(t, &TenantJoinTestSuite{TenantJoinTestSuiteInjectParams: p})
})
}
func (s *TenantJoinTestSuite) truncateAll(ctx context.Context) {
So(database.Truncate(ctx, s.DB,
models.TableNameTenantInvite,
models.TableNameTenantJoinRequest,
models.TableNameTenantUser,
), ShouldBeNil)
}
func (s *TenantJoinTestSuite) seedInvite(ctx context.Context, tenantID, creatorUserID int64, code string, status consts.TenantInviteStatus, maxUses, usedCount int32, expiresAt time.Time) *models.TenantInvite {
inv := &models.TenantInvite{
TenantID: tenantID,
UserID: creatorUserID,
Code: code,
Status: status,
MaxUses: maxUses,
UsedCount: usedCount,
ExpiresAt: expiresAt,
Remark: "seed",
CreatedAt: time.Now().UTC(),
UpdatedAt: time.Now().UTC(),
}
So(_db.WithContext(ctx).Create(inv).Error, ShouldBeNil)
return inv
}
func (s *TenantJoinTestSuite) seedJoinRequest(ctx context.Context, tenantID, userID int64, status consts.TenantJoinRequestStatus) *models.TenantJoinRequest {
req := &models.TenantJoinRequest{
TenantID: tenantID,
UserID: userID,
Status: status,
Reason: "seed",
DecidedAt: time.Time{},
DecidedReason: "",
CreatedAt: time.Now().UTC(),
UpdatedAt: time.Now().UTC(),
}
So(_db.WithContext(ctx).Omit("decided_at", "decided_operator_user_id").Create(req).Error, ShouldBeNil)
return req
}
func (s *TenantJoinTestSuite) Test_AdminCreateInvite() {
Convey("Tenant.AdminCreateInvite", s.T(), func() {
ctx := s.T().Context()
s.truncateAll(ctx)
tenantID := int64(1)
adminUserID := int64(10)
Convey("参数非法应返回参数错误", func() {
_, err := Tenant.AdminCreateInvite(ctx, 0, adminUserID, &tenantdto.AdminTenantInviteCreateForm{})
So(err, ShouldNotBeNil)
var appErr *errorx.AppError
So(errors.As(err, &appErr), ShouldBeTrue)
So(appErr.Code, ShouldEqual, errorx.CodeInvalidParameter)
_, err = Tenant.AdminCreateInvite(ctx, tenantID, 0, &tenantdto.AdminTenantInviteCreateForm{})
So(err, ShouldNotBeNil)
})
Convey("code 为空应自动生成并创建成功", func() {
out, err := Tenant.AdminCreateInvite(ctx, tenantID, adminUserID, &tenantdto.AdminTenantInviteCreateForm{
Code: "",
MaxUses: loToPtr(0),
Remark: "test",
})
So(err, ShouldBeNil)
So(out, ShouldNotBeNil)
So(out.ID, ShouldBeGreaterThan, 0)
So(out.Code, ShouldNotBeBlank)
So(out.Status, ShouldEqual, consts.TenantInviteStatusActive)
})
Convey("重复 code 应返回重复错误", func() {
_, err := Tenant.AdminCreateInvite(ctx, tenantID, adminUserID, &tenantdto.AdminTenantInviteCreateForm{
Code: "dup_code",
MaxUses: loToPtr(0),
})
So(err, ShouldBeNil)
_, err = Tenant.AdminCreateInvite(ctx, tenantID, adminUserID, &tenantdto.AdminTenantInviteCreateForm{
Code: "dup_code",
MaxUses: loToPtr(0),
})
So(err, ShouldNotBeNil)
var appErr *errorx.AppError
So(errors.As(err, &appErr), ShouldBeTrue)
So(appErr.Code, ShouldEqual, errorx.CodeRecordDuplicated)
})
})
}
func (s *TenantJoinTestSuite) Test_AdminDisableInvite() {
Convey("Tenant.AdminDisableInvite", s.T(), func() {
ctx := s.T().Context()
s.truncateAll(ctx)
tenantID := int64(1)
adminUserID := int64(10)
Convey("邀请码不存在应返回记录不存在", func() {
_, err := Tenant.AdminDisableInvite(ctx, tenantID, adminUserID, 999, "x")
So(err, ShouldNotBeNil)
var appErr *errorx.AppError
So(errors.As(err, &appErr), ShouldBeTrue)
So(appErr.Code, ShouldEqual, errorx.CodeRecordNotFound)
})
Convey("禁用应成功且幂等", func() {
inv := s.seedInvite(ctx, tenantID, adminUserID, "c1", consts.TenantInviteStatusActive, 0, 0, time.Time{})
out, err := Tenant.AdminDisableInvite(ctx, tenantID, adminUserID, inv.ID, "reason")
So(err, ShouldBeNil)
So(out.Status, ShouldEqual, consts.TenantInviteStatusDisabled)
So(out.DisabledOperatorUserID, ShouldEqual, adminUserID)
So(out.DisabledAt.IsZero(), ShouldBeFalse)
out2, err := Tenant.AdminDisableInvite(ctx, tenantID, adminUserID, inv.ID, "reason2")
So(err, ShouldBeNil)
So(out2.Status, ShouldEqual, consts.TenantInviteStatusDisabled)
})
})
}
func (s *TenantJoinTestSuite) Test_AdminInvitePage() {
Convey("Tenant.AdminInvitePage", s.T(), func() {
ctx := s.T().Context()
s.truncateAll(ctx)
tenantID := int64(1)
adminUserID := int64(10)
s.seedInvite(ctx, tenantID, adminUserID, "aaa", consts.TenantInviteStatusActive, 0, 0, time.Time{})
s.seedInvite(ctx, tenantID, adminUserID, "bbb", consts.TenantInviteStatusDisabled, 0, 0, time.Time{})
s.seedInvite(ctx, tenantID, adminUserID, "ccc", consts.TenantInviteStatusActive, 0, 0, time.Time{})
Convey("按 status 过滤应只返回匹配项", func() {
st := consts.TenantInviteStatusActive
pager, err := Tenant.AdminInvitePage(ctx, tenantID, &tenantdto.AdminTenantInviteListFilter{
Status: &st,
})
So(err, ShouldBeNil)
So(pager.Total, ShouldEqual, 2)
})
Convey("按 code 模糊过滤应生效", func() {
code := "bb"
pager, err := Tenant.AdminInvitePage(ctx, tenantID, &tenantdto.AdminTenantInviteListFilter{
Code: &code,
})
So(err, ShouldBeNil)
So(pager.Total, ShouldEqual, 1)
})
})
}
func (s *TenantJoinTestSuite) Test_JoinByInvite() {
Convey("Tenant.JoinByInvite", s.T(), func() {
ctx := s.T().Context()
s.truncateAll(ctx)
tenantID := int64(1)
adminUserID := int64(10)
userID := int64(20)
Convey("邀请码不存在应返回记录不存在", func() {
_, err := Tenant.JoinByInvite(ctx, tenantID, userID, "not_exist")
So(err, ShouldNotBeNil)
var appErr *errorx.AppError
So(errors.As(err, &appErr), ShouldBeTrue)
So(appErr.Code, ShouldEqual, errorx.CodeRecordNotFound)
})
Convey("成功加入应创建成员并消耗邀请码次数", func() {
inv := s.seedInvite(ctx, tenantID, adminUserID, "code1", consts.TenantInviteStatusActive, 1, 0, time.Now().UTC().Add(10*time.Minute))
tu, err := Tenant.JoinByInvite(ctx, tenantID, userID, "code1")
So(err, ShouldBeNil)
So(tu, ShouldNotBeNil)
So(tu.TenantID, ShouldEqual, tenantID)
So(tu.UserID, ShouldEqual, userID)
var inv2 models.TenantInvite
So(_db.WithContext(ctx).Where("id = ?", inv.ID).First(&inv2).Error, ShouldBeNil)
So(inv2.UsedCount, ShouldEqual, 1)
So(inv2.Status, ShouldEqual, consts.TenantInviteStatusExpired)
Convey("重复加入应幂等且不再消耗次数", func() {
_, err := Tenant.JoinByInvite(ctx, tenantID, userID, "code1")
So(err, ShouldBeNil)
var inv3 models.TenantInvite
So(_db.WithContext(ctx).Where("id = ?", inv.ID).First(&inv3).Error, ShouldBeNil)
So(inv3.UsedCount, ShouldEqual, 1)
})
})
})
}
func (s *TenantJoinTestSuite) Test_CreateJoinRequest() {
Convey("Tenant.CreateJoinRequest", s.T(), func() {
ctx := s.T().Context()
s.truncateAll(ctx)
tenantID := int64(1)
userID := int64(20)
Convey("已是成员应返回前置条件失败", func() {
So(_db.WithContext(ctx).Create(&models.TenantUser{
TenantID: tenantID,
UserID: userID,
Role: types.NewArray([]consts.TenantUserRole{consts.TenantUserRoleMember}),
Status: consts.UserStatusVerified,
}).Error, ShouldBeNil)
_, err := Tenant.CreateJoinRequest(ctx, tenantID, userID, &tenantjoindto.JoinRequestCreateForm{Reason: "x"})
So(err, ShouldNotBeNil)
var appErr *errorx.AppError
So(errors.As(err, &appErr), ShouldBeTrue)
So(appErr.Code, ShouldEqual, errorx.CodePreconditionFailed)
})
Convey("重复提交应返回同一个 pending 申请(幂等)", func() {
s.truncateAll(ctx)
out1, err := Tenant.CreateJoinRequest(ctx, tenantID, userID, &tenantjoindto.JoinRequestCreateForm{Reason: "a"})
So(err, ShouldBeNil)
So(out1, ShouldNotBeNil)
out2, err := Tenant.CreateJoinRequest(ctx, tenantID, userID, &tenantjoindto.JoinRequestCreateForm{Reason: "b"})
So(err, ShouldBeNil)
So(out2, ShouldNotBeNil)
So(out2.ID, ShouldEqual, out1.ID)
So(out2.Status, ShouldEqual, consts.TenantJoinRequestStatusPending)
})
})
}
func (s *TenantJoinTestSuite) Test_AdminJoinRequestPage() {
Convey("Tenant.AdminJoinRequestPage", s.T(), func() {
ctx := s.T().Context()
s.truncateAll(ctx)
tenantID := int64(1)
s.seedJoinRequest(ctx, tenantID, 11, consts.TenantJoinRequestStatusPending)
s.seedJoinRequest(ctx, tenantID, 22, consts.TenantJoinRequestStatusRejected)
s.seedJoinRequest(ctx, tenantID, 33, consts.TenantJoinRequestStatusPending)
Convey("按 status 过滤应生效", func() {
st := consts.TenantJoinRequestStatusPending
pager, err := Tenant.AdminJoinRequestPage(ctx, tenantID, &tenantdto.AdminTenantJoinRequestListFilter{Status: &st})
So(err, ShouldBeNil)
So(pager.Total, ShouldEqual, 2)
})
})
}
func (s *TenantJoinTestSuite) Test_AdminApproveJoinRequest() {
Convey("Tenant.AdminApproveJoinRequest", s.T(), func() {
ctx := s.T().Context()
s.truncateAll(ctx)
tenantID := int64(1)
adminUserID := int64(10)
userID := int64(20)
Convey("申请不存在应返回记录不存在", func() {
_, err := Tenant.AdminApproveJoinRequest(ctx, tenantID, adminUserID, 999, "x")
So(err, ShouldNotBeNil)
var appErr *errorx.AppError
So(errors.As(err, &appErr), ShouldBeTrue)
So(appErr.Code, ShouldEqual, errorx.CodeRecordNotFound)
})
Convey("通过 pending 申请应成功且幂等", func() {
req := s.seedJoinRequest(ctx, tenantID, userID, consts.TenantJoinRequestStatusPending)
out, err := Tenant.AdminApproveJoinRequest(ctx, tenantID, adminUserID, req.ID, "ok")
So(err, ShouldBeNil)
So(out.Status, ShouldEqual, consts.TenantJoinRequestStatusApproved)
So(out.DecidedOperatorUserID, ShouldEqual, adminUserID)
So(out.DecidedAt.IsZero(), ShouldBeFalse)
var tu models.TenantUser
So(_db.WithContext(ctx).Where("tenant_id = ? AND user_id = ?", tenantID, userID).First(&tu).Error, ShouldBeNil)
out2, err := Tenant.AdminApproveJoinRequest(ctx, tenantID, adminUserID, req.ID, "ok2")
So(err, ShouldBeNil)
So(out2.Status, ShouldEqual, consts.TenantJoinRequestStatusApproved)
})
})
}
func (s *TenantJoinTestSuite) Test_AdminRejectJoinRequest() {
Convey("Tenant.AdminRejectJoinRequest", s.T(), func() {
ctx := s.T().Context()
s.truncateAll(ctx)
tenantID := int64(1)
adminUserID := int64(10)
userID := int64(20)
Convey("拒绝 pending 申请应成功且幂等", func() {
req := s.seedJoinRequest(ctx, tenantID, userID, consts.TenantJoinRequestStatusPending)
out, err := Tenant.AdminRejectJoinRequest(ctx, tenantID, adminUserID, req.ID, "no")
So(err, ShouldBeNil)
So(out.Status, ShouldEqual, consts.TenantJoinRequestStatusRejected)
So(out.DecidedOperatorUserID, ShouldEqual, adminUserID)
out2, err := Tenant.AdminRejectJoinRequest(ctx, tenantID, adminUserID, req.ID, "no2")
So(err, ShouldBeNil)
So(out2.Status, ShouldEqual, consts.TenantJoinRequestStatusRejected)
})
})
}
func loToPtr[T any](v T) *T { return &v }