admin/worker/models.go

package worker

import (
	"sync"
	"time"

	"gorm.io/gorm"
)

type QorJob struct {
	gorm.Model

	Job    string
	Status string      `sql:"default:'new'"`
	Args   interface{} `sql:"-" gorm:"-"`
}

type QorJobInstance struct {
	gorm.Model

	QorJobID uint `gorm:"index"`

	Operator string

	Job     string
	Status  string `sql:"default:'new'"`
	Args    string
	Context string

	Progress     uint
	ProgressText string

	jb          *JobBuilder `sql:"-"`
	mutex       sync.Mutex  `sql:"-"`
	stopRefresh bool        `sql:"-"`
	inRefresh   bool        `sql:"-"`
}

type QorJobLog struct {
	ID        uint      `gorm:"primarykey"`
	CreatedAt time.Time `gorm:"index"`

	QorJobInstanceID uint `gorm:"index"`
	Log              string
}

type Scheduler interface {
	GetScheduleTime() *time.Time
	SetScheduleTime(t *time.Time)
}

// Schedule could be embedded as job argument, then the job will get run as scheduled feature
type Schedule struct {
	ScheduleTime *time.Time
}

// GetScheduleTime get scheduled time
func (schedule *Schedule) GetScheduleTime() *time.Time {
	if scheduleTime := schedule.ScheduleTime; scheduleTime != nil {
		if scheduleTime.After(time.Now().Add(time.Minute)) {
			return scheduleTime
		}
	}
	return nil
}

func (schedule *Schedule) SetScheduleTime(t *time.Time) {
	schedule.ScheduleTime = t
}

type GoQueError struct {
	gorm.Model
	Error string
}