Katyusha's blog

1
// Package report 将多场景的诊断结果汇总为一份结构化 Markdown 报告
2
//（赛题鼓励项：自动生成诊断报告 / 多维关联）。
3
package report
4

5
import (
6
  "fmt"
7
  "io"
8
  "os"
9
  "time"
10

11
  "github.com/os2026/ebpf-rca/internal/output"
12
  "github.com/os2026/ebpf-rca/internal/schema"
13
)
14

15
// Aggregator 收集去重后的诊断结果。
16
type Aggregator struct {
17
  reports []schema.AnomalyReport
18
  seen    map[string]bool
19
}
20

21
// New 构造聚合器。
22
func New() *Aggregator {
23
  return &Aggregator{seen: make(map[string]bool)}
24
}
25

26
// Add 添加一条结果（按异常类型+关联对象去重）。
27
func (a *Aggregator) Add(r schema.AnomalyReport) {
28
  k := fmt.Sprintf("%s|%d|%s|%s|%v",
29
    r.AnomalyType, r.RelatedObject.Pid, r.RelatedObject.Comm,
30
    r.RelatedObject.Device, r.KeyMetrics["syscall"])
31
  if a.seen[k] {
32
    return
33
  }
34
  a.seen[k] = true
35
  a.reports = append(a.reports, r)
36
}
37

38
// Count 返回已收集的结果数。
39
func (a *Aggregator) Count() int { return len(a.reports) }
40

41
// Render 输出一份汇总 Markdown 报告：概要表 + 各项详细诊断。
42
func (a *Aggregator) Render(w io.Writer, dur time.Duration) error {
43
  host, _ := os.Hostname()
44
  fmt.Fprintf(w, "# 系统异常诊断报告\n\n")
45
  fmt.Fprintf(w, "- 主机：%s\n", host)
46
  fmt.Fprintf(w, "- 采集时长：%s\n", dur.Round(time.Second))
47
  fmt.Fprintf(w, "- 生成时间：%s\n", time.Now().Format(time.RFC3339))
48
  fmt.Fprintf(w, "- 发现异常：%d 项\n\n", len(a.reports))
49

50
  if len(a.reports) == 0 {
51
    fmt.Fprintf(w, "未发现异常。系统在采集窗口内各项指标均低于告警阈值。\n")
52
    return nil
53
  }
54

55
  fmt.Fprintf(w, "## 概要\n\n")
56
  fmt.Fprintf(w, "| # | 异常类型 | 关联对象 | 疑似根因 | 置信度 |\n")
57
  fmt.Fprintf(w, "|---|----------|----------|----------|--------|\n")
58
  for i, r := range a.reports {
59
    fmt.Fprintf(w, "| %d | %s | %s | %s | %.2f |\n",
60
      i+1, r.AnomalyType, objStr(r.RelatedObject), r.SuspectedRootCause, r.Confidence)
61
  }
62
  fmt.Fprintf(w, "\n## 详细诊断\n\n")
63
  for i, r := range a.reports {
64
    fmt.Fprintf(w, "### %d. %s\n\n", i+1, r.AnomalyType)
65
    if err := output.Write(w, r, "md"); err != nil {
66
      return err
67
    }
68
  }
69
  return nil
70
}
71

72
func objStr(o schema.RelatedObject) string {
73
  if o.Device != "" {
74
    return "设备 " + o.Device
75
  }
76
  if o.Comm != "" {
77
    return fmt.Sprintf("%s(pid=%d)", o.Comm, o.Pid)
78
  }
79
  return fmt.Sprintf("pid=%d", o.Pid)
80
}