feat: control number of workers from config

pnicto · Apr 10, 2024 · 0e46486 · 0e46486
1 parent 181f635
commit 0e46486
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/config.go b/config.go
@@ -17,6 +17,7 @@ type Config struct {
 	DownloadLocation string
 	Token            string
 	TempDirLocation  string
+	NumWorkers       int
 	Slides           bool
 }
 
@@ -41,6 +42,9 @@ func parseConfig(configLocation string) *Config {
 		config.TempDirLocation = "./temp"
 	}
 
+	if config.NumWorkers == 0 {
+		config.NumWorkers = 5
+	}
 
 	return &config
 }

diff --git a/main.go b/main.go
@@ -66,7 +66,7 @@ func main() {
 		log.Fatalln("Could not create temp dir")
 	}
 
-	const numWorkers = 10
+	numWorkers := config.NumWorkers
 	playlistJobs := make(chan ParsedPlaylist, numWorkers)
 
 	var joinWg sync.WaitGroup

diff --git a/sample.config.json b/sample.config.json
@@ -6,5 +6,6 @@
   "views": "both",
   "downloadLocation": "./downloads",
   "tempDirLocation": "./.temp",
-  "slides": false
+  "slides": false,
+  "numWorkers": 5
 }