mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-18 01:06:39 +08:00
123 lines
3.0 KiB
Go
123 lines
3.0 KiB
Go
package admin
|
|
|
|
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
|
// * Copyright 2023 The Geek-AI Authors. All rights reserved.
|
|
// * Use of this source code is governed by a Apache-2.0 license
|
|
// * that can be found in the LICENSE file.
|
|
// * @Author yangjian102621@163.com
|
|
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
|
|
|
import (
|
|
"geekai/core"
|
|
"geekai/core/types"
|
|
"geekai/handler"
|
|
"geekai/store/model"
|
|
"geekai/store/vo"
|
|
"geekai/utils"
|
|
"geekai/utils/resp"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type OrderHandler struct {
|
|
handler.BaseHandler
|
|
}
|
|
|
|
func NewOrderHandler(app *core.AppServer, db *gorm.DB) *OrderHandler {
|
|
return &OrderHandler{BaseHandler: handler.BaseHandler{App: app, DB: db}}
|
|
}
|
|
|
|
func (h *OrderHandler) List(c *gin.Context) {
|
|
var data struct {
|
|
OrderNo string `json:"order_no"`
|
|
Status int `json:"status"`
|
|
PayTime []string `json:"pay_time"`
|
|
Page int `json:"page"`
|
|
PageSize int `json:"page_size"`
|
|
}
|
|
if err := c.ShouldBindJSON(&data); err != nil {
|
|
resp.ERROR(c, types.InvalidArgs)
|
|
return
|
|
}
|
|
|
|
session := h.DB.Session(&gorm.Session{})
|
|
if data.OrderNo != "" {
|
|
session = session.Where("order_no", data.OrderNo)
|
|
}
|
|
if len(data.PayTime) == 2 {
|
|
start := utils.Str2stamp(data.PayTime[0] + " 00:00:00")
|
|
end := utils.Str2stamp(data.PayTime[1] + " 00:00:00")
|
|
session = session.Where("pay_time >= ? AND pay_time <= ?", start, end)
|
|
}
|
|
if data.Status >= 0 {
|
|
session = session.Where("status", data.Status)
|
|
}
|
|
var total int64
|
|
session.Model(&model.Order{}).Count(&total)
|
|
var items []model.Order
|
|
var list = make([]vo.Order, 0)
|
|
offset := (data.Page - 1) * data.PageSize
|
|
res := session.Order("id DESC").Offset(offset).Limit(data.PageSize).Find(&items)
|
|
if res.Error == nil {
|
|
for _, item := range items {
|
|
var order vo.Order
|
|
err := utils.CopyObject(item, &order)
|
|
if err == nil {
|
|
order.Id = item.Id
|
|
order.CreatedAt = item.CreatedAt.Unix()
|
|
order.UpdatedAt = item.UpdatedAt.Unix()
|
|
payMethod, ok := types.PayMethods[item.PayWay]
|
|
if !ok {
|
|
payMethod = item.PayWay
|
|
}
|
|
payName, ok := types.PayNames[item.PayType]
|
|
if !ok {
|
|
payName = item.PayWay
|
|
}
|
|
order.PayMethod = payMethod
|
|
order.PayName = payName
|
|
list = append(list, order)
|
|
} else {
|
|
logger.Error(err)
|
|
}
|
|
}
|
|
}
|
|
resp.SUCCESS(c, vo.NewPage(total, data.Page, data.PageSize, list))
|
|
}
|
|
|
|
func (h *OrderHandler) Remove(c *gin.Context) {
|
|
id := h.GetInt(c, "id", 0)
|
|
|
|
if id > 0 {
|
|
var item model.Order
|
|
res := h.DB.First(&item, id)
|
|
if res.Error != nil {
|
|
resp.ERROR(c, "记录不存在！")
|
|
return
|
|
}
|
|
|
|
if item.Status == types.OrderPaidSuccess {
|
|
resp.ERROR(c, "已支付订单不允许删除！")
|
|
return
|
|
}
|
|
|
|
err := h.DB.Unscoped().Where("id = ?", id).Delete(&model.Order{}).Error
|
|
if err != nil {
|
|
resp.ERROR(c, err.Error())
|
|
return
|
|
}
|
|
}
|
|
resp.SUCCESS(c)
|
|
}
|
|
|
|
func (h *OrderHandler) Clear(c *gin.Context) {
|
|
|
|
err := h.DB.Unscoped().Where("status <> ?", 2).Where("pay_time", 0).Delete(&model.Order{}).Error
|
|
if err != nil {
|
|
resp.ERROR(c, err.Error())
|
|
return
|
|
}
|
|
resp.SUCCESS(c)
|
|
}
|