132ce8f2ec
Move the 'shutdown' state code to the 'offline' state, to match the removal of the 'shutdown' state from the OpenAPI definition.
53 lines
1.4 KiB
Go
53 lines
1.4 KiB
Go
package worker
|
|
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
|
|
"git.blender.org/flamenco/pkg/api"
|
|
)
|
|
|
|
func (w *Worker) gotoStateOffline(context.Context) {
|
|
w.stateMutex.Lock()
|
|
defer w.stateMutex.Unlock()
|
|
|
|
w.state = api.WorkerStatusOffline
|
|
|
|
logger := log.With().Int("pid", os.Getpid()).Logger()
|
|
proc, err := os.FindProcess(os.Getpid())
|
|
if err != nil {
|
|
logger.Fatal().Err(err).Msg("unable to find our own process for clean shutdown")
|
|
}
|
|
|
|
logger.Warn().Msg("sending our own process an interrupt signal")
|
|
err = proc.Signal(os.Interrupt)
|
|
if err != nil {
|
|
logger.Fatal().Err(err).Msg("unable to send interrupt signal to our own process")
|
|
}
|
|
}
|
|
|
|
// 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")
|
|
|
|
resp, err := w.client.SignOffWithResponse(ctx)
|
|
if err != nil {
|
|
logger.Error().Err(err).Msg("unable to sign off at Manager")
|
|
return
|
|
}
|
|
if resp.JSONDefault != nil {
|
|
logger.Error().Interface("error", resp.JSONDefault).Msg("error received when signing off at Manager")
|
|
return
|
|
}
|
|
}
|