pprof subcommand

2018-09-07 00:04:03 -07:00
parent bf5099baac
commit 0c2ac3a168
2 changed files with 71 additions and 0 deletions
@@ -0,0 +1,35 @@
+package client
+
+import (
+	"github.com/zrepl/zrepl/config"
+	"github.com/zrepl/zrepl/daemon"
+	"log"
+	"os"
+)
+
+type PProfArgs struct {
+	daemon.PprofServerControlMsg
+}
+
+func RunPProf(conf *config.Config, args PProfArgs) {
+	log := log.New(os.Stderr, "", 0)
+
+	die := func() {
+		log.Printf("exiting after error")
+		os.Exit(1)
+	}
+
+	log.Printf("connecting to zrepl daemon")
+
+	httpc, err := controlHttpClient(conf.Global.Control.SockPath)
+	if err != nil {
+		log.Printf("error creating http client: %s", err)
+		die()
+	}
+	err = jsonRequestResponse(httpc, daemon.ControlJobEndpointPProf, args.PprofServerControlMsg, struct{}{})
+	if err != nil {
+		log.Printf("error sending control message: %s", err)
+		die()
+	}
+	log.Printf("finished")
+}