liucaijun
/
voucher
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

还原代码

This commit is contained in:
李子铭 2025-03-21 14:35:59 +08:00
parent 9d0e5e6f76
commit 1a8343cda8
2 changed files with 31 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
internal/biz/cmb.go
View File

@ -8,16 +8,19 @@ import (
v1 "voucher/api/v1" v1 "voucher/api/v1"
"voucher/internal/biz/bo" "voucher/internal/biz/bo"
"voucher/internal/biz/vo" "voucher/internal/biz/vo"
"voucher/internal/pkg/helper"
"voucher/internal/pkg/lock" "voucher/internal/pkg/lock"
) )
func (v *VoucherBiz) CmbOrder(ctx context.Context, req *bo.OrderCreateReqBo) (orderNo string, err error) { func (v *VoucherBiz) CmbOrder(ctx context.Context, req *bo.OrderCreateReqBo) (orderNo string, err error) {
bizCtx := helper.CopyValueCtx(ctx)
c := vo.CmbOrderLockKey.BuildCache([]string{req.OutBizNo, req.Type.String()}) c := vo.CmbOrderLockKey.BuildCache([]string{req.OutBizNo, req.Type.String()})
err = lock.NewMutex(v.rdb.Rdb, c.TTL).Lock(ctx, c.Key, func(ctx context.Context) error { err = lock.NewMutex(v.rdb.Rdb, c.TTL).Lock(bizCtx, c.Key, func(bizCtx context.Context) error {
order, err := v.OrderRepo.GetByOutBizNo(ctx, vo.OrderTypeCmb, req.OutBizNo) order, err := v.OrderRepo.GetByOutBizNo(bizCtx, vo.OrderTypeCmb, req.OutBizNo)
if err != nil && !err2.IsDbNotFound(err) { if err != nil && !err2.IsDbNotFound(err) {
return err return err
@ -27,7 +30,7 @@ func (v *VoucherBiz) CmbOrder(ctx context.Context, req *bo.OrderCreateReqBo) (or
if order.Status.IsFail() { if order.Status.IsFail() {
if err = v.orderRetry(ctx, order); err != nil { if err = v.orderRetry(bizCtx, order); err != nil {
return err return err
} }
} }
@ -36,7 +39,7 @@ func (v *VoucherBiz) CmbOrder(ctx context.Context, req *bo.OrderCreateReqBo) (or
return nil return nil
} }
product, err := v.ProductRepo.GetByProductNo(ctx, req.ProductNo) product, err := v.ProductRepo.GetByProductNo(bizCtx, req.ProductNo)
if err != nil { if err != nil {
return err return err
} }
@ -45,7 +48,7 @@ func (v *VoucherBiz) CmbOrder(ctx context.Context, req *bo.OrderCreateReqBo) (or
return err2.ErrorCmbProductNotSupported("只支持微信") return err2.ErrorCmbProductNotSupported("只支持微信")
} }
order, err = v.order(ctx, req, product) order, err = v.order(bizCtx, req, product)
if err != nil { if err != nil {
return err return err
} }

22
internal/biz/order.go
View File

@ -18,11 +18,10 @@ func (v *VoucherBiz) order(ctx context.Context, req *bo.OrderCreateReqBo, produc
return nil, err return nil, err
} }
voucherNo := ""
if product.ProductNo == "001" { if product.ProductNo == "001" {
// 压测商品 // 压测商品-直接返回
voucherNo = order.OrderNo return order, nil
} else { }
// 注册通知标签 order.MerchantNo 批次创建商户, order.BatchNo 商品批次号 // 注册通知标签 order.MerchantNo 批次创建商户, order.BatchNo 商品批次号
if err = v.registerNotifyTag(ctx, order.MerchantNo, order.BatchNo); err != nil { if err = v.registerNotifyTag(ctx, order.MerchantNo, order.BatchNo); err != nil {
@ -30,7 +29,7 @@ func (v *VoucherBiz) order(ctx context.Context, req *bo.OrderCreateReqBo, produc
} }
// 真实发放 // 真实发放
voucherNo, err = v.WechatCpnRepo.Order(ctx, order) voucherNo, err := v.WechatCpnRepo.Order(ctx, order)
if err != nil { if err != nil {
if err3 := v.fail(ctx, order, err.Error()); err3 != nil { if err3 := v.fail(ctx, order, err.Error()); err3 != nil {
return nil, err3 return nil, err3
@ -38,8 +37,6 @@ func (v *VoucherBiz) order(ctx context.Context, req *bo.OrderCreateReqBo, produc
return nil, err return nil, err
} }
}
if err = v.success(ctx, order, voucherNo); err != nil { if err = v.success(ctx, order, voucherNo); err != nil {
return nil, err return nil, err
} }
@ -63,7 +60,7 @@ func (v *VoucherBiz) orderRetry(ctx context.Context, order *bo.OrderBo) error {
func (v *VoucherBiz) create(ctx context.Context, req *bo.OrderCreateReqBo, product *bo.ProductBo) (*bo.OrderBo, error) { func (v *VoucherBiz) create(ctx context.Context, req *bo.OrderCreateReqBo, product *bo.ProductBo) (*bo.OrderBo, error) {
return v.OrderRepo.Create(ctx, &bo.OrderBo{ o := &bo.OrderBo{
OrderNo: v.GenerateMixRepo.GeneratorString(ctx, fmt.Sprintf("%d%s", req.Type, req.OutBizNo)), OrderNo: v.GenerateMixRepo.GeneratorString(ctx, fmt.Sprintf("%d%s", req.Type, req.OutBizNo)),
OutBizNo: req.OutBizNo, OutBizNo: req.OutBizNo,
ProductNo: req.ProductNo, ProductNo: req.ProductNo,
@ -77,7 +74,14 @@ func (v *VoucherBiz) create(ctx context.Context, req *bo.OrderCreateReqBo, produ
Type: req.Type, Type: req.Type,
Status: vo.OrderStatusIng, // 同步发放，状态至为发放中 Status: vo.OrderStatusIng, // 同步发放，状态至为发放中
Attach: req.Attach, Attach: req.Attach,
}) }
if product.ProductNo == "001" {
// 压测商品
o.Status = vo.OrderStatusSuccess
}
return v.OrderRepo.Create(ctx, o)
} }
func (v *VoucherBiz) registerNotifyTag(ctx context.Context, stockCreatorMchID, stockID string) error { func (v *VoucherBiz) registerNotifyTag(ctx context.Context, stockCreatorMchID, stockID string) error {