opensource/filestash
1
0
Fork 0
mirror of https://github.com/mickael-kerjean/filestash.git synced 2025-11-03 04:50:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

improve (profiling): create a flag to enable profiling

Browse Source
This commit is contained in:
Mickael KERJEAN
2018-11-19 14:27:41 +11:00
parent edd6ba9d12
commit 0ed846f5f3
2 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
server/main.go
View File

@ -7,6 +7,8 @@ import (
. "github.com/mickael-kerjean/nuage/server/middleware"
_ "github.com/mickael-kerjean/nuage/server/plugin"
"net/http"
"net/http/pprof"
"os"
"strconv"
)
@ -20,6 +22,20 @@ func main() {
func Init(a *App) {
r := mux.NewRouter()
// Profiling - handy to indentify nasty leaks and/or bugs!
if os.Getenv("DEBUG") == "true" {
r.HandleFunc("/debug/pprof/", pprof.Index)
r.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
r.HandleFunc("/debug/pprof/profile", pprof.Profile)
r.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
r.HandleFunc("/debug/pprof/trace", pprof.Trace)
r.Handle("/debug/pprof/goroutine", pprof.Handler("goroutine"))
r.Handle("/debug/pprof/heap", pprof.Handler("heap"))
r.Handle("/debug/pprof/threadcreate", pprof.Handler("threadcreate"))
r.Handle("/debug/pprof/block", pprof.Handler("block"))
r.Handle("/debug/pprof/heap", pprof.Handler("heap"))
}
// API
session := r.PathPrefix("/api/session").Subrouter()
session.HandleFunc("", APIHandler(SessionGet, *a)).Methods("GET")
@ -44,7 +60,7 @@ func Init(a *App) {
// WEBDAV
r.PathPrefix("/s/{share}").Handler(CtxInjector(WebdavHandler, *a))
// APP
r.HandleFunc("/api/config", CtxInjector(ConfigHandler, *a)).Methods("GET")
r.PathPrefix("/assets").Handler(StaticHandler(FILE_ASSETS, *a)).Methods("GET")