Yggdrasil demo: Add Stop functions

2 files changed, 14 insertions, 0 deletions

diff --git a/build/gobind/monolith.go b/build/gobind/monolith.go
index b73e47f8..d415143e 100644
--- a/build/gobind/monolith.go
+++ b/build/gobind/monolith.go
@@ -180,3 +180,10 @@ func (m *DendriteMonolith) Start(staticPeer string, enableMulticast bool) {
 		logger.Fatal(httpServer.Serve(m.listener))
 	}()
 }
+
+func (m *DendriteMonolith) Stop() {
+	if err := m.listener.Close(); err != nil {
+		logrus.Warn("Error stopping listener:", err)
+	}
+	m.YggdrasilNode.Stop()
+}
diff --git a/cmd/dendrite-demo-yggdrasil/yggconn/node.go b/cmd/dendrite-demo-yggdrasil/yggconn/node.go
index 4b08f1f3..c89b0dc1 100644
--- a/cmd/dendrite-demo-yggdrasil/yggconn/node.go
+++ b/cmd/dendrite-demo-yggdrasil/yggconn/node.go
@@ -156,6 +156,13 @@ func Setup(instanceName, instancePeer, storageDirectory string, enableMulticast
 	return n, nil
 }
 
+func (n *Node) Stop() {
+	if err := n.multicast.Stop(); err != nil {
+		n.log.Println("Error stopping multicast:", err)
+	}
+	n.core.Stop()
+}
+
 func (n *Node) DerivedServerName() string {
 	return hex.EncodeToString(n.SigningPublicKey())
 }