lib9p

commit a5853aaca3aaa3749f406e5c8562b6edf70cd1ff
parent 0a44d21271d4d7ad7064bf30ebf779478f90e7df
Author: Matsuda Kenji <info@mtkn.jp>
Date:   Fri, 15 Dec 2023 11:41:05 +0900

add mem.prof in defer

Diffstat:
M
cmd/diskfs/main.go
 | 
13
+++++++++++++

1 file changed, 13 insertions(+), 0 deletions(-)
diff --git a/cmd/diskfs/main.go b/cmd/diskfs/main.go
@@ -13,6 +13,8 @@ import (
 	_ "net/http/pprof"
 	"os"
 	"os/signal"
+	"runtime"
+	"runtime/pprof"
 	"sync"
 
 	"git.mtkn.jp/lib9p"
@@ -24,6 +26,17 @@ var aFlag = flag.String("a", "127.0.0.1", "Address the server listens to.")
 var pFlag = flag.Int("p", 5640, "Port number the server listens to.")
 
 func main() {
+	f, err := os.Create("mem.prof")
+	if err != nil {
+		log.Fatal("mem.prof: %v", err)
+	}
+	defer func() {
+		runtime.GC()
+		if err := pprof.WriteHeapProfile(f); err != nil {
+			log.Fatal("write mem.prof: %v", err)
+		}
+		f.Close()
+	}()
 	go func() {
 		log.Println(http.ListenAndServe("localhost:8000", nil))
 	}()