diagnostics: added func to grab heap profile (#11643) (#11649)

- Added functionality to grab the heap profile by calling the
diagnostics endpoint
- Added support to pass heap profile file to diagnostics UI through
WebSocket

diagnostics/sysinfo.go

@@ -18,7 +18,9 @@ package diagnostics
 
 import (
 	"encoding/json"
+	"fmt"
 	"net/http"
+	"runtime/pprof"
 
 	diaglib "github.com/ledgerwatch/erigon-lib/diagnostics"
 	"github.com/ledgerwatch/erigon-lib/sysutils"
@@ -52,6 +54,20 @@ func SetupSysInfoAccess(metricsMux *http.ServeMux, diag *diaglib.DiagnosticClien
 		w.Header().Set("Content-Type", "application/json")
 		writeMemoryInfo(w)
 	})
+
+	metricsMux.HandleFunc("/heap-profile", func(w http.ResponseWriter, r *http.Request) {
+		w.Header().Set("Access-Control-Allow-Origin", "*")
+		w.Header().Set("Content-Type", "aplication/profile")
+		writeHeapProfile(w)
+	})
+}
+
+func writeHeapProfile(w http.ResponseWriter) {
+	err := pprof.Lookup("heap").WriteTo(w, 0)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("Failed to write profile: %v", err), http.StatusInternalServerError)
+		return
+	}
 }
 
 func writeHardwareInfo(w http.ResponseWriter, diag *diaglib.DiagnosticClient) {

turbo/app/support_cmd.go

@@ -408,6 +408,37 @@ func tunnel(ctx context.Context, cancel context.CancelFunc, sigs chan os.Signal,
 						})
 					}
 
+				case "aplication/profile":
+					if _, err := io.Copy(buffer, debugResponse.Body); err != nil {
+						return codec.WriteJSON(ctx1, &nodeResponse{
+							Id: requestId,
+							Error: &responseError{
+								Code:    http.StatusInternalServerError,
+								Message: fmt.Sprintf("Request for metrics method [%s] failed: %v", debugURL, err),
+							},
+							Last: true,
+						})
+					}
+
+					data, err := json.Marshal(struct {
+						Data []byte `json:"chunk"`
+					}{
+						Data: buffer.Bytes(),
+					})
+
+					buffer = bytes.NewBuffer(data)
+
+					if err != nil {
+						return codec.WriteJSON(ctx1, &nodeResponse{
+							Id: requestId,
+							Error: &responseError{
+								Code:    int64(http.StatusInternalServerError),
+								Message: fmt.Sprintf("Can't copy metrics response for [%s]: %s", debugURL, err),
+							},
+							Last: true,
+						})
+					}
+
 				default:
 					return codec.WriteJSON(ctx1, &nodeResponse{
 						Id: requestId,