You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
92 lines
2.6 KiB
Go
92 lines
2.6 KiB
Go
package model
|
|
|
|
import (
|
|
"context"
|
|
"git.noahlan.cn/northlan/ntools-go/gorm-zero/gormx"
|
|
"github.com/pkg/errors"
|
|
"gorm.io/gorm"
|
|
"gorm.io/plugin/optimisticlock"
|
|
"live-service/common/nerr"
|
|
)
|
|
|
|
var _ UserIntegralModel = (*customUserIntegralModel)(nil)
|
|
|
|
type (
|
|
// UserIntegralModel is an interface to be customized, add more methods here,
|
|
// and implement the added methods in customUserIntegralModel.
|
|
UserIntegralModel interface {
|
|
userIntegralModel
|
|
UpdateIntegral(ctx context.Context, tx *gorm.DB, integral *UserIntegral) error
|
|
// ChangeIntegral 用户积分变动
|
|
ChangeIntegral(ctx context.Context, tx *gorm.DB, userId int64, change int64) (int64, error)
|
|
}
|
|
|
|
customUserIntegralModel struct {
|
|
*defaultUserIntegralModel
|
|
}
|
|
)
|
|
|
|
// NewUserIntegralModel returns a model for the database table.
|
|
func NewUserIntegralModel(conn *gorm.DB) UserIntegralModel {
|
|
return &customUserIntegralModel{
|
|
defaultUserIntegralModel: newUserIntegralModel(conn),
|
|
}
|
|
}
|
|
|
|
func (m *customUserIntegralModel) UpdateIntegral(ctx context.Context, tx *gorm.DB, integral *UserIntegral) error {
|
|
if integral.Integral < 0 {
|
|
return errors.New("无法将积分更新至负数")
|
|
}
|
|
db := gormx.WithTx(ctx, m.DB, tx)
|
|
|
|
result := db.Model(&integral).Updates(&UserIntegral{Integral: integral.Integral, Version: optimisticlock.Version{Int64: 1}})
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
if result.RowsAffected == 0 {
|
|
return gormx.ErrRowsAffectedZero
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *customUserIntegralModel) ChangeIntegral(ctx context.Context, tx *gorm.DB, userId int64, change int64) (int64, error) {
|
|
resp := change
|
|
err := gormx.WithRetry(VersionRetryCount, func() error {
|
|
return m.Transact(tx, func(tx *gorm.DB) error {
|
|
data, err := m.FindOne(ctx, tx, userId)
|
|
if err != nil {
|
|
if errors.Is(err, ErrNotFound) {
|
|
if change < 0 {
|
|
return nerr.NewWithCode(nerr.UserIntegralNotEnoughError)
|
|
}
|
|
// 用户积分记录不存在,进行插入
|
|
if err = m.Insert(ctx, tx, &UserIntegral{
|
|
UserId: userId,
|
|
Integral: change,
|
|
}); err != nil {
|
|
return errors.Wrap(err, "插入用户积分失败")
|
|
}
|
|
return nil
|
|
} else {
|
|
return errors.Wrap(err, "获取当前用户积分失败")
|
|
}
|
|
}
|
|
if data.Integral+change < 0 {
|
|
return errors.New("用户积分不足")
|
|
}
|
|
data.Integral += change
|
|
if err = m.UpdateIntegral(ctx, tx, data); err != nil {
|
|
if errors.Is(err, gormx.ErrRowsAffectedZero) {
|
|
return err
|
|
}
|
|
return errors.Wrap(err, "更新用户积分失败")
|
|
}
|
|
resp = data.Integral
|
|
return nil
|
|
})
|
|
}, func(err error) bool {
|
|
return errors.Is(err, gormx.ErrRowsAffectedZero)
|
|
})
|
|
return resp, err
|
|
}
|