109 lines
2.8 KiB
Go
109 lines
2.8 KiB
Go
package lotterylgmod
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"91porn-server/common/db"
|
|
"91porn-server/common/log"
|
|
"91porn-server/common/timeutil/timerange"
|
|
"91porn-server/models"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
const table = models.LotteryLog
|
|
|
|
var mdb *db.MongoDB
|
|
|
|
func coll(t *db.MongoTool) *db.MongoTool {
|
|
if t == nil {
|
|
return mdb.Coll(table)
|
|
}
|
|
return t.Coll(table)
|
|
}
|
|
|
|
// initIndexAds 初始化索引
|
|
func initIndex() {
|
|
many := []mongo.IndexModel{
|
|
{
|
|
Keys: bson.D{{Key: "uid", Value: 1}, {Key: "createdAt", Value: 1}},
|
|
},
|
|
{
|
|
Keys: bson.D{{Key: "code", Value: 1}},
|
|
},
|
|
{
|
|
Keys: bson.D{{Key: "createdAt", Value: 1}},
|
|
},
|
|
}
|
|
if _, err := coll(nil).CreateIndex(many); err != nil {
|
|
panic(fmt.Sprintf("%s model set index err ==>[%+v]", table, err))
|
|
}
|
|
}
|
|
|
|
// InsertLog 记录一个抽奖号行为
|
|
func InsertLog(uid uint64, code int, gateName string) error {
|
|
ll := LotteryLog{
|
|
UID: uid,
|
|
Code: code,
|
|
GateName: gateName,
|
|
CreatedAt: time.Now(),
|
|
}
|
|
if _, err := coll(nil).InsertOne(ll); err != nil {
|
|
log.Error("InsertClickLog error", log.Any("ll", ll), log.E(err))
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// UserTodayChances 获取用户当日抽奖号数量
|
|
func UserTodayChances(uid uint64) (int, error) {
|
|
sumDate := timerange.LocDayRange(time.Now()).Head
|
|
ll := []LotteryLog{}
|
|
cond := bson.M{"uid": uid, "createdAt": bson.M{"$gte": sumDate}}
|
|
if err := coll(nil).Find(&ll, cond); err != nil {
|
|
log.Error("UserTodayChances error", log.Any("uid", uid), log.E(err))
|
|
return 0, err
|
|
}
|
|
return len(ll), nil
|
|
}
|
|
|
|
// LottnumList 获取摇号列表
|
|
func LottnumList(page, size uint64, start *time.Time, end *time.Time, uid *uint64, code *int) ([]LotteryLog, int64, error) {
|
|
back := []LotteryLog{}
|
|
cond := bson.M{}
|
|
if uid != nil {
|
|
cond["uid"] = *uid
|
|
}
|
|
if code != nil {
|
|
cond["code"] = *code
|
|
}
|
|
if start != nil && end != nil {
|
|
cond["createdAt"] = bson.M{"$gte": *start, "$lt": *end}
|
|
}
|
|
sort := bson.D{{Key: "createdAt", Value: -1}}
|
|
skip := (page - 1) * size
|
|
opts := options.FindOptions{}
|
|
opts.SetSort(sort).SetSkip(int64(skip)).SetLimit(int64(size))
|
|
if err := coll(nil).Find(&back, bson.M(cond), &opts); err != nil {
|
|
log.Error("LottnumList error", log.Any("page", page), log.Any("size", size), log.Any("cond", cond), log.E(err))
|
|
return back, 0, err
|
|
}
|
|
total, _ := coll(nil).Count(cond)
|
|
return back, total, nil
|
|
}
|
|
|
|
// UserTodayNum 获取用户当日抽奖号
|
|
func UserTodayNum(uid uint64) ([]LotteryLog, error) {
|
|
sumDate := timerange.LocDayRange(time.Now()).Head
|
|
ll := []LotteryLog{}
|
|
cond := bson.M{"uid": uid, "createdAt": bson.M{"$gte": sumDate}}
|
|
if err := coll(nil).Find(&ll, cond); err != nil {
|
|
log.Error("UserTodayChances error", log.Any("uid", uid), log.E(err))
|
|
return ll, err
|
|
}
|
|
return ll, nil
|
|
}
|