1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
package flamegraph
import (
"fmt"
"io"
"os"
"strings"
"iter"
)
type NativeSVG struct {
fields []string
countField string
config SVGConfig
}
func NewNativeSVG(fields []string, countField string) NativeSVG {
return NativeSVG{
fields: fields,
countField: countField,
config: defaultSVGConfig(),
}
}
func (n NativeSVG) WriteSVGFromFile(iorDataFile string) (string, error) {
outFile := fmt.Sprintf("%s.%s-by-%s.svg",
strings.TrimSuffix(iorDataFile, ".ior.zst"),
strings.Join(n.fields, ":"),
n.countField,
)
iod, err := newIorDataFromFile(iorDataFile)
if err != nil {
return outFile, fmt.Errorf("read ior data: %w", err)
}
fd, err := os.Create(outFile)
if err != nil {
return outFile, fmt.Errorf("create output %s: %w", outFile, err)
}
defer fd.Close()
if err := n.WriteSVGFromIter(iod.iter(), fd); err != nil {
return outFile, err
}
return outFile, nil
}
func (n NativeSVG) WriteSVGFromIter(records iter.Seq[IterRecord], w io.Writer) error {
tr := newTrie()
for record := range records {
frames, err := n.recordFrames(record)
if err != nil {
return err
}
tr.add(frames, record.Cnt.ValueByName(n.countField))
}
tr.computeTotals()
return WriteSVG(w, tr, n.config)
}
func (n NativeSVG) recordFrames(record IterRecord) ([]string, error) {
var frames []string
for _, fieldName := range n.fields {
value, err := record.StringByName(fieldName)
if err != nil {
return nil, fmt.Errorf("field %s: %w", fieldName, err)
}
for _, part := range strings.Split(value, ";") {
if part != "" {
frames = append(frames, part)
}
}
}
return frames, nil
}
|
