bedf10e435
Instead of logging "sleep aborted", the message is now "sleep command aborted", to make it clear that it's about the sleep command, and not the "asleep" worker state.
70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
package worker
|
|
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
/* This file contains the commands in the "misc" type group. */
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
|
|
"git.blender.org/flamenco/pkg/api"
|
|
)
|
|
|
|
// cmdEcho executes the "echo" command.
|
|
func (ce *CommandExecutor) cmdEcho(ctx context.Context, logger zerolog.Logger, taskID string, cmd api.Command) error {
|
|
message, ok := cmd.Parameters["message"]
|
|
if !ok {
|
|
return fmt.Errorf("missing 'message' setting")
|
|
}
|
|
messageStr := fmt.Sprintf("%v", message)
|
|
|
|
logger.Info().Str("message", messageStr).Msg("echo")
|
|
if err := ce.listener.LogProduced(ctx, taskID, fmt.Sprintf("echo: %q", messageStr)); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// cmdSleep executes the "sleep" command.
|
|
func (ce *CommandExecutor) cmdSleep(ctx context.Context, logger zerolog.Logger, taskID string, cmd api.Command) error {
|
|
|
|
sleepTime, ok := cmd.Parameters["duration_in_seconds"]
|
|
if !ok {
|
|
return errors.New("missing setting 'duration_in_seconds'")
|
|
}
|
|
|
|
var duration time.Duration
|
|
switch v := sleepTime.(type) {
|
|
case int:
|
|
duration = time.Duration(v) * time.Second
|
|
case float64:
|
|
duration = time.Duration(v) * time.Second
|
|
default:
|
|
log.Warn().Interface("duration_in_seconds", v).Msg("bad type for setting 'duration_in_seconds', expected int")
|
|
return fmt.Errorf("bad type for setting 'duration_in_seconds', expected int, not %T", v)
|
|
}
|
|
|
|
log.Info().Str("duration", duration.String()).Msg("sleep")
|
|
|
|
select {
|
|
case <-ctx.Done():
|
|
err := ctx.Err()
|
|
log.Warn().Msg("sleep command aborted due to context shutdown")
|
|
return fmt.Errorf("sleep command aborted due to context shutdown: %w", err)
|
|
case <-ce.timeService.After(duration):
|
|
log.Debug().Msg("sleeping done")
|
|
}
|
|
|
|
if err := ce.listener.LogProduced(ctx, taskID, fmt.Sprintf("slept %v", duration)); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|