Merge pull request #17729 from testwill/close_profile

fix: close profile in the mvccPutFunc func
etcd-io · Apr 11, 2024 · fa08f01 · fa08f01
2 parents 8d3abed + 90152fe
commit fa08f01
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/benchmark/cmd/mvcc-put.go b/tools/benchmark/cmd/mvcc-put.go
@@ -77,7 +77,7 @@ func mvccPutFunc(_ *cobra.Command, _ []string) {
 			fmt.Fprintln(os.Stderr, "Failed to create a file for storing cpu profile result: ", err)
 			os.Exit(1)
 		}
-
+		defer f.Close()
 		err = pprof.StartCPUProfile(f)
 		if err != nil {
 			fmt.Fprintln(os.Stderr, "Failed to start cpu profile: ", err)
@@ -92,7 +92,7 @@ func mvccPutFunc(_ *cobra.Command, _ []string) {
 			fmt.Fprintln(os.Stderr, "Failed to create a file for storing heap profile result: ", err)
 			os.Exit(1)
 		}
-
+		defer f.Close()
 		defer func() {
 			err := pprof.WriteHeapProfile(f)
 			if err != nil {