mirror of https://github.com/portainer/portainer
117 lines
3.2 KiB
Go
117 lines
3.2 KiB
Go
package cron
|
|
|
|
import (
|
|
"github.com/portainer/portainer/api"
|
|
"github.com/robfig/cron"
|
|
)
|
|
|
|
// JobScheduler represents a service for managing crons
|
|
type JobScheduler struct {
|
|
cron *cron.Cron
|
|
}
|
|
|
|
// NewJobScheduler initializes a new service
|
|
func NewJobScheduler() *JobScheduler {
|
|
return &JobScheduler{
|
|
cron: cron.New(cron.WithSeconds()),
|
|
}
|
|
}
|
|
|
|
// ScheduleJob schedules the execution of a job via a runner
|
|
func (scheduler *JobScheduler) ScheduleJob(runner portainer.JobRunner) error {
|
|
_, err := scheduler.cron.AddJob(runner.GetSchedule().CronExpression, runner)
|
|
return err
|
|
}
|
|
|
|
// UpdateSystemJobSchedule updates the first occurence of the specified
|
|
// scheduled job based on the specified job type.
|
|
// It does so by re-creating a new cron
|
|
// and adding all the existing jobs. It will then re-schedule the new job
|
|
// with the update cron expression passed in parameter.
|
|
// NOTE: the cron library do not support updating schedules directly
|
|
// hence the work-around
|
|
func (scheduler *JobScheduler) UpdateSystemJobSchedule(jobType portainer.JobType, newCronExpression string) error {
|
|
cronEntries := scheduler.cron.Entries()
|
|
newCron := cron.New()
|
|
|
|
for _, entry := range cronEntries {
|
|
if entry.Job.(portainer.JobRunner).GetSchedule().JobType == jobType {
|
|
_, err := newCron.AddJob(newCronExpression, entry.Job)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
|
|
newCron.Schedule(entry.Schedule, entry.Job)
|
|
}
|
|
|
|
scheduler.cron.Stop()
|
|
scheduler.cron = newCron
|
|
scheduler.cron.Start()
|
|
return nil
|
|
}
|
|
|
|
// UpdateJobSchedule updates a specific scheduled job by re-creating a new cron
|
|
// and adding all the existing jobs. It will then re-schedule the new job
|
|
// via the specified JobRunner parameter.
|
|
// NOTE: the cron library do not support updating schedules directly
|
|
// hence the work-around
|
|
func (scheduler *JobScheduler) UpdateJobSchedule(runner portainer.JobRunner) error {
|
|
cronEntries := scheduler.cron.Entries()
|
|
newCron := cron.New()
|
|
|
|
for _, entry := range cronEntries {
|
|
|
|
if entry.Job.(portainer.JobRunner).GetSchedule().ID == runner.GetSchedule().ID {
|
|
|
|
var jobRunner cron.Job = runner
|
|
if entry.Job.(portainer.JobRunner).GetSchedule().JobType == portainer.SnapshotJobType {
|
|
jobRunner = entry.Job
|
|
}
|
|
|
|
_, err := newCron.AddJob(runner.GetSchedule().CronExpression, jobRunner)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
continue
|
|
}
|
|
|
|
newCron.Schedule(entry.Schedule, entry.Job)
|
|
}
|
|
|
|
scheduler.cron.Stop()
|
|
scheduler.cron = newCron
|
|
scheduler.cron.Start()
|
|
return nil
|
|
}
|
|
|
|
// UnscheduleJob remove a scheduled job by re-creating a new cron
|
|
// and adding all the existing jobs except for the one specified via scheduleID.
|
|
// NOTE: the cron library do not support removing schedules directly
|
|
// hence the work-around
|
|
func (scheduler *JobScheduler) UnscheduleJob(scheduleID portainer.ScheduleID) {
|
|
cronEntries := scheduler.cron.Entries()
|
|
newCron := cron.New()
|
|
|
|
for _, entry := range cronEntries {
|
|
|
|
if entry.Job.(portainer.JobRunner).GetSchedule().ID == scheduleID {
|
|
continue
|
|
}
|
|
|
|
newCron.Schedule(entry.Schedule, entry.Job)
|
|
}
|
|
|
|
scheduler.cron.Stop()
|
|
scheduler.cron = newCron
|
|
scheduler.cron.Start()
|
|
}
|
|
|
|
// Start starts the scheduled jobs
|
|
func (scheduler *JobScheduler) Start() {
|
|
if len(scheduler.cron.Entries()) > 0 {
|
|
scheduler.cron.Start()
|
|
}
|
|
}
|