More perf tweaks

1 files changed, 10 insertions, 12 deletions

diff --git a/json/profile/json.go b/json/profile/json.go
index a03d047..0579fce 100644
--- a/json/profile/json.go
+++ b/json/profile/json.go
@@ -34,7 +34,16 @@ func main() {
 	max := 1000000
 	if *memprofile != "" {
 		runtime.MemProfileRate = 1
-		max = 10000
+		max = 100000
+		defer func() {
+			f, err := os.Create(*memprofile)
+			if err != nil {
+				log.Fatal(err)
+			}
+
+			pprof.WriteHeapProfile(f)
+			f.Close()
+		}()
 	}
 
 	for i := 0; i < max; i++ {
@@ -43,15 +52,4 @@ func main() {
 			panic(err)
 		}
 	}
-
-	if *memprofile != "" {
-		f, err := os.Create(*memprofile)
-		if err != nil {
-			log.Fatal(err)
-		}
-
-		pprof.WriteHeapProfile(f)
-		f.Close()
-		return
-	}
 }