81 lines
2.0 KiB
Go
81 lines
2.0 KiB
Go
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)
|
|
}
|