Allow optimistic mirror of kbuckets with read lock only

src/discv5.rs

@@ -453,6 +453,17 @@ impl<P: ProtocolIdentity> Discv5<P> {
             .collect()
     }
 
+    /// Takes only a read lock on the kbuckets. This means pending entries aren't added. Otherwise
+    /// same as [`Self::table_entries_id`]. Returns an iterator over all ENR node IDs of nodes
+    /// currently contained in the routing table.
+    pub fn table_entries_id_rlock(&self) -> Vec<NodeId> {
+        self.kbuckets
+            .read()
+            .iter_ref()
+            .map(|entry| *entry.node.key.preimage())
+            .collect()
+    }
+
     /// Returns an iterator over all the ENR's of nodes currently contained in the routing table.
     pub fn table_entries_enr(&self) -> Vec<Enr> {
         self.kbuckets
@@ -477,6 +488,23 @@ impl<P: ProtocolIdentity> Discv5<P> {
             .collect()
     }
 
+    /// Takes only a read lock on the kbuckets. This means pending entries aren't added. Otherwise
+    /// same as [`Self::table_entries`]. Returns an iterator over all ENR node IDs of nodes
+    /// currently contained in the routing table.
+    pub fn table_entries_rlock(&self) -> Vec<(NodeId, Enr, NodeStatus)> {
+        self.kbuckets
+            .read()
+            .iter_ref()
+            .map(|entry| {
+                (
+                    *entry.node.key.preimage(),
+                    entry.node.value.clone(),
+                    entry.status,
+                )
+            })
+            .collect()
+    }
+
     /// Requests the ENR of a node corresponding to multiaddr or multi-addr string.
     ///
     /// Only `ed25519` and `secp256k1` key types are currently supported.

src/handler/request_call.rs

@@ -1,4 +1,4 @@
-pub use crate::node_info::{NodeAddress, NodeContact};
+pub use crate::node_info::NodeContact;
 use crate::{
     packet::Packet,
     rpc::{Request, RequestBody},

src/kbucket.rs

@@ -533,7 +533,7 @@ where
 
     /// Returns an iterator over all the entries in the routing table.
     /// Does not add pending node to kbucket to get an iterator which
-    /// takes a reference instead of a mutable reference.
+    /// takes a mutable reference instead of a reference.
     pub fn iter_ref(&self) -> impl Iterator<Item = EntryRefView<'_, TNodeId, TVal>> {
         self.buckets.iter().flat_map(move |table| {
             table.iter().map(move |n| EntryRefView {

src/kbucket/entry.rs

@@ -25,9 +25,7 @@
 //! representing the nodes participating in the Kademlia DHT.
 
 pub use super::{
-    bucket::{
-        AppliedPending, ConnectionState, InsertResult, Node, NodeStatus, MAX_NODES_PER_BUCKET,
-    },
+    bucket::{AppliedPending, ConnectionState, InsertResult, Node, NodeStatus},
     key::*,
     ConnectionDirection,
 };