aiforge/models/point_account.go

package models

import (
	"code.gitea.io/gitea/modules/timeutil"
)

type PointAccountStatus int

// Possible PointAccountStatus types.
const (
	PointAccountNormal  int = iota + 1 // 1
	PointAccountFreeze                 // 2
	PointAccountDeleted                // 3
)

type PointAccount struct {
	ID            int64              `xorm:"pk autoincr"`
	AccountCode   string             `xorm:"INDEX NOT NULL"`
	Balance       int64              `xorm:"NOT NULL DEFAULT 0"`
	TotalEarned   int64              `xorm:"NOT NULL DEFAULT 0"`
	TotalConsumed int64              `xorm:"NOT NULL DEFAULT 0"`
	UserId        int64              `xorm:"INDEX NOT NULL"`
	Status        int                `xorm:"NOT NULL"`
	Version       int64              `xorm:"NOT NULL"`
	CreatedUnix   timeutil.TimeStamp `xorm:"INDEX created"`
	UpdatedUnix   timeutil.TimeStamp `xorm:"INDEX updated"`
}

func (account *PointAccount) Increase(amount int64, sourceId string) error {
	sess := x.NewSession()
	defer sess.Close()
	sql := "update point_account set balance = balance + ?,total_earned = total_earned + ? ,version = version + 1 where account_code = ? "
	_, err := sess.Exec(sql, amount, amount, account.AccountCode)
	if err != nil {
		sess.Rollback()
		return err
	}
	accountLog := &PointAccountLog{
		AccountCode:    account.AccountCode,
		UserId:         account.UserId,
		Type:           IncreaseAccountBalance,
		SourceId:       sourceId,
		PointsAmount:   amount,
		BalanceBefore:  account.Balance,
		BalanceAfter:   account.Balance + amount,
		AccountVersion: account.Version,
	}
	_, err = sess.Insert(accountLog)
	if err != nil {
		sess.Rollback()
		return err
	}
	sess.Commit()
	return nil
}

func (account *PointAccount) Decrease(amount int64, sourceId string) error {
	sess := x.NewSession()
	defer sess.Close()
	sql := "update point_account set balance = balance - ?,total_consumed = total_consumed + ? ,version = version + 1 where account_code = ? "
	_, err := sess.Exec(sql, amount, amount, account.AccountCode)
	if err != nil {
		sess.Rollback()
		return err
	}
	accountLog := &PointAccountLog{
		AccountCode:    account.AccountCode,
		UserId:         account.UserId,
		Type:           DecreaseAccountBalance,
		SourceId:       sourceId,
		PointsAmount:   amount,
		BalanceBefore:  account.Balance,
		BalanceAfter:   account.Balance - amount,
		AccountVersion: account.Version,
	}
	_, err = sess.Insert(accountLog)
	if err != nil {
		sess.Rollback()
		return err
	}
	sess.Commit()
	return nil
}

func GetAccountByUserId(userId int64) (*PointAccount, error) {
	p := &PointAccount{}
	has, err := x.Where("user_id = ?", userId).Get(p)
	if err != nil {
		return nil, err
	}
	if !has {
		return nil, ErrRecordNotExist{}
	}
	return p, nil
}

func InsertAccount(tl *PointAccount) (int64, error) {
	return x.Insert(tl)
}