Redmage/api/scheduler/rebalance.go

package scheduler

import (
	"context"
	"errors"

	"github.com/stephenafamo/bob"
	"github.com/tigorlazuardi/redmage/models"
	"github.com/tigorlazuardi/redmage/pkg/errs"
)

// Sync empties the scheduler and re-adds all enabled jobs from the database.
//
// This is costly but ensures that the scheduler is in sync with the database.
//
// For simpler operation consider using Put and Delete instead.
func (scheduler *Scheduler) Sync(ctx context.Context, db bob.Executor) (err error) {
	ctx, span := tracer.Start(ctx, "*Scheduler.Rebalance")
	defer span.End()

	subs, err := models.Subreddits.Query(ctx, db, models.SelectWhere.Subreddits.EnableSchedule.EQ(1)).All()
	if err != nil {
		return errs.Wrapw(err, "scheduler: rebalance: failed to query subreddits")
	}

	scheduler.mu.Lock()
	defer scheduler.mu.Unlock()

	for _, entry := range scheduler.scheduler.Entries() {
		scheduler.scheduler.Remove(entry.ID)
	}

	var errcoll error

	for _, sub := range subs {
		_, err := scheduler.put(sub.Name, sub.Schedule, false)
		if err != nil {
			errcoll = errors.Join(errcoll, errs.Wrapw(err, "scheduler: rebalance: failed to add job", "subreddit", sub.Name, "schedule", sub.Schedule))
		}
	}

	if errcoll != nil {
		return errs.Wrapw(errcoll, "scheduler: encountered errors while rebalancing jobs")
	}

	return nil
}