Changes in benchmark/readyQ/bench.go [2c7eee0:f4f79dd]

File:

• benchmark/readyQ/bench.go (modified) (3 diffs)

benchmark/readyQ/bench.go

@@ -5 +5 @@
         "flag"
         "fmt"
+        "log"
         "os"
         "runtime"
+        "runtime/pprof"
         "sync/atomic"
         "time"
@@ -43 +45 @@
 }
 
-func bench_init() {
+func bench_init() func() {
         nprocsOpt := flag.Int("p", 1, "The number of processors")
         nthreadsOpt := flag.Int("t", 1, "The number of threads")
         durationOpt := flag.Float64("d", 0, "Duration of the experiment in seconds")
         stopOpt := flag.Uint64("i", 0, "Duration of the experiment in iterations")
+        cpuprofile := flag.String("cpuprofile", "", "write cpu profile to file")
 
         flag.Parse()
@@ -72 +75 @@
 
         runtime.GOMAXPROCS(nprocs)
+
+        if (*cpuprofile) != "" {
+                f, err := os.Create(*cpuprofile)
+                if err != nil {
+                    log.Fatal(err)
+                }
+                pprof.StartCPUProfile(f)
+        }
+
+        return func() {
+                if (*cpuprofile) != "" {
+                        pprof.StopCPUProfile()
+                }
+        }
 }