Merge pull request studygolang#53 from meission/master

polaris1119 · web-flow · commit 71486a20458c · 2018-01-28T09:31:23.000+08:00
update add pprof for studygolang
diff --git a/config/env.sample.ini b/config/env.sample.ini
@@ -1,6 +1,7 @@
 [global]
 env = dev
 
+pprof = 127.0.0.1:8096
 log_level = DEBUG
 domain = xxx
 
diff --git a/src/server/studygolang/main.go b/src/server/studygolang/main.go
@@ -54,6 +54,8 @@ func main() {
 	go logic.Book.ClearRedisUser()
 
 	go ServeBackGround()
+	// go pprof
+	Pprof(ConfigFile.MustValue("global", "pprof", "127.0.0.1:8096"))
 
 	e := echo.New()
 
diff --git a/src/server/studygolang/pprof.go b/src/server/studygolang/pprof.go
@@ -0,0 +1,26 @@
+// Copyright 2018 The StudyGolang Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+// http://studygolang.com
+// Author: meission	meission@aliyun.com
+
+package main
+
+import (
+	"net/http"
+	"net/http/pprof"
+)
+
+// Pprof start http pprof.
+func Pprof(addr string) {
+	ps := http.NewServeMux()
+	ps.HandleFunc("/debug/pprof/", pprof.Index)
+	ps.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
+	ps.HandleFunc("/debug/pprof/profile", pprof.Profile)
+	ps.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
+	go func() {
+		if err := http.ListenAndServe(addr, ps); err != nil {
+			panic(err)
+		}
+	}()
+}
