From 043bd4593b2b1b509bc2b8b259f8982b2b56f4fb Mon Sep 17 00:00:00 2001
From: guoguangwu <guoguangwug@gmail.com>
Date: Wed, 3 Apr 2024 13:51:10 +0800
Subject: [PATCH] fix: close cpu profile

Signed-off-by: guoguangwu <guoguangwug@gmail.com>
---
 session/session.go | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/session/session.go b/session/session.go
index cab5eb06..0e9b1008 100644
--- a/session/session.go
+++ b/session/session.go
@@ -120,9 +120,12 @@ func New() (*Session, error) {
 	}
 
 	if *s.Options.CpuProfile != "" {
-		if f, err := os.Create(*s.Options.CpuProfile); err != nil {
+		f, err := os.Create(*s.Options.CpuProfile)
+		if err != nil {
 			return nil, err
-		} else if err := pprof.StartCPUProfile(f); err != nil {
+		}
+		defer  f.Close()
+		if err := pprof.StartCPUProfile(f); err != nil {
 			return nil, err
 		}
 	}