From 20324763d25dceefe28cd72c024e91c28bf46d6f Mon Sep 17 00:00:00 2001
From: henrygd <hank@henrygd.me>
Date: Thu, 29 Jan 2026 19:34:44 -0500
Subject: [PATCH] remove stale systemd services from tracking after deletion
 (#1594)

---
 agent/systemd.go | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/agent/systemd.go b/agent/systemd.go
index 43351298..ac9b0ff2 100644
--- a/agent/systemd.go
+++ b/agent/systemd.go
@@ -144,13 +144,27 @@ func (sm *systemdManager) getServiceStats(conn *dbus.Conn, refresh bool) []*syst
 		return nil
 	}
 
+	// Track which units are currently present to remove stale entries
+	currentUnits := make(map[string]struct{}, len(units))
+
 	for _, unit := range units {
+		currentUnits[unit.Name] = struct{}{}
 		service, err := sm.updateServiceStats(conn, unit)
 		if err != nil {
 			continue
 		}
 		services = append(services, service)
 	}
+
+	// Remove services that no longer exist in systemd
+	sm.Lock()
+	for unitName := range sm.serviceStatsMap {
+		if _, exists := currentUnits[unitName]; !exists {
+			delete(sm.serviceStatsMap, unitName)
+		}
+	}
+	sm.Unlock()
+
 	sm.hasFreshStats = true
 	return services
 }