92 lines
2.8 KiB
Go
92 lines
2.8 KiB
Go
|
// Package service 沉淀跨接口复用的业务逻辑：积分变动、站点配置读写
|
|||
|
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"errors"
|
|||
|
|
"strconv"
|
|||
|
|
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
"gorm.io/gorm/clause"
|
|||
|
|
|
|||
|
|
"nl-game-api-gin/internal/database"
|
|||
|
|
"nl-game-api-gin/internal/model"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// ChangePoints 变动用户积分并写流水（事务保证，正数=增加、负数=扣减）
|
|||
|
|
// tx 传 nil 时自动开启新事务；传入已有事务则复用（用于下单等复合操作）
|
|||
|
|
func ChangePoints(tx *gorm.DB, userID, change, ptype, relatedID int, remark string) (int, error) {
|
|||
|
|
run := func(db *gorm.DB) (int, error) {
|
|||
|
|
var user model.User
|
|||
|
|
// 行级锁读取（SELECT ... FOR UPDATE），防止并发变动基于同一余额快照造成丢失更新
|
|||
|
|
// 注意：GORM v2 必须用 clause.Locking，旧的 Set("gorm:query_option") 会被静默忽略
|
|||
|
|
if err := db.Clauses(clause.Locking{Strength: "UPDATE"}).First(&user, userID).Error; err != nil {
|
|||
|
|
return 0, errors.New("用户不存在")
|
|||
|
|
}
|
|||
|
|
newBalance := user.Points + change
|
|||
|
|
if newBalance < 0 {
|
|||
|
|
return 0, errors.New("积分不足")
|
|||
|
|
}
|
|||
|
|
// 更新余额；增加时同步累计 total_points（排行榜依据）
|
|||
|
|
updates := map[string]any{"points": newBalance}
|
|||
|
|
if change > 0 {
|
|||
|
|
updates["total_points"] = user.TotalPoints + change
|
|||
|
|
}
|
|||
|
|
if err := db.Model(&model.User{}).Where("id = ?", userID).Updates(updates).Error; err != nil {
|
|||
|
|
return 0, err
|
|||
|
|
}
|
|||
|
|
// 写积分流水（审计与个人中心展示）
|
|||
|
|
record := model.PointRecord{
|
|||
|
|
UserID: userID, ChangePoints: change, Balance: newBalance,
|
|||
|
|
Type: ptype, RelatedID: relatedID, Remark: remark,
|
|||
|
|
}
|
|||
|
|
if err := db.Create(&record).Error; err != nil {
|
|||
|
|
return 0, err
|
|||
|
|
}
|
|||
|
|
return newBalance, nil
|
|||
|
|
}
|
|||
|
|
// 复用外部事务或自行开启事务
|
|||
|
|
if tx != nil {
|
|||
|
|
return run(tx)
|
|||
|
|
}
|
|||
|
|
var balance int
|
|||
|
|
err := database.DB.Transaction(func(db *gorm.DB) error {
|
|||
|
|
b, err := run(db)
|
|||
|
|
balance = b
|
|||
|
|
return err
|
|||
|
|
})
|
|||
|
|
return balance, err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetConfig 读取站点配置，不存在时返回默认值
|
|||
|
|
func GetConfig(key, def string) string {
|
|||
|
|
var conf model.SiteConfig
|
|||
|
|
if err := database.DB.Where("config_key = ?", key).First(&conf).Error; err != nil {
|
|||
|
|
return def
|
|||
|
|
}
|
|||
|
|
return conf.ConfigValue
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetConfigInt 读取整数类型站点配置，解析失败返回默认值
|
|||
|
|
func GetConfigInt(key string, def int) int {
|
|||
|
|
v := GetConfig(key, "")
|
|||
|
|
if v == "" {
|
|||
|
|
return def
|
|||
|
|
}
|
|||
|
|
n, err := strconv.Atoi(v)
|
|||
|
|
if err != nil {
|
|||
|
|
return def
|
|||
|
|
}
|
|||
|
|
return n
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SetConfig 写入站点配置（不存在则创建）
|
|||
|
|
func SetConfig(key, value string) error {
|
|||
|
|
var conf model.SiteConfig
|
|||
|
|
err := database.DB.Where("config_key = ?", key).First(&conf).Error
|
|||
|
|
if err != nil {
|
|||
|
|
conf = model.SiteConfig{ConfigKey: key, ConfigValue: value}
|
|||
|
|
return database.DB.Create(&conf).Error
|
|||
|
|
}
|
|||
|
|
return database.DB.Model(&conf).Update("config_value", value).Error
|
|||
|
|
}
|