02fac6a4df
Change the package base name of the Go code, from `git.blender.org/flamenco` to `projects.blender.org/studio/flamenco`. The old location, `git.blender.org`, has no longer been use since the [migration to Gitea][1]. The new package names now reflect the actual location where Flamenco is hosted. [1]: https://code.blender.org/2023/02/new-blender-development-infrastructure/
82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
package worker
|
|
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
|
|
"projects.blender.org/studio/flamenco/pkg/api"
|
|
)
|
|
|
|
// queryManagerForStateChange asks the Manager whether we should go to another state or not.
|
|
// Any error communicating with the Manager is logged but otherwise ignored.
|
|
// Returns nil when no state change is requested.
|
|
func (w *Worker) queryManagerForStateChange(ctx context.Context) *api.WorkerStatus {
|
|
resp, err := w.client.WorkerStateWithResponse(ctx)
|
|
if err != nil {
|
|
log.Warn().Err(err).Msg("error checking upstream state changes")
|
|
return nil
|
|
}
|
|
switch {
|
|
case resp.JSON200 != nil:
|
|
log.Info().
|
|
Str("requestedStatus", string(resp.JSON200.StatusRequested)).
|
|
Msg("Manager requests status change")
|
|
return &resp.JSON200.StatusRequested
|
|
case resp.StatusCode() == http.StatusNoContent:
|
|
log.Debug().Msg("we can stay in the current state")
|
|
default:
|
|
log.Warn().
|
|
Int("code", resp.StatusCode()).
|
|
Str("error", string(resp.Body)).
|
|
Msg("unable to obtain requested state for unknown reason")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// mayIKeepRunning asks the Manager whether we can keep running a certain task.
|
|
// Any error communicating with the Manager is logged but otherwise ignored.
|
|
func (w *Worker) mayIKeepRunning(ctx context.Context, taskID string) api.MayKeepRunning {
|
|
resp, err := w.client.MayWorkerRunWithResponse(ctx, taskID)
|
|
if err != nil {
|
|
if errors.Is(err, context.Canceled) {
|
|
log.Debug().
|
|
Str("task", taskID).
|
|
Msg("may-I-keep-running query interrupted by shutdown")
|
|
} else {
|
|
log.Warn().
|
|
Err(err).
|
|
Str("task", taskID).
|
|
Msg("error asking Manager may-I-keep-running task")
|
|
}
|
|
return api.MayKeepRunning{MayKeepRunning: true}
|
|
}
|
|
|
|
switch {
|
|
case resp.JSON200 != nil:
|
|
mkr := *resp.JSON200
|
|
logCtx := log.With().
|
|
Str("task", taskID).
|
|
Bool("mayKeepRunning", mkr.MayKeepRunning).
|
|
Bool("statusChangeRequested", mkr.StatusChangeRequested)
|
|
if mkr.Reason != "" {
|
|
logCtx = logCtx.Str("reason", mkr.Reason)
|
|
}
|
|
logger := logCtx.Logger()
|
|
logger.Debug().Msg("may-i-keep-running response")
|
|
return mkr
|
|
default:
|
|
log.Warn().
|
|
Str("task", taskID).
|
|
Int("code", resp.StatusCode()).
|
|
Str("error", string(resp.Body)).
|
|
Msg("unable to check may-i-keep-running for unknown reason")
|
|
return api.MayKeepRunning{MayKeepRunning: true}
|
|
}
|
|
}
|