speatzle/morffix
1
0
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Wiki Activity Actions

Move Task Starting into task

Browse source
This commit is contained in:
Samuel Lorch 2024-05-09 04:45:54 +02:00
parent bce27eddf3
commit b17e55de7d
2 changed files with 94 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

51
worker/task.go
View file

@ -4,20 +4,16 @@ import (
"context"
"encoding/json"
"fmt"
"sync"
"git.lastassault.de/speatzle/morffix/constants"
"git.lastassault.de/speatzle/morffix/rpc"
"git.lastassault.de/speatzle/morffix/task"
"git.lastassault.de/speatzle/morffix/types"
)
var tasks map[int]*task.Task
var taskMutex sync.Mutex
func init() {
rpcServer.RegisterMethod("task-start", taskStart)
rpcServer.RegisterMethod("task-status", taskStatus)
rpcServer.RegisterMethod("task-delete", taskDelete)
}
func taskStart(ctx context.Context, req rpc.Request) (any, error) {
@ -27,40 +23,31 @@ func taskStart(ctx context.Context, req rpc.Request) (any, error) {
return nil, fmt.Errorf("Unmarshal Task Start Params: %w", err)
}
// TODO move to task module
taskMutex.Lock()
defer taskMutex.Unlock()
switch data.Type {
case constants.TASK_TYPE_HEALTHCHECK:
tasks[data.ID] = &task.Task{
ID: data.ID,
Type: data.Type,
FileID: data.FileID,
}
var hData task.HealthCheckData
err := json.Unmarshal(data.Data, &hData)
if err != nil {
return nil, fmt.Errorf("Unmarshal Healthcheck Data: %w", err)
}
tasks[data.ID].Status = constants.TASK_STATUS_RUNNING
go tasks[data.ID].RunHealthCheck(hData)
case constants.TASK_TYPE_TRANSCODE:
return nil, fmt.Errorf("Transcode Task Not Implemented")
default:
return nil, fmt.Errorf("Unknown Task Type %v", data.Type)
err = task.StartTask(conf, data)
if err != nil {
return nil, fmt.Errorf("Starting Task: %w", err)
}
return nil, nil
}
func taskStatus(ctx context.Context, req rpc.Request) (any, error) {
var id int
err := json.Unmarshal(*req.Params, &id)
return types.TaskStatus{
Tasks: task.Get(),
}, nil
}
func taskDelete(ctx context.Context, req rpc.Request) (any, error) {
var data int
err := json.Unmarshal(*req.Params, &data)
if err != nil {
return nil, fmt.Errorf("Unmarshal Task Status Params: %w", err)
return nil, fmt.Errorf("Unmarshal Task Delete ID: %w", err)
}
err = task.DeleteTask(data)
if err != nil {
return nil, fmt.Errorf("Deleting Task: %w", err)
}
return nil, nil
}