feat: add Free methods to Graph and P2P for resetting state and enhance eachStop for peer management

Author: elecbug

v2/graph/graph.go

@@ -34,6 +34,13 @@ func New(directed bool, weighted bool) *Graph {
 	}
 }
 
+// Free clears all nodes and edges from the graph, effectively resetting it to an empty state.
+func (g *Graph) Free() {
+	for id := range g.nodes {
+		delete(g.nodes, id)
+	}
+}
+
 /* Node */
 
 // AddNode adds a node to the graph.

v2/p2p/p2p.go

@@ -68,6 +68,14 @@ func New(source *graph.Graph, cfg *Config) (*P2P, error) {
 	return &P2P{peers: nodes, cfg: cfg}, nil
 }
 
+// Free clears all peers from the P2P network, effectively resetting it to an empty state.
+func (p *P2P) Free() {
+	for id := range p.peers {
+		p.peers[id].eachStop()
+		delete(p.peers, id)
+	}
+}
+
 /* Basic Actions */
 
 // Run starts the message handling routines for all peers in the network.

v2/p2p/peer.go

@@ -146,3 +146,12 @@ func (p *peer) eachPublish(network *P2P, msg Message) {
 		}(edgeCopy)
 	}
 }
+
+// eachStop marks the peer as inactive and closes its message queue.
+func (p *peer) eachStop() {
+	p.mu.Lock()
+	defer p.mu.Unlock()
+
+	p.alive = false
+	close(p.msgQueue)
+}