Initial commit

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
2026-09-15 13:57:10 +08:00
commit 8679200f41
1897 changed files with 257900 additions and 0 deletions
@@ -0,0 +1,108 @@
+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
+}