99 lines
2.1 KiB
Go
99 lines
2.1 KiB
Go
package model
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/codinl/go-logger"
|
|
)
|
|
|
|
const (
|
|
//VmEventExchangeGoods = "exchange_goods"
|
|
VmEventBuyGoods = "buy_goods"
|
|
VmEventOpenMember = "open_member"
|
|
VmEventInvite1Member = "invite_1_member"
|
|
VmEventInvite2Member = "invite_2_member"
|
|
VmEventAttendance = "attendance"
|
|
)
|
|
|
|
// 用户积分
|
|
//go:generate goqueryset -in user_vm.go
|
|
// gen:qs
|
|
type UserVm struct {
|
|
Model
|
|
|
|
Uid uint32 `json:"uid" gorm:"column:uid;unique_index"`
|
|
Vm uint32 `json:"vm"`
|
|
}
|
|
|
|
// 用户积分-变动记录
|
|
// gen:qs
|
|
type UserVmRecord struct {
|
|
Model
|
|
|
|
Uid uint32 `json:"uid" gorm:"column:uid;index"`
|
|
BeforeVm uint32 `json:"before_vm"` // 变动前
|
|
AfterVm uint32 `json:"after_vm"` // 变动后
|
|
Alter int `json:"alter"` // 数值
|
|
Event string `json:"event" gorm:"type:varchar(100)"` // 事件
|
|
Describe string `json:"describe" gorm:"type:text"` // 描述
|
|
|
|
}
|
|
|
|
func UserVmUpdate(uid uint32, amount int, event, describe string) error {
|
|
var userVm UserVm
|
|
err := NewUserVmQuerySet(DB).UidEq(uid).One(&userVm)
|
|
if err != nil && err != RecordNotFound {
|
|
logger.Error("err:", err)
|
|
return err
|
|
}
|
|
begin := DB.Begin()
|
|
if err == RecordNotFound {
|
|
if amount < 0 {
|
|
begin.Rollback()
|
|
logger.Error("amount lt 0")
|
|
return errors.New("amount lt 0")
|
|
}
|
|
userVm = UserVm{
|
|
Uid: uid,
|
|
Vm: uint32(amount),
|
|
}
|
|
err = begin.Create(&userVm).Error
|
|
if err != nil {
|
|
begin.Rollback()
|
|
logger.Error("err:", err)
|
|
return err
|
|
}
|
|
} else {
|
|
sql := fmt.Sprintf("UPDATE user_vm SET vm = vm+? WHERE uid=?;")
|
|
err = begin.Exec(sql, amount, uid).Error
|
|
if err != nil {
|
|
begin.Rollback()
|
|
logger.Error("err:", err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
vmRecord := &UserVmRecord{
|
|
Uid: uid,
|
|
BeforeVm: uint32(userVm.Vm),
|
|
AfterVm: uint32(int(userVm.Vm) + amount),
|
|
Alter: amount,
|
|
Event: event,
|
|
Describe: describe,
|
|
}
|
|
err = begin.Create(vmRecord).Error
|
|
if err != nil {
|
|
begin.Rollback()
|
|
logger.Error("err:", err)
|
|
return err
|
|
}
|
|
|
|
err = begin.Commit().Error
|
|
if err != nil {
|
|
begin.Rollback()
|
|
logger.Error("err:", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|