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
78
79
80
81
82
83
84
85
86
87
88
|
// Copyright 2025 syzkaller project authors. All rights reserved.
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
package report
import (
"encoding/json"
"errors"
"fmt"
"maps"
"os"
)
func AddTitleStat(file string, reps []*Report) error {
var titles []string
for _, rep := range reps {
titles = append(titles, rep.Title)
}
stat, err := ReadStatFile(file)
if err != nil {
return fmt.Errorf("report.ReadStatFile: %w", err)
}
stat.add(titles)
if err := writeStatFile(file, stat); err != nil {
return fmt.Errorf("writeStatFile: %w", err)
}
return nil
}
func ReadStatFile(file string) (*titleStat, error) {
stat := &titleStat{}
if _, err := os.Stat(file); errors.Is(err, os.ErrNotExist) {
return stat, nil
}
data, err := os.ReadFile(file)
if err != nil {
return nil, err
}
if len(data) == 0 {
return stat, nil
}
if err := json.Unmarshal(data, stat); err != nil {
return nil, err
}
return stat, nil
}
func writeStatFile(file string, stat *titleStat) error {
data, err := json.MarshalIndent(stat, "", "\t")
if err != nil {
return err
}
if err := os.WriteFile(file, data, 0644); err != nil {
return err
}
return nil
}
type titleStatNodes map[string]*titleStat
type titleStat struct {
Count int
Nodes titleStatNodes
}
func (ts *titleStat) add(reps []string) {
if len(reps) == 0 {
return
}
if ts.Nodes == nil {
ts.Nodes = make(titleStatNodes)
}
if ts.Nodes[reps[0]] == nil {
ts.Nodes[reps[0]] = &titleStat{}
}
ts.Nodes[reps[0]].Count++
ts.Nodes[reps[0]].add(reps[1:])
}
func (ts *titleStat) visit(cb func(int, ...string), titles ...string) {
if len(ts.Nodes) == 0 {
cb(ts.Count, titles...)
return
}
for title := range maps.Keys(ts.Nodes) {
ts.Nodes[title].visit(cb, append(titles, title)...)
}
}
|
