Extend gops trace to allow duration parameter

`gops trace :port 10m` will now run a trace for 10 minutes,
as opposed to always doing 5 seconds.

For backwards compatibility both server and the client
default to 5s if nothing is specified.

Author: glibsm

agent/agent.go

@@ -269,10 +269,21 @@ func handle(conn io.ReadWriter, msg []byte) error {
 		_, err = bufio.NewReader(f).WriteTo(conn)
 		return err
 	case signal.Trace:
+		intd, err := binary.ReadVarint(bufio.NewReader(conn))
+		if err != nil {
+			return err
+		}
+		d := time.Duration(intd)
+		if intd == 0 {
+			// also default to 5 seconds if the client didn't send any duration to
+			// keep compatibility between older clients contacting a newer server.
+			d = 5 * time.Second
+		}
+
 		if err := trace.Start(conn); err != nil {
 			return err
 		}
-		time.Sleep(5 * time.Second)
+		time.Sleep(d)
 		trace.Stop()
 	case signal.SetGCPercent:
 		perc, err := binary.ReadVarint(bufio.NewReader(conn))

cmd.go

@@ -15,6 +15,7 @@ import (
 	"os/exec"
 	"strconv"
 	"strings"
+	"time"
 
 	"github.com/google/gops/internal"
 	"github.com/google/gops/signal"
@@ -71,9 +72,22 @@ func pprofCPU(addr net.TCPAddr, _ []string) error {
 	return pprof(addr, signal.CPUProfile, "cpu")
 }
 
-func trace(addr net.TCPAddr, _ []string) error {
-	fmt.Println("Tracing now, will take 5 secs...")
-	out, err := cmd(addr, signal.Trace)
+func trace(addr net.TCPAddr, params []string) error {
+	// keep the original duration of 5 seconds by default
+	duration := 5 * time.Second
+	buf := make([]byte, binary.MaxVarintLen64)
+
+	if len(params) > 0 {
+		d, err := time.ParseDuration(params[0])
+		if err != nil {
+			return fmt.Errorf("failed to parse duration: %v", params[0])
+		}
+		duration = d
+		binary.PutVarint(buf, int64(duration))
+	}
+
+	fmt.Printf("Tracing now, will take %v...\n", duration)
+	out, err := cmd(addr, signal.Trace, buf...)
 	if err != nil {
 		return err
 	}

main.go

@@ -35,7 +35,7 @@ Commands:
   memstats   Prints the allocation and garbage collection stats.
   version    Prints the Go version used to build the program.
   stats      Prints runtime stats.
-  trace      Runs the runtime tracer for 5 secs and launches "go tool trace".
+  trace      Runs the runtime tracer and launches "go tool trace".
   pprof-heap Reads the heap profile and launches "go tool pprof".
   pprof-cpu  Reads the CPU profile and launches "go tool pprof".