Add scheduler to clean old sessions

1 files changed, 3 insertions, 1 deletions

diff --git a/main.go b/main.go
index 02f53c0..a252c61 100644
--- a/main.go
+++ b/main.go
@@ -47,13 +47,15 @@ func run(cfg *config.Config, store *storage.Storage) {
 	pool := scheduler.NewWorkerPool(feedHandler, cfg.GetInt("WORKER_POOL_SIZE", config.DefaultWorkerPoolSize))
 	server := server.NewServer(cfg, store, pool, feedHandler)
 
-	scheduler.NewScheduler(
+	scheduler.NewFeedScheduler(
 		store,
 		pool,
 		cfg.GetInt("POLLING_FREQUENCY", config.DefaultPollingFrequency),
 		cfg.GetInt("BATCH_SIZE", config.DefaultBatchSize),
 	)
 
+	scheduler.NewSessionScheduler(store, config.DefaultSessionCleanupFrequency)
+
 	<-stop
 	logger.Info("Shutting down the server...")
 	ctx, _ := context.WithTimeout(context.Background(), 5*time.Second)