eta_mini_crm_ht/models/trade_order.go

package models

import (
	"github.com/beego/beego/v2/client/orm"
	"time"
)

type PaymentType string
type PaymentStatus string

const (
	PaymentTypePay          PaymentType   = "pay"
	PaymentTypeRefund       PaymentType   = "refund"
	PaymentStatusDone       PaymentStatus = "done"
	PaymentStatusFailed     PaymentStatus = "failed"
	PaymentStatusProcessing PaymentStatus = "processing"
)

type TradeOrderView struct {
	RealName       string
	Mobile         string
	ProductName    string
	TransactionID  string `gorm:"column:transaction_id;size:255;default:null;comment:'第三方平台ID'" json:"transaction_id"`
	ProductOrderID string `gorm:"column:product_order_id;size:255;default:null;comment:'商品订单号'" json:"product_order_id"`
	PaymentAccount string `gorm:"column:payment_account;size:255;default:null;comment:'支付账号'" json:"payment_account"`
	PaymentWay     string `gorm:"column:payment_way;type:enum('wechat');default:null;comment:'支付渠道'" json:"payment_way"`
	Amount         string `gorm:"column:amount;size:20;default:null;comment:'支付金额'" json:"amount"`
	Currency       string `gorm:"column:currency;size:255;default:null;comment:'货币'" json:"currency"`
	MerchantID     string `gorm:"column:merchant_id;size:255;default:null;comment:'商户id'" json:"merchant_id"`
	UserID         int    `gorm:"column:user_id;default:null;comment:'用户id'" json:"user_id"`
	TemplateUserID int    `gorm:"column:template_user_id;default:null;comment:'临时用户id'" json:"template_user_id"`
	PaymentType    string `gorm:"column:payment_type;type:enum('pay','refund');default:null;comment:'订单类型'" json:"payment_type"`
	PaymentStatus  string `gorm:"column:payment_status;type:enum('pending','done','failed');default:null;comment:'支付状态'" json:"payment_status"`
	DealTime       string
	CreatedTime    string
}
type TradeOrder struct {
	Id               uint          `gorm:"primaryKey;autoIncrement" json:"id"`
	TransactionId    string        `gorm:"column:transaction_id;size:255;default:null;comment:'第三方平台ID'" json:"transaction_id"`
	OrgTransactionId string        `gorm:"column:org_transaction_id;size:255;default:null;comment:'原支付第三方平台ID'" json:"org_transaction_id"`
	ProductOrderId   string        `gorm:"column:product_order_id;size:255;default:null;comment:'商品订单号'" json:"product_order_id"`
	PaymentAccount   string        `gorm:"column:payment_account;size:255;default:null;comment:'支付账号'" json:"payment_account"`
	PaymentWay       PaymentWay    `gorm:"column:payment_way;type:enum('wechat');default:null;comment:'支付渠道'" json:"payment_way"`
	Amount           string        `gorm:"column:amount;size:20;default:null;comment:'支付金额'" json:"amount"`
	Currency         string        `gorm:"column:currency;size:255;default:null;comment:'货币'" json:"currency"`
	MerchantId       string        `gorm:"column:merchant_id;size:255;default:null;comment:'商户id'" json:"merchant_id"`
	UserId           int           `gorm:"column:user_id;default:null;comment:'用户id'" json:"user_id"`
	TemplateUserId   int           `gorm:"column:template_user_id;default:null;comment:'临时用户id'" json:"template_user_id"`
	PaymentType      PaymentType   `gorm:"column:payment_type;type:enum('pay','refund');default:null;comment:'订单类型'" json:"payment_type"`
	PaymentStatus    PaymentStatus `gorm:"column:payment_status;type:enum('pending','done','failed');default:null;comment:'支付状态'" json:"payment_status"`
	DealTime         time.Time     `gorm:"column:deal_time;default:null;comment:'完成时间'" json:"deal_time"`
	CreatedTime      time.Time     `gorm:"column:created_time;default:null" json:"created_time"`
	UpdatedTime      time.Time     `gorm:"column:updated_time;default:null" json:"updated_time"`
}

func (TradeOrder) TableName() string {
	return "trade_orders"
}

func GetTradeOrderByCondition(condition string, sortCondition string, startSize int, pageSize int) (list []*TradeOrder, err error) {
	o := orm.NewOrm()
	sql := `select * from trade_orders where 1=1 `
	if condition != "" {
		sql += condition
	}
	if sortCondition != "" {
		sql += sortCondition
	}
	sql += ` LIMIT ?,?`
	_, err = o.Raw(sql, startSize, pageSize).QueryRows(&list)
	return
}

func GetTradeOrderListByCondition(condition string, sortCondition string) (list []*TradeOrder, err error) {
	o := orm.NewOrm()
	sql := `select * from trade_orders where 1=1 `
	if condition != "" {
		sql += condition
	}
	if sortCondition != "" {
		sql += sortCondition
	}
	_, err = o.Raw(sql).QueryRows(&list)
	return
}
func GetTradeOrderCountByCondition(condition string) (total int, err error) {
	o := orm.NewOrm()
	err = o.Raw("select count(*) from trade_orders where 1=1 " + condition).QueryRow(&total)
	return
}

func GetTradeOrderByNo(tradeNo string) (tradeOrder TradeOrder, err error) {
	o := orm.NewOrm()
	err = o.Raw("select * from trade_orders where transaction_id=? ", tradeNo).QueryRow(&tradeOrder)
	return
}

func (to *TradeOrder) Refund(productOrder *ProductOrder) (err error) {
	o := orm.NewOrm()
	tx, _ := o.Begin()
	defer func() {
		if err != nil {
			tx.Rollback()
		} else {
			tx.Commit()
		}
	}()
	_, err = tx.Insert(to)
	_, err = tx.Update(productOrder, "status", "refund_status", "refund_trade_id", "remark", "refund_amount")
	return
}