senaduka/old-vegeta
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Parameterize number of CPUs to be used

Browse Source
This commit is contained in:
Tomás Senart
2013-09-08 21:02:04 +01:00
parent 3eab2de2c1
commit b53c5f7a72
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
main.go
View File

@@ -11,10 +11,6 @@ import (
"time" "time"
) )
func init() {
runtime.GOMAXPROCS(runtime.NumCPU())
}
func main() { func main() {
var ( var (
rate = flag.Uint64("rate", 50, "Requests per second") rate = flag.Uint64("rate", 50, "Requests per second")
@@ -23,6 +19,7 @@ func main() {
duration = flag.Duration("duration", 10*time.Second, "Duration of the test") duration = flag.Duration("duration", 10*time.Second, "Duration of the test")
reporter = flag.String("reporter", "text", "Reporter to use [text, plot:timings]") reporter = flag.String("reporter", "text", "Reporter to use [text, plot:timings]")
output = flag.String("output", "stdout", "Reporter output file") output = flag.String("output", "stdout", "Reporter output file")
cpus = flag.Int("cpus", runtime.NumCPU(), "Number of CPUs to use")
) )
flag.Parse() flag.Parse()
@@ -31,6 +28,8 @@ func main() {
return return
} }
runtime.GOMAXPROCS(*cpus)
if err := run(*rate, *duration, *targetsf, *ordering, *reporter, *output); err != nil { if err := run(*rate, *duration, *targetsf, *ordering, *reporter, *output); err != nil {
log.Fatal(err) log.Fatal(err)
} }