statsengine: add top-n snapshot helpers for overview

1 files changed, 48 insertions, 0 deletions

diff --git a/internal/statsengine/snapshot.go b/internal/statsengine/snapshot.go
index 6e38739..1cc2118 100644
--- a/internal/statsengine/snapshot.go
+++ b/internal/statsengine/snapshot.go
@@ -183,6 +183,12 @@ func (s Snapshot) SyscallsCount() int {
 	return len(s.syscalls)
 }
 
+// TopNSyscalls returns at most n per-syscall rows in ranking order.
+// Callers must treat returned data as read-only.
+func (s Snapshot) TopNSyscalls(n int) []SyscallSnapshot {
+	return topNSyscalls(s.syscalls, n)
+}
+
 // Files returns per-file snapshot rows.
 // Callers must treat returned data as read-only.
 func (s Snapshot) Files() []FileSnapshot {
@@ -194,6 +200,12 @@ func (s Snapshot) FilesCount() int {
 	return len(s.files)
 }
 
+// TopNFiles returns at most n file rows in ranking order.
+// Callers must treat returned data as read-only.
+func (s Snapshot) TopNFiles(n int) []FileSnapshot {
+	return topNFiles(s.files, n)
+}
+
 // Processes returns per-process snapshot rows.
 // Callers must treat returned data as read-only.
 func (s Snapshot) Processes() []ProcessSnapshot {
@@ -205,8 +217,44 @@ func (s Snapshot) ProcessesCount() int {
 	return len(s.processes)
 }
 
+// TopNProcesses returns at most n process rows in ranking order.
+// Callers must treat returned data as read-only.
+func (s Snapshot) TopNProcesses(n int) []ProcessSnapshot {
+	return topNProcesses(s.processes, n)
+}
+
 // Buckets returns histogram buckets.
 // Callers must treat returned data as read-only.
 func (h HistogramSnapshot) Buckets() []HistogramBucketSnapshot {
 	return h.buckets
 }
+
+func topNSyscalls(rows []SyscallSnapshot, n int) []SyscallSnapshot {
+	if n <= 0 || len(rows) == 0 {
+		return nil
+	}
+	if n > len(rows) {
+		n = len(rows)
+	}
+	return rows[:n:n]
+}
+
+func topNFiles(rows []FileSnapshot, n int) []FileSnapshot {
+	if n <= 0 || len(rows) == 0 {
+		return nil
+	}
+	if n > len(rows) {
+		n = len(rows)
+	}
+	return rows[:n:n]
+}
+
+func topNProcesses(rows []ProcessSnapshot, n int) []ProcessSnapshot {
+	if n <= 0 || len(rows) == 0 {
+		return nil
+	}
+	if n > len(rows) {
+		n = len(rows)
+	}
+	return rows[:n:n]
+}