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,205 @@
+package prizemod
+
+import (
+	"errors"
+	"fmt"
+	"time"
+
+	"91porn-server/common/db"
+	"91porn-server/common/log"
+	"91porn-server/models"
+
+	"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"
+)
+
+var mdb *db.MongoDB
+
+func coll(t *db.MongoTool) *db.MongoTool {
+	if t == nil {
+		return mdb.Coll(models.ActivityPrize)
+	}
+	return t.Coll(models.ActivityPrize)
+}
+
+// initIndex 索引设置
+func initIndex() {
+	many := []mongo.IndexModel{ //batch set indexes   //value is the type 1 or -1
+		{
+			Keys: bson.D{{Key: "activityId", Value: 1}, {Key: "status", Value: 1}, {Key: "sort", Value: -1}},
+		},
+		{
+			Keys: bson.D{{Key: "activityId", Value: 1}, {Key: "status", Value: 1}},
+		},
+		{
+			Keys: bson.D{{Key: "status", Value: 1}, {Key: "type", Value: 1}},
+		},
+		{
+			Keys: bson.D{{Key: "status", Value: 1}},
+		},
+		{
+			Keys: bson.D{{Key: "type", Value: 1}},
+		},
+		{
+			Keys: bson.D{{Key: "level", Value: 1}},
+		},
+		{
+			Keys: bson.D{{Key: "createTime", Value: -1}},
+		},
+	}
+	if _, err := coll(nil).CreateIndex(many); err != nil {
+		panic(fmt.Sprintf("%s model set index err ==>[%+v]", models.ActivityPrize, err))
+	}
+}
+
+// 获取指定活动奖品列表
+func GetPrizeListByActivityID(id primitive.ObjectID) ([]*Prize, error) {
+	out := []*Prize{}
+	if err := coll(nil).Find(&out, bson.M{"activityId": id, "status": true}, options.Find().
+		SetSort(bson.M{"sort": 1})); err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetPrizeListByActivityID", models.ActivityPrize, "Find", err),
+			log.Any("id", id),
+		)
+		return nil, err
+	}
+	return out, nil
+}
+
+// 获取指定活动大奖列表
+func GetHighPrizeListByActivityID(id primitive.ObjectID) ([]*Prize, error) {
+	out := []*Prize{}
+	if err := coll(nil).Find(&out, bson.M{"activityId": id, "status": true, "level": High}, options.Find().
+		SetSort(bson.M{"sort": 1})); err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetPrizeListByActivityID", models.ActivityPrize, "Find", err),
+			log.Any("id", id),
+		)
+		return nil, err
+	}
+	return out, nil
+}
+
+// 获取优惠卷列表
+func GetCouponList(couponType int) ([]*Prize, error) {
+	out := []*Prize{}
+	if err := coll(nil).Find(&out, bson.M{"status": true, "type": couponType}); err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetCouponList", models.ActivityPrize, "Find", err),
+			log.Any("couponType", couponType),
+		)
+		return nil, err
+	}
+	return out, nil
+}
+
+// 通过ID获取奖品列表
+func GetPrizeListByIDs(ids []primitive.ObjectID) ([]*Prize, error) {
+	out := make([]*Prize, 0)
+	if len(ids) == 0 {
+		return out, nil
+	}
+	if err := coll(nil).Find(&out, bson.M{"_id": bson.M{"$in": ids}, "status": true}, options.Find().SetSort(bson.M{"type": 1})); err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetByUID", models.ActivityPrize, "Find", err),
+			log.Any("ids", ids),
+		)
+		return nil, err
+	}
+	return out, nil
+}
+
+// 获取所有开启的奖品
+func GetPrizeList() (map[primitive.ObjectID]Prize, error) {
+	out := make([]Prize, 0)
+	if err := coll(nil).Find(&out, bson.M{"status": true}, options.Find()); err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetPrizeList", models.ActivityPrize, "Find", err))
+		return nil, err
+	}
+	var m map[primitive.ObjectID]Prize = make(map[primitive.ObjectID]Prize, len(out))
+	for _, v := range out {
+		m[v.ID] = v
+	}
+	return m, nil
+}
+
+// 新增奖品
+func AddPrize(in AddPrizeCond) (interface{}, error) {
+	result, err := coll(nil).InsertOne(&Prize{
+		ActivityID:   in.ActivityID,
+		VIPCardID:    in.VIPCardID,
+		Name:         in.Name,
+		Type:         in.Type,
+		Price:        in.Price,
+		Count:        in.Count,
+		Value:        in.Value,
+		Desc:         in.Desc,
+		Weights:      in.Weights,
+		Level:        in.Level,
+		Status:       false,
+		ValidityTime: in.ValidityTime,
+		Image:        in.Image,
+		Sort:         in.Sort,
+		UpdateTime:   time.Now(),
+		CreateTime:   time.Now(),
+	})
+	if err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "AddPrize", models.ActivityPrize, "InsertOne", err),
+			log.Any("in", in),
+		)
+		return nil, err
+	}
+	return result, nil
+}
+
+// 修改活动奖品
+func ModifyPrize(filter, update primitive.M) error {
+	result, err := coll(nil).UpdateOne(filter, update)
+	if err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "ModifyPrize", models.ActivityPrize, "UpdateOne", err),
+			log.Any("filter", filter),
+			log.Any("update", update),
+		)
+		return err
+	}
+	if result.ModifiedCount == 0 {
+		log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail", "ModifyPrize", models.ActivityPrize, "result.ModifiedCount == 0"),
+			log.Any("filter", filter),
+			log.Any("update", update),
+		)
+		return errors.New("activity UpdateOne ModifiedCount err")
+	}
+	return nil
+}
+
+// 查询活动奖品列表
+func QueryAllPrize(filter primitive.M, opts ...*options.FindOptions) ([]*Prize, error) {
+	out := []*Prize{}
+	if err := coll(nil).Find(&out, filter, opts...); err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "QueryAllPrize", models.ActivityPrize, "Find", err),
+			log.Any("filter", filter),
+		)
+		return nil, err
+	}
+	return out, nil
+}
+
+// CountPrize 查询奖品总条数
+func CountPrize(filter primitive.M) (int64, error) {
+	count, err := coll(nil).Count(filter)
+	if err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "CountPrize", models.ActivityPrize, "Count", err),
+			log.Any("filter", filter),
+		)
+		return 0, err
+	}
+	return count, nil
+}
+
+// QueryDropDownBox 奖品下拉框
+func QueryDropDownBox() ([]*Prize, error) {
+	out := []*Prize{}
+	if err := coll(nil).Find(&out, bson.M{"status": true}); err != nil {
+		log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "QueryDropDownBox", models.ActivityPrize, "Find", err))
+		return nil, err
+	}
+	return out, nil
+}