forked from MartialBE/one-hub
-
Notifications
You must be signed in to change notification settings - Fork 0
/
order.go
145 lines (120 loc) · 4.28 KB
/
order.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
package model
import (
"time"
"gorm.io/gorm"
)
type OrderStatus string
const (
OrderStatusPending OrderStatus = "pending"
OrderStatusSuccess OrderStatus = "success"
OrderStatusFailed OrderStatus = "failed"
OrderStatusClosed OrderStatus = "closed"
)
type Order struct {
ID int `json:"id"`
UserId int `json:"user_id"`
GatewayId int `json:"gateway_id"`
TradeNo string `json:"trade_no" gorm:"type:varchar(50);uniqueIndex"`
GatewayNo string `json:"gateway_no" gorm:"type:varchar(100)"`
Amount int `json:"amount" gorm:"default:0"`
OrderAmount float64 `json:"order_amount" gorm:"type:decimal(10,2);default:0"`
OrderCurrency CurrencyType `json:"order_currency" gorm:"type:varchar(16)"`
Quota int `json:"quota" gorm:"type:int;default:0"`
Fee float64 `json:"fee" gorm:"type:decimal(10,2);default:0"`
Discount float64 `json:"discount" gorm:"type:decimal(10,2);default:0"`
Status OrderStatus `json:"status" gorm:"type:varchar(32)"`
CreatedAt int `json:"created_at"`
UpdatedAt int `json:"-"`
DeletedAt gorm.DeletedAt `json:"-" gorm:"index"`
}
// 查询并关闭未完成的订单
func CloseUnfinishedOrder() error {
// 关闭超过 3 小时未支付的订单
unixTime := time.Now().Unix() - 3*3600
return DB.Model(&Order{}).Where("status = ? AND created_at < ?", OrderStatusPending, unixTime).Update("status", OrderStatusClosed).Error
}
func GetOrderByTradeNo(tradeNo string) (*Order, error) {
var order Order
err := DB.Where("trade_no = ?", tradeNo).First(&order).Error
return &order, err
}
func GetUserOrder(userId int, tradeNo string) (*Order, error) {
var order Order
err := DB.Where("user_id = ? AND trade_no = ?", userId, tradeNo).First(&order).Error
return &order, err
}
func (o *Order) Insert() error {
return DB.Create(o).Error
}
func (o *Order) Update() error {
return DB.Save(o).Error
}
var allowedOrderFields = map[string]bool{
"id": true,
"gateway_id": true,
"user_id": true,
"status": true,
"created_at": true,
}
type SearchOrderParams struct {
UserId int `form:"user_id"`
GatewayId int `form:"gateway_id"`
TradeNo string `form:"trade_no"`
GatewayNo string `form:"gateway_no"`
Status string `form:"status"`
StartTimestamp int64 `form:"start_timestamp"`
EndTimestamp int64 `form:"end_timestamp"`
PaginationParams
}
func GetOrderList(params *SearchOrderParams) (*DataResult[Order], error) {
var orders []*Order
db := DB.Omit("key")
if params.GatewayId != 0 {
db = db.Where("gateway_id = ?", params.GatewayId)
}
if params.UserId != 0 {
db = db.Where("user_id = ?", params.UserId)
}
if params.TradeNo != "" {
db = db.Where("trade_no = ?", params.TradeNo)
}
if params.GatewayNo != "" {
db = db.Where("gateway_no = ?", params.GatewayNo)
}
if params.Status != "" {
db = db.Where("status = ?", params.Status)
}
if params.StartTimestamp != 0 {
db = db.Where("created_at >= ?", params.StartTimestamp)
}
if params.EndTimestamp != 0 {
db = db.Where("created_at <= ?", params.EndTimestamp)
}
return PaginateAndOrder(db, ¶ms.PaginationParams, &orders, allowedOrderFields)
}
type OrderStatistics struct {
Quota int64 `json:"quota"`
Money float64 `json:"money"`
OrderCurrency string `json:"order_currency"`
}
func GetStatisticsOrder() (orderStatistics []*OrderStatistics, err error) {
err = DB.Model(&Order{}).Select("sum(quota) as quota, sum(order_amount) as money, order_currency").Where("status = ?", OrderStatusSuccess).Group("order_currency").Scan(&orderStatistics).Error
return orderStatistics, err
}
type OrderStatisticsGroup struct {
Date string `json:"date"`
OrderAmount float64 `json:"order_amount"`
}
func GetStatisticsOrderByPeriod(startTimestamp, endTimestamp int64) (orderStatistics []*OrderStatisticsGroup, err error) {
groupSelect := getTimestampGroupsSelect("created_at", "day", "date")
err = DB.Raw(`
SELECT `+groupSelect+`,
sum(order_amount) as order_amount
FROM orders
WHERE status= ?
AND created_at BETWEEN ? AND ?
GROUP BY date
ORDER BY date
`, OrderStatusSuccess, startTimestamp, endTimestamp).Scan(&orderStatistics).Error
return orderStatistics, err
}