OliveTin/internal/oncron/cron.go

package oncron

import (
	"github.com/OliveTin/OliveTin/internal/acl"
	"github.com/OliveTin/OliveTin/internal/config"
	"github.com/OliveTin/OliveTin/internal/executor"
	"github.com/robfig/cron/v3"
	log "github.com/sirupsen/logrus"
)

func Schedule(cfg *config.Config, ex *executor.Executor) {
	var scheduler *cron.Cron

	if cfg.CronSupportForSeconds {
		scheduler = cron.New(cron.WithSeconds())
	} else {
		scheduler = cron.New()
	}

	for _, action := range cfg.Actions {
		for _, cronline := range action.ExecOnCron {
			scheduleAction(cfg, scheduler, cronline, ex, action)
		}
	}

	scheduler.Start()
}

func scheduleAction(cfg *config.Config, scheduler *cron.Cron, cronline string, ex *executor.Executor, action *config.Action) {
	log.WithFields(log.Fields{
		"action":   action.Title,
		"cronline": cronline,
	}).Infof("Scheduling Action for cron")

	_, err := scheduler.AddFunc(cronline, func() {
		req := &executor.ExecutionRequest{
			ActionTitle:       action.Title,
			Cfg:               cfg,
			Tags:              []string{},
			AuthenticatedUser: acl.UserFromSystem(cfg, "cron"),
		}

		ex.ExecRequest(req)
	})

	if err != nil {
		log.WithFields(log.Fields{
			"action":    action.Title,
			"cronError": err,
		}).Errorf("CRON schedule error")
		return
	}

}