Initial commit

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

models/l/payvidlgmod/discStat.go

@@ -0,0 +1,114 @@
+package payvidlgmod
+
+import (
+	"fmt"
+	"strings"
+	"time"
+
+	"91porn-server/common/log"
+	"91porn-server/common/pageopt"
+	"91porn-server/common/ysinterface/disc"
+
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+type DistrictStatKey struct {
+	DiscSeqe `bson:",inline"` //商区码
+	SysType  string           `bson:"sysType"` //系统类型 iOS Android
+}
+
+func (d DistrictStatKey) GetDiscCode() string {
+	return d.DistrictCode
+}
+
+func (d DistrictStatKey) GetPromSeqe() string {
+	return d.PromSeqe
+}
+
+func (d DistrictStatKey) GetSysType() string {
+	return d.SysType
+}
+
+func (d DistrictStatKey) String() string {
+	if d.DiscSeqe.String() == "" {
+		return ""
+	}
+	return strings.ToUpper(strings.Join([]string{d.DiscSeqe.String(), d.SysType}, "-"))
+}
+
+type DistrictStater = disc.DistrictStater
+
+// DiscSeqeTransCount
+func DiscSeqeBuyVidCount(start, end time.Time, mats ...Matcher) (map[DistrictStater]int64, error) {
+	mats = append(mats, (&CreatedAtGTEAndLTMatch{GTE: &start, LT: &end}).New())
+	filter := pageopt.MergeM(mats)
+	list := []DistrictStatKey{}
+	opt := (&options.FindOptions{}).SetProjection(M{
+		"districtCode": 1,
+		"promSeqe":     1,
+		"sysType":      1,
+	})
+	if err := coll(nil).Find(&list, filter, opt); err != nil {
+		log.Warn(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "DiscSeqeTransCount", table, "Find", err))
+		return nil, err
+	}
+	m := make(map[DistrictStater]int64)
+	for _, v := range list {
+		if v.String() != "" {
+			m[v] += 1
+		}
+	}
+	return m, nil
+}
+
+// DiscSeqeBuyVidCoins
+func DiscSeqeBuyVidCoins(start, end time.Time, mats ...Matcher) (map[DistrictStater]int64, error) {
+	mats = append(mats, (&CreatedAtGTEAndLTMatch{GTE: &start, LT: &end}).New())
+	filter := pageopt.MergeM(mats)
+	list := []struct {
+		DistrictStatKey `bson:",inline"` //商区码
+		Coins           int64            `bson:"coins"`
+	}{}
+	opt := (&options.FindOptions{}).SetProjection(M{
+		"districtCode": 1,
+		"promSeqe":     1,
+		"sysType":      1,
+		"coins":        1,
+	})
+	if err := coll(nil).Find(&list, filter, opt); err != nil {
+		return nil, err
+	}
+	m := make(map[DistrictStater]int64, len(list))
+	for _, v := range list {
+		if v.String() != "" {
+			m[v.DistrictStatKey] += v.Coins
+		}
+	}
+	return m, nil
+}
+
+// DiscSeqeTaxAmount
+func DiscSeqeTaxAmount(start, end time.Time, mats ...Matcher) (map[DistrictStater]float64, error) {
+	mats = append(mats, (&CreatedAtGTEAndLTMatch{GTE: &start, LT: &end}).New())
+	filter := pageopt.MergeM(mats)
+	list := []struct {
+		DistrictStatKey `bson:",inline"` //商区码
+		TaxAmount       float64          `bson:"taxAmount"` //税额
+	}{}
+	opt := (&options.FindOptions{}).SetProjection(M{
+		"districtCode": 1,
+		"promSeqe":     1,
+		"sysType":      1,
+		"taxAmount":    1,
+	})
+	if err := coll(nil).Find(&list, filter, opt); err != nil {
+		return nil, err
+	}
+	m := make(map[DistrictStater]float64, len(list))
+	for _, v := range list {
+		if v.String() != "" {
+			m[v.DistrictStatKey] += v.TaxAmount
+		}
+	}
+	return m, nil
+}