shanshanzhong147
1022160ff8
P01: GET /v1/admin/user/invite/list 增加 search/enable/user_id_search 筛选参数，响应新增 order_count/has_purchased/inviter_commission/ inviter_gift_days/invitee_gift_days 权益字段。 P02: 新增 GET /v1/admin/invite/list 全局邀请管理接口，支持按 邀请人/被邀请人筛选和搜索，返回邀请关系及权益聚合数据。 共用 QueryBenefits 批量查询防 N+1，分页上限 100。 Co-authored-by: multica-agent <github@multica.ai>
106 lines
3.4 KiB
Go
106 lines
3.4 KiB
Go
package user
|
|
|
|
import (
|
|
"context"
|
|
|
|
adminInvite "github.com/perfect-panel/server/internal/logic/admin/invite"
|
|
"github.com/perfect-panel/server/internal/svc"
|
|
"github.com/perfect-panel/server/internal/types"
|
|
"github.com/perfect-panel/server/pkg/logger"
|
|
"github.com/perfect-panel/server/pkg/xerr"
|
|
"github.com/pkg/errors"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type GetAdminUserInviteListLogic struct {
|
|
logger.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewGetAdminUserInviteListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetAdminUserInviteListLogic {
|
|
return &GetAdminUserInviteListLogic{
|
|
Logger: logger.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
func (l *GetAdminUserInviteListLogic) GetAdminUserInviteList(req *types.GetAdminUserInviteListRequest) (resp *types.GetAdminUserInviteListResponse, err error) {
|
|
req.Page, req.Size = adminInvite.NormalizePage(req.Page, req.Size)
|
|
|
|
type InvitedUser struct {
|
|
Id int64 `gorm:"column:id"`
|
|
Avatar string `gorm:"column:avatar"`
|
|
Identifier string `gorm:"column:identifier"`
|
|
Enable bool `gorm:"column:enable"`
|
|
CreatedAt int64 `gorm:"column:created_at"`
|
|
}
|
|
|
|
var total int64
|
|
baseQuery := applyAdminUserInviteFilters(l.svcCtx.DB.WithContext(l.ctx).
|
|
Table("user u").
|
|
Where("u.referer_id = ? AND u.deleted_at IS NULL", req.UserId), req)
|
|
|
|
if err = baseQuery.Count(&total).Error; err != nil {
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "count invited users failed: %v", err)
|
|
}
|
|
|
|
var rows []InvitedUser
|
|
err = applyAdminUserInviteFilters(l.svcCtx.DB.WithContext(l.ctx).
|
|
Table("user u").
|
|
Select("u.id, u.avatar, u.enable, UNIX_TIMESTAMP(u.created_at) as created_at, COALESCE((SELECT uam.auth_identifier FROM user_auth_methods uam WHERE uam.user_id = u.id ORDER BY uam.id ASC LIMIT 1), '') as identifier").
|
|
Where("u.referer_id = ? AND u.deleted_at IS NULL", req.UserId), req).
|
|
Order("u.created_at DESC").
|
|
Limit(req.Size).
|
|
Offset((req.Page - 1) * req.Size).
|
|
Scan(&rows).Error
|
|
if err != nil {
|
|
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "query invited users failed: %v", err)
|
|
}
|
|
|
|
relations := make([]adminInvite.InviteRelation, 0, len(rows))
|
|
for _, r := range rows {
|
|
relations = append(relations, adminInvite.InviteRelation{InviteeId: r.Id, InviterId: req.UserId})
|
|
}
|
|
benefits, err := adminInvite.QueryBenefits(l.ctx, l.svcCtx.DB, relations)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
list := make([]types.AdminInvitedUser, 0, len(rows))
|
|
for _, r := range rows {
|
|
benefit := benefits[r.Id]
|
|
list = append(list, types.AdminInvitedUser{
|
|
Id: r.Id,
|
|
Avatar: r.Avatar,
|
|
Identifier: r.Identifier,
|
|
Enable: r.Enable,
|
|
CreatedAt: r.CreatedAt,
|
|
OrderCount: benefit.OrderCount,
|
|
HasPurchased: benefit.HasPurchased,
|
|
InviterCommission: benefit.InviterCommission,
|
|
InviterGiftDays: benefit.InviterGiftDays,
|
|
InviteeGiftDays: benefit.InviteeGiftDays,
|
|
})
|
|
}
|
|
|
|
return &types.GetAdminUserInviteListResponse{
|
|
Total: total,
|
|
List: list,
|
|
}, nil
|
|
}
|
|
|
|
func applyAdminUserInviteFilters(db *gorm.DB, req *types.GetAdminUserInviteListRequest) *gorm.DB {
|
|
if req.Search != "" {
|
|
db = db.Where("EXISTS (SELECT 1 FROM user_auth_methods uam WHERE uam.user_id = u.id AND uam.auth_identifier LIKE ?)", "%"+req.Search+"%")
|
|
}
|
|
if req.Enable != nil {
|
|
db = db.Where("u.enable = ?", *req.Enable)
|
|
}
|
|
if req.UserIdSearch > 0 {
|
|
db = db.Where("u.id = ?", req.UserIdSearch)
|
|
}
|
|
return db
|
|
}
|