Worker: implement fail/completed/log task updates
This commit is contained in:
parent
8424e690ee
commit
f6437be844
@ -5,6 +5,7 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@ -77,18 +78,51 @@ func (l *Listener) Wait() {
|
|||||||
l.doneWg.Wait()
|
l.doneWg.Wait()
|
||||||
}
|
}
|
||||||
|
|
||||||
// TaskStarted tells the Manager that task execution has started.
|
func ptr[T any](value T) *T {
|
||||||
func (l *Listener) TaskStarted(ctx context.Context, taskID string) error {
|
return &value
|
||||||
activity := "Started"
|
|
||||||
status := api.TaskStatusActive
|
|
||||||
update := api.TaskUpdateJSONRequestBody{
|
|
||||||
Activity: &activity,
|
|
||||||
TaskStatus: &status,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TaskStarted tells the Manager that task execution has started.
|
||||||
|
func (l *Listener) TaskStarted(ctx context.Context, taskID string) error {
|
||||||
|
return l.sendTaskUpdate(ctx, taskID, api.TaskUpdateJSONRequestBody{
|
||||||
|
Activity: ptr("Started"),
|
||||||
|
TaskStatus: ptr(api.TaskStatusActive),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// TaskFailed tells the Manager the task failed for some reason.
|
||||||
|
func (l *Listener) TaskFailed(ctx context.Context, taskID string, reason string) error {
|
||||||
|
return l.sendTaskUpdate(ctx, taskID, api.TaskUpdateJSONRequestBody{
|
||||||
|
Activity: ptr(fmt.Sprintf("Failed: %v", reason)),
|
||||||
|
TaskStatus: ptr(api.TaskStatusFailed),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// TaskCompleted tells the Manager the task has been completed.
|
||||||
|
func (l *Listener) TaskCompleted(ctx context.Context, taskID string) error {
|
||||||
|
return l.sendTaskUpdate(ctx, taskID, api.TaskUpdateJSONRequestBody{
|
||||||
|
Activity: ptr("Completed"),
|
||||||
|
TaskStatus: ptr(api.TaskStatusCompleted),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// LogProduced sends any logging to whatever service for storing logging.
|
||||||
|
func (l *Listener) LogProduced(ctx context.Context, taskID string, logLines ...string) error {
|
||||||
|
return l.sendTaskUpdate(ctx, taskID, api.TaskUpdateJSONRequestBody{
|
||||||
|
Log: ptr(strings.Join(logLines, "\n")),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// OutputProduced tells the Manager there has been some output (most commonly a rendered frame or video).
|
||||||
|
func (l *Listener) OutputProduced(ctx context.Context, taskID string, outputLocation string) error {
|
||||||
|
// TODO: implement
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Listener) sendTaskUpdate(ctx context.Context, taskID string, update api.TaskUpdateJSONRequestBody) error {
|
||||||
resp, err := l.client.TaskUpdateWithResponse(ctx, string(taskID), update)
|
resp, err := l.client.TaskUpdateWithResponse(ctx, string(taskID), update)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("error notifying Manager of task start: %w", err)
|
return fmt.Errorf("error sending task update: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
switch resp.StatusCode() {
|
switch resp.StatusCode() {
|
||||||
@ -100,23 +134,3 @@ func (l *Listener) TaskStarted(ctx context.Context, taskID string) error {
|
|||||||
return fmt.Errorf("unknown error from Manager, code %d: %v", resp.StatusCode(), resp.JSONDefault)
|
return fmt.Errorf("unknown error from Manager, code %d: %v", resp.StatusCode(), resp.JSONDefault)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// TaskFailed tells the Manager the task failed for some reason.
|
|
||||||
func (l *Listener) TaskFailed(ctx context.Context, taskID string, reason string) error {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// TaskCompleted tells the Manager the task has been completed.
|
|
||||||
func (l *Listener) TaskCompleted(ctx context.Context, taskID string) error {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// LogProduced sends any logging to whatever service for storing logging.
|
|
||||||
func (l *Listener) LogProduced(ctx context.Context, taskID string, logLines ...string) error {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// OutputProduced tells the Manager there has been some output (most commonly a rendered frame or video).
|
|
||||||
func (l *Listener) OutputProduced(ctx context.Context, taskID string, outputLocation string) error {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user