From 7225cb4ac541ebd4a77338b5053b9dbd177a8ecd Mon Sep 17 00:00:00 2001
From: Chris Lu <chris.lu@gmail.com>
Date: Sun, 13 Jun 2021 16:15:54 -0700
Subject: [PATCH] add block and mutex profiling

---
 weed/util/grace/pprof.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/weed/util/grace/pprof.go b/weed/util/grace/pprof.go
index 14686bfc8..0406b762c 100644
--- a/weed/util/grace/pprof.go
+++ b/weed/util/grace/pprof.go
@@ -14,9 +14,30 @@ func SetupProfiling(cpuProfile, memProfile string) {
 		if err != nil {
 			glog.Fatal(err)
 		}
+		runtime.SetBlockProfileRate(1)
+		runtime.SetMutexProfileFraction(1)
 		pprof.StartCPUProfile(f)
 		OnInterrupt(func() {
 			pprof.StopCPUProfile()
+
+			// write block pprof
+			blockF, err := os.Create(cpuProfile+".block")
+			if err != nil {
+				return
+			}
+			p := pprof.Lookup("block")
+			p.WriteTo(blockF,0)
+			blockF.Close()
+
+			// write mutex pprof
+			mutexF, err := os.Create(cpuProfile+".mutex")
+			if err != nil {
+				return
+			}
+			p = pprof.Lookup("mutex")
+			p.WriteTo(mutexF,0)
+			mutexF.Close()
+
 		})
 	}
 	if memProfile != "" {