shanshanzhong147
b162022d39
MySQL DATETIME 带小数秒精度时, UNIX_TIMESTAMP() 返回 "1779966403.631" 形式的 DECIMAL, Go 端 int64 字段无法 Scan, 触发 100 条 Scan error, 前端拿到的 invited_at/created_at 全为 0。 改用 CAST(UNIX_TIMESTAMP(...) AS SIGNED) 在 SQL 层截断转 BIGINT, 与 getInviteRecordsLogic / getInviteSalesLogic 的现有约定一致。
106 lines
3.5 KiB
Go
106 lines
3.5 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, CAST(UNIX_TIMESTAMP(u.created_at) AS SIGNED) 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
|
|
}
|