Merge pull request #99 from kochd/master

add -no-limit parameter
broderickhyman · Dec 1, 2021 · 6cb6c2f · 6cb6c2f
2 parents a86bc07 + aaf0d1f
commit 6cb6c2f
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 5 deletions.
diff --git a/albiondata-client.go b/albiondata-client.go
@@ -128,6 +128,13 @@ func init() {
 		false,
 		"Ignore the decoding errors when debugging",
 	)
+
+	flag.BoolVar(
+		&client.ConfigGlobal.NoCPULimit,
+		"no-limit",
+		false,
+		"Use all available CPU cores",
+	)
 }
 
 func main() {

diff --git a/client/config.go b/client/config.go
@@ -21,6 +21,7 @@ type config struct {
 	RecordPath                     string
 	PrivateIngestBaseUrls          string
 	PublicIngestBaseUrls           string
+	NoCPULimit                     bool
 }
 
 //ConfigGlobal global config data

diff --git a/client/uploader_http_pow.go b/client/uploader_http_pow.go
@@ -28,12 +28,14 @@ type Pow struct {
 // newHTTPUploaderPow creates a new HTTP uploader
 func newHTTPUploaderPow(url string) uploader {
 
-	// Limit to 25% of available cpu cores
-	procs := runtime.NumCPU() / 4
-	if procs < 1 {
-		procs = 1
+	if !ConfigGlobal.NoCPULimit {
+		// Limit to 25% of available cpu cores
+		procs := runtime.NumCPU() / 4
+		if procs < 1 {
+			procs = 1
+		}
+		runtime.GOMAXPROCS(procs)
 	}
-	runtime.GOMAXPROCS(procs)
 
 	return &httpUploaderPow{
 		baseURL:   strings.Replace(url, "http+pow", "http", -1),