115 lines
3.0 KiB
Go
115 lines
3.0 KiB
Go
package walletservicelogic
|
||
|
||
import (
|
||
"context"
|
||
"database/sql"
|
||
"errors"
|
||
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
||
"tianyuan-api/apps/sentinel/client/product"
|
||
"tianyuan-api/apps/user/internal/model"
|
||
"time"
|
||
|
||
"tianyuan-api/apps/user/internal/svc"
|
||
"tianyuan-api/apps/user/user"
|
||
|
||
"github.com/zeromicro/go-zero/core/logx"
|
||
)
|
||
|
||
type UpdateWalletLogic struct {
|
||
ctx context.Context
|
||
svcCtx *svc.ServiceContext
|
||
logx.Logger
|
||
}
|
||
|
||
func NewUpdateWalletLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateWalletLogic {
|
||
return &UpdateWalletLogic{
|
||
ctx: ctx,
|
||
svcCtx: svcCtx,
|
||
Logger: logx.WithContext(ctx),
|
||
}
|
||
}
|
||
|
||
const maxRetries = 5 // 最大重试次数
|
||
// 修改钱包余额
|
||
func (l *UpdateWalletLogic) UpdateWallet(in *user.UpdateWalletRequest) (*user.UpdateWalletResponse, error) {
|
||
if !in.Charge {
|
||
return nil, errors.New("不需要扣款")
|
||
}
|
||
// 获取产品信息
|
||
consumeProduct, getProductByCodeErr := l.svcCtx.ProductRpc.GetProductByCode(l.ctx, &product.GetRecordByCodeRequest{Code: in.ProductCode})
|
||
if getProductByCodeErr != nil {
|
||
return nil, getProductByCodeErr
|
||
}
|
||
|
||
// 检查是否已经扣款
|
||
deduction, FindOneByTransactionIdErr := l.svcCtx.DeductionsModel.FindOneByTransactionId(l.ctx, in.TransactionId)
|
||
if FindOneByTransactionIdErr != nil {
|
||
if FindOneByTransactionIdErr == sql.ErrNoRows {
|
||
} else {
|
||
// 其他错误，直接返回
|
||
return nil, FindOneByTransactionIdErr
|
||
}
|
||
} else if deduction != nil {
|
||
// 找到扣款记录，不能重复扣款
|
||
return nil, errors.New("该订单已扣款")
|
||
}
|
||
|
||
// 启动事务
|
||
err := l.svcCtx.WalletsModel.TransCtx(l.ctx, func(ctx context.Context, session sqlx.Session) error {
|
||
var err error
|
||
// 尝试多次更新余额（用于乐观锁）
|
||
for i := 0; i < maxRetries; i++ {
|
||
err = l.svcCtx.WalletsModel.UpdateBalance(session, l.ctx, in.UserId, -consumeProduct.ProductPrice)
|
||
if err == nil {
|
||
// 成功，退出循环
|
||
break
|
||
}
|
||
if errors.Is(err, model.ErrVersionMismatch) {
|
||
// 版本号不匹配，重试
|
||
time.Sleep(100 * time.Millisecond) // 重试前的延迟
|
||
continue
|
||
} else {
|
||
// 其他错误，直接返回
|
||
return err
|
||
}
|
||
}
|
||
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// 更新余额成功后，插入扣款记录
|
||
_, err = l.svcCtx.DeductionsModel.InsertDeductionsTrans(ctx, &model.Deductions{
|
||
TransactionId: in.TransactionId,
|
||
UserId: in.UserId,
|
||
Amount: consumeProduct.ProductPrice,
|
||
}, session)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
})
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
// 获取最新的用户余额
|
||
wallet, err := l.svcCtx.WalletsModel.FindOneByUserId(l.ctx, in.UserId)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
// 如果余额小于 0，更新 QuotaExceeded 字段为 1
|
||
if wallet.Balance < 0 {
|
||
users, findOneErr := l.svcCtx.UserModel.FindOne(l.ctx, in.UserId)
|
||
if findOneErr != nil {
|
||
return nil, findOneErr
|
||
}
|
||
users.QuotaExceeded = 1
|
||
updateErr := l.svcCtx.UserModel.Update(l.ctx, users)
|
||
if updateErr != nil {
|
||
return nil, updateErr
|
||
}
|
||
}
|
||
return &user.UpdateWalletResponse{}, nil
|
||
}
|