live-service/app/user_center/model/user_integral_model.go

package model

import (
	"context"
	"git.noahlan.cn/northlan/ntools-go/gorm-zero/gormc"
	"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
		Transact(ctx context.Context, tx *gorm.DB, fn func(tx *gorm.DB) error) error
		InsertTx(ctx context.Context, tx *gorm.DB, data *UserIntegral) error
		FindOneTx(ctx context.Context, tx *gorm.DB, userId int64) (*UserIntegral, error)
		UpdateTx(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) Transact(ctx context.Context, tx *gorm.DB, fn func(tx *gorm.DB) error) error {
	return withTx(ctx, m.conn, tx).Transaction(fn)
}

func (m *customUserIntegralModel) InsertTx(ctx context.Context, tx *gorm.DB, data *UserIntegral) error {
	err := withTx(ctx, m.conn, tx).Create(&data).Error
	return err
}

func (m *customUserIntegralModel) UpdateTx(ctx context.Context, tx *gorm.DB, integral *UserIntegral) error {
	if integral.Integral < 0 {
		return errors.New("无法将积分更新至负数")
	}
	db := withTx(ctx, m.conn, 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 ErrRowsAffectedZero
	}
	return nil
}

func (m *customUserIntegralModel) FindOneTx(ctx context.Context, tx *gorm.DB, userId int64) (*UserIntegral, error) {
	var resp UserIntegral
	err := withTx(ctx, m.conn, tx).Model(&UserIntegral{}).
		Where("`user_id` = ?", userId).Take(&resp).Error
	switch err {
	case nil:
		return &resp, nil
	case gormc.ErrNotFound:
		return nil, ErrNotFound
	default:
		return nil, err
	}
}

func (m *customUserIntegralModel) ChangeIntegral(ctx context.Context, tx *gorm.DB, userId int64, change int64) (int64, error) {
	resp := change
	var err error
	for i := VersionRetryCount; i > 0; i-- {
		err = withTx(ctx, m.conn, tx).Transaction(func(tx *gorm.DB) error {
			data, err := m.FindOneTx(ctx, tx, userId)
			if err != nil {
				if errors.Is(err, ErrNotFound) {
					if change < 0 {
						return nerr.NewWithCode(nerr.UserIntegralNotEnoughError)
					}
					// 用户积分记录不存在,进行插入
					if err = m.InsertTx(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.UpdateTx(ctx, tx, data); err != nil {
				if errors.Is(err, ErrRowsAffectedZero) {
					return err
				}
				return errors.Wrap(err, "更新用户积分失败")
			}
			resp = data.Integral
			return nil
		})
		if err != nil && errors.Is(err, ErrRowsAffectedZero) {
			// 未能正确更新,直接重试
			continue
		} else {
			// 其它错误退出循环
			break
		}
	}

	return resp, err
}