override maxprocs, defaulting to 1

Change-Id: I0ae4b788ba5a08ff1b6bf3efb4cbf79371f842e3

Author: mattcary

cmd/gce-pd-csi-driver/main.go

@@ -20,6 +20,7 @@ import (
 	"flag"
 	"math/rand"
 	"os"
+	"runtime"
 	"time"
 
 	"k8s.io/klog"
@@ -40,6 +41,7 @@ var (
 	httpEndpoint         = flag.String("http-endpoint", "", "The TCP network address where the prometheus metrics endpoint will listen (example: `:8080`). The default is empty string, which means metrics endpoint is disabled.")
 	metricsPath          = flag.String("metrics-path", "/metrics", "The HTTP path where prometheus metrics will be exposed. Default is `/metrics`.")
 	extraVolumeLabelsStr = flag.String("extra-labels", "", "Extra labels to attach to each PD created. It is a comma separated list of key value pairs like '<key1>=<value1>,<key2>=<value2>'. See https://cloud.google.com/compute/docs/labeling-resources for details")
+	maxprocs             = flag.Int("maxprocs", 1, "GOMAXPROCS override")
 	version              string
 )
 
@@ -67,6 +69,9 @@ func main() {
 func handle() {
 	var err error
 
+	runtime.GOMAXPROCS(*maxprocs)
+	klog.Infof("Sys info: NumCPU: %v MAXPROC: %v", runtime.NumCPU(), runtime.GOMAXPROCS(0))
+
 	if version == "" {
 		klog.Fatalf("version must be set at compile time")
 	}