add type checking to ensure that both int and float types can be used…

… in retry-interval.
elvis972602 · Mar 3, 2023 · 01dd43a · 01dd43a
1 parent 328ee30
commit 01dd43a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/main/main.go b/main/main.go
@@ -546,7 +546,13 @@ func setFlag() {
 		retry = config["retry"].(int)
 	}
 	if !isFlagPassed("retry-interval") && config["retry-interval"] != nil {
-		retryInterval = config["retry-interval"].(float64)
+		// check if retry-interval is float64 or int
+		_, ok := config["retry-interval"].(float64)
+		if !ok {
+			retryInterval = float64(config["retry-interval"].(int))
+		} else {
+			retryInterval = config["retry-interval"].(float64)
+		}
 	}
 	if !isFlagPassed("max-download-parallel") && config["max-download-parallel"] != nil {
 		maxDownloadParallel = config["max-download-parallel"].(int)