Initial commit

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>

models/v/videocoupon/videocoupon.go

@@ -0,0 +1,80 @@
+package videocoupon
+
+import (
+	"errors"
+	"sort"
+	"time"
+
+	"91porn-server/common/db"
+	"91porn-server/common/stderr"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+// InsertOne ...
+func InsertOne(coupon UserGoldVideoCoupon) error {
+	_, err := coll(nil).InsertOne(coupon)
+	return err
+}
+
+// InsertMany batch insert
+func InsertMany(coupons []UserGoldVideoCoupon) error {
+	_, err := coll(nil).InsertMany(coupons)
+	return err
+}
+
+func InsertManyTrans(t *db.MongoTool, coupons []UserGoldVideoCoupon) error {
+	_, err := coll(t).InsertMany(coupons)
+	return err
+}
+
+// GetUserUnusedCouponCount 获取用户未使用的观影券总数
+func GetUserUnusedCouponCount(uid uint64) (int64, error) {
+	filter := bson.M{"uid": uid, "used": false}
+	return coll(nil).Count(filter)
+}
+
+// GetCouponsByUID 根据uid获取观影券
+func GetCouponsByUID(uid uint64) (list []UserGoldVideoCoupon, err error) {
+	filter := bson.M{"uid": uid}
+	err = coll(nil).Find(&list, filter)
+	return
+}
+
+// GetByUIDAndCouponNum 根据uid和观影券面值查询观影券
+func GetByUIDAndCouponNum(uid uint64, num int) (list []UserGoldVideoCoupon, err error) {
+	filter := bson.M{"uid": uid, "num": num, "used": false}
+	opts := options.FindOptions{}
+	opts.SetSort(bson.D{{Key: "createdAt", Value: -1}})
+	err = coll(nil).Find(&list, filter)
+	return
+}
+
+// UpdateCoupon 更新用户观影券
+func UpdateCoupon(edit EditSelector) error {
+	res, err := coll(nil).UpdateOne(bson.M{"_id": edit.ID}, bson.M{"$set": edit})
+	if err != nil {
+		return err
+	}
+	if res.ModifiedCount+res.UpsertedCount < 1 {
+		return errors.New("update failed")
+	}
+	return nil
+}
+
+// UseOneCoupon 使用一张观影券
+func UseOneCoupon(coupons []UserGoldVideoCoupon) error {
+	if len(coupons) == 0 {
+		return stderr.GoldVideoCoupleNotExist
+	}
+	sort.Slice(coupons, func(i, j int) bool { // 优先使用最早创建的观影券
+		return coupons[i].CreatedAt.After(coupons[j].CreatedAt)
+	})
+	edit := EditSelector{
+		ID:        coupons[0].ID,
+		Used:      true,
+		UpdatedAt: time.Now(),
+	}
+	return UpdateCoupon(edit)
+}