diff options
Diffstat (limited to 'internal/flamegraph/flamegraph.go')
| -rw-r--r-- | internal/flamegraph/flamegraph.go | 25 |
1 files changed, 17 insertions, 8 deletions
diff --git a/internal/flamegraph/flamegraph.go b/internal/flamegraph/flamegraph.go index d5d7e96..9e1e14b 100644 --- a/internal/flamegraph/flamegraph.go +++ b/internal/flamegraph/flamegraph.go @@ -4,6 +4,7 @@ import ( "context" "fmt" "ior/internal/event" + "ior/internal/flags" "runtime" "sync" ) @@ -11,12 +12,13 @@ import ( // TODO: Add Command in path! Make it configurable? comm/syscall/path, or path/syscall/comm, etc... // TODO: Idea, show time spent between the syscalls (off syscalls) as well, but in a different color type Flamegraph struct { + flags flags.Flags Ch chan *event.Pair Done chan struct{} workers []worker } -func New() Flamegraph { +func New(flags flags.Flags) Flamegraph { f := Flamegraph{ Ch: make(chan *event.Pair, 4096), Done: make(chan struct{}), @@ -39,17 +41,24 @@ func (f Flamegraph) Start(ctx context.Context) { for i, worker := range f.workers { fmt.Println("Starting flamegraph worker", i) - go worker.run(ctx, &wg, f.Ch) + if f.flags.FlamegraphName == "" { // Empty string means: old style collapsed + go worker.runCollapsed(ctx, &wg, f.Ch) + } else { + go worker.run(ctx, &wg, f.Ch) + } } wg.Wait() - collapsed := f.workers[0].collapsed - if len(f.workers) > 1 { - for i, c := range f.workers[1:] { - fmt.Println("Worker", i+1, "merged", collapsed.merge(c.collapsed), - "counters =>", len(collapsed), "total counters") + // COLLAPSED: Will be removed, once migrated to iorData + if f.flags.FlamegraphName == "" { // Empty string means: old style collapsed + collapsed := f.workers[0].collapsed + if len(f.workers) > 1 { + for i, c := range f.workers[1:] { + fmt.Println("Worker", i+1, "merged", collapsed.merge(c.collapsed), + "counters =>", len(collapsed), "total counters") + } } + collapsed.dump() } - collapsed.dump() }() } |