aiforge/models/reward_operate_record.go

package models

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

const (
	SourceTypeAccomplishTask    string = "ACCOMPLISH_TASK"
	SourceTypeAdminOperate             = "ADMIN_OPERATE"
	SourceTypeRunCloudbrainTask        = "RUN_CLOUBRAIN_TASK"
)

type RewardType string

const (
	RewardTypePoint RewardType = "POINT"
)

func (r *RewardType) String() string {
	return fmt.Sprint(r)
}

const (
	OperateTypeIncrease = "INCREASE_POINT"
	OperateTypeDecrease = "DECREASE_POINT"
)

const (
	OperateStatusOperating = "OPERATING"
	OperateStatusSucceeded = "SUCCEEDED"
	OperateStatusFailed    = "FAILED"
)

type RewardOperateRecord struct {
	ID                   int64  `xorm:"pk autoincr"`
	RecordId             string `xorm:"INDEX NOT NULL"`
	UserId               int64  `xorm:"INDEX NOT NULL"`
	Amount               int64  `xorm:"NOT NULL"`
	RewardType           string `xorm:"NOT NULL"`
	SourceType           string `xorm:"NOT NULL"`
	SourceId             string `xorm:"INDEX NOT NULL"`
	RequestId            string `xorm:"INDEX NOT NULL"`
	OperateType          string `xorm:"NOT NULL"`
	CycleIntervalSeconds int64  `xorm:"NOT NULL default 0"`
	Status               string `xorm:"NOT NULL"`
	Remark               string
	CreatedUnix          timeutil.TimeStamp `xorm:"INDEX created"`
	UpdatedUnix          timeutil.TimeStamp `xorm:"INDEX updated"`
}

func getPointOperateRecord(tl *RewardOperateRecord) (*RewardOperateRecord, error) {
	has, err := x.Get(tl)
	if err != nil {
		return nil, err
	} else if !has {
		return nil, ErrRecordNotExist{}
	}
	return tl, nil
}

func GetPointOperateRecordBySourceTypeAndRequestId(sourceType, requestId string) (*RewardOperateRecord, error) {
	t := &RewardOperateRecord{
		SourceType: sourceType,
		RequestId:  requestId,
	}
	return getPointOperateRecord(t)
}

func InsertAwardOperateRecord(tl *RewardOperateRecord) (int64, error) {
	return x.Insert(tl)
}

func UpdateAwardOperateRecordStatus(sourceType, requestId, oldStatus, newStatus string) (int64, error) {
	r := &RewardOperateRecord{
		Status: newStatus,
	}
	return x.Cols("status").Where("source_type=? and request_id=? and status=?", sourceType, requestId, oldStatus).Update(r)
}

type RewardOperateContext struct {
	SourceType           string
	SourceId             string
	Remark               string
	Reward               Reward
	TargetUserId         int64
	RequestId            string
	OperateType          string
	CycleIntervalSeconds int64
}

type Reward struct {
	Amount int64
	Type   string
}