Add feature to refresh all feeds from the user interface

1 files changed, 9 insertions, 15 deletions

diff --git a/main.go b/main.go
index 9ab7cdc..4393472 100644
--- a/main.go
+++ b/main.go
@@ -44,21 +44,15 @@ func run(cfg *config.Config, store *storage.Storage) {
 	signal.Notify(stop, os.Interrupt)
 
 	feedHandler := feed.NewFeedHandler(store)
-	server := server.NewServer(cfg, store, feedHandler)
-
-	go func() {
-		pool := scheduler.NewWorkerPool(
-			feedHandler,
-			cfg.GetInt("WORKER_POOL_SIZE", config.DefaultWorkerPoolSize),
-		)
-
-		scheduler.NewScheduler(
-			store,
-			pool,
-			cfg.GetInt("POLLING_FREQUENCY", config.DefaultPollingFrequency),
-			cfg.GetInt("BATCH_SIZE", config.DefaultBatchSize),
-		)
-	}()
+	pool := scheduler.NewWorkerPool(feedHandler, cfg.GetInt("WORKER_POOL_SIZE", config.DefaultWorkerPoolSize))
+	server := server.NewServer(cfg, store, pool, feedHandler)
+
+	scheduler.NewScheduler(
+		store,
+		pool,
+		cfg.GetInt("POLLING_FREQUENCY", config.DefaultPollingFrequency),
+		cfg.GetInt("BATCH_SIZE", config.DefaultBatchSize),
+	)
 
 	<-stop
 	log.Println("Shutting down the server...")