feat: add tasks closes #74

logic/tasks/utils/utils.go

@@ -0,0 +1,68 @@
+package tasks_utils
+
+import (
+	"time"
+
+	. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
+	"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"`
+	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
+)
+
+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
+}