267 lines
8.3 KiB
Go
267 lines
8.3 KiB
Go
|
package model
|
||
|
|
||
|
import (
|
||
|
"errors"
|
||
|
"fmt"
|
||
|
"github.com/codinl/go-logger"
|
||
|
"math/rand"
|
||
|
"qiniupkg.com/x/log.v7"
|
||
|
"time"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
OrderCardStatusUnPick = 1 // 待取货中
|
||
|
OrderCardStatusPlaying = 2 // 游玩中
|
||
|
OrderCardStatusReturning = 3 // 归还中
|
||
|
OrderCardStatusCompleted = 4 // 已完成
|
||
|
|
||
|
DeliveryTypeStorePick = 1 // 门店取货
|
||
|
DeliveryTypeExpress = 2 // 快递取货
|
||
|
|
||
|
ExpressFeeProvince = 10 * 100 // 省内
|
||
|
ExpressFeeOutsideProvince = 15 * 100 // 省外
|
||
|
|
||
|
PayStatusUnPay = 1 // 未支付
|
||
|
PayStatusPaid = 2 // 已支付
|
||
|
)
|
||
|
|
||
|
//go:generate goqueryset -in order.go
|
||
|
// gen:qs
|
||
|
type Order struct {
|
||
|
Model
|
||
|
|
||
|
Uid uint64 `json:"uid" gorm:"index"`
|
||
|
GameCardId uint64 `json:"game_card_id" gorm:"index"` // 游戏id
|
||
|
GameCardGoodsId uint64 `json:"game_card_goods_id" gorm:"index"` // 游戏卡id
|
||
|
StoreId uint64 `json:"store_id"` // 门店id
|
||
|
UserAddressId uint64 `json:"user_address_id"` // 地址id
|
||
|
DeliveryType uint8 `json:"delivery_type"` // 取货类型 1-门店自取 2-快递
|
||
|
Count uint32 `json:"count"` // 数量
|
||
|
PickupCode string `json:"pickup_code"` // 取货码
|
||
|
CardStatus uint8 `json:"card_status"` // 1-待取货中 2-游玩中 3-归还中 4-已完成
|
||
|
DeliveryTime time.Time `json:"delivery_time"` // 发货时间
|
||
|
ReceiptTime time.Time `json:"receipt_time"` // 收货时间
|
||
|
PayTime time.Time `json:"pay_time"` // 支付时间
|
||
|
PayStatus uint8 `json:"pay_status"` // 支付状态 1-未支付 2-已支付
|
||
|
OrderSn string `json:"order_sn" gorm:"type:varchar(255);not null;COMMENT:'订单编号'"` // 订单编号
|
||
|
PayPrice uint32 `json:"pay_price"` // 实际付款金额(包含运费)
|
||
|
Postage uint32 `json:"postage"` // 邮费
|
||
|
ExpressCompany string `json:"express_company"` // 物流公司
|
||
|
ExpressCompanyNo string `json:"express_company_no"` // 物流公司编号
|
||
|
ExpressNo string `json:"express_no"` // 物流单号
|
||
|
ExpressDuration uint64 `json:"express_duration"` // 物流时长
|
||
|
RevertStoreId uint64 `json:"revert_store_id"` // 归还门店id
|
||
|
RevertTime time.Time `json:"revert_time"` // 归还时间
|
||
|
RevertExpressCompany string `json:"revert_express_company"` // 归还物流公司
|
||
|
RevertExpressCompanyNo string `json:"revert_express_company_no"` // 归还物流公司编号
|
||
|
RevertExpressNo string `json:"revert_express_no"` // 归还物流单号
|
||
|
RevertExpressDuration uint64 `json:"revert_express_duration"` // 归还物流时长
|
||
|
}
|
||
|
|
||
|
func (*Order) TableName() string {
|
||
|
return "order"
|
||
|
}
|
||
|
|
||
|
func (m *Order) ToInfo(card *GameCard) OrderInfo {
|
||
|
return OrderInfo{
|
||
|
Order: *m,
|
||
|
GameName: card.Name,
|
||
|
GamePrice: card.Price,
|
||
|
GameCoverImg: card.CoverImg,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
type OrderInfo struct {
|
||
|
Order
|
||
|
GameName string `json:"name" gorm:"index"` // 名称
|
||
|
GamePrice uint32 `json:"price" gorm:"index"` // 价格
|
||
|
GameCoverImg string `json:"cover_img"` // 封面
|
||
|
}
|
||
|
|
||
|
func GetOrderList(uid uint64, page, pageSize int) ([]OrderInfo, uint32, error) {
|
||
|
var (
|
||
|
orders []Order
|
||
|
totalPage uint32
|
||
|
orderInfos = make([]OrderInfo, 0, pageSize)
|
||
|
)
|
||
|
page -= 1
|
||
|
if page < 0 {
|
||
|
page = 0
|
||
|
}
|
||
|
|
||
|
oq := NewOrderQuerySet(DB).UidEq(uid).OrderDescByCreatedAt().PayStatusEq(PayStatusPaid)
|
||
|
count, err := oq.Count()
|
||
|
if err != nil {
|
||
|
log.Error("NewGameCardQuerySet err:", err)
|
||
|
return orderInfos, 0, err
|
||
|
}
|
||
|
err = oq.Offset(page * pageSize).Limit(pageSize).All(&orders)
|
||
|
if err != nil && err != RecordNotFound {
|
||
|
log.Error("err:", err)
|
||
|
return orderInfos, 0, err
|
||
|
}
|
||
|
totalPage = uint32(count/pageSize + 1)
|
||
|
|
||
|
var games []GameCard
|
||
|
gameIds := make([]uint32, 0, len(orders))
|
||
|
for i, _ := range orders {
|
||
|
gameIds = append(gameIds, uint32(orders[i].GameCardId))
|
||
|
}
|
||
|
if len(gameIds) == 0 {
|
||
|
log.Error("gameIds is null")
|
||
|
return orderInfos, 0, err
|
||
|
}
|
||
|
err = NewGameCardQuerySet(DB).IDIn(gameIds...).All(&games)
|
||
|
if err != nil && err != RecordNotFound {
|
||
|
log.Error("err:", err)
|
||
|
return orderInfos, 0, err
|
||
|
}
|
||
|
|
||
|
gamesMap := make(map[uint32]*GameCard)
|
||
|
for i, _ := range games {
|
||
|
gamesMap[games[i].ID] = &games[i]
|
||
|
}
|
||
|
for i, _ := range orders {
|
||
|
card, ok := gamesMap[uint32(orders[i].GameCardId)]
|
||
|
if !ok {
|
||
|
continue
|
||
|
}
|
||
|
orderInfos = append(orderInfos, orders[i].ToInfo(card))
|
||
|
}
|
||
|
|
||
|
return orderInfos, totalPage, nil
|
||
|
}
|
||
|
|
||
|
func (m *Order) Info() (*OrderInfo, error) {
|
||
|
var (
|
||
|
order Order
|
||
|
card GameCard
|
||
|
)
|
||
|
err := NewOrderQuerySet(DB).IDEq(m.ID).One(&order)
|
||
|
if err != nil {
|
||
|
log.Error("err:", err)
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
err = NewGameCardQuerySet(DB).IDEq(uint32(order.GameCardId)).One(&card)
|
||
|
if err != nil && err != RecordNotFound {
|
||
|
log.Error("err:", err)
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
info := order.ToInfo(&card)
|
||
|
return &info, nil
|
||
|
}
|
||
|
|
||
|
func (m *Order) Revert() (*OrderInfo, error) {
|
||
|
var (
|
||
|
order Order
|
||
|
card GameCard
|
||
|
)
|
||
|
err := NewOrderQuerySet(DB).IDEq(m.ID).GetUpdater().
|
||
|
SetCardStatus(OrderCardStatusReturning).
|
||
|
SetRevertTime(time.Now()).
|
||
|
SetRevertStoreId(m.RevertStoreId).
|
||
|
SetRevertExpressCompany(m.RevertExpressCompany).
|
||
|
SetRevertExpressCompanyNo(m.RevertExpressCompanyNo).
|
||
|
SetRevertExpressNo(m.RevertExpressNo).Update()
|
||
|
if err != nil {
|
||
|
log.Error("err:", err)
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
//err = NewGameCardQuerySet(DB).IDEq(uint32(order.GameCardId)).One(&card)
|
||
|
//if err != nil && err != RecordNotFound {
|
||
|
// log.Error("err:", err)
|
||
|
// return nil, err
|
||
|
//}
|
||
|
info := order.ToInfo(&card)
|
||
|
return &info, nil
|
||
|
}
|
||
|
|
||
|
func (m *Order) OrderCreate() error {
|
||
|
m.PayStatus = PayStatusUnPay
|
||
|
err := m.Create(DB)
|
||
|
if err != nil {
|
||
|
log.Error("err:", err)
|
||
|
return err
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (m *Order) Modify() (*OrderInfo, error) {
|
||
|
if m.ID == 0 {
|
||
|
return nil, errors.New("orderId is null")
|
||
|
}
|
||
|
fields := make([]OrderDBSchemaField, 0)
|
||
|
if m.CardStatus != 0 {
|
||
|
fields = append(fields, OrderDBSchema.CardStatus)
|
||
|
}
|
||
|
|
||
|
if len(fields) > 0 {
|
||
|
err := m.Update(DB.Where(&Order{Model: Model{ID: m.ID}, Uid: m.Uid}), fields...)
|
||
|
if err != nil {
|
||
|
logger.Error("err:", err)
|
||
|
return nil, err
|
||
|
}
|
||
|
}
|
||
|
|
||
|
var card GameCard
|
||
|
err := NewGameCardQuerySet(DB).IDEq(uint32(m.GameCardId)).One(&card)
|
||
|
if err != nil && err != RecordNotFound {
|
||
|
log.Error("err:", err)
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
info := m.ToInfo(&card)
|
||
|
|
||
|
return &info, nil
|
||
|
}
|
||
|
|
||
|
func IsHaveUnreturnedOrders(uid uint32) (bool, error) {
|
||
|
count, err := NewOrderQuerySet(DB).UidEq(uint64(uid)).CardStatusNe(OrderCardStatusCompleted).Count()
|
||
|
if err != nil {
|
||
|
logger.Error("err:", err)
|
||
|
return false, err
|
||
|
}
|
||
|
return count != 0, nil
|
||
|
}
|
||
|
|
||
|
// gen:qs
|
||
|
type ExpressCompany struct {
|
||
|
Model
|
||
|
|
||
|
CompanyName string `json:"company_name"`
|
||
|
CompanyCode string `json:"company_code"`
|
||
|
}
|
||
|
|
||
|
func (*ExpressCompany) TableName() string {
|
||
|
return "express_company"
|
||
|
}
|
||
|
|
||
|
func (*ExpressCompany) List() ([]ExpressCompany, error) {
|
||
|
companys := make([]ExpressCompany, 0)
|
||
|
err := NewExpressCompanyQuerySet(DB).OrderAscByID().All(&companys)
|
||
|
if err != nil && err != RecordNotFound {
|
||
|
logger.Error("err:", err)
|
||
|
return companys, err
|
||
|
}
|
||
|
return companys, nil
|
||
|
}
|
||
|
|
||
|
func GetPickupCode() string {
|
||
|
for {
|
||
|
rand.Seed(time.Now().UnixNano())
|
||
|
code := fmt.Sprintf("%d", rand.Intn(1000000)+100000)
|
||
|
count, err := NewOrderQuerySet(DB).PickupCodeEq(code).Count()
|
||
|
if err != nil {
|
||
|
logger.Error("err:", err)
|
||
|
}
|
||
|
if count == 0 {
|
||
|
return code
|
||
|
}
|
||
|
}
|
||
|
//将时间戳设置成种子数
|
||
|
|
||
|
}
|