@@ -0,0 +1,9 @@
|
||||
package backpackmod
|
||||
|
||||
import "91porn-server/models/commod"
|
||||
|
||||
type QueryAllCond struct {
|
||||
UID uint64
|
||||
Status *GoodsStatus `form:"status" json:"status" binding:"required,min=1"` // 物品状态
|
||||
commod.Page
|
||||
}
|
||||
@@ -0,0 +1,221 @@
|
||||
package backpackmod
|
||||
|
||||
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"
|
||||
)
|
||||
|
||||
const table = models.Backpack
|
||||
|
||||
func coll(t *db.MongoTool) *db.MongoTool {
|
||||
if t == nil {
|
||||
return mdb.Coll(table)
|
||||
}
|
||||
return t.Coll(table)
|
||||
}
|
||||
|
||||
// Init 索引设置
|
||||
func initIndex() {
|
||||
many := []mongo.IndexModel{ //batch set indexes //value is the type 1 or -1
|
||||
{
|
||||
Keys: bson.D{
|
||||
{Key: "uid", Value: 1},
|
||||
{Key: "status", Value: 1},
|
||||
{Key: "createTimt", Value: -1},
|
||||
},
|
||||
},
|
||||
{
|
||||
Keys: bson.D{
|
||||
{Key: "uid", Value: 1},
|
||||
{Key: "status", Value: 1},
|
||||
{Key: "useTime", Value: -1},
|
||||
},
|
||||
},
|
||||
{
|
||||
Keys: bson.D{
|
||||
{Key: "uid", Value: 1},
|
||||
{Key: "expiredTime", Value: -1},
|
||||
},
|
||||
},
|
||||
}
|
||||
if _, err := coll(nil).CreateIndex(many); err != nil {
|
||||
panic(fmt.Sprintf("backpack model set index err ==>[%+v]", err))
|
||||
}
|
||||
}
|
||||
|
||||
// 获取个人拥有可用优惠卷列表
|
||||
func GetCouponListByUID(uid uint64, goodsType int) ([]*Backpack, error) {
|
||||
var out []*Backpack = make([]*Backpack, 0)
|
||||
if err := coll(nil).Find(&out, bson.M{
|
||||
"uid": uid,
|
||||
"goodsType": goodsType,
|
||||
"status": Unused,
|
||||
"expiredTime": bson.M{"$gt": time.Now()},
|
||||
}); err != nil {
|
||||
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetCouponListByUID", models.Backpack, "Find", err),
|
||||
log.Any("uid", uid),
|
||||
log.Any("goodsType", goodsType),
|
||||
)
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
// 添加物品到背包
|
||||
func AddGoods(t *db.MongoTool, uid uint64, b ...Backpack) error {
|
||||
for _, item := range b {
|
||||
item.GoodsName = item.GoodsType.GetName()
|
||||
item.GoodsDesc = item.GoodsType.GetDesc()
|
||||
}
|
||||
if _, err := coll(t).InsertMany(&b); err != nil {
|
||||
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "AddGoods", models.Backpack, "InsertMany", err),
|
||||
log.Any("uid", uid),
|
||||
)
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// AddGoodsMany 添加物品到背包
|
||||
func AddGoodsMany(t *db.MongoTool, uid uint64, b []Backpack) error {
|
||||
if _, err := coll(t).InsertMany(&b); err != nil {
|
||||
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "AddGoodsMany", models.Backpack, "InsertMany", err),
|
||||
log.Any("uid", uid),
|
||||
)
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// 使用物品
|
||||
func UseGoods(t *db.MongoTool, id primitive.ObjectID) error {
|
||||
result, err := coll(t).UpdateOne(bson.M{"_id": id}, bson.M{"$set": bson.M{"status": Used, "useTime": time.Now()}})
|
||||
if err != nil {
|
||||
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "UseGoods", models.Backpack, "UpdateOne", err),
|
||||
log.Any("id", id),
|
||||
)
|
||||
return err
|
||||
}
|
||||
if result.ModifiedCount == 0 {
|
||||
err = errors.New("backpack UpdateOne ModifiedCount err")
|
||||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "UseGoods", models.Backpack, "result.ModifiedCount == 0", err),
|
||||
log.Any("id", id),
|
||||
)
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// 查询背包列表
|
||||
func GetCouponList(uid uint64, types, status int, limit, page int64) ([]*Backpack, error) {
|
||||
var out []*Backpack = make([]*Backpack, 0)
|
||||
cond := bson.M{"uid": uid}
|
||||
sort := bson.M{}
|
||||
switch GoodsStatus(status) {
|
||||
case Used:
|
||||
sort["useTime"] = -1
|
||||
cond["status"] = status
|
||||
case Unused:
|
||||
cond["status"] = status
|
||||
cond["expiredTime"] = bson.M{"$gt": time.Now()}
|
||||
sort["createTime"] = 1
|
||||
case Expired:
|
||||
cond["expiredTime"] = bson.M{"$lte": time.Now()}
|
||||
sort["expiredTime"] = -1
|
||||
}
|
||||
if types > 0 {
|
||||
cond["goodsType"] = types
|
||||
}
|
||||
opts := options.Find().SetLimit(limit).SetSkip((page - 1) * limit).SetSort(sort)
|
||||
if err := coll(nil).Find(&out, cond, opts); err != nil {
|
||||
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetGoodsList", models.Backpack, "Find", err),
|
||||
log.Any("uid", uid),
|
||||
)
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
// 获取物品信息
|
||||
func GetGoodsDetail(id primitive.ObjectID) (*Backpack, error) {
|
||||
var out *Backpack
|
||||
if err := coll(nil).FindOne(&out, bson.M{"_id": id}); err != nil {
|
||||
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetGoodsDetail", models.Backpack, "FindOne", err),
|
||||
log.Any("id", id),
|
||||
)
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
// 根据id获取券信息
|
||||
// 如果未找到, 返回nil
|
||||
func GetByID(t *db.MongoTool, id primitive.ObjectID) (*Backpack, error) {
|
||||
var bp Backpack
|
||||
if err := coll(t).FindOne(&bp, bson.M{"_id": id}); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if bp.ID != id {
|
||||
return nil, nil
|
||||
}
|
||||
return &bp, nil
|
||||
}
|
||||
|
||||
// UseManyGoods 使用物品(多件)
|
||||
func UseManyGoods(t *db.MongoTool, ids []primitive.ObjectID) (int64, error) {
|
||||
filter := bson.M{"_id": bson.M{"$in": ids}, "status": Unused}
|
||||
result, err := coll(t).UpdateMany(filter, bson.M{"$set": bson.M{"status": Used, "useTime": time.Now()}})
|
||||
if err != nil {
|
||||
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "UseManyGoods", models.Backpack, "UpdateOne", err),
|
||||
log.Any("id", ids),
|
||||
)
|
||||
return 0, err
|
||||
}
|
||||
return result.ModifiedCount, nil
|
||||
}
|
||||
|
||||
// UnuseManyGoods 退换物品(多件)
|
||||
func UnuseManyGoods(t *db.MongoTool, ids []primitive.ObjectID) (int64, error) {
|
||||
result, err := coll(t).UpdateMany(bson.M{"_id": bson.M{"$in": ids}, "status": Used}, bson.M{"$set": bson.M{"status": Unused}})
|
||||
if err != nil {
|
||||
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "UseManyGoods", models.Backpack, "UpdateOne", err),
|
||||
log.Any("id", ids),
|
||||
)
|
||||
return 0, err
|
||||
}
|
||||
return result.ModifiedCount, nil
|
||||
}
|
||||
|
||||
// QueryAllList 分页查询文档
|
||||
func QueryAllList(filter primitive.M, opts ...*options.FindOptions) (out []*Backpack, err error) {
|
||||
if err = coll(nil).Find(&out, filter, opts...); err != nil {
|
||||
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "QueryAllList", table, "Find", err),
|
||||
log.Any("filter", filter),
|
||||
log.Any("opts", opts),
|
||||
)
|
||||
return nil, err
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// QueryAllCount 查询文档条目数
|
||||
func QueryAllCount(filter primitive.M) (int64, error) {
|
||||
if count, err := coll(nil).Count(filter); err != nil {
|
||||
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "QueryAllCount", table, "Count", err),
|
||||
log.Any("filter", filter),
|
||||
)
|
||||
return 0, err
|
||||
} else {
|
||||
return count, nil
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,76 @@
|
||||
package backpackmod
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"91porn-server/common/db"
|
||||
"91porn-server/models"
|
||||
|
||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||
)
|
||||
|
||||
var mdb *db.MongoDB
|
||||
|
||||
var itemNames = map[GoodsType]string{
|
||||
LouFengUnLock: "楼风解锁折扣卷",
|
||||
VIPDiscount: "会员折扣卷",
|
||||
AiChangeFaceDiscount: "AI换脸折扣券",
|
||||
GoldBonusCoupon: "金币加赠券",
|
||||
ExchangeCoupon: "兑换码观影券",
|
||||
SIGNCoupon: "签到赠观影券",
|
||||
}
|
||||
|
||||
var itemDesc = map[GoodsType]string{
|
||||
LouFengUnLock: "楼风解锁专享折扣",
|
||||
VIPDiscount: "购买会员专享折扣",
|
||||
AiChangeFaceDiscount: "AI换脸专享折扣",
|
||||
GoldBonusCoupon: "充值加送金币",
|
||||
ExchangeCoupon: "兑换码观影券",
|
||||
SIGNCoupon: "签到赠观影券",
|
||||
}
|
||||
|
||||
type GoodsType int
|
||||
|
||||
const (
|
||||
LouFengUnLock GoodsType = 1 // 1、楼风解锁折扣卷
|
||||
VIPDiscount GoodsType = 2 // 2、会员折扣卷
|
||||
AiChangeFaceDiscount GoodsType = 3 // 3、AI换脸折扣券
|
||||
GameCode GoodsType = 4 // 4、游戏码
|
||||
GoldBonusCoupon GoodsType = 5 // 5、金币加赠券(用于购买金币时额外赠送额度)
|
||||
ExchangeCoupon GoodsType = 6 // 6、兑换赠观影券
|
||||
SIGNCoupon GoodsType = 7 // 7、签到赠观影券
|
||||
)
|
||||
|
||||
func (t GoodsType) GetName() string {
|
||||
return itemNames[t]
|
||||
}
|
||||
func (t GoodsType) GetDesc() string {
|
||||
return itemDesc[t]
|
||||
}
|
||||
|
||||
type GoodsStatus int
|
||||
|
||||
const (
|
||||
Used GoodsStatus = iota + 1 // 已使用
|
||||
Unused // 未使用
|
||||
Expired // 过期
|
||||
)
|
||||
|
||||
type Backpack struct {
|
||||
ID primitive.ObjectID `json:"id" bson:"_id,omitempty"` // 文档ID
|
||||
UID uint64 `json:"uid" bson:"uid"` // 用户ID
|
||||
GoodsName string `json:"goodsName" bson:"goodsName"` // 物品名称
|
||||
GoodsType GoodsType `json:"goodsType" bson:"goodsType"` // 物品类型
|
||||
GoodsValue int64 `json:"goodsValue" bson:"goodsValue"` // 物品价值
|
||||
GoodsOrigin string `json:"goodsOrigin" bson:"goodsOrigin"` // 物品来源
|
||||
GoodsDesc string `json:"goodsDesc" bson:"goodsDesc"` // 物品描述
|
||||
Status GoodsStatus `json:"status" bson:"status"` // 物品状态
|
||||
ExpiredTime time.Time `json:"expiredTime" bson:"expiredTime"` // 物品过期时间
|
||||
UseTime time.Time `json:"useTime" bson:"useTime"` // 物品使用时间
|
||||
CreateTime time.Time `json:"createTime" bson:"createTime"` // 创建时间
|
||||
}
|
||||
|
||||
func Init() {
|
||||
mdb = db.Init(models.Backpack)
|
||||
initIndex()
|
||||
}
|
||||
@@ -0,0 +1,41 @@
|
||||
package backpackmod
|
||||
|
||||
import (
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/mongo/options"
|
||||
)
|
||||
|
||||
type QueryAllWebCond struct {
|
||||
Page int64 `form:"page" binding:"required"`
|
||||
Limit int64 `form:"limit" binding:"required"`
|
||||
Uid *int `form:"uid"` // 用户ID
|
||||
Status *int `form:"status"` // 使用状态 1、已使用 2、未使用 3、过期
|
||||
GoodsType *int `form:"goodsType"` // 类型
|
||||
GoodsDesc *string `form:"goodsDesc"` // 描述
|
||||
}
|
||||
|
||||
type QueryAllRes struct {
|
||||
List []*Backpack `json:"list"`
|
||||
Total int64 `json:"total"`
|
||||
}
|
||||
|
||||
func (receiver *QueryAllWebCond) Filter() bson.M {
|
||||
var query = bson.M{}
|
||||
if receiver.Uid != nil {
|
||||
query["uid"] = receiver.Uid
|
||||
}
|
||||
if receiver.GoodsType != nil {
|
||||
query["goodsType"] = receiver.GoodsType
|
||||
}
|
||||
if receiver.GoodsDesc != nil {
|
||||
query["goodsDesc"] = receiver.GoodsDesc
|
||||
}
|
||||
if receiver.Status != nil {
|
||||
query["status"] = receiver.Status
|
||||
}
|
||||
return query
|
||||
}
|
||||
|
||||
func (receiver QueryAllWebCond) Options() *options.FindOptions {
|
||||
return options.Find().SetSkip((receiver.Page - 1) * receiver.Limit).SetLimit(receiver.Limit).SetSort(bson.D{{"createTime", -1}})
|
||||
}
|
||||
Reference in New Issue
Block a user