68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
|
package api
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/stephenafamo/bob"
|
||
|
|
"github.com/tigorlazuardi/redmage/models"
|
||
|
|
"github.com/tigorlazuardi/redmage/pkg/errs"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ScheduleStatus int8
|
||
|
|
|
||
|
|
const (
|
||
|
|
ScheduleStatusDisabled ScheduleStatus = iota
|
||
|
|
ScheduleStatusStandby
|
||
|
|
ScheduleStatusEnqueued
|
||
|
|
ScheduleStatusDownloading
|
||
|
|
ScheduleStatusError
|
||
|
|
)
|
||
|
|
|
||
|
|
func (ss ScheduleStatus) String() string {
|
||
|
|
switch ss {
|
||
|
|
case ScheduleStatusDisabled:
|
||
|
|
return "Disabled"
|
||
|
|
case ScheduleStatusStandby:
|
||
|
|
return "Standby"
|
||
|
|
case ScheduleStatusEnqueued:
|
||
|
|
return "Enqueued"
|
||
|
|
case ScheduleStatusDownloading:
|
||
|
|
return "Downloading"
|
||
|
|
case ScheduleStatusError:
|
||
|
|
return "Error"
|
||
|
|
}
|
||
|
|
return "Unknown"
|
||
|
|
}
|
||
|
|
|
||
|
|
func (ss ScheduleStatus) Int8() int8 {
|
||
|
|
return int8(ss)
|
||
|
|
}
|
||
|
|
|
||
|
|
type ScheduleSetParams struct {
|
||
|
|
Subreddit string
|
||
|
|
Status ScheduleStatus
|
||
|
|
ErrorMessage string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (api *API) ScheduleSet(ctx context.Context, params ScheduleSetParams) (schedule *models.ScheduleStatus, err error) {
|
||
|
|
ctx, span := tracer.Start(ctx, "*API.ScheduleSet")
|
||
|
|
defer span.End()
|
||
|
|
|
||
|
|
errTx := api.withTransaction(ctx, func(exec bob.Executor) error {
|
||
|
|
schedule, err = api.ScheduleStatusUpsert(ctx, params)
|
||
|
|
if err != nil {
|
||
|
|
return errs.Wrapw(err, "failed to set schedule status", "params", params)
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = api.ScheduleHistoryInsert(ctx, params)
|
||
|
|
if err != nil {
|
||
|
|
return errs.Wrapw(err, "failed to insert schedule history", "params", params)
|
||
|
|
}
|
||
|
|
|
||
|
|
// TODO: Create cron job schedule rebalancing
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
|
||
|
|
return schedule, errTx
|
||
|
|
}
|