feat(cluster): Explicitly delete host during the node unregistering

ab180 · Mar 29, 2023 · 797d819 · 797d819
1 parent 9f236d2
commit 797d819
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/cluster/cluster.go b/cluster/cluster.go
@@ -339,5 +339,12 @@ func (n *nodeRegistration) States() node.State {
 
 // Unregister removes node from the cluster's node list, and clears all NodeState.
 func (n *nodeRegistration) Unregister() {
+	_, err := n.cluster.States().Delete(n.ctx, path.Join(nodeNs, n.node.Host))
+	if err != nil {
+		log.Warn().
+			Err(err).
+			Msg("failed to delete node")
+	}
+
 	n.cancel()
 }