noth/surrealpatch
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add debugging option to codebase

Browse source
This commit is contained in:
Tobie Morgan Hitchcock 2016-09-06 12:52:11 +01:00
parent f53a30130b
commit 94f5a72e26
1 changed files with 20 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
main.go
View file

@ -14,8 +14,27 @@
package main
import "github.com/abcum/surreal/cli"
import (
"os"
"github.com/pkg/profile"
"github.com/abcum/surreal/cli"
)
func main() {
cli.Init()
switch os.Getenv("DEBUG") {
case "cpu":
defer profile.Start(profile.CPUProfile, profile.ProfilePath("."), profile.NoShutdownHook).Stop()
case "mem":
defer profile.Start(profile.MemProfile, profile.ProfilePath("."), profile.NoShutdownHook).Stop()
case "block":
defer profile.Start(profile.BlockProfile, profile.ProfilePath("."), profile.NoShutdownHook).Stop()
case "trace":
defer profile.Start(profile.TraceProfile, profile.ProfilePath("."), profile.NoShutdownHook).Stop()
}
cli.Init()
}