3e72391cbf
When the worker is started with `-restart-exit-code 47` or has `restart_exit_code=47` in `flamenco-worker.yaml`, it's marked as 'restartable'. This will enable two worker actions 'Restart (immediately)' and 'Restart (after task is finished)' in the Manager web interface. When a worker is asked to restart, it will exit with exit code `47`. Of course any positive exit code can be used here.
56 lines
1.6 KiB
Go
56 lines
1.6 KiB
Go
package worker
|
|
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
|
|
"projects.blender.org/studio/flamenco/pkg/api"
|
|
)
|
|
|
|
func (w *Worker) gotoStateOffline(context.Context) {
|
|
w.stateMutex.Lock()
|
|
defer w.stateMutex.Unlock()
|
|
|
|
w.state = api.WorkerStatusOffline
|
|
w.requestShutdown(false)
|
|
}
|
|
|
|
// requestShutdown closes the w.shutdown channel, to indicate to the main
|
|
// function that it should proceed with the shutdown procedure.
|
|
func (w *Worker) requestShutdown(requestRestart bool) {
|
|
w.restartAfterShutdown = requestRestart
|
|
|
|
// Signal that the Worker should shut down.
|
|
log.Debug().Msg("closing the shutdown channel")
|
|
close(w.shutdown)
|
|
}
|
|
|
|
// SignOff forces the worker in shutdown state and acknlowedges this to the Manager.
|
|
// Does NOT actually peform a shutdown; is intended to be called while shutdown is in progress.
|
|
func (w *Worker) SignOff(ctx context.Context) {
|
|
w.stateMutex.Lock()
|
|
w.state = api.WorkerStatusOffline
|
|
logger := log.With().Str("state", string(w.state)).Logger()
|
|
w.stateMutex.Unlock()
|
|
|
|
logger.Info().Msg("signing off at Manager")
|
|
|
|
SignOff(ctx, logger, w.client)
|
|
}
|
|
|
|
// SignOff sends a signoff request to the Manager.
|
|
// Any error is logged but not returned.
|
|
func SignOff(ctx context.Context, logger zerolog.Logger, client FlamencoClient) {
|
|
resp, err := client.SignOffWithResponse(ctx)
|
|
switch {
|
|
case err != nil:
|
|
logger.Error().Err(err).Msg("unable to sign off at Manager")
|
|
case resp.JSONDefault != nil:
|
|
logger.Error().Interface("error", resp.JSONDefault).Msg("error received when signing off at Manager")
|
|
}
|
|
}
|