MCorange/wings
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add support for both memory and cpu profiling

Browse Source
This commit is contained in:
Dane Everitt 2020-11-08 11:59:04 -08:00
parent 4b17ac4f1c
commit 536f00a5e5
No known key found for this signature in database
GPG Key ID: EEA66103B3D71F53
2 changed files with 23 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

@ -47,3 +47,5 @@ test_*/
debug debug
data/.states.json data/.states.json
.DS_Store .DS_Store
*.pprof
*.pdf

30
cmd/root.go
View File

@ -30,12 +30,15 @@ import (
"github.com/spf13/cobra" "github.com/spf13/cobra"
) )
var configPath = config.DefaultLocation var (
var debug = false configPath = config.DefaultLocation
var shouldRunProfiler = false debug = false
var useAutomaticTls = false runCPUProfile = false
var tlsHostname = "" runMemProfile = false
var showVersion = false useAutomaticTls = false
tlsHostname = ""
showVersion = false
)
var root = &cobra.Command{ var root = &cobra.Command{
Use: "wings", Use: "wings",
@ -54,7 +57,8 @@ func init() {
root.PersistentFlags().BoolVar(&showVersion, "version", false, "show the version and exit") root.PersistentFlags().BoolVar(&showVersion, "version", false, "show the version and exit")
root.PersistentFlags().StringVar(&configPath, "config", config.DefaultLocation, "set the location for the configuration file") root.PersistentFlags().StringVar(&configPath, "config", config.DefaultLocation, "set the location for the configuration file")
root.PersistentFlags().BoolVar(&debug, "debug", false, "pass in order to run wings in debug mode") root.PersistentFlags().BoolVar(&debug, "debug", false, "pass in order to run wings in debug mode")
root.PersistentFlags().BoolVar(&shouldRunProfiler, "profile", false, "pass in order to profile wings") root.PersistentFlags().BoolVar(&runCPUProfile, "profile-cpu", false, "pass in order to profile wings CPU usage")
root.PersistentFlags().BoolVar(&runMemProfile, "profile-mem", false, "pass in order to profile wings memory usage")
root.PersistentFlags().BoolVar(&useAutomaticTls, "auto-tls", false, "pass in order to have wings generate and manage it's own SSL certificates using Let's Encrypt") root.PersistentFlags().BoolVar(&useAutomaticTls, "auto-tls", false, "pass in order to have wings generate and manage it's own SSL certificates using Let's Encrypt")
root.PersistentFlags().StringVar(&tlsHostname, "tls-hostname", "", "required with --auto-tls, the FQDN for the generated SSL certificate") root.PersistentFlags().StringVar(&tlsHostname, "tls-hostname", "", "required with --auto-tls, the FQDN for the generated SSL certificate")
@ -89,8 +93,16 @@ func rootCmdRun(*cobra.Command, []string) {
os.Exit(0) os.Exit(0)
} }
if shouldRunProfiler { if runCPUProfile {
defer profile.Start().Stop() defer profile.Start(func(p *profile.Profile) {
profile.CPUProfile(p)
}).Stop()
}
if runMemProfile {
defer profile.Start(func(p *profile.Profile) {
profile.MemProfile(p)
}).Stop()
} }
// Only attempt configuration file relocation if a custom location has not // Only attempt configuration file relocation if a custom location has not