|
- package model
-
- import (
- "errors"
- "fmt"
-
- "github.com/QuantumNous/new-api/common"
- "github.com/QuantumNous/new-api/logger"
-
- "github.com/shopspring/decimal"
- "gorm.io/gorm"
- )
-
- // RechargeWechat 微信支付充值完成(由回调触发)
- // 与 Recharge/RechargeCreem 类似,使用事务+行锁保证幂等
- func RechargeWechat(tradeNo string) error {
- if tradeNo == "" {
- return errors.New("未提供支付单号")
- }
-
- var quotaToAdd int64
- var payMoney float64
- var userId int
-
- refCol := "`trade_no`"
- if common.UsingPostgreSQL {
- refCol = `"trade_no"`
- }
-
- err := DB.Transaction(func(tx *gorm.DB) error {
- topUp := &TopUp{}
- if err := tx.Set("gorm:query_option", "FOR UPDATE").Where(refCol+" = ?", tradeNo).First(topUp).Error; err != nil {
- return errors.New("充值订单不存在")
- }
-
- if topUp.Status == common.TopUpStatusSuccess {
- // 已处理,幂等返回
- return nil
- }
-
- if topUp.Status != common.TopUpStatusPending {
- return errors.New("充值订单状态错误")
- }
-
- topUp.CompleteTime = common.GetTimestamp()
- topUp.Status = common.TopUpStatusSuccess
- if err := tx.Save(topUp).Error; err != nil {
- return err
- }
-
- // 微信支付充值额度计算:
- // topUp.Money = req.Amount * topUpGroupRatio(经分组倍率调整后的数量)
- // 充值额度 = topUp.Money * QuotaPerUnit(与 Stripe 的 Recharge 逻辑一致)
- dMoney := decimal.NewFromFloat(topUp.Money)
- dQuotaPerUnit := decimal.NewFromFloat(common.QuotaPerUnit)
- quotaToAdd = dMoney.Mul(dQuotaPerUnit).IntPart()
-
- if quotaToAdd <= 0 {
- return errors.New("无效的充值额度")
- }
-
- if err := tx.Model(&User{}).Where("id = ?", topUp.UserId).Update("quota", gorm.Expr("quota + ?", quotaToAdd)).Error; err != nil {
- return err
- }
-
- userId = topUp.UserId
- payMoney = topUp.Money
- return nil
- })
-
- if err != nil {
- return err
- }
-
- if quotaToAdd > 0 {
- RecordLog(userId, LogTypeTopup, fmt.Sprintf("使用微信支付充值成功,充值金额: %v,支付金额:%.2f", logger.FormatQuota(int(quotaToAdd)), payMoney))
- }
-
- return nil
- }
|