From 2ca19d04eb3ec1dc36959d693bb7d0cdb55c8d84 Mon Sep 17 00:00:00 2001
From: Lautaro Emanuel <emlautarom1@gmail.com>
Date: Tue, 21 Apr 2026 13:42:36 -0300
Subject: [PATCH] Use a single P2PContext

---
 crates/dkg/examples/bcast.rs | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/crates/dkg/examples/bcast.rs b/crates/dkg/examples/bcast.rs
index ce3f006f..395bd7d9 100644
--- a/crates/dkg/examples/bcast.rs
+++ b/crates/dkg/examples/bcast.rs
@@ -82,7 +82,6 @@ use pluto_p2p::{
     config::P2PConfig,
     gater, k1,
     p2p::{Node, NodeType},
-    p2p_context::P2PContext,
     relay::{MutableRelayReservation, RelayRouter},
 };
 use pluto_tracing::TracingConfig;
@@ -578,16 +577,10 @@ async fn main() -> Result<()> {
         disable_reuse_port: args.disable_reuse_port,
     };
 
-    let p2p_context = P2PContext::new(known_peers.clone());
-    let (bcast_behaviour, component) =
-        bcast::Behaviour::new(cluster_peers.clone(), p2p_context.clone(), key.clone());
-    register_message(&component, local_node_number)
-        .await
-        .expect("Failed to register demo bcast message");
-
+    let mut component = None;
     let mut node: Node<ExampleBehaviour> = Node::new(
         p2p_config,
-        key,
+        key.clone(),
         NodeType::QUIC,
         args.filter_private_addrs,
         known_peers,
@@ -595,6 +588,10 @@ async fn main() -> Result<()> {
             let p2p_context = builder.p2p_context();
             let local_peer_id = keypair.public().to_peer_id();
 
+            let (bcast_behaviour, c) =
+                bcast::Behaviour::new(cluster_peers.clone(), p2p_context.clone(), key.clone());
+            component = Some(c);
+
             builder
                 .with_p2p_context(p2p_context.clone())
                 .with_gater(conn_gater)
@@ -607,6 +604,11 @@ async fn main() -> Result<()> {
         },
     )?;
 
+    let component = component.expect("BCast component was not initialized");
+    register_message(&component, local_node_number)
+        .await
+        .expect("Failed to register demo bcast message");
+
     info!(
         local_peer_id = %local_peer_id,
         local_node = local_node_number,