104 lines
3.4 KiB
Go
104 lines
3.4 KiB
Go
package model
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
// TradeStatus 交易状态
|
|
type TradeStatus string
|
|
|
|
const (
|
|
TradeStatusCreating TradeStatus = "CREATING"
|
|
TradeStatusPaying TradeStatus = "PAYING"
|
|
TradeStatusPaid TradeStatus = "PAID"
|
|
TradeStatusClosed TradeStatus = "CLOSED"
|
|
TradeStatusFailed TradeStatus = "FAILED"
|
|
TradeStatusCreateFailed TradeStatus = "CREATE_FAILED"
|
|
TradeStatusRefunded TradeStatus = "REFUNDED"
|
|
)
|
|
|
|
// PayMethod 支付方式
|
|
type PayMethod string
|
|
|
|
const (
|
|
PayMethodWechatJSAPI PayMethod = "WECHAT_JSAPI"
|
|
PayMethodWechatH5 PayMethod = "WECHAT_H5"
|
|
PayMethodWechatNative PayMethod = "WECHAT_NATIVE"
|
|
PayMethodWechatMini PayMethod = "WECHAT_MINI"
|
|
PayMethodAlipay PayMethod = "ALIPAY"
|
|
PayMethodQuickPay PayMethod = "QUICK_PAY"
|
|
PayMethodTransfer PayMethod = "TRANSFER" // 对公转账
|
|
)
|
|
|
|
// JSONMap JSON 字段类型
|
|
type JSONMap map[string]any
|
|
|
|
func (j JSONMap) Value() (driver.Value, error) {
|
|
if j == nil {
|
|
return nil, nil
|
|
}
|
|
b, err := json.Marshal(j)
|
|
return string(b), err
|
|
}
|
|
|
|
func (j *JSONMap) Scan(value any) error {
|
|
if value == nil {
|
|
*j = nil
|
|
return nil
|
|
}
|
|
var bytes []byte
|
|
switch v := value.(type) {
|
|
case string:
|
|
bytes = []byte(v)
|
|
case []byte:
|
|
bytes = v
|
|
default:
|
|
return fmt.Errorf("unsupported type: %T", value)
|
|
}
|
|
return json.Unmarshal(bytes, j)
|
|
}
|
|
|
|
// TradeOrder 交易订单
|
|
type TradeOrder struct {
|
|
ID uint64 `gorm:"column:id;primaryKey;autoIncrement"`
|
|
TradeNo string `gorm:"column:trade_no;uniqueIndex;size:32;not null"`
|
|
MerchantOrderNo string `gorm:"column:merchant_order_no;size:64;not null"`
|
|
AppID string `gorm:"column:app_id;size:32;not null;index:idx_app_merchant,unique"`
|
|
ChannelCode string `gorm:"column:channel_code;size:32;not null"`
|
|
ChannelTradeNo string `gorm:"column:channel_trade_no;size:64;index"`
|
|
PayMethod PayMethod `gorm:"column:pay_method;size:32;not null"`
|
|
Amount int64 `gorm:"column:amount;not null"`
|
|
ProfitSharingAmount int64 `gorm:"column:profit_sharing_amount;not null;default:0"`
|
|
ServiceFeeAmount int64 `gorm:"column:service_fee_amount;not null;default:0"`
|
|
Subject string `gorm:"column:subject;size:256;not null"`
|
|
NotifyURL string `gorm:"column:notify_url;size:512;not null"`
|
|
Status TradeStatus `gorm:"column:status;size:20;not null;default:CREATING"`
|
|
Extra JSONMap `gorm:"column:extra;type:json"`
|
|
ChannelExtra JSONMap `gorm:"column:channel_extra;type:json"`
|
|
ExpireTime time.Time `gorm:"column:expire_time;not null"`
|
|
PayTime *time.Time `gorm:"column:pay_time"`
|
|
CreatedAt time.Time `gorm:"column:created_at;autoCreateTime:milli"`
|
|
UpdatedAt time.Time `gorm:"column:updated_at;autoUpdateTime:milli"`
|
|
}
|
|
|
|
func (TradeOrder) TableName() string { return "trade_order" }
|
|
|
|
// CanTransitTo 校验状态流转是否合法
|
|
func (t TradeStatus) CanTransitTo(next TradeStatus) bool {
|
|
allowed := map[TradeStatus][]TradeStatus{
|
|
TradeStatusCreating: {TradeStatusPaying, TradeStatusCreateFailed},
|
|
TradeStatusPaying: {TradeStatusPaid, TradeStatusClosed, TradeStatusFailed},
|
|
TradeStatusPaid: {TradeStatusRefunded},
|
|
TradeStatusCreateFailed: {TradeStatusPaying},
|
|
}
|
|
for _, s := range allowed[t] {
|
|
if s == next {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|