feat: automatically adjust the maximum concurrency to the container C…

…PU quota (#4441)
golangci · Mar 3, 2024 · b1eed50 · b1eed50
1 parent 61f2f70
commit b1eed50
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/go.mod b/go.mod
@@ -124,6 +124,7 @@ require (
 	gitlab.com/bosi/decorder v0.4.1
 	go-simpler.org/musttag v0.9.0
 	go-simpler.org/sloglint v0.4.0
+	go.uber.org/automaxprocs v1.5.3
 	golang.org/x/exp v0.0.0-20240103183307-be819d1f06fc
 	golang.org/x/tools v0.18.0
 	gopkg.in/yaml.v3 v3.0.1

diff --git a/go.sum b/go.sum
diff --git a/pkg/commands/run.go b/pkg/commands/run.go
@@ -23,6 +23,7 @@ import (
 	"github.com/spf13/cobra"
 	"github.com/spf13/pflag"
 	"github.com/spf13/viper"
+	"go.uber.org/automaxprocs/maxprocs"
 	"golang.org/x/exp/maps"
 	"gopkg.in/yaml.v3"
 
@@ -154,7 +155,12 @@ func (c *runCommand) persistentPreRunE(cmd *cobra.Command, _ []string) error {
 		return fmt.Errorf("can't load config: %w", err)
 	}
 
-	runtime.GOMAXPROCS(c.cfg.Run.Concurrency)
+	if c.cfg.Run.Concurrency == 0 {
+		// Automatically set GOMAXPROCS to match Linux container CPU quota.
+		_, _ = maxprocs.Set(nil)
+	} else {
+		runtime.GOMAXPROCS(c.cfg.Run.Concurrency)
+	}
 
 	return c.startTracing()
 }