Пример #1
0
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)
}
Пример #2
0
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)
}
Пример #3
0
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)
}