214 lines
7.1 KiB
Go
214 lines
7.1 KiB
Go
package prdcthsomod
|
||
|
||
import (
|
||
"fmt"
|
||
"time"
|
||
|
||
"91porn-server/common/db"
|
||
"91porn-server/common/log"
|
||
"91porn-server/models"
|
||
"91porn-server/models/commod"
|
||
"91porn-server/models/v/productmod"
|
||
|
||
"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.ProductHistory
|
||
|
||
// initIndex 设置index
|
||
func initIndex() {
|
||
coll := coll(nil)
|
||
many := []mongo.IndexModel{
|
||
{
|
||
Keys: bson.D{{Key: "uid", Value: 1}},
|
||
},
|
||
{
|
||
Keys: bson.D{{Key: "productID", Value: 1}},
|
||
},
|
||
{
|
||
Keys: bson.D{{Key: "name", Value: 1}},
|
||
},
|
||
{
|
||
Keys: bson.D{{Key: "amount", Value: 1}},
|
||
},
|
||
{
|
||
Keys: bson.D{{Key: "productType", Value: 1}},
|
||
},
|
||
{
|
||
Keys: bson.D{{Key: "status", Value: 1}},
|
||
},
|
||
{
|
||
Keys: bson.D{{Key: "createdAt", Value: 1}},
|
||
},
|
||
{
|
||
Keys: bson.D{{Key: "updatedAt", Value: 1}},
|
||
},
|
||
// 用户运营导出预售状态相关的订单
|
||
{
|
||
Keys: bson.D{{Key: "advanceOrderStatus", Value: 1}, {Key: "createdAt", Value: -1}},
|
||
},
|
||
{
|
||
Keys: bson.D{{Key: "deductType", Value: 1}, {Key: "createdAt", Value: -1}},
|
||
Options: options.Index().SetSparse(true),
|
||
},
|
||
{
|
||
Keys: bson.D{
|
||
{Key: "experimentId", Value: 1},
|
||
{Key: "experimentVariant", Value: 1},
|
||
{Key: "productID", Value: 1},
|
||
},
|
||
Options: options.Index().
|
||
SetName("vip_experiment_gold_order_statistics").
|
||
SetPartialFilterExpression(bson.M{"experimentId": bson.M{"$gt": ""}}),
|
||
},
|
||
}
|
||
if _, err := coll.CreateIndex(many); err != nil {
|
||
panic(fmt.Sprintf("%s model set index err ==>[%+v]", table, err))
|
||
}
|
||
}
|
||
|
||
func coll(t *db.MongoTool) *db.MongoTool {
|
||
if t == nil {
|
||
return mdb.Coll(table)
|
||
}
|
||
return t.Coll(table)
|
||
}
|
||
|
||
func IsBoughtVIP(uid uint64) (bool, error) {
|
||
p := ProductHistory{}
|
||
if err := coll(nil).FindOne(&p, bson.M{"uid": uid, "productType": VIP}); err != nil {
|
||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "IsBoughtVIP", table, "FindOne", err), log.Any("uid", uid))
|
||
return false, err
|
||
}
|
||
return p.UID != 0, nil
|
||
}
|
||
|
||
// GetUserLastVip 获取用户最后充值的VIP
|
||
func GetUserLastVip(uid uint64) (productID primitive.ObjectID, productName string, amount int64) {
|
||
p := ProductHistory{}
|
||
opt := options.FindOne().SetSort(bson.D{{Key: "createdAt", Value: -1}})
|
||
if err := coll(nil).FindOne(&p, bson.M{"uid": uid, "productType": bson.M{"$in": []commod.ProductType{productmod.VIP, productmod.NewUser, productmod.AdvanceCard}}}, opt); err != nil {
|
||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "IsBoughtVIP", table, "FindOne", err), log.Any("uid", uid))
|
||
return primitive.NilObjectID, "", 0
|
||
}
|
||
if p.ProductType == productmod.AdvanceCard {
|
||
return productID, "", 0
|
||
}
|
||
if p.IsUpgrade {
|
||
// 假设A：100 B：300 C：500 如果A连续升到B，不这么处理，就会变成B=》C要付款300，实际应该给200差价就好
|
||
return p.ProductID, p.Name, p.PurchasePrice
|
||
}
|
||
return p.ProductID, p.Name, p.Amount
|
||
}
|
||
|
||
// InsertProductHistory 插入一条数据
|
||
func InsertProductHistory(t *db.MongoTool, p *ProductHistory) error {
|
||
p.CreatedAt = time.Now()
|
||
p.UpdatedAt = time.Now()
|
||
res, err := coll(t).InsertOne(p)
|
||
if err != nil {
|
||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "InsertProductHistory", table, "InsertOne", err))
|
||
return err
|
||
}
|
||
p.ID = res.InsertedID.(primitive.ObjectID)
|
||
return nil
|
||
}
|
||
|
||
func InsertManyProductHistory(t *db.MongoTool, p []ProductHistory) error {
|
||
if _, err := coll(t).InsertMany(p); err != nil {
|
||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "InsertProductHistory", table, "InsertOne", err))
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// FindProductHistorys 查询所有ProductHistory类型
|
||
func FindProductHistorys(cond bson.M, opts *options.FindOptions) (total int64, data []*ProductHistory, err error) {
|
||
data = make([]*ProductHistory, 0)
|
||
if err = coll(nil).Find(&data, cond, opts); err != nil {
|
||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "FindProductHistorys", table, "Find", err), log.Any("cond", cond))
|
||
return
|
||
}
|
||
total, err = coll(nil).Count(cond)
|
||
if err != nil {
|
||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "FindProductHistorys", table, "Count", err), log.Any("cond", cond))
|
||
}
|
||
return
|
||
}
|
||
|
||
// FindList
|
||
func FindList(cond bson.M, opts *options.FindOptions) (data []*ProductHistory, err error) {
|
||
data = make([]*ProductHistory, 0)
|
||
if err = coll(nil).Find(&data, cond, opts); err != nil {
|
||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "FindList", table, "Find", err), log.Any("cond", cond))
|
||
return
|
||
}
|
||
return
|
||
}
|
||
|
||
func FindUserProductHistory(uid uint64) (list []ProductHistory, err error) {
|
||
f := bson.M{"uid": uid, "productType": bson.M{"$in": []commod.ProductType{productmod.VIP, productmod.OTHER,
|
||
productmod.MeetingCard, productmod.VideoDiscount,
|
||
productmod.VideoFreeCard, productmod.VideoDiscount, productmod.VideoFreeCard,
|
||
}}}
|
||
opts := options.Find().SetSort(bson.D{{Key: "createdAt", Value: -1}})
|
||
if err = coll(nil).Find(&list, f, opts); err != nil {
|
||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "FindProductHistorys", table, "Find", err), log.Any("query", f))
|
||
return
|
||
}
|
||
return
|
||
}
|
||
|
||
func FindUserProductHistoryByType(uid uint64) (list []ProductHistory, err error) {
|
||
f := bson.M{"uid": uid, "productType": bson.M{"$in": []commod.ProductType{productmod.GameAdvanceCard}}}
|
||
opts := options.Find().SetSort(bson.D{{Key: "createdAt", Value: -1}})
|
||
if err = coll(nil).Find(&list, f, opts); err != nil {
|
||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "FindProductHistorys", table, "Find", err), log.Any("query", f))
|
||
return
|
||
}
|
||
return
|
||
}
|
||
|
||
// FindProductHistorysByUID 查询所有ProductHistory类型
|
||
func FindProductHistorysByUID(uid, pageNumber, pageSizse uint64) (total int64, data []*ProductHistory, hasNext bool, err error) {
|
||
data = make([]*ProductHistory, 0)
|
||
f := bson.M{"uid": uid, "productType": bson.M{"$in": []commod.ProductType{productmod.VIP, productmod.OTHER,
|
||
productmod.MeetingCard, productmod.VideoDiscount,
|
||
productmod.VideoFreeCard, productmod.VideoDiscount, productmod.VideoFreeCard,
|
||
}}}
|
||
skip := int64(pageSizse * (pageNumber - 1))
|
||
limit := int64(pageSizse + 1)
|
||
opts := options.FindOptions{
|
||
Skip: &skip,
|
||
Limit: &limit,
|
||
Sort: bson.D{{Key: "createdAt", Value: -1}},
|
||
}
|
||
total, err = coll(nil).Count(f)
|
||
if err != nil {
|
||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %ss fail error:%+v:", "FindProductHistorys", table, "Count", err), log.Any("query", f))
|
||
}
|
||
if err = coll(nil).Find(&data, f, &opts); err != nil {
|
||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "FindProductHistorys", table, "Find", err), log.Any("query", f))
|
||
return
|
||
}
|
||
if len(data) > int(pageSizse) {
|
||
hasNext = true
|
||
data = data[:pageSizse]
|
||
}
|
||
return
|
||
}
|
||
|
||
// FindMany FindMany
|
||
func FindMany(start, end time.Time, opts *options.FindOptions) (data []*ProductStat, err error) {
|
||
cond := bson.M{"createdAt": bson.M{"$gte": start, "$lt": end}}
|
||
data = make([]*ProductStat, 0)
|
||
if err = coll(nil).Find(&data, cond, opts); err != nil {
|
||
log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "FindProductHistorys", table, "Find", err), log.Any("cond", cond))
|
||
return
|
||
}
|
||
return
|
||
}
|