MyFavMirrors/minio
1
0
Fork 0
mirror of https://github.com/minio/minio.git synced 2025-11-07 12:52:58 -05:00
Code Issues Packages Projects Releases Wiki Activity

profiler: Fix how we generate profiles. (#2596)

Browse Source 
Fixes #2594
This commit is contained in:
Harshavardhana
2016-09-01 03:50:31 -07:00
committed by GitHub
parent 790ad68d4d
commit 511a8cbe04
3 changed files with 34 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
cmd/main.go
View File

@@ -140,6 +140,11 @@ func checkMainSyntax(c *cli.Context) {
}
}
// Global profiler to be cleanly set and saved inside graceful shutdown.
var globalProfiler interface {
Stop()
}
// Main - main for minio server.
func Main() {
app := registerApp()
@@ -181,11 +186,11 @@ func Main() {
// Enable profiler if ``MINIO_PROFILER`` is set. Supported options are [cpu, mem, block].
switch os.Getenv("MINIO_PROFILER") {
case "cpu":
defer profile.Start(profile.CPUProfile, profile.ProfilePath(profileDir)).Stop()
globalProfiler = profile.Start(profile.CPUProfile, profile.ProfilePath(profileDir))
case "mem":
defer profile.Start(profile.MemProfile, profile.ProfilePath(profileDir)).Stop()
globalProfiler = profile.Start(profile.MemProfile, profile.ProfilePath(profileDir))
case "block":
defer profile.Start(profile.BlockProfile, profile.ProfilePath(profileDir)).Stop()
globalProfiler = profile.Start(profile.BlockProfile, profile.ProfilePath(profileDir))
}
// Run the app - exit on error.

10
cmd/utils.go
View File

@@ -201,8 +201,18 @@ func startMonitorShutdownSignal(onExitFn onExitFunc) error {
if err != nil {
errorIf(errors.New("Unable to reboot."), err.Error())
}
// Successfully forked.
onExitFn(int(exitSuccess))
}
// Enable profiler if ``MINIO_PROFILER`` is set.
switch os.Getenv("MINIO_PROFILER") {
case "cpu", "mem", "block":
// Stop any running profiler.
globalProfiler.Stop()
}
// Exit as success if no errors.
onExitFn(int(exitSuccess))
}
}