fyp/logic/tasks/utils/utils.go

package tasks_utils

import (
	"time"

	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/db_types"
	"github.com/goccy/go-json"
)

type Task struct {
	Id            string    `db:"id" json:"id"`
	UserId        string    `db:"user_id" json:"user_id"`
	ModelId       string    `db:"model_id" json:"model_id"`
	Status        int       `db:"status" json:"status"`
	StatusMessage string    `db:"status_message" json:"status_message"`
	UserConfirmed int       `db:"user_confirmed" json:"user_confirmed"`
	Compacted     int       `db:"compacted" json:"compacted"`
	TaskType      int       `db:"task_type" json:"type"`
	ExtraTaskInfo string    `db:"extra_task_info" json:"extra_task_info"`
	Result        string    `db:"result" json:"result"`
	CreatedOn     time.Time `db:"created_on" json:"created"`
}

type TaskStatus int

const (
	TASK_FAILED_RUNNING  TaskStatus = -2
	TASK_FAILED_CREATION            = -1
	TASK_PREPARING                  = 0
	TASK_TODO                       = 1
	TASK_PICKED_UP                  = 2
	TASK_RUNNING                    = 3
	TASK_DONE                       = 4
)

type TaskType int

const (
	TASK_TYPE_CLASSIFICATION TaskType = 1 + iota
	TASK_TYPE_TRAINING
)

func (t Task) UpdateStatus(base BasePack, status TaskStatus, message string) (err error) {
	return UpdateTaskStatus(base, t.Id, status, message)
}

/**
* Call the UpdateStatus function and logs on the case of failure!
* This varient does not return any error message
 */
func (t Task) UpdateStatusLog(base BasePack, status TaskStatus, message string) {
	err := t.UpdateStatus(base, status, message)
	if err != nil {
		base.GetLogger().Error("Failed to update task status", "error", err, "task", t.Id)
	}
}

func UpdateTaskStatus(base BasePack, id string, status TaskStatus, message string) (err error) {
	_, err = base.GetDb().Exec("update tasks set status=$1, status_message=$2 where id=$3", status, message, id)
	return
}

func (t Task) SetResult(base BasePack, result any) (err error) {
	text, err := json.Marshal(result)
	if err != nil {
		return
	}
	_, err = base.GetDb().Exec("update tasks set result=$1 where id=$2", text, t.Id)
	return
}