flamegraph: add live baseline reset hotkey

1 files changed, 16 insertions, 0 deletions

diff --git a/internal/flamegraph/liveserver.go b/internal/flamegraph/liveserver.go
index 6b9a72b..5790cb0 100644
--- a/internal/flamegraph/liveserver.go
+++ b/internal/flamegraph/liveserver.go
@@ -12,6 +12,7 @@ func ServeLive(ctx context.Context, lt *LiveTrie, interval time.Duration) error
 	mux := http.NewServeMux()
 	mux.HandleFunc("/", handleLivePage())
 	mux.HandleFunc("/events", handleSSE(lt, interval))
+	mux.HandleFunc("/reset", handleReset(lt))
 	srv := &http.Server{Handler: mux}
 
 	listener, err := listenRandomPort()
@@ -96,6 +97,21 @@ func handleSSE(lt *LiveTrie, interval time.Duration) http.HandlerFunc {
 	}
 }
 
+func handleReset(lt *LiveTrie) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		if r.Method != http.MethodPost {
+			w.Header().Set("Allow", http.MethodPost)
+			http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
+			return
+		}
+
+		lt.Reset()
+		payload, _ := lt.SnapshotJSON()
+		w.Header().Set("Content-Type", "application/json")
+		_, _ = w.Write(payload)
+	}
+}
+
 func sendSnapshot(w http.ResponseWriter, flusher http.Flusher, lt *LiveTrie, lastVersion uint64) (uint64, error) {
 	payload, version := lt.SnapshotJSON()
 	if version == lastVersion {