lbsadmin/Beego
1
0
Fork 0
mirror of https://github.com/astaxie/beego.git synced 2024-11-22 08:50:56 +00:00
Code Issues Releases Activity

fix the responseWriter

Browse Source
This commit is contained in:
astaxie 2014-08-15 15:24:46 +08:00
parent b611b9dab6
commit c1234e7c6d
1 changed files with 6 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
toolbox/profile.go
View File

@ -13,7 +13,6 @@ import (
"fmt"
"io"
"log"
"net/http"
"os"
"path"
"runtime"
@ -46,7 +45,7 @@ func ProcessInput(input string, w io.Writer) {
p := pprof.Lookup("block")
p.WriteTo(w, 2)
case "get cpuprof":
GetCPUProfile(w.(http.ResponseWriter))
GetCPUProfile(w)
case "get memprof":
MemProf(w)
case "gc summary":
@ -71,24 +70,21 @@ func MemProf(w io.Writer) {
}
// start cpu profile monitor
func GetCPUProfile(rw http.ResponseWriter) {
func GetCPUProfile(w io.Writer) {
sec := 30
rw.Header().Set("Content-Type", "application/octet-stream")
filename := "cpu-" + strconv.Itoa(pid) + ".pprof"
f, err := os.Create(filename)
if err != nil {
rw.Header().Set("Content-Type", "text/plain; charset=utf-8")
rw.WriteHeader(http.StatusInternalServerError)
fmt.Fprintf(rw, "Could not enable CPU profiling: %s\n", err)
fmt.Fprintf(w, "Could not enable CPU profiling: %s\n", err)
log.Fatal("record cpu profile failed: ", err)
}
fmt.Fprintf(rw, "start cpu profileing\n")
pprof.StartCPUProfile(f)
time.Sleep(time.Duration(sec) * time.Second)
pprof.StopCPUProfile()
fmt.Fprintf(rw, "create cpu profile %s \n", filename)
fmt.Fprintf(w, "create cpu profile %s \n", filename)
_, fl := path.Split(os.Args[0])
fmt.Fprintf(rw, "Now you can use this to check it: go tool pprof %s %s\n", fl, filename)
fmt.Fprintf(w, "Now you can use this to check it: go tool pprof %s %s\n", fl, filename)
}
// print gc information to io.Writer