1
0
mirror of https://github.com/google/gops.git synced 2024-11-27 08:31:17 +02:00
gops/cmd.go
2016-11-28 10:45:02 -08:00

116 lines
2.2 KiB
Go

package main
import (
"errors"
"fmt"
"io/ioutil"
"net"
"os"
"os/exec"
"github.com/google/gops/internal"
"github.com/google/gops/signal"
ps "github.com/keybase/go-ps"
)
var cmds = map[string](func(pid int) error){
"stack": stackTrace,
"gc": gc,
"memstats": memStats,
"version": version,
"pprof-heap": pprofHeap,
"pprof-cpu": pprofCPU,
"vitals": vitals,
}
func stackTrace(pid int) error {
return cmdWithPrint(pid, signal.StackTrace)
}
func gc(pid int) error {
_, err := cmd(pid, signal.GC)
return err
}
func memStats(pid int) error {
return cmdWithPrint(pid, signal.MemStats)
}
func version(pid int) error {
return cmdWithPrint(pid, signal.Version)
}
func pprofHeap(pid int) error {
return pprof(pid, signal.HeapProfile)
}
func pprofCPU(pid int) error {
fmt.Println("Profiling CPU now, will take 30 secs...")
return pprof(pid, signal.CPUProfile)
}
func pprof(pid int, p byte) error {
out, err := cmd(pid, p)
if err != nil {
return err
}
if out == "" {
return errors.New("failed to read the profile")
}
tmpfile, err := ioutil.TempFile("", "heap-profile")
if err != nil {
return err
}
defer os.Remove(tmpfile.Name())
if err := ioutil.WriteFile(tmpfile.Name(), []byte(out), 0); err != nil {
return err
}
process, err := ps.FindProcess(pid)
if err != nil {
// TODO(jbd): add context to the error
return err
}
binary, err := process.Path()
if err != nil {
return fmt.Errorf("cannot the binary for the PID: %v", err)
}
cmd := exec.Command("go", "tool", "pprof", binary, tmpfile.Name())
cmd.Env = os.Environ()
cmd.Stdin = os.Stdin
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
return cmd.Run()
}
func vitals(pid int) error {
return cmdWithPrint(pid, signal.Vitals)
}
func cmdWithPrint(pid int, c byte) error {
out, err := cmd(pid, c)
if err != nil {
return err
}
fmt.Printf(out)
return nil
}
func cmd(pid int, c byte) (string, error) {
port, err := internal.GetPort(pid)
if err != nil {
return "", err
}
conn, err := net.Dial("tcp", "127.0.0.1:"+port)
if err != nil {
return "", err
}
if _, err := conn.Write([]byte{c}); err != nil {
return "", err
}
all, err := ioutil.ReadAll(conn)
if err != nil {
return "", err
}
return string(all), nil
}