Claude Opus 5
@@ -0,0 +1,95 @@
|
||||
package statordermod
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"91porn-server/common/log"
|
||||
"91porn-server/common/pageopt"
|
||||
|
||||
"go.mongodb.org/mongo-driver/mongo/options"
|
||||
)
|
||||
|
||||
type (
|
||||
Matcher = pageopt.Matcher
|
||||
|
||||
// ChannelMatch
|
||||
ChannelMatch struct {
|
||||
Type *string
|
||||
}
|
||||
)
|
||||
|
||||
func (t *ChannelMatch) New() Matcher {
|
||||
return pageopt.NewAssignMatch("channel", t.Type)
|
||||
}
|
||||
|
||||
// TypeMatch
|
||||
type TypeMatch struct {
|
||||
Type *OrderType
|
||||
}
|
||||
|
||||
func (t *TypeMatch) New() Matcher {
|
||||
return pageopt.NewAssignMatch("type", t.Type)
|
||||
}
|
||||
|
||||
// SumDateGTEAndLTMatch
|
||||
type SumDateGTEAndLTMatch struct {
|
||||
GTE *time.Time
|
||||
LT *time.Time
|
||||
}
|
||||
|
||||
func (s *SumDateGTEAndLTMatch) New() Matcher {
|
||||
return pageopt.NewGTEAndLTMatch("sumDate", s.GTE, s.LT)
|
||||
}
|
||||
|
||||
var Sort_SumDate_n1 = Sort{{Key: "sumDate", Value: -1}}
|
||||
|
||||
func List(sort Sort, skip, limit *int64, matchers ...Matcher) ([]OrderState, error) {
|
||||
filter := pageopt.MergeM(matchers)
|
||||
opt := (&options.FindOptions{})
|
||||
if len(sort) != 0 {
|
||||
opt.SetSort(sort)
|
||||
}
|
||||
if skip != nil {
|
||||
opt.SetSkip(*skip)
|
||||
}
|
||||
if limit != nil {
|
||||
opt.SetLimit(*limit)
|
||||
}
|
||||
list := []OrderState{}
|
||||
if err := coll(nil).Find(&list, filter, opt); err != nil {
|
||||
log.Error("statordermod List error", log.E(err))
|
||||
return nil, err
|
||||
}
|
||||
return list, nil
|
||||
}
|
||||
|
||||
func Count(matchers ...Matcher) (int64, error) {
|
||||
filter := pageopt.MergeM(matchers)
|
||||
count, err := coll(nil).Count(filter)
|
||||
if err != nil {
|
||||
log.Error("statordermod Count error", log.E(err))
|
||||
return 0, err
|
||||
}
|
||||
return count, nil
|
||||
}
|
||||
|
||||
// Sum
|
||||
func Sum(matchers ...Matcher) (Items, error) {
|
||||
filter := pageopt.MergeM(matchers)
|
||||
pipeline := []M{}
|
||||
pipeline = append(pipeline, M{"$match": filter})
|
||||
pipeline = append(pipeline, M{"$group": M{"_id": nil,
|
||||
"orderCount": M{"$sum": "$orderCount"},
|
||||
"completeOrderCount": M{"$sum": "$completeOrderCount"},
|
||||
"orderAmount": M{"$sum": "$orderAmount"},
|
||||
"completeOrderAmount": M{"$sum": "$completeOrderAmount"},
|
||||
"paidOrderCount": M{"$sum": "$paidOrderCount"},
|
||||
"paidAmount": M{"$sum": "$paidAmount"},
|
||||
}})
|
||||
items := Items{}
|
||||
if err := coll(nil).AggregateDecode(&items, pipeline); err != nil {
|
||||
log.Error("statordermod Sum error", log.E(err))
|
||||
return Items{}, err
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
@@ -0,0 +1,123 @@
|
||||
package statordermod
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"91porn-server/common"
|
||||
"91porn-server/common/db"
|
||||
"91porn-server/common/log"
|
||||
"91porn-server/models"
|
||||
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/mongo"
|
||||
"go.mongodb.org/mongo-driver/mongo/options"
|
||||
)
|
||||
|
||||
var mdb *db.MongoDB
|
||||
|
||||
const table = models.OrderStat
|
||||
|
||||
func coll(t *db.MongoTool) *db.MongoTool {
|
||||
if t == nil {
|
||||
return mdb.Coll(table)
|
||||
}
|
||||
return t.Coll(table)
|
||||
}
|
||||
|
||||
// initIndex 初始化索引
|
||||
func initIndex() {
|
||||
many := []mongo.IndexModel{
|
||||
{
|
||||
Keys: bson.D{{Key: "sumDate", Value: -1}},
|
||||
},
|
||||
{
|
||||
Keys: bson.D{{Key: "channel", Value: 1}, {Key: "type", Value: 1}},
|
||||
},
|
||||
{
|
||||
Keys: bson.D{{Key: "sumDate", Value: -1}, {Key: "channel", Value: 1}, {Key: "type", Value: 1}},
|
||||
Options: options.Index().SetUnique(true),
|
||||
},
|
||||
}
|
||||
if _, err := coll(nil).CreateIndex(many); err != nil {
|
||||
panic(fmt.Sprintf("%s model set index err ==>[%+v]", table, err))
|
||||
}
|
||||
}
|
||||
|
||||
func ChangeRechargeStatTrans(trans *db.MongoTool, sumDate, recordTime time.Time, stateMap RecOrderStateMap) error {
|
||||
writes := make([]mongo.WriteModel, len(stateMap))
|
||||
i := 0
|
||||
for channel, statDoc := range stateMap {
|
||||
model, err := toMgoModel(sumDate, recordTime, channel, Recharge, statDoc.RechargeOrderStateIncDoc, statDoc.RechargeOrderStateSetDoc)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
writes[i] = model
|
||||
i++
|
||||
}
|
||||
if len(writes) == 0 {
|
||||
return nil
|
||||
}
|
||||
//bulkWrite 不是原子操作 不具备事务性
|
||||
opt := (&options.BulkWriteOptions{}).SetOrdered(false) //设为无序,触发并行写,提升写效率
|
||||
_, err := coll(trans).Bulk(writes, opt)
|
||||
return err
|
||||
}
|
||||
|
||||
func ChangeWithDrawStatTrans(trans *db.MongoTool, sumDate, recordTime time.Time, stateMap WitOrderStateMap) error {
|
||||
writes := make([]mongo.WriteModel, len(stateMap))
|
||||
i := 0
|
||||
for channel, statDoc := range stateMap {
|
||||
model, err := toMgoModel(sumDate, recordTime, channel, WithDraw, statDoc.WithDrawOrderStateIncDoc, statDoc.WithDrawOrderStateSetDoc)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
writes[i] = model
|
||||
i++
|
||||
}
|
||||
if len(writes) == 0 {
|
||||
return nil
|
||||
}
|
||||
//bulkWrite 不是原子操作 不具备事务性
|
||||
opt := (&options.BulkWriteOptions{}).SetOrdered(false) //设为无序,触发并行写,提升写效率
|
||||
_, err := coll(trans).Bulk(writes, opt)
|
||||
return err
|
||||
}
|
||||
|
||||
func toMgoModel(sumDate, recordTime time.Time, channel string, typ OrderType, inc orderStateIncDoc, set orderStateSetDoc) (mongo.WriteModel, error) {
|
||||
filter := M{
|
||||
"sumDate": sumDate,
|
||||
"channel": channel,
|
||||
"type": typ,
|
||||
}
|
||||
setM, err := common.ToBsonM(set)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
setM["recordAt"] = recordTime
|
||||
setM["updatedAt"] = time.Now()
|
||||
incM, err := common.ToBsonM(inc)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
update := M{
|
||||
"$setOnInsert": M{
|
||||
"sumDate": sumDate,
|
||||
"channel": channel,
|
||||
"type": typ,
|
||||
"createdAt": time.Now(),
|
||||
},
|
||||
}
|
||||
if len(incM) != 0 {
|
||||
update["$inc"] = incM
|
||||
}
|
||||
if len(setM) != 0 {
|
||||
update["$set"] = setM
|
||||
}
|
||||
model := mongo.NewUpdateOneModel().
|
||||
SetFilter(filter).
|
||||
SetUpdate(update).
|
||||
SetUpsert(true)
|
||||
log.Debug(fmt.Sprintf("sumDate:%s table:%s [ typ:%s channel:%s filter:%+v update:%+v ]\n", sumDate, table, typ.String(), channel, filter, update))
|
||||
return model, nil
|
||||
}
|
||||
@@ -0,0 +1,90 @@
|
||||
package statordermod
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"91porn-server/common/db"
|
||||
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||
)
|
||||
|
||||
type (
|
||||
OrderType int
|
||||
|
||||
M = bson.M
|
||||
|
||||
Sort = bson.D
|
||||
)
|
||||
|
||||
const (
|
||||
Recharge OrderType = 1
|
||||
WithDraw OrderType = 2
|
||||
)
|
||||
|
||||
func (o *OrderType) String() string {
|
||||
switch *o {
|
||||
case Recharge:
|
||||
return "recharge"
|
||||
case WithDraw:
|
||||
return "withDraw"
|
||||
default:
|
||||
return "unknow"
|
||||
}
|
||||
}
|
||||
|
||||
type Items struct {
|
||||
OrderCount int64 `json:"orderCount" bson:"orderCount"` //生产订单数
|
||||
CompleteOrderCount int64 `json:"completeOrderCount" bson:"completeOrderCount"` //完成订单数
|
||||
OrderAmount int64 `json:"orderAmount" bson:"orderAmount"` //生产订单金额
|
||||
CompleteOrderAmount int64 `json:"completeOrderAmount" bson:"completeOrderAmount"` //完成订单的实际支付金额
|
||||
PaidOrderCount int64 `json:"paidOrderCount" bson:"paidOrderCount"` //用户已支付金额/三方已出款金额
|
||||
PaidAmount int64 `json:"paidAmount" bson:"paidAmount"` //用户已支付订单数/三方已出款订单数
|
||||
}
|
||||
|
||||
// OrderState 充值提现统计
|
||||
type OrderState struct {
|
||||
ID primitive.ObjectID `json:"id" bson:"_id,omitempty"` // ID
|
||||
SumDate time.Time `json:"sumDate" bson:"sumDate"`
|
||||
|
||||
Items `bson:",inline"`
|
||||
|
||||
Channel string `json:"channel" bson:"channel"` //充值渠道
|
||||
Type OrderType `json:"type" bson:"type"` //订单类型， 1，充值；2，提现
|
||||
RecordAt time.Time `json:"recordAt" bson:"recordAt"` //数据记录时间
|
||||
CreatedAt time.Time `json:"createdAt" bson:"createdAt"` //文档创建时间
|
||||
UpdatedAt time.Time `json:"updatedAt" bson:"updatedAt"` //文档更新时间
|
||||
}
|
||||
|
||||
func Init() {
|
||||
mdb = db.Init(table)
|
||||
initIndex()
|
||||
}
|
||||
|
||||
type orderStateSetDoc struct {
|
||||
}
|
||||
|
||||
type orderStateIncDoc struct {
|
||||
OrderCount *int64 `bson:"orderCount,omitempty"` //订单数
|
||||
CompleteOrderCount *int64 `bson:"completeOrderCount,omitempty"` //完成订单数
|
||||
OrderAmount *int64 `bson:"orderAmount,omitempty"` //订单金额
|
||||
CompleteOrderAmount *int64 `bson:"completeOrderAmount,omitempty"` //实际支付金额
|
||||
PaidOrderCount *int64 `bson:"paidOrderCount,omitempty"` //用户已支付订单数/三方已出款订单数
|
||||
PaidAmount *int64 `bson:"paidAmount,omitempty"` //用户已支付金额/三方已出款金额
|
||||
}
|
||||
|
||||
type RechargeOrderStateIncDoc = orderStateIncDoc
|
||||
type RechargeOrderStateSetDoc = orderStateSetDoc
|
||||
type RechargeOrderStateDoc struct {
|
||||
RechargeOrderStateIncDoc
|
||||
RechargeOrderStateSetDoc
|
||||
}
|
||||
type RecOrderStateMap = map[string]*RechargeOrderStateDoc
|
||||
|
||||
type WithDrawOrderStateIncDoc = orderStateIncDoc
|
||||
type WithDrawOrderStateSetDoc = orderStateSetDoc
|
||||
type WithDrawOrderStateDoc struct {
|
||||
WithDrawOrderStateIncDoc
|
||||
WithDrawOrderStateSetDoc
|
||||
}
|
||||
type WitOrderStateMap = map[string]*WithDrawOrderStateDoc
|
||||
Reference in New Issue
Block a user