mirror of
https://github.com/DeBrosOfficial/orama.git
synced 2026-03-17 06:23:00 +00:00
124 lines
3.3 KiB
Go
124 lines
3.3 KiB
Go
package report
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// oramaProcessNames lists command substrings that identify orama-related processes.
|
|
var oramaProcessNames = []string{
|
|
"orama", "rqlite", "olric", "ipfs", "caddy", "coredns",
|
|
}
|
|
|
|
// collectProcesses gathers zombie/orphan process info and panic counts from logs.
|
|
func collectProcesses() *ProcessReport {
|
|
r := &ProcessReport{}
|
|
|
|
// Collect known systemd-managed PIDs to avoid false positive orphan detection.
|
|
// Processes with PPID=1 that are systemd-managed daemons are NOT orphans.
|
|
managedPIDs := collectManagedPIDs()
|
|
|
|
// Run ps once and reuse the output for both zombies and orphans.
|
|
ctx, cancel := context.WithTimeout(context.Background(), 4*time.Second)
|
|
defer cancel()
|
|
|
|
out, err := runCmd(ctx, "ps", "-eo", "pid,ppid,state,comm", "--no-headers")
|
|
if err == nil {
|
|
for _, line := range strings.Split(out, "\n") {
|
|
line = strings.TrimSpace(line)
|
|
if line == "" {
|
|
continue
|
|
}
|
|
|
|
fields := strings.Fields(line)
|
|
if len(fields) < 4 {
|
|
continue
|
|
}
|
|
|
|
pid, _ := strconv.Atoi(fields[0])
|
|
ppid, _ := strconv.Atoi(fields[1])
|
|
state := fields[2]
|
|
command := strings.Join(fields[3:], " ")
|
|
|
|
proc := ProcessInfo{
|
|
PID: pid,
|
|
PPID: ppid,
|
|
State: state,
|
|
Command: command,
|
|
}
|
|
|
|
// Zombies: state == "Z"
|
|
if state == "Z" {
|
|
r.Zombies = append(r.Zombies, proc)
|
|
}
|
|
|
|
// Orphans: PPID == 1 and command is orama-related,
|
|
// but NOT a known systemd-managed service PID.
|
|
if ppid == 1 && isOramaProcess(command) && !managedPIDs[pid] {
|
|
r.Orphans = append(r.Orphans, proc)
|
|
}
|
|
}
|
|
}
|
|
|
|
r.ZombieCount = len(r.Zombies)
|
|
r.OrphanCount = len(r.Orphans)
|
|
|
|
// PanicCount: check journal for panic/fatal in last hour.
|
|
{
|
|
ctx2, cancel2 := context.WithTimeout(context.Background(), 4*time.Second)
|
|
defer cancel2()
|
|
|
|
out, err := runCmd(ctx2, "bash", "-c",
|
|
`journalctl -u orama-node --no-pager -n 500 --since "1 hour ago" 2>/dev/null | grep -ciE "(panic|fatal)" || echo 0`)
|
|
if err == nil {
|
|
if n, err := strconv.Atoi(strings.TrimSpace(out)); err == nil {
|
|
r.PanicCount = n
|
|
}
|
|
}
|
|
}
|
|
|
|
return r
|
|
}
|
|
|
|
// managedServiceUnits lists systemd units whose MainPID should be excluded from orphan detection.
|
|
var managedServiceUnits = []string{
|
|
"orama-node", "orama-gateway", "orama-olric",
|
|
"orama-ipfs", "orama-ipfs-cluster",
|
|
"orama-anyone-relay", "orama-anyone-client",
|
|
"coredns", "caddy", "rqlited",
|
|
}
|
|
|
|
// collectManagedPIDs queries systemd for the MainPID of each known service.
|
|
// Returns a set of PIDs that are legitimately managed by systemd (not orphans).
|
|
func collectManagedPIDs() map[int]bool {
|
|
pids := make(map[int]bool)
|
|
for _, unit := range managedServiceUnits {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
|
|
out, err := runCmd(ctx, "systemctl", "show", unit, "--property=MainPID")
|
|
cancel()
|
|
if err != nil {
|
|
continue
|
|
}
|
|
props := parseProperties(out)
|
|
if pidStr, ok := props["MainPID"]; ok {
|
|
if pid, err := strconv.Atoi(pidStr); err == nil && pid > 0 {
|
|
pids[pid] = true
|
|
}
|
|
}
|
|
}
|
|
return pids
|
|
}
|
|
|
|
// isOramaProcess checks if a command string contains any orama-related process name.
|
|
func isOramaProcess(command string) bool {
|
|
lower := strings.ToLower(command)
|
|
for _, name := range oramaProcessNames {
|
|
if strings.Contains(lower, name) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|