runner-go (#102)
Reviewed-on: andr3/fyp#102 Co-authored-by: Andre Henriques <andr3h3nriqu3s@gmail.com> Co-committed-by: Andre Henriques <andr3h3nriqu3s@gmail.com>
This commit was merged in pull request #102.
This commit is contained in:
29
logic/tasks/utils/runner.go
Normal file
29
logic/tasks/utils/runner.go
Normal file
@@ -0,0 +1,29 @@
|
||||
package tasks_utils
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"git.andr3h3nriqu3s.com/andr3/fyp/logic/db"
|
||||
dbtypes "git.andr3h3nriqu3s.com/andr3/fyp/logic/db_types"
|
||||
)
|
||||
|
||||
type RunnerType int64
|
||||
|
||||
const (
|
||||
RUNNER_TYPE_GPU RunnerType = iota + 1
|
||||
)
|
||||
|
||||
type Runner struct {
|
||||
Id string `json:"id" db:"ru.id"`
|
||||
UserId string `json:"user_id" db:"ru.user_id"`
|
||||
Token string `json:"token" db:"ru.token"`
|
||||
Type RunnerType `json:"type" db:"ru.type"`
|
||||
CreateOn time.Time `json:"createOn" db:"ru.created_on"`
|
||||
}
|
||||
|
||||
func GetRunner(db db.Db, id string) (ru *Runner, err error) {
|
||||
var runner Runner
|
||||
err = dbtypes.GetDBOnce(db, &runner, "remote_runner as ru where ru.id=$1", id)
|
||||
ru = &runner
|
||||
return
|
||||
}
|
||||
@@ -101,7 +101,11 @@ func (t Task) SetResult(base BasePack, result any) (err error) {
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
_, err = base.GetDb().Exec("update tasks set result=$1 where id=$2", text, t.Id)
|
||||
return t.SetResultText(base, string(text))
|
||||
}
|
||||
|
||||
func (t Task) SetResultText(base BasePack, text string) (err error) {
|
||||
_, err = base.GetDb().Exec("update tasks set result=$1 where id=$2", []byte(text), t.Id)
|
||||
return
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user