add flag for setting runtime.GOMAXPROCS (#35)

* add flag for setting runtime.GOMAXPROCS

Author: coocood

cmd/go-tpc/main.go

@@ -6,7 +6,6 @@ import (
 	"fmt"
 	"os"
 	"os/signal"
-	"runtime"
 	"strings"
 	"syscall"
 	"time"
@@ -34,6 +33,7 @@ var (
 	silence        bool
 	summaryReport  bool
 	pprofAddr      string
+	maxProcs       int
 
 	globalDB  *sql.DB
 	globalCtx context.Context
@@ -83,8 +83,7 @@ func main() {
 		Use:   "go-tpc",
 		Short: "Benchmark database with different workloads",
 	}
-	runtime.GOMAXPROCS(runtime.NumCPU())
-
+	rootCmd.PersistentFlags().IntVar(&maxProcs, "max-procs", 0, "runtime.GOMAXPROCS")
 	rootCmd.PersistentFlags().StringVar(&pprofAddr, "pprof", "", "Address of pprof endpoint")
 	rootCmd.PersistentFlags().StringVarP(&dbName, "db", "D", "test", "Database name")
 	rootCmd.PersistentFlags().StringVarP(&host, "host", "H", "127.0.0.1", "Database host")

cmd/go-tpc/tpcc.go

@@ -6,6 +6,7 @@ import (
 	"net/http"
 	_ "net/http/pprof"
 	"os"
+	"runtime"
 
 	"github.com/pingcap/go-tpc/tpcc"
 	"github.com/spf13/cobra"
@@ -19,6 +20,7 @@ func executeTpcc(action string) {
 			http.ListenAndServe(pprofAddr, http.DefaultServeMux)
 		}()
 	}
+	runtime.GOMAXPROCS(maxProcs)
 
 	openDB()
 	defer closeDB()