fix(discovery): properly delete nodes when targets disappear

cryostatio · Apr 22, 2024 · 4d97568 · 4d97568
1 parent a05895c
commit 4d97568
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 0 deletions.
diff --git a/src/main/java/io/cryostat/discovery/ContainerDiscovery.java b/src/main/java/io/cryostat/discovery/ContainerDiscovery.java
@@ -399,6 +399,7 @@ public void handleContainerEvent(ContainerSpec desc, EventKind evtKind) {
             } else {
                 realm.children.remove(t.discoveryNode);
             }
+            t.discoveryNode.parent = null;
             realm.persist();
             t.delete();
         }

diff --git a/src/main/java/io/cryostat/discovery/CustomDiscovery.java b/src/main/java/io/cryostat/discovery/CustomDiscovery.java
@@ -209,6 +209,7 @@ public Response delete(@RestPath long id) throws URISyntaxException {
         Target target = Target.find("id", id).singleResult();
         DiscoveryNode realm = DiscoveryNode.getRealm(REALM).orElseThrow();
         realm.children.remove(target.discoveryNode);
+        target.discoveryNode = null;
         realm.persist();
         target.delete();
         return Response.noContent().build();

diff --git a/src/main/java/io/cryostat/discovery/JDPDiscovery.java b/src/main/java/io/cryostat/discovery/JDPDiscovery.java
@@ -152,6 +152,7 @@ public synchronized void handleJdpEvent(JvmDiscoveryEvent evt) {
             case LOST:
                 Target t = Target.getTargetByConnectUrl(connectUrl);
                 realm.children.remove(t.discoveryNode);
+                t.discoveryNode.parent = null;
                 realm.persist();
                 t.delete();
                 break;

diff --git a/src/main/java/io/cryostat/discovery/KubeApiDiscovery.java b/src/main/java/io/cryostat/discovery/KubeApiDiscovery.java
@@ -245,6 +245,7 @@ public void handleEndpointEvent(String namespace) {
 
             if (!nsNode.hasChildren()) {
                 realm.children.remove(nsNode);
+                nsNode.parent = null;
             } else if (!realm.children.contains(nsNode)) {
                 realm.children.add(nsNode);
                 nsNode.parent = realm;
@@ -275,6 +276,7 @@ private void pruneOwnerChain(DiscoveryNode nsNode, Target target) {
             }
 
             parent.children.remove(child);
+            child.parent = null;
             parent.persist();
 
             if (parent.hasChildren()