Blizzard
119 lines
3.2 KiB
Go
119 lines
3.2 KiB
Go
package logic
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
plantModel "sundynix-micro-go/app/plant/model"
|
|
"sundynix-micro-go/app/plant/rpc/internal/svc"
|
|
"sundynix-micro-go/app/plant/rpc/plant"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type CreateExchangeOrderLogic struct {
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
logx.Logger
|
|
}
|
|
|
|
func NewCreateExchangeOrderLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateExchangeOrderLogic {
|
|
return &CreateExchangeOrderLogic{
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
Logger: logx.WithContext(ctx),
|
|
}
|
|
}
|
|
|
|
// 创建兑换订单（阳光值扣减+库存扣减事务）
|
|
func (l *CreateExchangeOrderLogic) CreateExchangeOrder(in *plant.CreateExchangeOrderReq) (*plant.CommonResp, error) {
|
|
txErr := l.svcCtx.DB.Transaction(func(tx *gorm.DB) error {
|
|
// 1. 查询并锁定商品
|
|
var item plantModel.ExchangeItem
|
|
if err := tx.Set("gorm:query_option", "FOR UPDATE").
|
|
Where("id = ? AND status = 1", in.ItemId).First(&item).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return errors.New("商品不存在或已下架")
|
|
}
|
|
return err
|
|
}
|
|
|
|
quantity := int(in.Quantity)
|
|
if quantity <= 0 {
|
|
quantity = 1
|
|
}
|
|
cost := item.CostSunlight
|
|
if cost == 0 {
|
|
cost = item.Cost
|
|
}
|
|
totalCost := cost * int64(quantity)
|
|
now := time.Now()
|
|
if item.StartTime != nil && now.Before(*item.StartTime) {
|
|
return errors.New("兑换尚未开始")
|
|
}
|
|
if item.EndTime != nil && now.After(*item.EndTime) {
|
|
return errors.New("兑换已结束")
|
|
}
|
|
if item.LimitPerUser > 0 {
|
|
var count int64
|
|
tx.Model(&plantModel.ExchangeOrder{}).
|
|
Where("user_id = ? AND item_id = ? AND status <> ?", in.UserId, in.ItemId, 5).
|
|
Count(&count)
|
|
if int(count)+quantity > item.LimitPerUser {
|
|
return errors.New("已达到兑换上限")
|
|
}
|
|
}
|
|
|
|
// 2. 检查库存（-1 表示无限库存）
|
|
if item.Stock >= 0 && item.Stock < quantity {
|
|
return errors.New("库存不足")
|
|
}
|
|
|
|
// 3. 扣减阳光值（原子操作，RowsAffected==0 说明阳光不足）
|
|
result := tx.Model(&plantModel.UserProfile{}).
|
|
Where("user_id = ? AND current_sunlight >= ?", in.UserId, totalCost).
|
|
Update("current_sunlight", gorm.Expr("current_sunlight - ?", totalCost))
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
if result.RowsAffected == 0 {
|
|
return errors.New("阳光值不足")
|
|
}
|
|
|
|
// 4. 扣减库存（-1无限库存时跳过）
|
|
if item.Stock >= 0 {
|
|
if err := tx.Model(&plantModel.ExchangeItem{}).
|
|
Where("id = ? AND stock >= ?", in.ItemId, quantity).
|
|
Update("stock", gorm.Expr("stock - ?", quantity)).Error; err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// 5. 创建订单
|
|
order := plantModel.ExchangeOrder{
|
|
UserID: in.UserId,
|
|
ItemID: in.ItemId,
|
|
ItemName: item.Name,
|
|
Cost: totalCost,
|
|
CostSunlight: totalCost,
|
|
Quantity: quantity,
|
|
ItemType: item.Type,
|
|
RecipientName: in.RecipientName,
|
|
Phone: in.Phone,
|
|
Address: in.Address,
|
|
Status: 1,
|
|
}
|
|
if item.Type == "VIRTUAL" {
|
|
order.Status = 4
|
|
order.CompletedAt = &now
|
|
}
|
|
return tx.Create(&order).Error
|
|
})
|
|
if txErr != nil {
|
|
return nil, txErr
|
|
}
|
|
return &plant.CommonResp{Code: 0, Msg: "ok"}, nil
|
|
}
|