GithubBackup/s3-balance
1
0
Fork 0
mirror of https://github.com/DullJZ/s3-balance.git synced 2026-06-28 06:21:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: Use transaction for atomic increment and read operation

Browse Source
This commit is contained in:
DullJZ
2025-11-03 23:03:09 +08:00
parent 86d2bc1f3e
commit de019d56d8
1 changed files with 22 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
internal/storage/service.go
View File

@@ -304,18 +304,29 @@ func (s *Service) IncrementBucketOperation(bucketName, category string) (int64,
return 0, fmt.Errorf("unknown operation category: %s", category)
}
if err := s.db.Model(&BucketStats{}).
Where("bucket_name = ?", bucketName).
UpdateColumn(field, gorm.Expr(field+" + ?", 1)).Error; err != nil {
return 0, fmt.Errorf("failed to increment %s for bucket %s: %w", field, bucketName, err)
}
// 使用事务确保原子性
var count int64
if err := s.db.Model(&BucketStats{}).
Where("bucket_name = ?", bucketName).
Select(field).
Scan(&count).Error; err != nil {
return 0, fmt.Errorf("failed to fetch updated %s for bucket %s: %w", field, bucketName, err)
err := s.db.Transaction(func(tx *gorm.DB) error {
// 原子递增
if err := tx.Model(&BucketStats{}).
Where("bucket_name = ?", bucketName).
UpdateColumn(field, gorm.Expr(field+" + ?", 1)).Error; err != nil {
return fmt.Errorf("failed to increment %s for bucket %s: %w", field, bucketName, err)
}
// 在同一事务中读取最新值
if err := tx.Model(&BucketStats{}).
Where("bucket_name = ?", bucketName).
Select(field).
Scan(&count).Error; err != nil {
return fmt.Errorf("failed to fetch updated %s for bucket %s: %w", field, bucketName, err)
}
return nil
})
if err != nil {
return 0, err
}
return count, nil