func reqInfoHandler(r *http.Request, param map[string]string, reply *web.Reply) {
stream := reply.OpenStream()
fmt.Fprintf(stream, "Method: %s\n", r.Method)
fmt.Fprintf(stream, "Protocol: %s\n", r.Proto)
fmt.Fprintf(stream, "Host: %s\n", r.Host)
fmt.Fprintf(stream, "RemoteAddr: %s\n", r.RemoteAddr)
fmt.Fprintf(stream, "RequestURI: %q\n", r.RequestURI)
fmt.Fprintf(stream, "URL: %#v\n", r.URL)
fmt.Fprintf(stream, "Body.ContentLength: %d (-1 means unknown)\n", r.ContentLength)
fmt.Fprintf(stream, "Close: %v (relevant for HTTP/1 only)\n", r.Close)
fmt.Fprintf(stream, "TLS: %#v\n", r.TLS)
fmt.Fprintf(stream, "\nHeaders:\n")
r.Header.Write(stream)
}
func (this *Proxy) DoProxy(request *http.Request, reply *web.Reply, chain web.FilterChain) {
request.URL.Scheme = this.scheme
request.URL.Host = this.host
request.Host = this.host
request.RequestURI = ""
resp, err := this.client.Do(request)
if err != nil {
reply.SetCode(500)
reply.With(fmt.Sprintf("代理错误:%s", err))
return
}
reply.SetCode(resp.StatusCode)
for k, v := range resp.Header {
reply.SetHeader(k, v[0])
}
reply.With(resp.Body)
}
func TestService(request *http.Request, param map[string]string, reply *web.Reply) {
stream := reply.OpenStream()
clientGone := stream.CloseNotify()
ticker := time.NewTicker(1 * time.Second)
defer ticker.Stop()
fmt.Fprintf(stream, "# ~1KB of junk to force browsers to start rendering immediately: \n")
io.WriteString(stream, strings.Repeat("# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n", 13))
for {
fmt.Fprintf(stream, "%v\n", time.Now())
stream.Flush()
select {
case <-ticker.C:
case <-clientGone:
log.Printf("Client %v disconnected from the clock", request.RemoteAddr)
return
}
}
}
// Index responds with the pprof-formatted profile named by the request.
// For example, "/debug/pprof/heap" serves the "heap" profile.
// Index responds to a request for "/debug/pprof/" with an HTML page
// listing the available profiles.
func Index(request *http.Request, pathFragments map[string]string, reply *web.Reply) {
if strings.HasPrefix(request.URL.Path, "/debug/pprof/") {
name := strings.TrimPrefix(request.URL.Path, "/debug/pprof/")
if name != "" {
handler(name).RequestHandler(request, pathFragments, reply)
return
}
}
r, w := io.Pipe()
go func() {
defer w.Close()
profiles := pprof.Profiles()
if err := indexTmpl.Execute(w, profiles); err != nil {
logger.Error("出现错误:%s", err)
}
}()
reply.With(r)
}
func Profile(request *http.Request, pathFragments map[string]string, reply *web.Reply) {
sec, _ := strconv.ParseInt(request.FormValue("seconds"), 10, 64)
if sec == 0 {
sec = 30
}
reply.SetContentType("application/octet-stream")
r, w := io.Pipe()
if err := pprof.StartCPUProfile(w); err != nil {
reply.SetContentType("text/plain; charset=utf-8")
reply.SetCode(http.StatusInternalServerError)
reply.With(fmt.Sprintf("Could not enable CPU profiling: %s\n", err))
return
}
go func() {
time.Sleep(time.Duration(sec) * time.Second)
pprof.StopCPUProfile()
w.Close()
}()
reply.With(r)
}
// Symbol looks up the program counters listed in the request,
// responding with a table mapping program counters to function names.
// The package initialization registers it as /debug/pprof/symbol.
func Symbol(request *http.Request, pathFragments map[string]string, reply *web.Reply) {
reply.SetContentType("text/plain; charset=utf-8")
// We have to read the whole POST body before
// writing any output. Buffer the output here.
var buf bytes.Buffer
fmt.Fprintf(&buf, "num_symbols: 1\n")
var b *bufio.Reader
if request.Method == "POST" {
b = bufio.NewReader(request.Body)
} else {
b = bufio.NewReader(strings.NewReader(request.URL.RawQuery))
}
for {
word, err := b.ReadSlice('+')
if err == nil {
word = word[0 : len(word)-1] // trim +
}
pc, _ := strconv.ParseUint(string(word), 0, 64)
if pc != 0 {
f := runtime.FuncForPC(uintptr(pc))
if f != nil {
fmt.Fprintf(&buf, "%#x %s\n", pc, f.Name())
}
}
// Wait until here to check for err; the last
// symbol will have an err because it doesn't end in +.
if err != nil {
if err != io.EOF {
fmt.Fprintf(&buf, "reading request: %v\n", err)
}
break
}
}
reply.With(string(buf.Bytes()))
}
func (name handler) RequestHandler(request *http.Request, pathFragments map[string]string, reply *web.Reply) {
reply.SetContentType("text/plain; charset=utf-8")
debug, _ := strconv.Atoi(request.FormValue("debug"))
p := pprof.Lookup(string(name))
if p == nil {
reply.SetCode(404).With(fmt.Sprintf("Unknown profile: %s\n", name))
return
}
gc, _ := strconv.Atoi(request.FormValue("gc"))
if name == "heap" && gc > 0 {
runtime.GC()
}
r, w := io.Pipe()
go func() {
defer w.Close()
p.WriteTo(w, debug)
}()
reply.With(r)
}
func Service(request *http.Request, param map[string]string, reply *web.Reply) {
reply.With("123" + param["name"])
panic(errors.New("test error"))
}
func Cmdline(request *http.Request, pathFragments map[string]string, reply *web.Reply) {
reply.SetContentType("text/plain; charset=utf-8").With(strings.Join(os.Args, "\x00"))
}
