init: web_hgcg_server 初始化提交

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-10-27 16:44:55 +08:00
commit 189556e702
627 changed files with 126228 additions and 0 deletions
@@ -0,0 +1,171 @@
+package vipordermod
+
+import (
+	"context"
+	"fmt"
+	"time"
+	"videoweb/pub/common/log"
+	"videoweb/pub/common/srvdef"
+	"videoweb/pub/mongos"
+	"videoweb/pub/pubdef"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+var coll *mongos.Collection
+
+func Init(ctx context.Context) {
+	coll = mongos.Shop.Coll(TN.VipOrder)
+	initIndex(ctx)
+}
+
+func initIndex(ctx context.Context) {
+	indexes := []mongo.IndexModel{
+		{Keys: bson.D{{Key: FN.VipOrder.UserId, Value: -1}}},
+		{Keys: bson.D{{Key: FN.VipOrder.TradeNo, Value: 1}}, Options: options.Index().SetUnique(true)},
+		{Keys: bson.D{{Key: FN.VipOrder.Status, Value: 1}}},
+		{Keys: bson.D{{Key: FN.VipOrder.DistrictCode, Value: 1}}},
+		{Keys: bson.D{{Key: FN.VipOrder.CreatedAt, Value: -1}}},
+	}
+	if _, err := coll.CreateIndex(ctx, indexes); err != nil {
+		panic(fmt.Sprintf("(vipordermod:initIndex) Create Index failed! err:%+v:", err))
+	}
+}
+
+func InsertOneVipOrder(ctx context.Context, doc *VipOrder) *srvdef.SrvErr {
+	if _, err := coll.InsertOne(ctx, doc); err != nil {
+		log.Error("vipordermod:InsertOneVipOrder failed", log.Any("doc", doc), log.E(err))
+		return srvdef.New(srvdef.ErrDbInsertError, "", "")
+	}
+	return nil
+}
+
+func FindOneVipOrder(ctx context.Context, filter bson.M) (*VipOrder, *srvdef.SrvErr) {
+	var doc VipOrder
+	if err := coll.FindOneEx(ctx, &doc, filter); err != nil {
+		if err != mongo.ErrNoDocuments {
+			log.Warn("vipordermod:FindOneVipOrder failed", log.E(err), log.Any("filter", filter))
+			return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
+		}
+		return nil, srvdef.New(srvdef.ErrNoData, "", "")
+	}
+	return &doc, nil
+}
+
+func FindManyVipOrder(ctx context.Context, filter bson.M, opts ...*options.FindOptions) (VipOrderList, *srvdef.SrvErr) {
+	var data VipOrderList
+	if err := coll.Find(ctx, &data, filter, opts...); err != nil {
+		log.Warn("vipordermod:FindManyVipOrder failed", log.E(err), log.Any("filter", filter))
+		return nil, srvdef.New(srvdef.ErrDbQueryError, "", "")
+	}
+	return data, nil
+}
+
+func CountVipOrder(ctx context.Context, filter bson.M) (int64, *srvdef.SrvErr) {
+	count, err := coll.Count(ctx, filter)
+	if err != nil {
+		log.Warn("vipordermod:CountVipOrder failed", log.E(err), log.Any("filter", filter))
+		return 0, srvdef.New(srvdef.ErrDbQueryError, "", "")
+	}
+	return count, nil
+}
+
+// StatSyncSuccessOrderList 增量拉取支付成功订单（统计中心同步）
+func StatSyncSuccessOrderList(ctx context.Context, successTime time.Time, size int64) (VipOrderList, error) {
+	opt := (&options.FindOptions{}).SetLimit(size).SetSort(bson.M{FN.VipOrder.FinishedAt: 1})
+	// 预留足够时间让注册数据先到达统计中心，避免充值记录找不到对应注册用户
+	endTime := time.Now().Add(-time.Minute * 4)
+	filter := bson.M{
+		FN.VipOrder.FinishedAt: bson.M{"$gt": successTime, "$lt": endTime},
+		FN.VipOrder.Status:     pubdef.VipOrderSuccess,
+	}
+	list := VipOrderList{}
+	if err := coll.Find(ctx, &list, filter, opt); err != nil {
+		return nil, err
+	}
+	return list, nil
+}
+
+// StatSyncAllOrderList 增量拉取非失败订单（统计中心同步）
+func StatSyncAllOrderList(ctx context.Context, createTime time.Time, size int64) (VipOrderList, error) {
+	opt := (&options.FindOptions{}).SetLimit(size).SetSort(bson.M{FN.VipOrder.CreatedAt: 1})
+	endTime := time.Now().Add(-time.Minute * 1)
+	filter := bson.M{
+		FN.VipOrder.CreatedAt: bson.M{"$gt": createTime, "$lt": endTime},
+		FN.VipOrder.Status:    bson.M{"$ne": pubdef.VipOrderFailed},
+	}
+	list := VipOrderList{}
+	if err := coll.Find(ctx, &list, filter, opt); err != nil {
+		return nil, err
+	}
+	return list, nil
+}
+
+// StatSyncVipOrderList 增量拉取会员购买成功订单（统计中心同步）
+func StatSyncVipOrderList(ctx context.Context, successTime time.Time, size int64) (VipOrderList, error) {
+	opt := (&options.FindOptions{}).SetLimit(size).SetSort(bson.M{FN.VipOrder.FinishedAt: 1})
+	endTime := time.Now().Add(-time.Minute * 1)
+	filter := bson.M{
+		FN.VipOrder.FinishedAt: bson.M{"$gt": successTime, "$lt": endTime},
+		FN.VipOrder.Status:     pubdef.VipOrderSuccess,
+		FN.VipOrder.OrderType:  OrderTypeVip,
+	}
+	list := VipOrderList{}
+	if err := coll.Find(ctx, &list, filter, opt); err != nil {
+		return nil, err
+	}
+	return list, nil
+}
+
+func EmptyDistrictFilter() bson.M {
+	return bson.M{
+		"$or": []bson.M{
+			{FN.VipOrder.DistrictCode: bson.M{"$exists": false}},
+			{FN.VipOrder.DistrictCode: ""},
+			{FN.VipOrder.DistrictCode: nil},
+		},
+	}
+}
+
+// AggregateVipOrder 聚合查询充值订单
+func AggregateVipOrder(ctx context.Context, bind interface{}, pipeline []bson.M) *srvdef.SrvErr {
+	if err := coll.Aggregate(ctx, bind, pipeline); err != nil {
+		log.Warn("vipordermod:AggregateVipOrder failed", log.E(err), log.Any("pipeline", pipeline))
+		return srvdef.New(srvdef.ErrDbQueryError, "", "")
+	}
+	return nil
+}
+
+func UpdateManyVipOrder(ctx context.Context, filter bson.M, update bson.M) (int64, *srvdef.SrvErr) {
+	result, err := coll.UpdateMany(ctx, filter, update)
+	if err != nil {
+		log.Warn("vipordermod:UpdateManyVipOrder failed", log.E(err), log.Any("filter", filter))
+		return 0, srvdef.New(srvdef.ErrDbUpdateError, "", "")
+	}
+	return result.ModifiedCount, nil
+}
+
+// UpdateOrderSuccess 将订单标记为支付成功
+func UpdateOrderSuccess(ctx context.Context, tradeNo string, oid string, realAmount int64, paymentAt time.Time) *srvdef.SrvErr {
+	now := time.Now()
+	result, err := coll.UpdateOne(ctx,
+		bson.M{FN.VipOrder.TradeNo: tradeNo},
+		bson.M{"$set": bson.M{
+			FN.VipOrder.Status:     pubdef.VipOrderSuccess,
+			FN.VipOrder.Oid:        oid,
+			FN.VipOrder.Amount:     realAmount,
+			FN.VipOrder.FinishedAt: paymentAt,
+			FN.VipOrder.UpdatedAt:  now,
+		}},
+	)
+	if err != nil {
+		log.Warn("vipordermod:UpdateOrderSuccess failed", log.E(err), log.Any("tradeNo", tradeNo))
+		return srvdef.New(srvdef.ErrDbUpdateError, "", "")
+	}
+	if result.MatchedCount == 0 {
+		return srvdef.New(srvdef.ErrNoData, "订单不存在", "")
+	}
+	return nil
+}
@@ -0,0 +1,64 @@
+package vipordermod
+
+type TagVipOrder struct {
+	Id           string
+	OrderType    string
+	UserId       string
+	CardId       string
+	CardName     string
+	CoinCfgId    string
+	Coin         string
+	BaseCoin     string
+	GiftCoin     string
+	Amount       string
+	PayMode      string
+	TradeNo      string
+	Oid          string
+	Status       string
+	PayUrl       string
+	PayUrlMode   string
+	PayChannel   string
+	PayRate      string
+	IPAddr       string
+	SysType      string
+	DistrictCode string
+	CreatedAt    string
+	UpdatedAt    string
+	FinishedAt   string
+}
+
+var FN struct {
+	VipOrder TagVipOrder
+}
+var TN struct {
+	VipOrder string
+}
+
+func init() {
+	FN.VipOrder.Id = "_id"
+	FN.VipOrder.OrderType = "orderType"
+	FN.VipOrder.UserId = "userId"
+	FN.VipOrder.CardId = "cardId"
+	FN.VipOrder.CardName = "cardName"
+	FN.VipOrder.CoinCfgId = "coinCfgId"
+	FN.VipOrder.Coin = "coin"
+	FN.VipOrder.BaseCoin = "baseCoin"
+	FN.VipOrder.GiftCoin = "giftCoin"
+	FN.VipOrder.Amount = "amount"
+	FN.VipOrder.PayMode = "payMode"
+	FN.VipOrder.TradeNo = "tradeNo"
+	FN.VipOrder.Oid = "oid"
+	FN.VipOrder.Status = "status"
+	FN.VipOrder.PayUrl = "payUrl"
+	FN.VipOrder.PayUrlMode = "payUrlMode"
+	FN.VipOrder.PayChannel = "payChannel"
+	FN.VipOrder.PayRate = "payRate"
+	FN.VipOrder.IPAddr = "ipAddr"
+	FN.VipOrder.SysType = "sysType"
+	FN.VipOrder.DistrictCode = "districtCode"
+	FN.VipOrder.CreatedAt = "createdAt"
+	FN.VipOrder.UpdatedAt = "updatedAt"
+	FN.VipOrder.FinishedAt = "finishedAt"
+
+	TN.VipOrder = "vip_order"
+}
@@ -0,0 +1,46 @@
+package vipordermod
+
+import (
+	"time"
+	"videoweb/pub/pubdef"
+)
+
+// OrderType 订单类型
+type OrderType int8
+
+const (
+	OrderTypeVip  OrderType = 1 // 会员购买
+	OrderTypeCoin OrderType = 2 // 金币购买
+)
+
+// @table: vip_order
+type VipOrder struct {
+	Id        int64     `json:"id" bson:"_id"`              // 唯一ID
+	OrderType OrderType `json:"orderType" bson:"orderType"` // 订单类型：1会员 2金币
+	UserId    int64     `json:"userId" bson:"userId"`       // 用户ID
+	// ── VIP 专用 ──
+	CardId   int64  `json:"cardId" bson:"cardId"`     // 购买套餐ID（会员畅享）
+	CardName string `json:"cardName" bson:"cardName"` // 套餐名快照
+	// ── 金币专用 ──
+	CoinCfgId    int64                 `json:"coinCfgId" bson:"coinCfgId"`       // 购买套餐ID（金币充值）
+	Coin         int64                 `json:"coin" bson:"coin"`                 // 到账金币数（快照）
+	BaseCoin     int64                 `json:"baseCoin" bson:"baseCoin"`         // 基础金币数（快照）
+	GiftCoin     int64                 `json:"giftCoin" bson:"giftCoin"`         // 赠送金币数（快照）
+	Amount       int64                 `json:"amount" bson:"amount"`             // 支付金额（分）
+	PayMode      string                `json:"payMode" bson:"payMode"`           // 支付方式：overage/alipay/wechat
+	TradeNo      string                `json:"tradeNo" bson:"tradeNo"`           // 本地订单号
+	Oid          string                `json:"oid" bson:"oid"`                   // 第三方订单号
+	Status       pubdef.VipOrderStatus `json:"status" bson:"status"`             // 订单状态
+	PayUrl       string                `json:"payUrl" bson:"payUrl"`             // 支付链接
+	PayUrlMode   string                `json:"payUrlMode" bson:"payUrlMode"`     // 链接打开方式
+	PayChannel   string                `json:"payChannel" bson:"payChannel"`     // 支付渠道
+	PayRate      string                `json:"payRate" bson:"payRate"`           // 费率
+	IPAddr       string                `json:"ipAddr" bson:"ipAddr"`             // 客户端 IP
+	SysType      string                `json:"sysType" bson:"sysType"`           // ios/android
+	DistrictCode string                `json:"districtCode" bson:"districtCode"` // 下单时用户渠道码
+	CreatedAt    time.Time             `json:"createdAt" bson:"createdAt"`       // 创建时间
+	UpdatedAt    time.Time             `json:"updatedAt" bson:"updatedAt"`       // 更新时间
+	FinishedAt   time.Time             `json:"finishedAt" bson:"finishedAt"`     // 完成时间
+}
+
+type VipOrderList []*VipOrder