flamegraph: improve interactive zoom and serve svg over embedded http

1 files changed, 9 insertions, 1 deletions

diff --git a/internal/ior.go b/internal/ior.go
index ee1f988..599736d 100644
--- a/internal/ior.go
+++ b/internal/ior.go
@@ -99,6 +99,7 @@ func Run() error {
 
 	if iorFile != "" {
 		noTraceRun = true
+		var svgFile string
 		if cfg.FlamegraphTool != "" {
 			collapsed := flamegraph.NewCollapsed(iorFile, cfg.CollapsedFields, cfg.CountField)
 			collapsedFile, err := collapsed.Write(iorFile)
@@ -113,12 +114,19 @@ func Run() error {
 			if err := tool.WriteSVG(); err != nil {
 				return err
 			}
+			svgFile = tool.OutFile()
 		} else {
 			native := flamegraph.NewNativeSVG(cfg.CollapsedFields, cfg.CountField)
-			if err := native.WriteSVGFromFile(iorFile); err != nil {
+			var err error
+			svgFile, err = native.WriteSVGFromFile(iorFile)
+			if err != nil {
 				return err
 			}
 		}
+
+		if err := flamegraph.ServeSVG(svgFile); err != nil {
+			return err
+		}
 	}
 
 	if noTraceRun {