增加退单管理

2026-10-09 19:50:49 +08:00
parent 22def0b5cb
commit 3d9bdb6f6e
11 changed files with 591 additions and 0 deletions
@@ -0,0 +1,101 @@
+package refundOrderMod
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"stat/common/log"
+	"stat/common/mongos"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+func Init(ctx context.Context) {
+	db = mongos.New(mongos.MgoManager, TN.RefundOrder)
+	indexes := []mongo.IndexModel{
+		// (appId, orderId) 唯一,从数据库层杜绝重复添加 —— 应用层的"先查后插"
+		// 存在并发竞态,两个请求可能同时通过检查。与 log_recharge 的唯一键对齐。
+		// orderId 为空的历史数据不参与唯一约束,否则多条空值会互相冲突。
+		{
+			Keys: bson.D{
+				{Key: FN.RefundOrder.AppId, Value: 1},
+				{Key: FN.RefundOrder.OrderId, Value: 1},
+			},
+			Options: options.Index().SetUnique(true).
+				SetPartialFilterExpression(bson.M{FN.RefundOrder.OrderId: bson.M{"$gt": ""}}),
+		},
+		{Keys: bson.D{{Key: FN.RefundOrder.CID, Value: 1}}},
+		{Keys: bson.D{{Key: FN.RefundOrder.UserId, Value: 1}}},
+		{Keys: bson.D{{Key: FN.RefundOrder.ChannelName, Value: 1}}},
+		{Keys: bson.D{{Key: FN.RefundOrder.DistrictCode, Value: 1}}},
+		{Keys: bson.D{{Key: FN.RefundOrder.CreatedAt, Value: -1}}},
+	}
+	if _, err := db.CreateIndex(ctx, indexes); err != nil {
+		panic(fmt.Sprintf("(refundOrderMod:Init) CreateIndex failed: %+v", err))
+	}
+}
+
+// InsertOneRefundOrder 插入一条。
+// 必须用 InsertOneTiny: db.InsertOne 会调 addCreatedAt 把 CreatedAt 无条件
+// 改写成 time.Now(),而本表的 createdAt 存的是原订单创建时间,不能被覆盖。
+// 命中唯一索引时返回的 error 可用 IsDuplicateKey 判定。
+func InsertOneRefundOrder(ctx context.Context, doc *RefundOrder) error {
+	if err := db.InsertOneTiny(ctx, doc); err != nil {
+		if !IsDuplicateKey(err) {
+			log.Warn("refundOrderMod InsertOne failed", log.E(err))
+		}
+		return err
+	}
+	return nil
+}
+
+// IsDuplicateKey 判断是否唯一索引冲突。
+// 驱动为 v1.4.1,没有 mongo.IsDuplicateKeyError(1.11 才引入),按错误码 11000 判定。
+func IsDuplicateKey(err error) bool {
+	var we mongo.WriteException
+	if errors.As(err, &we) {
+		for _, e := range we.WriteErrors {
+			if e.Code == 11000 {
+				return true
+			}
+		}
+	}
+	return false
+}
+
+func FindOneRefundOrder(ctx context.Context, filter bson.M) (*RefundOrder, error) {
+	var doc RefundOrder
+	if err := db.FindOne(ctx, &doc, filter); err != nil {
+		return nil, err
+	}
+	return &doc, nil
+}
+
+func FindManyRefundOrders(ctx context.Context, filter bson.M, opts ...*options.FindOptions) (RefundOrderList, error) {
+	list := make(RefundOrderList, 0)
+	if err := db.Find(ctx, &list, filter, opts...); err != nil {
+		log.Warn("refundOrderMod Find failed", log.E(err), log.Any("filter", filter))
+		return nil, err
+	}
+	return list, nil
+}
+
+func CountRefundOrders(ctx context.Context, filter bson.M) (int64, error) {
+	count, err := db.Count(ctx, filter)
+	if err != nil {
+		log.Warn("refundOrderMod Count failed", log.E(err), log.Any("filter", filter))
+		return 0, err
+	}
+	return count, nil
+}
+
+func DelOneRefundOrder(ctx context.Context, filter bson.M) (int64, error) {
+	count, err := db.DeleteOne(ctx, filter)
+	if err != nil {
+		log.Warn("refundOrderMod DeleteOne failed", log.E(err), log.Any("filter", filter))
+		return 0, err
+	}
+	return count, nil
+}
@@ -0,0 +1,37 @@
+package refundOrderMod
+
+type TagRefundOrder struct {
+	ID           string
+	AppId        string
+	CID          string
+	DistrictCode string
+	Money        string
+	OrderId      string
+	UserId       string
+	Type         string
+	ChannelName  string
+	CreatedAt    string
+}
+
+var FN struct {
+	RefundOrder TagRefundOrder
+}
+
+var TN struct {
+	RefundOrder string
+}
+
+func init() {
+	FN.RefundOrder.ID = "_id"
+	FN.RefundOrder.AppId = "appId"
+	FN.RefundOrder.CID = "cid"
+	FN.RefundOrder.DistrictCode = "districtCode"
+	FN.RefundOrder.Money = "money"
+	FN.RefundOrder.OrderId = "orderId"
+	FN.RefundOrder.UserId = "userId"
+	FN.RefundOrder.Type = "type"
+	FN.RefundOrder.ChannelName = "channelName"
+	FN.RefundOrder.CreatedAt = "createdAt"
+
+	TN.RefundOrder = CollRefundOrder
+}
@@ -0,0 +1,31 @@
+package refundOrderMod
+
+import (
+	"stat/common/mongos"
+	"time"
+
+	"go.mongodb.org/mongo-driver/bson/primitive"
+)
+
+// 表名
+const CollRefundOrder = "refund_order"
+
+var db *mongos.MongoTool
+
+// RefundOrder 退款订单
+// 字段取自 log_recharge 的同名字段,按 cid 回填时逐字段拷贝,是下单时点的快照
+// @table:refund_order
+type RefundOrder struct {
+	ID           primitive.ObjectID `json:"id,omitempty" bson:"_id,omitempty"`
+	AppId        int32              `json:"appId" bson:"appId"`               // appId
+	CID          string             `json:"cid" bson:"cid"`                   // 渠道id
+	DistrictCode string             `json:"districtCode" bson:"districtCode"` // 渠道码
+	Money        int64              `json:"money" bson:"money"`               // 订单金额
+	OrderId      string             `json:"orderId" bson:"orderId"`           // 流水id
+	UserId       uint64             `json:"userId" bson:"userId"`             // 用户ID
+	Type         string             `json:"type" bson:"type"`                 // 充值方式
+	ChannelName  string             `json:"channelName" bson:"channelName"`   // 支付渠道名字
+	CreatedAt    time.Time          `json:"createdAt" bson:"createdAt"`       // 原订单创建时间(取自 log_recharge.createdAt)
+}
+
+type RefundOrderList []*RefundOrder