diff options
Diffstat (limited to 'main.go')
-rw-r--r-- | main.go | 4 |
1 files changed, 3 insertions, 1 deletions
@@ -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) |