135 lines
3.9 KiB
Go
135 lines
3.9 KiB
Go
package task
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"server/internal/dao"
|
||
"server/internal/model"
|
||
"server/internal/service"
|
||
"server/utility/ecode"
|
||
"sort"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
type sTask struct{}
|
||
|
||
func New() service.ITask {
|
||
return &sTask{}
|
||
}
|
||
|
||
func init() {
|
||
service.RegisterTask(New())
|
||
}
|
||
|
||
func (s *sTask) UserTaskRankingList(ctx context.Context, in *model.UserTaskRankingIn) (out *model.UserTaskRankingOut, err error) {
|
||
|
||
// 判断排行类型
|
||
var start, end time.Time
|
||
if in.Type == 1 {
|
||
// 日
|
||
now := time.Now()
|
||
// 当天开始时间
|
||
start = time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location())
|
||
// 当天结束时间
|
||
end = time.Date(now.Year(), now.Month(), now.Day(), 23, 59, 59, 0, now.Location())
|
||
|
||
fmt.Println("开始时间:", start)
|
||
fmt.Println("结束时间:", end)
|
||
|
||
} else if in.Type == 2 {
|
||
// 周
|
||
now := time.Now()
|
||
loc := now.Location()
|
||
weekday := int(now.Weekday())
|
||
if weekday == 0 { // 周日是0
|
||
weekday = 7
|
||
}
|
||
// 本周开始时间（周一 00:00:00）
|
||
start = time.Date(now.Year(), now.Month(), now.Day()-weekday+1, 0, 0, 0, 0, loc)
|
||
// 本周结束时间（周日 23:59:59）
|
||
end = time.Date(now.Year(), now.Month(), now.Day()-weekday+7, 23, 59, 59, 0, loc)
|
||
} else if in.Type == 3 {
|
||
// 月
|
||
now := time.Now()
|
||
loc := now.Location()
|
||
// 本月开始时间（1号 00:00:00）
|
||
start = time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, loc)
|
||
// 下月1号
|
||
nextMonth := start.AddDate(0, 1, 0)
|
||
// 本月结束时间（本月最后一天 23:59:59）
|
||
end = nextMonth.Add(-time.Second)
|
||
} else {
|
||
return nil, ecode.Params.Sub("排行方式错误")
|
||
}
|
||
m := dao.UserTasks.Ctx(ctx)
|
||
|
||
// 构建查询条件
|
||
if in.Page == 0 {
|
||
in.Page = 1
|
||
}
|
||
|
||
if in.Size == 0 {
|
||
in.Size = 10
|
||
}
|
||
|
||
if in.StoreId > 0 {
|
||
m = m.Where(dao.UserTasks.Columns().StoreId, in.StoreId)
|
||
}
|
||
|
||
list := make([]model.UserTaskRankingArgs, 0)
|
||
var total int
|
||
// SELECT `uid`,`nickname` FROM `user` ORDER BY `uid` asc
|
||
err = m.Page(in.Page, in.Size).LeftJoin(dao.Users.Table(), fmt.Sprintf("`%s`.`id` = `%s`.`user_id`", dao.Users.Table(), dao.UserTasks.Table())).
|
||
Fields("username,avatar,count(*) num").Where(dao.UserTasks.Columns().Status, 3).
|
||
WhereBetween(dao.UserTasks.Columns().CompletedAt, start, end).OrderDesc("num").OrderDesc("username").Group("user_id").
|
||
ScanAndCount(&list, &total, false)
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("查询排行榜失败")
|
||
}
|
||
|
||
// 判断登录查询当前任务完成数量，排名
|
||
var rankingNum int
|
||
var loginUserRanking []model.LoginUserRanking
|
||
var loginUserRankingNum model.LoginUserRankingNum
|
||
if in.OperatorId != 0 {
|
||
err := dao.UserTasks.Ctx(ctx).Fields("count(*) num").Where(dao.UserTasks.Columns().UserId, in.OperatorId).Where(dao.UserTasks.Columns().Status, 3).
|
||
WhereBetween(dao.UserTasks.Columns().CompletedAt, start, end).Group("user_id").
|
||
Scan(&loginUserRankingNum)
|
||
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("查询当前登录用户完成数失败")
|
||
}
|
||
|
||
err = dao.UserTasks.Ctx(ctx).LeftJoin(dao.Users.Table(), fmt.Sprintf("`%s`.`id` = `%s`.`user_id`", dao.Users.Table(), dao.UserTasks.Table())).
|
||
Fields("username,count(*) num").Where(dao.UserTasks.Columns().Status, 3).
|
||
WhereBetween(dao.UserTasks.Columns().CompletedAt, start, end).OrderDesc("num").OrderDesc("username").Group("user_id").
|
||
Scan(&loginUserRanking)
|
||
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("查询当前登录用户排名失败")
|
||
}
|
||
|
||
// 查找当前登录用户排名
|
||
//index := sort.Search(len(loginUserRanking), func(i int) bool {
|
||
// return loginUserRanking[i].Num <= loginUserRankingNum.Num
|
||
//})
|
||
|
||
index, flag := sort.Find(len(loginUserRanking), func(i int) int {
|
||
return strings.Compare(strconv.Itoa(loginUserRanking[i].Num), strconv.Itoa(loginUserRankingNum.Num))
|
||
})
|
||
|
||
if flag {
|
||
rankingNum = index + 1
|
||
}
|
||
}
|
||
|
||
return &model.UserTaskRankingOut{
|
||
List: list,
|
||
Total: total,
|
||
CompletedNum: loginUserRankingNum.Num,
|
||
RankingNum: rankingNum,
|
||
}, nil
|
||
}
|