微信支付

internal/domains/finance/entities/recharge_record.go

@@ -14,6 +14,7 @@ type RechargeType string
 
 const (
 	RechargeTypeAlipay   RechargeType = "alipay"   // 支付宝充值
+	RechargeTypeWechat   RechargeType = "wechat"   // 微信充值
 	RechargeTypeTransfer RechargeType = "transfer" // 对公转账
 	RechargeTypeGift     RechargeType = "gift"     // 赠送
 )
@@ -42,6 +43,7 @@ type RechargeRecord struct {
 
 	// 订单号字段（根据充值类型使用不同字段）
 	AlipayOrderID   *string `gorm:"type:varchar(64);uniqueIndex" json:"alipay_order_id,omitempty" comment:"支付宝订单号"`
+	WechatOrderID   *string `gorm:"type:varchar(64);uniqueIndex" json:"wechat_order_id,omitempty" comment:"微信订单号"`
 	TransferOrderID *string `gorm:"type:varchar(64);uniqueIndex" json:"transfer_order_id,omitempty" comment:"转账订单号"`
 
 	// 通用字段
@@ -104,14 +106,24 @@ func (r *RechargeRecord) MarkCancelled() {
 // ValidatePaymentMethod 验证支付方式：支付宝订单号和转账订单号只能有一个存在
 func (r *RechargeRecord) ValidatePaymentMethod() error {
 	hasAlipay := r.AlipayOrderID != nil && *r.AlipayOrderID != ""
+	hasWechat := r.WechatOrderID != nil && *r.WechatOrderID != ""
 	hasTransfer := r.TransferOrderID != nil && *r.TransferOrderID != ""
 
-	if hasAlipay && hasTransfer {
-		return errors.New("支付宝订单号和转账订单号不能同时存在")
+	count := 0
+	if hasAlipay {
+		count++
 	}
-
-	if !hasAlipay && !hasTransfer {
-		return errors.New("必须提供支付宝订单号或转账订单号")
+	if hasWechat {
+		count++
+	}
+	if hasTransfer {
+		count++
+	}
+	if count > 1 {
+		return errors.New("支付宝、微信或转账订单号只能存在一个")
+	}
+	if count == 0 {
+		return errors.New("必须提供支付宝、微信或转账订单号")
 	}
 
 	return nil
@@ -124,6 +136,10 @@ func (r *RechargeRecord) GetOrderID() string {
 		if r.AlipayOrderID != nil {
 			return *r.AlipayOrderID
 		}
+	case RechargeTypeWechat:
+		if r.WechatOrderID != nil {
+			return *r.WechatOrderID
+		}
 	case RechargeTypeTransfer:
 		if r.TransferOrderID != nil {
 			return *r.TransferOrderID
@@ -137,6 +153,11 @@ func (r *RechargeRecord) SetAlipayOrderID(orderID string) {
 	r.AlipayOrderID = &orderID
 }
 
+// SetWechatOrderID 设置微信订单号
+func (r *RechargeRecord) SetWechatOrderID(orderID string) {
+	r.WechatOrderID = &orderID
+}
+
 // SetTransferOrderID 设置转账订单号
 func (r *RechargeRecord) SetTransferOrderID(orderID string) {
 	r.TransferOrderID = &orderID
@@ -153,6 +174,17 @@ func NewAlipayRechargeRecord(userID string, amount decimal.Decimal, alipayOrderI
 	}
 }
 
+// NewWechatRechargeRecord 工厂方法 - 创建微信充值记录
+func NewWechatRechargeRecord(userID string, amount decimal.Decimal, wechatOrderID string) *RechargeRecord {
+	return &RechargeRecord{
+		UserID:        userID,
+		Amount:        amount,
+		RechargeType:  RechargeTypeWechat,
+		Status:        RechargeStatusPending,
+		WechatOrderID: &wechatOrderID,
+	}
+}
+
 // NewTransferRechargeRecord 工厂方法 - 创建对公转账充值记录
 func NewTransferRechargeRecord(userID string, amount decimal.Decimal, transferOrderID, notes string) *RechargeRecord {
 	return &RechargeRecord{