aiforge/models/task_config.go

package models

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

type TaskType string

const (
	TaskTypeComment TaskType = "COMMENT"
)

func (t *TaskType) String() string {
	return fmt.Sprint(t)
}

const (
	PeriodNotCycle = "NOT_CYCLE"
	PeriodDaily    = "DAILY"
)

//PointTaskConfig Only add and delete are allowed, edit is not allowed
//so if you want to edit config for some task code,please delete first and add new one
type TaskConfig struct {
	ID          int64              `xorm:"pk autoincr"`
	TaskCode    string             `xorm:"NOT NULL"`
	Tittle      string             `xorm:"NOT NULL"`
	RefreshRate string             `xorm:"NOT NULL"`
	Times       int64              `xorm:"NOT NULL"`
	AwardType   string             `xorm:"NOT NULL"`
	AwardAmount int64              `xorm:"NOT NULL"`
	Creator     int64              `xorm:"NOT NULL"`
	CreatedUnix timeutil.TimeStamp `xorm:"created"`
	DeletedAt   timeutil.TimeStamp `xorm:"deleted"`
}

func getTaskConfig(t *TaskConfig) (*TaskConfig, error) {
	has, err := x.Get(t)
	if err != nil {
		return nil, err
	} else if !has {
		return nil, ErrRecordNotExist{}
	}
	return t, nil
}

func GetTaskConfigByTaskCode(taskCode string) (*TaskConfig, error) {
	t := &TaskConfig{
		TaskCode: taskCode,
	}
	return getTaskConfig(t)
}