Katyusha's blog

1
package detector
2

3
import (
4
  "time"
5

6
  "github.com/os2026/ebpf-rca/internal/collector"
7
)
8

9
// MemSignal 表示一次已确认的内存抖动 / 回收压力异常。
10
type MemSignal struct {
11
  Snapshot    collector.MemSnapshot
12
  Culprit     collector.MemProc // 主要肇事进程（直接回收最多者）
13
  WindowStart time.Time
14
  WindowEnd   time.Time
15
}
16

17
// MemDetector 检测"可用内存持续低于阈值"的系统级内存压力。
18
type MemDetector struct {
19
  AvailPctFloor float64 // 可用内存占比下限(%)
20
  SustainTicks  int
21
  count         int
22
  firstSeen     time.Time
23
  fired         bool
24
}
25

26
// NewMemDetector 构造检测器（阈值单位为百分比）。
27
func NewMemDetector(availPctFloor float64, sustain int) *MemDetector {
28
  if sustain < 1 {
29
    sustain = 1
30
  }
31
  return &MemDetector{AvailPctFloor: availPctFloor, SustainTicks: sustain}
32
}
33

34
// Detect 处理一个窗口的内存快照，返回新触发的异常信号（0 或 1 个）。
35
func (d *MemDetector) Detect(snap collector.MemSnapshot, now time.Time) []MemSignal {
36
  if snap.MemAvailablePct >= d.AvailPctFloor || snap.MemTotalKB == 0 {
37
    d.count = 0
38
    d.fired = false
39
    return nil
40
  }
41
  if d.count == 0 {
42
    d.firstSeen = now
43
  }
44
  d.count++
45
  if d.count >= d.SustainTicks && !d.fired {
46
    d.fired = true
47
    return []MemSignal{{
48
      Snapshot:    snap,
49
      Culprit:     pickCulprit(snap),
50
      WindowStart: d.firstSeen,
51
      WindowEnd:   now,
52
    }}
53
  }
54
  return nil
55
}
56

57
// pickCulprit 选取压力贡献最大的进程：优先直接回收次数，其次 major fault。
58
func pickCulprit(snap collector.MemSnapshot) collector.MemProc {
59
  var best collector.MemProc
60
  for _, p := range snap.Procs {
61
    if p.DirectReclaimCount > best.DirectReclaimCount ||
62
      (p.DirectReclaimCount == best.DirectReclaimCount && p.MajFlt > best.MajFlt) {
63
      best = p
64
    }
65
  }
66
  return best
67
}