ddced5a823
Replace old used-to-be-GORM datastructures (#104305) with sqlc-generated structs. This also makes it possible to use more specific structs that are more taylored to the specific queries, increasing efficiency. This commit deals with the worker sleep schedule. Functional changes are kept to a minimum, as the API still serves the same data. Because this work covers so much of Flamenco's code, it's been split up into different commits. Each commit brings Flamenco to a state where it compiles and unit tests pass. Only the result of the final commit has actually been tested properly. Ref: #104343
89 lines
3.2 KiB
Go
89 lines
3.2 KiB
Go
package api_impl
|
|
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
"projects.blender.org/studio/flamenco/internal/manager/persistence"
|
|
"projects.blender.org/studio/flamenco/internal/uuid"
|
|
"projects.blender.org/studio/flamenco/pkg/api"
|
|
)
|
|
|
|
func (f *Flamenco) FetchWorkerSleepSchedule(e echo.Context, workerUUID string) error {
|
|
if !uuid.IsValid(workerUUID) {
|
|
return sendAPIError(e, http.StatusBadRequest, "not a valid UUID")
|
|
}
|
|
|
|
ctx := e.Request().Context()
|
|
logger := requestLogger(e)
|
|
logger = logger.With().Str("worker", workerUUID).Logger()
|
|
schedule, err := f.sleepScheduler.FetchSchedule(ctx, workerUUID)
|
|
|
|
switch {
|
|
case errors.Is(err, persistence.ErrWorkerNotFound):
|
|
logger.Warn().Msg("FetchWorkerSleepSchedule: worker does not exist")
|
|
return sendAPIError(e, http.StatusNotFound, "worker %q does not exist", workerUUID)
|
|
case err != nil:
|
|
logger.Error().Err(err).Msg("FetchWorkerSleepSchedule: error fetching sleep schedule")
|
|
return sendAPIError(e, http.StatusInternalServerError, "error fetching sleep schedule: %v", err)
|
|
case schedule == nil:
|
|
return e.NoContent(http.StatusNoContent)
|
|
}
|
|
|
|
apiSchedule := api.WorkerSleepSchedule{
|
|
DaysOfWeek: schedule.DaysOfWeek,
|
|
EndTime: schedule.EndTime.String(),
|
|
IsActive: schedule.IsActive,
|
|
StartTime: schedule.StartTime.String(),
|
|
}
|
|
return e.JSON(http.StatusOK, apiSchedule)
|
|
}
|
|
|
|
func (f *Flamenco) SetWorkerSleepSchedule(e echo.Context, workerUUID string) error {
|
|
if !uuid.IsValid(workerUUID) {
|
|
return sendAPIError(e, http.StatusBadRequest, "not a valid UUID")
|
|
}
|
|
|
|
ctx := e.Request().Context()
|
|
logger := requestLogger(e)
|
|
logger = logger.With().Str("worker", workerUUID).Logger()
|
|
|
|
var req api.SetWorkerSleepScheduleJSONRequestBody
|
|
err := e.Bind(&req)
|
|
if err != nil {
|
|
logger.Warn().Err(err).Msg("bad request received")
|
|
return sendAPIError(e, http.StatusBadRequest, "invalid format")
|
|
}
|
|
schedule := api.WorkerSleepSchedule(req)
|
|
|
|
// Create a sleep schedule that can be persisted.
|
|
dbSchedule := persistence.SleepSchedule{
|
|
IsActive: schedule.IsActive,
|
|
DaysOfWeek: schedule.DaysOfWeek,
|
|
}
|
|
if err := dbSchedule.StartTime.Scan(schedule.StartTime); err != nil {
|
|
logger.Warn().Interface("schedule", schedule).Err(err).Msg("bad request received, cannot parse schedule start time")
|
|
return sendAPIError(e, http.StatusBadRequest, "invalid format for schedule start time")
|
|
}
|
|
if err := dbSchedule.EndTime.Scan(schedule.EndTime); err != nil {
|
|
logger.Warn().Interface("schedule", schedule).Err(err).Msg("bad request received, cannot parse schedule end time")
|
|
return sendAPIError(e, http.StatusBadRequest, "invalid format for schedule end time")
|
|
}
|
|
|
|
// Send the sleep schedule to the scheduler.
|
|
err = f.sleepScheduler.SetSchedule(ctx, workerUUID, dbSchedule)
|
|
switch {
|
|
case errors.Is(err, persistence.ErrWorkerNotFound):
|
|
logger.Warn().Msg("SetWorkerSleepSchedule: worker does not exist")
|
|
return sendAPIError(e, http.StatusNotFound, "worker %q does not exist", workerUUID)
|
|
case err != nil:
|
|
logger.Error().Err(err).Msg("SetWorkerSleepSchedule: error fetching sleep schedule")
|
|
return sendAPIError(e, http.StatusInternalServerError, "error fetching sleep schedule: %v", err)
|
|
}
|
|
|
|
return e.NoContent(http.StatusNoContent)
|
|
}
|