fix: close profile in the mvccPutFunc func

Signed-off-by: guoguangwu <guoguangwug@gmail.com>
etcd-io · Apr 8, 2024 · 90152fe · 90152fe
1 parent 84e67ff
commit 90152fe
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 {