Add tracing

Author: ancientlore

http.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"fmt"
 	"io"
 	"io/ioutil"
 	"log"
@@ -10,6 +11,7 @@ import (
 	"os"
 	"path/filepath"
 	"regexp"
+	"runtime/trace"
 	"strconv"
 	"strings"
 	"sync"
@@ -61,12 +63,14 @@ func posterThread(ctx context.Context, ch <-chan hRequest, wg *sync.WaitGroup) {
 				return
 			}
 			//log.Printf("%#v", i)
+			tr := trace.StartRegion(ctx, fmt.Sprintf("#%d %s %s", i.LoopNum, i.Method, i.URL))
 			var f io.ReadCloser
 			var err error
 			if i.Filename != "" {
 				f, err = os.Open(i.Filename)
 				if err != nil {
 					log.Printf("Unable to open %s", i.Filename)
+					tr.End()
 					continue
 				}
 			}
@@ -75,6 +79,7 @@ func posterThread(ctx context.Context, ch <-chan hRequest, wg *sync.WaitGroup) {
 				if f != nil {
 					f.Close()
 				}
+				tr.End()
 				log.Fatal(err)
 			}
 			if i.Filename != "" {
@@ -107,6 +112,7 @@ func posterThread(ctx context.Context, ch <-chan hRequest, wg *sync.WaitGroup) {
 			if err != nil {
 				kubismus.Metric("Error", 1, 0)
 				log.Print("HTTP error ", i.URL, ": ", err)
+				tr.End()
 				continue
 			}
 			kubismus.Metric("Sent", 1, float64(i.Size))
@@ -150,6 +156,7 @@ func posterThread(ctx context.Context, ch <-chan hRequest, wg *sync.WaitGroup) {
 			if outfile != nil {
 				outfile.Close()
 			}
+			tr.End()
 		case <-done:
 			return
 		}

main.go

@@ -12,6 +12,7 @@ import (
 	"os/signal"
 	"runtime"
 	"runtime/pprof"
+	"runtime/trace"
 	"strconv"
 	"strings"
 	"time"
@@ -45,6 +46,7 @@ var (
 	filesPat     string
 	cpuProfile   string
 	memProfile   string
+	traceFile    string
 	cpus         int
 	workingDir   string
 	discard      bool
@@ -84,6 +86,7 @@ func init() {
 	// profiling
 	flag.StringVar(&cpuProfile, "cpuprofile", "", "Write CPU profile to given file.")
 	flag.StringVar(&memProfile, "memprofile", "", "Write memory profile to given file.")
+	flag.StringVar(&traceFile, "trace", "", "Write a trace file.")
 
 	// runtime
 	flag.IntVar(&cpus, "cpu", 0, "Number of CPUs to use.")
@@ -197,6 +200,23 @@ func main() {
 		}()
 	}
 
+	if traceFile != "" {
+		f, err := os.Create(traceFile)
+		if err != nil {
+			log.Fatalf("failed to create trace output file: %v", err)
+		}
+		defer func() {
+			if err := f.Close(); err != nil {
+				log.Fatalf("failed to close trace file: %v", err)
+			}
+		}()
+
+		if err := trace.Start(f); err != nil {
+			log.Fatalf("failed to start trace: %v", err)
+		}
+		defer trace.Stop()
+	}
+
 	// create HTTP transport and client
 	transport = &http.Transport{Proxy: http.ProxyFromEnvironment, DisableKeepAlives: noKeepAlive, MaxIdleConnsPerHost: conns, DisableCompression: noCompress, ResponseHeaderTimeout: timeout}
 	client = &http.Client{Transport: transport, Timeout: timeout}