v1.0
This commit is contained in:
@@ -2,6 +2,7 @@ package agent
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
"ycc-server/app/main/model"
|
||||
"ycc-server/common/ctxdata"
|
||||
"ycc-server/common/globalkey"
|
||||
@@ -45,21 +46,68 @@ func (l *GetTeamStatisticsLogic) GetTeamStatistics() (resp *types.TeamStatistics
|
||||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err)
|
||||
}
|
||||
|
||||
// 2. 确定团队首领ID
|
||||
teamLeaderId := agent.Id
|
||||
if agent.TeamLeaderId.Valid {
|
||||
teamLeaderId = agent.TeamLeaderId.Int64
|
||||
// 2. 递归查询所有下级（直接+间接）
|
||||
allSubordinateIds := make(map[int64]bool)
|
||||
directSubordinateIds := make(map[int64]bool)
|
||||
|
||||
// 递归函数：收集所有下级ID
|
||||
var collectSubordinates func(int64) error
|
||||
collectSubordinates = func(parentId int64) error {
|
||||
// 查询直接下级
|
||||
builder := l.svcCtx.AgentRelationModel.SelectBuilder().
|
||||
Where("parent_id = ? AND relation_type = ? AND del_state = ?", parentId, 1, globalkey.DelStateNo)
|
||||
relations, err := l.svcCtx.AgentRelationModel.FindAll(l.ctx, builder, "")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, relation := range relations {
|
||||
// 如果是第一层，标记为直接下级
|
||||
if parentId == agent.Id {
|
||||
directSubordinateIds[relation.ChildId] = true
|
||||
}
|
||||
// 添加到所有下级集合
|
||||
allSubordinateIds[relation.ChildId] = true
|
||||
// 递归查询下级的下级
|
||||
if err := collectSubordinates(relation.ChildId); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// 3. 查询团队所有成员（通过 team_leader_id）
|
||||
// 开始递归收集所有下级
|
||||
if err := collectSubordinates(agent.Id); err != nil {
|
||||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询下级关系失败, %v", err)
|
||||
}
|
||||
|
||||
// 3. 获取当前时间用于统计今日和本月新增
|
||||
now := time.Now()
|
||||
todayStart := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location())
|
||||
monthStart := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, now.Location())
|
||||
|
||||
// 4. 如果没有下级，返回空数据
|
||||
if len(allSubordinateIds) == 0 {
|
||||
return &types.TeamStatisticsResp{
|
||||
TotalCount: 0,
|
||||
DirectCount: 0,
|
||||
IndirectCount: 0,
|
||||
GoldCount: 0,
|
||||
NormalCount: 0,
|
||||
TodayNewMembers: 0,
|
||||
MonthNewMembers: 0,
|
||||
}, nil
|
||||
}
|
||||
|
||||
// 5. 将下级ID转换为切片用于查询
|
||||
subordinateIds := make([]int64, 0, len(allSubordinateIds))
|
||||
for id := range allSubordinateIds {
|
||||
subordinateIds = append(subordinateIds, id)
|
||||
}
|
||||
|
||||
// 6. 查询所有下级代理信息
|
||||
builder := l.svcCtx.AgentModel.SelectBuilder().
|
||||
Where(squirrel.Or{
|
||||
squirrel.Eq{"team_leader_id": teamLeaderId},
|
||||
squirrel.And{
|
||||
squirrel.Eq{"id": teamLeaderId},
|
||||
squirrel.Eq{"level": 3}, // 钻石代理自己也是团队首领
|
||||
},
|
||||
}).
|
||||
Where(squirrel.Eq{"id": subordinateIds}).
|
||||
Where("del_state = ?", globalkey.DelStateNo)
|
||||
|
||||
teamMembers, err := l.svcCtx.AgentModel.FindAll(l.ctx, builder, "")
|
||||
@@ -67,51 +115,43 @@ func (l *GetTeamStatisticsLogic) GetTeamStatistics() (resp *types.TeamStatistics
|
||||
return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询团队成员失败, %v", err)
|
||||
}
|
||||
|
||||
// 4. 统计
|
||||
totalCount := int64(len(teamMembers))
|
||||
// 7. 统计
|
||||
totalCount := int64(len(teamMembers)) // 下级总数（不包括自己）
|
||||
directCount := int64(len(directSubordinateIds))
|
||||
indirectCount := totalCount - directCount
|
||||
|
||||
level1Count := int64(0) // 普通
|
||||
level2Count := int64(0) // 黄金
|
||||
level3Count := int64(0) // 钻石
|
||||
// 不再统计钻石，因为下级不可能是钻石
|
||||
|
||||
// 统计直接下级
|
||||
directCount := int64(0)
|
||||
builder = l.svcCtx.AgentRelationModel.SelectBuilder().
|
||||
Where("parent_id = ? AND relation_type = ? AND del_state = ?", agent.Id, 1, globalkey.DelStateNo)
|
||||
directRelations, err := l.svcCtx.AgentRelationModel.FindAll(l.ctx, builder, "")
|
||||
if err == nil {
|
||||
directCount = int64(len(directRelations))
|
||||
}
|
||||
todayNewCount := int64(0)
|
||||
monthNewCount := int64(0)
|
||||
|
||||
for _, member := range teamMembers {
|
||||
// 排除自己
|
||||
if member.Id == agent.Id {
|
||||
continue
|
||||
}
|
||||
|
||||
// 统计等级（只统计普通和黄金）
|
||||
switch member.Level {
|
||||
case 1:
|
||||
level1Count++
|
||||
case 2:
|
||||
level2Count++
|
||||
case 3:
|
||||
level3Count++
|
||||
}
|
||||
|
||||
// 统计今日和本月新增
|
||||
if member.CreateTime.After(todayStart) {
|
||||
todayNewCount++
|
||||
}
|
||||
if member.CreateTime.After(monthStart) {
|
||||
monthNewCount++
|
||||
}
|
||||
}
|
||||
|
||||
// 间接下级 = 总人数 - 直接下级 - 自己
|
||||
indirectCount := totalCount - directCount - 1
|
||||
if indirectCount < 0 {
|
||||
indirectCount = 0
|
||||
}
|
||||
|
||||
return &types.TeamStatisticsResp{
|
||||
TotalCount: totalCount - 1, // 排除自己
|
||||
DirectCount: directCount,
|
||||
IndirectCount: indirectCount,
|
||||
ByLevel: types.TeamLevelStats{
|
||||
Normal: level1Count,
|
||||
Gold: level2Count,
|
||||
Diamond: level3Count,
|
||||
},
|
||||
TotalCount: totalCount, // 下级总数（不包括自己）
|
||||
DirectCount: directCount,
|
||||
IndirectCount: indirectCount,
|
||||
GoldCount: level2Count,
|
||||
NormalCount: level1Count,
|
||||
TodayNewMembers: todayNewCount,
|
||||
MonthNewMembers: monthNewCount,
|
||||
}, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user