From 41435a6c4ee0a5b52528890edf3fbf5a9dc0a6c8 Mon Sep 17 00:00:00 2001
From: kim <89579420+NyaaaWhatsUpDoc@users.noreply.github.com>
Date: Sat, 4 Nov 2023 20:21:20 +0000
Subject: [feature] support canceling scheduled tasks, some federation API
 performance improvements (#2329)

---
 internal/cleaner/cleaner.go | 31 +++++++++++++++----------------
 1 file changed, 15 insertions(+), 16 deletions(-)

(limited to 'internal/cleaner/cleaner.go')

diff --git a/internal/cleaner/cleaner.go b/internal/cleaner/cleaner.go
index 1139a85bb..a1209ae08 100644
--- a/internal/cleaner/cleaner.go
+++ b/internal/cleaner/cleaner.go
@@ -22,8 +22,6 @@ import (
 	"errors"
 	"time"
 
-	"codeberg.org/gruf/go-runners"
-	"codeberg.org/gruf/go-sched"
 	"codeberg.org/gruf/go-store/v2/storage"
 	"github.com/superseriousbusiness/gotosocial/internal/config"
 	"github.com/superseriousbusiness/gotosocial/internal/gtscontext"
@@ -150,26 +148,27 @@ func (c *Cleaner) ScheduleJobs() error {
 		firstCleanupAt = firstCleanupAt.Add(cleanupEvery)
 	}
 
-	// Get ctx associated with scheduler run state.
-	done := c.state.Workers.Scheduler.Done()
-	doneCtx := runners.CancelCtx(done)
-
-	// TODO: we'll need to do some thinking to make these
-	// jobs restartable if we want to implement reloads in
-	// the future that make call to Workers.Stop() -> Workers.Start().
+	fn := func(ctx context.Context, start time.Time) {
+		log.Info(ctx, "starting media clean")
+		c.Media().All(ctx, config.GetMediaRemoteCacheDays())
+		c.Emoji().All(ctx, config.GetMediaRemoteCacheDays())
+		log.Infof(ctx, "finished media clean after %s", time.Since(start))
+	}
 
 	log.Infof(nil,
 		"scheduling media clean to run every %s, starting from %s; next clean will run at %s",
 		cleanupEvery, cleanupFromStr, firstCleanupAt,
 	)
 
-	// Schedule the cleaning tasks to execute according to given schedule.
-	c.state.Workers.Scheduler.Schedule(sched.NewJob(func(start time.Time) {
-		log.Info(nil, "starting media clean")
-		c.Media().All(doneCtx, config.GetMediaRemoteCacheDays())
-		c.Emoji().All(doneCtx, config.GetMediaRemoteCacheDays())
-		log.Infof(nil, "finished media clean after %s", time.Since(start))
-	}).EveryAt(firstCleanupAt, cleanupEvery))
+	// Schedule the cleaning to execute according to schedule.
+	if !c.state.Workers.Scheduler.AddRecurring(
+		"@mediacleanup",
+		firstCleanupAt,
+		cleanupEvery,
+		fn,
+	) {
+		panic("failed to schedule @mediacleanup")
+	}
 
 	return nil
 }
-- 
cgit v1.2.3