修复余额与套餐用量并发更新可能导致数据错误的问题

web/services/channel.go

@@ -2,6 +2,7 @@ package services
 
 import (
 	"context"
+	"database/sql"
 	"encoding/json"
 	"errors"
 	"fmt"
@@ -244,6 +245,8 @@ func (s *channelService) CreateChannel(
 	count int,
 	nodeFilter ...NodeFilterConfig,
 ) ([]*PortInfo, error) {
+	var now = time.Now()
+
 	var step = time.Now()
 	var rid = ctx.Value(requestid.ConfigDefault.ContextKey).(string)
 
@@ -298,7 +301,6 @@ func (s *channelService) CreateChannel(
 		// 分配端口
 		step = time.Now()
 
-		now := time.Now()
 		expiration := common.LocalDateTime(now.Add(time.Duration(resource.Live) * time.Second))
 		_addr, channels, err := assignPort(q, edgeAssigns, auth.Payload.Id, protocol, authType, expiration, filter)
 		if err != nil {
@@ -315,7 +317,7 @@ func (s *channelService) CreateChannel(
 			Used:      resource.Used + int32(count),
 			DailyLast: common.LocalDateTime(now),
 		}
-		last := time.Time(resource.DailyLast)
+		var last = time.Time(resource.DailyLast)
 		if now.Year() != last.Year() || now.Month() != last.Month() || now.Day() != last.Day() {
 			toUpdate.DailyUsed = int32(count)
 		} else {
@@ -346,7 +348,7 @@ func (s *channelService) CreateChannel(
 		slog.Debug("缓存通道数据", "rid", rid, "step", time.Since(step))
 
 		return nil
-	})
+	}, &sql.TxOptions{Isolation: sql.LevelRepeatableRead})
 	if err != nil {
 		return nil, err
 	}
@@ -674,7 +676,7 @@ func assignPort(
 				q.Channel.NodeHost,
 				q.Channel.DeletedAt,
 			).
-			Save(channels...)
+			Create(channels...)
 		if err != nil {
 			return nil, nil, err
 		}

web/services/resource.go

@@ -2,6 +2,7 @@ package services
 
 import (
 	"context"
+	"database/sql"
 	"encoding/json"
 	"fmt"
 	"platform/pkg/rds"
@@ -162,13 +163,13 @@ func (s *resourceService) CreateResource(data *CreateResourceData, uid int32) er
 		// 更新用户余额
 		_, err = q.User.
 			Where(q.User.ID.Eq(uid)).
-			Update(q.User.Balance, user.Balance-amount)
+			UpdateSimple(q.User.Balance.Sub(amount))
 		if err != nil {
 			return err
 		}
 
 		return nil
-	})
+	}, &sql.TxOptions{Isolation: sql.LevelRepeatableRead})
 	if err != nil {
 		return err
 	}

web/services/user.go

@@ -31,7 +31,7 @@ func (s *userService) RechargeConfirm(ctx context.Context, tradeNo string, verif
 
 		_, err = tx.User.
 			Where(tx.User.ID.Eq(user.ID)).
-			Update(tx.User.Balance, user.Balance+result.Trade.Amount)
+			UpdateSimple(tx.User.Balance.Add(result.Trade.Amount))
 		if err != nil {
 			return err
 		}