fix: mem profiling (#368)

bietkul · web-flow · commit d9503e013352 · 2023-01-30T14:51:00.000+07:00
* fix: mem profiling

* fix: minor fixes
diff --git a/main.go b/main.go
@@ -212,13 +212,13 @@ func init() {
 }
 
 func main() {
-	// add cpu profilling
+	// add cpu profiling
 	if cpuprofile {
-		defer profile.Start(profile.NoShutdownHook).Stop()
+		defer profile.Start(profile.CPUProfile, profile.NoShutdownHook).Stop()
 	}
-	// add mem profilling
+	// add mem profiling
 	if memprofile {
-		defer profile.Start(profile.MemProfile).Stop()
+		defer profile.Start(profile.MemProfile, profile.NoShutdownHook).Stop()
 	}
 
 	log.SetReportCaller(true)

Original file line number	Diff line number	Diff line change
`@@ -212,13 +212,13 @@ func init() {`
`212`	`212`	`}`
`213`	`213`
`214`	`214`	`func main() {`
`215`		`- // add cpu profilling`
	`215`	`+ // add cpu profiling`
`216`	`216`	`if cpuprofile {`
`217`		`- defer profile.Start(profile.NoShutdownHook).Stop()`
	`217`	`+ defer profile.Start(profile.CPUProfile, profile.NoShutdownHook).Stop()`
`218`	`218`	`}`
`219`		`- // add mem profilling`
	`219`	`+ // add mem profiling`
`220`	`220`	`if memprofile {`
`221`		`- defer profile.Start(profile.MemProfile).Stop()`
	`221`	`+ defer profile.Start(profile.MemProfile, profile.NoShutdownHook).Stop()`
`222`	`222`	`}`
`223`	`223`
`224`	`224`	`log.SetReportCaller(true)`