405 lines
20 KiB
Go
405 lines
20 KiB
Go
package main
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"log"
|
||
"time"
|
||
|
||
"go.mongodb.org/mongo-driver/bson"
|
||
"go.mongodb.org/mongo-driver/bson/primitive"
|
||
"go.mongodb.org/mongo-driver/mongo"
|
||
"go.mongodb.org/mongo-driver/mongo/options"
|
||
)
|
||
|
||
const mongoURI = "mongodb://admin:TESTwfeZYP0kjBdy8DAUdwU@172.104.181.99:57017,172.104.181.99:57018,172.104.181.99:57019/91porn?replicaSet=yctest&maxPoolSize=200&connectTimeout=30&authSource=admin"
|
||
|
||
// PrizeType 与 prizemod.PrizeType 保持一致
|
||
const (
|
||
PrizeTypeCurrentValue = 1 // 活跃值
|
||
PrizeTypeGold = 2 // 金币
|
||
PrizeTypeVIPCard = 5 // 会员卡
|
||
PrizeTypeIntegral = 8 // 积分
|
||
PrizeTypeAIUndress = 9 // AI脱衣
|
||
PrizeTypeAIChangeFace = 10 // AI换脸
|
||
PrizeTypeVideoCoupon = 11 // 观影券
|
||
)
|
||
|
||
const CheckinTypeContinuously = 1
|
||
|
||
// activityPrizeDef 定义基础奖品
|
||
type activityPrizeDef struct {
|
||
Name string
|
||
Type int
|
||
Count int32
|
||
Price int64
|
||
Desc string
|
||
}
|
||
|
||
// checkinPrizeDef 定义签到奖品配置
|
||
type checkinPrizeDef struct {
|
||
Title string
|
||
CheckinDays int64
|
||
BigPrize bool
|
||
PrizeName string
|
||
PrizeRef string // 引用 activityPrizeDef 的 key
|
||
}
|
||
|
||
func main() {
|
||
ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
|
||
defer cancel()
|
||
|
||
client, err := mongo.Connect(ctx, options.Client().ApplyURI(mongoURI))
|
||
if err != nil {
|
||
log.Fatal("connect failed:", err)
|
||
}
|
||
defer client.Disconnect(ctx)
|
||
|
||
db := client.Database("91porn")
|
||
now := time.Now()
|
||
|
||
// ========================
|
||
// Step 0: 清理上次脚本插入的旧数据
|
||
// ========================
|
||
fmt.Println("=== Step 0: 清理旧签到配置数据 ===")
|
||
db.Collection("checkin_config").DeleteMany(ctx, bson.M{})
|
||
db.Collection("checkin_prize").DeleteMany(ctx, bson.M{})
|
||
// 清理上次脚本创建的 activity_prize（按名称前缀匹配）
|
||
delRes, _ := db.Collection("activity_prize").DeleteMany(ctx, bson.M{
|
||
"desc": bson.M{"$regex": "^签到奖励|^VIP签到"},
|
||
})
|
||
fmt.Printf(" 清理 activity_prize: %d, checkin_prize: all, checkin_config: all\n\n", delRes.DeletedCount)
|
||
|
||
// ========================
|
||
// Step 1: 创建 activity_prize 基础奖品
|
||
// ========================
|
||
fmt.Println("=== Step 1: 创建 activity_prize 基础奖品 ===")
|
||
|
||
activityPrizeColl := db.Collection("activity_prize")
|
||
|
||
// 定义所有需要的奖品规格
|
||
prizeDefs := map[string]activityPrizeDef{
|
||
// 积分类
|
||
"integral_3": {Name: "签到积分x3", Type: PrizeTypeIntegral, Count: 3, Price: 3, Desc: "签到奖励3积分"},
|
||
"integral_5": {Name: "签到积分x5", Type: PrizeTypeIntegral, Count: 5, Price: 5, Desc: "签到奖励5积分"},
|
||
"integral_8": {Name: "签到积分x8", Type: PrizeTypeIntegral, Count: 8, Price: 8, Desc: "签到奖励8积分"},
|
||
"integral_10": {Name: "签到积分x10", Type: PrizeTypeIntegral, Count: 10, Price: 10, Desc: "签到奖励10积分"},
|
||
"integral_12": {Name: "签到积分x12", Type: PrizeTypeIntegral, Count: 12, Price: 12, Desc: "签到奖励12积分"},
|
||
"integral_15": {Name: "签到积分x15", Type: PrizeTypeIntegral, Count: 15, Price: 15, Desc: "签到奖励15积分"},
|
||
"integral_18": {Name: "签到积分x18", Type: PrizeTypeIntegral, Count: 18, Price: 18, Desc: "签到奖励18积分"},
|
||
"integral_20": {Name: "签到积分x20", Type: PrizeTypeIntegral, Count: 20, Price: 20, Desc: "签到奖励20积分"},
|
||
"integral_25": {Name: "签到积分x25", Type: PrizeTypeIntegral, Count: 25, Price: 25, Desc: "签到奖励25积分"},
|
||
"integral_30": {Name: "签到积分x30", Type: PrizeTypeIntegral, Count: 30, Price: 30, Desc: "签到奖励30积分"},
|
||
"integral_50": {Name: "签到积分x50", Type: PrizeTypeIntegral, Count: 50, Price: 50, Desc: "签到奖励50积分"},
|
||
"integral_80": {Name: "签到积分x80", Type: PrizeTypeIntegral, Count: 80, Price: 80, Desc: "签到奖励80积分"},
|
||
"integral_100": {Name: "签到积分x100", Type: PrizeTypeIntegral, Count: 100, Price: 100, Desc: "签到奖励100积分"},
|
||
// 金币类
|
||
"gold_50": {Name: "签到金币x50", Type: PrizeTypeGold, Count: 50, Price: 50, Desc: "签到奖励50金币"},
|
||
"gold_100": {Name: "签到金币x100", Type: PrizeTypeGold, Count: 100, Price: 100, Desc: "签到奖励100金币"},
|
||
"gold_200": {Name: "签到金币x200", Type: PrizeTypeGold, Count: 200, Price: 200, Desc: "签到奖励200金币"},
|
||
"gold_500": {Name: "签到金币x500", Type: PrizeTypeGold, Count: 500, Price: 500, Desc: "签到奖励500金币"},
|
||
// 观影券
|
||
"video_coupon_1": {Name: "观影券x1", Type: PrizeTypeVideoCoupon, Count: 1, Price: 1, Desc: "签到奖励1张观影券"},
|
||
"video_coupon_2": {Name: "观影券x2", Type: PrizeTypeVideoCoupon, Count: 2, Price: 2, Desc: "签到奖励2张观影券"},
|
||
"video_coupon_3": {Name: "观影券x3", Type: PrizeTypeVideoCoupon, Count: 3, Price: 3, Desc: "签到奖励3张观影券"},
|
||
// AI
|
||
"ai_undress_1": {Name: "AI脱衣x1", Type: PrizeTypeAIUndress, Count: 1, Price: 1, Desc: "签到奖励AI脱衣1次"},
|
||
"ai_undress_2": {Name: "AI脱衣x2", Type: PrizeTypeAIUndress, Count: 2, Price: 2, Desc: "签到奖励AI脱衣2次"},
|
||
"ai_changeface_1": {Name: "AI换脸x1", Type: PrizeTypeAIChangeFace, Count: 1, Price: 1, Desc: "签到奖励AI换脸1次"},
|
||
"ai_changeface_2": {Name: "AI换脸x2", Type: PrizeTypeAIChangeFace, Count: 2, Price: 2, Desc: "签到奖励AI换脸2次"},
|
||
// VIP专属奖品
|
||
"vip_integral_5": {Name: "VIP积分x5", Type: PrizeTypeIntegral, Count: 5, Price: 5, Desc: "VIP签到额外5积分"},
|
||
"vip_integral_8": {Name: "VIP积分x8", Type: PrizeTypeIntegral, Count: 8, Price: 8, Desc: "VIP签到额外8积分"},
|
||
"vip_integral_10": {Name: "VIP积分x10", Type: PrizeTypeIntegral, Count: 10, Price: 10, Desc: "VIP签到额外10积分"},
|
||
"vip_integral_15": {Name: "VIP积分x15", Type: PrizeTypeIntegral, Count: 15, Price: 15, Desc: "VIP签到额外15积分"},
|
||
"vip_integral_20": {Name: "VIP积分x20", Type: PrizeTypeIntegral, Count: 20, Price: 20, Desc: "VIP签到额外20积分"},
|
||
"vip_integral_30": {Name: "VIP积分x30", Type: PrizeTypeIntegral, Count: 30, Price: 30, Desc: "VIP签到额外30积分"},
|
||
"vip_integral_50": {Name: "VIP积分x50", Type: PrizeTypeIntegral, Count: 50, Price: 50, Desc: "VIP签到额外50积分"},
|
||
"vip_gold_100": {Name: "VIP金币x100", Type: PrizeTypeGold, Count: 100, Price: 100, Desc: "VIP签到额外100金币"},
|
||
"vip_gold_200": {Name: "VIP金币x200", Type: PrizeTypeGold, Count: 200, Price: 200, Desc: "VIP签到额外200金币"},
|
||
"vip_gold_500": {Name: "VIP金币x500", Type: PrizeTypeGold, Count: 500, Price: 500, Desc: "VIP签到额外500金币"},
|
||
"vip_video_coupon_1": {Name: "VIP观影券x1", Type: PrizeTypeVideoCoupon, Count: 1, Price: 1, Desc: "VIP签到额外1张观影券"},
|
||
"vip_video_coupon_2": {Name: "VIP观影券x2", Type: PrizeTypeVideoCoupon, Count: 2, Price: 2, Desc: "VIP签到额外2张观影券"},
|
||
"vip_ai_undress_1": {Name: "VIP AI脱衣x1", Type: PrizeTypeAIUndress, Count: 1, Price: 1, Desc: "VIP签到额外AI脱衣1次"},
|
||
"vip_ai_undress_2": {Name: "VIP AI脱衣x2", Type: PrizeTypeAIUndress, Count: 2, Price: 2, Desc: "VIP签到额外AI脱衣2次"},
|
||
"vip_ai_changeface_1": {Name: "VIP AI换脸x1", Type: PrizeTypeAIChangeFace, Count: 1, Price: 1, Desc: "VIP签到额外AI换脸1次"},
|
||
"vip_ai_changeface_2": {Name: "VIP AI换脸x2", Type: PrizeTypeAIChangeFace, Count: 2, Price: 2, Desc: "VIP签到额外AI换脸2次"},
|
||
}
|
||
|
||
prizeIdMap := make(map[string]primitive.ObjectID)
|
||
for key, def := range prizeDefs {
|
||
doc := bson.M{
|
||
"name": def.Name, "type": def.Type, "count": def.Count, "price": def.Price,
|
||
"sort": 0, "desc": def.Desc, "value": 0, "weights": "0", "level": 1,
|
||
"status": true, "validityTime": 0, "image": "",
|
||
"updateTime": now, "createTimt": now,
|
||
}
|
||
result, err := activityPrizeColl.InsertOne(ctx, doc)
|
||
if err != nil {
|
||
log.Printf(" [WARN] 插入 '%s' 失败: %v", key, err)
|
||
continue
|
||
}
|
||
prizeIdMap[key] = result.InsertedID.(primitive.ObjectID)
|
||
}
|
||
fmt.Printf(" 共创建 %d 个 activity_prize\n\n", len(prizeIdMap))
|
||
|
||
// ========================
|
||
// Step 2: 创建 checkin_config
|
||
// ========================
|
||
fmt.Println("=== Step 2: 配置 checkin_config ===")
|
||
configColl := db.Collection("checkin_config")
|
||
configDoc := bson.M{
|
||
"enable": true,
|
||
"description": "每日签到领好礼！连续签到天数越多奖励越丰厚，第7天可获得双倍奖励！月卡VIP用户每天额外领取专属奖品。坚持签到31天有超级大礼！",
|
||
"backgroundImage": "",
|
||
"rewardBgVideos": []bson.M{
|
||
{"prizeType": PrizeTypeIntegral, "bgMediaUrl": ""},
|
||
{"prizeType": PrizeTypeGold, "bgMediaUrl": ""},
|
||
{"prizeType": PrizeTypeVideoCoupon, "bgMediaUrl": ""},
|
||
{"prizeType": PrizeTypeAIUndress, "bgMediaUrl": ""},
|
||
{"prizeType": PrizeTypeAIChangeFace, "bgMediaUrl": ""},
|
||
},
|
||
"integerExchangeList": []bson.M{
|
||
{"name": "月卡VIP", "icon": ""},
|
||
{"name": "观影券x10", "icon": ""},
|
||
{"name": "AI脱衣x5", "icon": ""},
|
||
{"name": "AI换脸x5", "icon": ""},
|
||
},
|
||
}
|
||
configColl.InsertOne(ctx, configDoc)
|
||
fmt.Println(" checkin_config 已创建（enable=true）\n")
|
||
|
||
// ========================
|
||
// Step 3: 创建 checkin_prize 31天连续签到（普通 + VIP）
|
||
// ========================
|
||
fmt.Println("=== Step 3: 创建 checkin_prize 31天签到奖品 ===")
|
||
checkinPrizeColl := db.Collection("checkin_prize")
|
||
|
||
// 31天普通奖品配置（奖励递增，每周有节奏感）
|
||
normalPrizes := []checkinPrizeDef{
|
||
// 第1周：基础奖励
|
||
{Title: "第1天", CheckinDays: 1, PrizeName: "积分x3", PrizeRef: "integral_3"},
|
||
{Title: "第2天", CheckinDays: 2, PrizeName: "积分x5", PrizeRef: "integral_5"},
|
||
{Title: "第3天", CheckinDays: 3, PrizeName: "积分x5", PrizeRef: "integral_5"},
|
||
{Title: "第4天", CheckinDays: 4, PrizeName: "积分x8", PrizeRef: "integral_8"},
|
||
{Title: "第5天", CheckinDays: 5, PrizeName: "积分x8", PrizeRef: "integral_8"},
|
||
{Title: "第6天", CheckinDays: 6, PrizeName: "积分x10", PrizeRef: "integral_10"},
|
||
{Title: "第7天", CheckinDays: 7, PrizeName: "观影券x1", PrizeRef: "video_coupon_1"},
|
||
// 第2周：稳步提升
|
||
{Title: "第8天", CheckinDays: 8, PrizeName: "积分x10", PrizeRef: "integral_10"},
|
||
{Title: "第9天", CheckinDays: 9, PrizeName: "积分x10", PrizeRef: "integral_10"},
|
||
{Title: "第10天", CheckinDays: 10, PrizeName: "积分x12", PrizeRef: "integral_12"},
|
||
{Title: "第11天", CheckinDays: 11, PrizeName: "积分x12", PrizeRef: "integral_12"},
|
||
{Title: "第12天", CheckinDays: 12, PrizeName: "积分x15", PrizeRef: "integral_15"},
|
||
{Title: "第13天", CheckinDays: 13, PrizeName: "金币x50", PrizeRef: "gold_50"},
|
||
{Title: "第14天", CheckinDays: 14, PrizeName: "观影券x1", PrizeRef: "video_coupon_1"},
|
||
// 第3周：中期奖励
|
||
{Title: "第15天", CheckinDays: 15, PrizeName: "积分x15", PrizeRef: "integral_15"},
|
||
{Title: "第16天", CheckinDays: 16, PrizeName: "积分x18", PrizeRef: "integral_18"},
|
||
{Title: "第17天", CheckinDays: 17, PrizeName: "积分x18", PrizeRef: "integral_18"},
|
||
{Title: "第18天", CheckinDays: 18, PrizeName: "积分x20", PrizeRef: "integral_20"},
|
||
{Title: "第19天", CheckinDays: 19, PrizeName: "AI脱衣x1", PrizeRef: "ai_undress_1"},
|
||
{Title: "第20天", CheckinDays: 20, PrizeName: "积分x20", PrizeRef: "integral_20"},
|
||
{Title: "第21天", CheckinDays: 21, PrizeName: "金币x100", PrizeRef: "gold_100"},
|
||
// 第4周：高级奖励
|
||
{Title: "第22天", CheckinDays: 22, PrizeName: "积分x25", PrizeRef: "integral_25"},
|
||
{Title: "第23天", CheckinDays: 23, PrizeName: "积分x25", PrizeRef: "integral_25"},
|
||
{Title: "第24天", CheckinDays: 24, PrizeName: "AI换脸x1", PrizeRef: "ai_changeface_1"},
|
||
{Title: "第25天", CheckinDays: 25, PrizeName: "积分x30", PrizeRef: "integral_30"},
|
||
{Title: "第26天", CheckinDays: 26, PrizeName: "观影券x2", PrizeRef: "video_coupon_2"},
|
||
{Title: "第27天", CheckinDays: 27, PrizeName: "积分x30", PrizeRef: "integral_30"},
|
||
{Title: "第28天", CheckinDays: 28, PrizeName: "金币x200", PrizeRef: "gold_200"},
|
||
// 最后冲刺
|
||
{Title: "第29天", CheckinDays: 29, PrizeName: "积分x50", PrizeRef: "integral_50"},
|
||
{Title: "第30天", CheckinDays: 30, PrizeName: "观影券x3", PrizeRef: "video_coupon_3"},
|
||
{Title: "第31天", CheckinDays: 31, PrizeName: "积分x100", PrizeRef: "integral_100"},
|
||
}
|
||
|
||
// 31天VIP专属奖品配置
|
||
vipPrizes := []checkinPrizeDef{
|
||
// 第1周
|
||
{Title: "VIP第1天", CheckinDays: 1, BigPrize: true, PrizeName: "VIP积分x5", PrizeRef: "vip_integral_5"},
|
||
{Title: "VIP第2天", CheckinDays: 2, BigPrize: true, PrizeName: "VIP积分x5", PrizeRef: "vip_integral_5"},
|
||
{Title: "VIP第3天", CheckinDays: 3, BigPrize: true, PrizeName: "VIP积分x8", PrizeRef: "vip_integral_8"},
|
||
{Title: "VIP第4天", CheckinDays: 4, BigPrize: true, PrizeName: "VIP积分x8", PrizeRef: "vip_integral_8"},
|
||
{Title: "VIP第5天", CheckinDays: 5, BigPrize: true, PrizeName: "VIP积分x10", PrizeRef: "vip_integral_10"},
|
||
{Title: "VIP第6天", CheckinDays: 6, BigPrize: true, PrizeName: "VIP积分x10", PrizeRef: "vip_integral_10"},
|
||
{Title: "VIP第7天", CheckinDays: 7, BigPrize: true, PrizeName: "VIP观影券x1", PrizeRef: "vip_video_coupon_1"},
|
||
// 第2周
|
||
{Title: "VIP第8天", CheckinDays: 8, BigPrize: true, PrizeName: "VIP积分x10", PrizeRef: "vip_integral_10"},
|
||
{Title: "VIP第9天", CheckinDays: 9, BigPrize: true, PrizeName: "VIP积分x10", PrizeRef: "vip_integral_10"},
|
||
{Title: "VIP第10天", CheckinDays: 10, BigPrize: true, PrizeName: "VIP积分x15", PrizeRef: "vip_integral_15"},
|
||
{Title: "VIP第11天", CheckinDays: 11, BigPrize: true, PrizeName: "VIP积分x15", PrizeRef: "vip_integral_15"},
|
||
{Title: "VIP第12天", CheckinDays: 12, BigPrize: true, PrizeName: "VIP金币x100", PrizeRef: "vip_gold_100"},
|
||
{Title: "VIP第13天", CheckinDays: 13, BigPrize: true, PrizeName: "VIP积分x15", PrizeRef: "vip_integral_15"},
|
||
{Title: "VIP第14天", CheckinDays: 14, BigPrize: true, PrizeName: "VIP AI脱衣x1", PrizeRef: "vip_ai_undress_1"},
|
||
// 第3周
|
||
{Title: "VIP第15天", CheckinDays: 15, BigPrize: true, PrizeName: "VIP积分x15", PrizeRef: "vip_integral_15"},
|
||
{Title: "VIP第16天", CheckinDays: 16, BigPrize: true, PrizeName: "VIP积分x20", PrizeRef: "vip_integral_20"},
|
||
{Title: "VIP第17天", CheckinDays: 17, BigPrize: true, PrizeName: "VIP积分x20", PrizeRef: "vip_integral_20"},
|
||
{Title: "VIP第18天", CheckinDays: 18, BigPrize: true, PrizeName: "VIP观影券x1", PrizeRef: "vip_video_coupon_1"},
|
||
{Title: "VIP第19天", CheckinDays: 19, BigPrize: true, PrizeName: "VIP积分x20", PrizeRef: "vip_integral_20"},
|
||
{Title: "VIP第20天", CheckinDays: 20, BigPrize: true, PrizeName: "VIP AI换脸x1", PrizeRef: "vip_ai_changeface_1"},
|
||
{Title: "VIP第21天", CheckinDays: 21, BigPrize: true, PrizeName: "VIP金币x200", PrizeRef: "vip_gold_200"},
|
||
// 第4周
|
||
{Title: "VIP第22天", CheckinDays: 22, BigPrize: true, PrizeName: "VIP积分x20", PrizeRef: "vip_integral_20"},
|
||
{Title: "VIP第23天", CheckinDays: 23, BigPrize: true, PrizeName: "VIP积分x30", PrizeRef: "vip_integral_30"},
|
||
{Title: "VIP第24天", CheckinDays: 24, BigPrize: true, PrizeName: "VIP AI脱衣x2", PrizeRef: "vip_ai_undress_2"},
|
||
{Title: "VIP第25天", CheckinDays: 25, BigPrize: true, PrizeName: "VIP积分x30", PrizeRef: "vip_integral_30"},
|
||
{Title: "VIP第26天", CheckinDays: 26, BigPrize: true, PrizeName: "VIP观影券x2", PrizeRef: "vip_video_coupon_2"},
|
||
{Title: "VIP第27天", CheckinDays: 27, BigPrize: true, PrizeName: "VIP积分x30", PrizeRef: "vip_integral_30"},
|
||
{Title: "VIP第28天", CheckinDays: 28, BigPrize: true, PrizeName: "VIP金币x500", PrizeRef: "vip_gold_500"},
|
||
// 最后冲刺
|
||
{Title: "VIP第29天", CheckinDays: 29, BigPrize: true, PrizeName: "VIP积分x50", PrizeRef: "vip_integral_50"},
|
||
{Title: "VIP第30天", CheckinDays: 30, BigPrize: true, PrizeName: "VIP AI换脸x2", PrizeRef: "vip_ai_changeface_2"},
|
||
{Title: "VIP第31天", CheckinDays: 31, BigPrize: true, PrizeName: "VIP积分x50", PrizeRef: "vip_integral_50"},
|
||
}
|
||
|
||
allCheckinPrizes := append(normalPrizes, vipPrizes...)
|
||
insertedCount := 0
|
||
|
||
for _, cp := range allCheckinPrizes {
|
||
prizeId, ok := prizeIdMap[cp.PrizeRef]
|
||
if !ok {
|
||
log.Printf(" [WARN] 找不到奖品引用 '%s'，跳过", cp.PrizeRef)
|
||
continue
|
||
}
|
||
doc := bson.M{
|
||
"title": cp.Title,
|
||
"image": "",
|
||
"checkinDays": cp.CheckinDays,
|
||
"checkinType": int64(CheckinTypeContinuously),
|
||
"prizeId": prizeId,
|
||
"prizeName": cp.PrizeName,
|
||
"status": true,
|
||
"bigPrize": cp.BigPrize,
|
||
"createdAt": now,
|
||
"updatedAt": now,
|
||
}
|
||
_, err := checkinPrizeColl.InsertOne(ctx, doc)
|
||
if err != nil {
|
||
log.Printf(" [WARN] 插入 '%s' 失败: %v", cp.Title, err)
|
||
continue
|
||
}
|
||
tag := ""
|
||
if cp.BigPrize {
|
||
tag = " [VIP]"
|
||
}
|
||
fmt.Printf(" Day%-2d %-16s => activity_prize(%s)%s\n",
|
||
cp.CheckinDays, cp.PrizeName, prizeId.Hex(), tag)
|
||
insertedCount++
|
||
}
|
||
fmt.Printf(" 共创建 %d 个 checkin_prize（普通%d + VIP%d）\n\n", insertedCount, len(normalPrizes), len(vipPrizes))
|
||
|
||
// ========================
|
||
// Step 4: 生成测试用户签到记录
|
||
// ========================
|
||
fmt.Println("=== Step 4: 生成测试用户签到记录 ===")
|
||
|
||
// ====== 在这里修改参数 ======
|
||
userIds := []uint64{300001}
|
||
n := 6 // 生成6天签到记录，今天签到为第7天（触发双倍奖励）
|
||
// ===========================
|
||
|
||
SeedCheckin(ctx, db, userIds, n)
|
||
|
||
fmt.Println("\n=== 全部配置完成 ===")
|
||
fmt.Println("数据关联关系：")
|
||
fmt.Println(" checkin_config (enable=true)")
|
||
fmt.Println(" checkin_prize (31天普通 + 31天VIP) --[prizeId]--> activity_prize")
|
||
fmt.Println(" user_checkin --[prizes]--> checkin_prize._id")
|
||
fmt.Printf(" 测试用户: %v (已签到%d天，今天第%d天)\n", userIds, n, n+1)
|
||
}
|
||
|
||
// SeedCheckin 为指定用户生成N天连续签到数据
|
||
func SeedCheckin(ctx context.Context, db *mongo.Database, userIds []uint64, n int) {
|
||
prizeColl := db.Collection("checkin_prize")
|
||
checkinColl := db.Collection("user_checkin")
|
||
|
||
loc, _ := time.LoadLocation("Asia/Shanghai")
|
||
now := time.Now().In(loc)
|
||
today := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, loc)
|
||
|
||
fmt.Printf(" Now: %v\n", now)
|
||
fmt.Printf(" Today (CST 00:00): %v\n", today)
|
||
fmt.Printf(" 生成 %d 天签到记录，今天签到为第 %d 天\n", n, n+1)
|
||
|
||
// 查奖励配置（非大奖）
|
||
cursor, err := prizeColl.Find(ctx, bson.M{
|
||
"status": true,
|
||
"checkinType": int64(1),
|
||
}, options.Find().SetSort(bson.M{"checkinDays": 1}))
|
||
if err != nil {
|
||
log.Printf(" query prizes failed: %v", err)
|
||
return
|
||
}
|
||
var prizeConfigs []bson.M
|
||
if err := cursor.All(ctx, &prizeConfigs); err != nil {
|
||
log.Printf(" decode prizes failed: %v", err)
|
||
return
|
||
}
|
||
|
||
dayPrizes := make(map[int64][]primitive.ObjectID)
|
||
for _, p := range prizeConfigs {
|
||
days := toInt64(p["checkinDays"])
|
||
bigPrize, _ := p["bigPrize"].(bool)
|
||
if !bigPrize {
|
||
if oid, ok := p["_id"].(primitive.ObjectID); ok {
|
||
dayPrizes[days] = append(dayPrizes[days], oid)
|
||
}
|
||
}
|
||
}
|
||
|
||
delResult, err := checkinColl.DeleteMany(ctx, bson.M{
|
||
"userId": bson.M{"$in": userIds},
|
||
})
|
||
if err != nil {
|
||
log.Printf(" delete failed: %v", err)
|
||
return
|
||
}
|
||
fmt.Printf(" Deleted %d existing records\n", delResult.DeletedCount)
|
||
|
||
for _, userId := range userIds {
|
||
fmt.Printf("\n --- userId=%d ---\n", userId)
|
||
for day := n; day >= 1; day-- {
|
||
date := today.AddDate(0, 0, -day)
|
||
continuouslyDays := int64(n - day + 1)
|
||
cumulativeDays := continuouslyDays
|
||
prizes := dayPrizes[continuouslyDays]
|
||
|
||
doc := bson.M{
|
||
"_id": primitive.NewObjectID(),
|
||
"date": date,
|
||
"userId": userId,
|
||
"prizes": prizes,
|
||
"gave": true,
|
||
"vipPrizeGave": false,
|
||
"continuouslyDays": continuouslyDays,
|
||
"cumulativeDays": cumulativeDays,
|
||
"isReset": false,
|
||
"createdAt": date.Add(10 * time.Hour),
|
||
}
|
||
_, err := checkinColl.InsertOne(ctx, doc)
|
||
if err != nil {
|
||
log.Printf(" Insert failed: %v", err)
|
||
continue
|
||
}
|
||
fmt.Printf(" date=%s contDays=%d prizes=%v\n",
|
||
date.Format("2006-01-02"), continuouslyDays, prizes)
|
||
}
|
||
}
|
||
}
|
||
|
||
func toInt64(v interface{}) int64 {
|
||
switch n := v.(type) {
|
||
case int64:
|
||
return n
|
||
case int32:
|
||
return int64(n)
|
||
case float64:
|
||
return int64(n)
|
||
default:
|
||
return 0
|
||
}
|
||
}
|