kad: Refactor GetRecord query and add tests (#97)

This PR refactors the `GetRecord` query to make it more robust. - Avoid panics on unwraps and unimplemented logic - Separate config immutable variables from main query logic - Simplifies logic of `next_action` method - Private methods for internal logic Builds on top of: #96 cc @paritytech/networking --------- Signed-off-by: Alexandru Vasile <[email protected]>
paritytech · May 16, 2024 · 5548223 · 5548223
1 parent 423a35d
commit 5548223
Show file tree

Hide file tree

Showing 3 changed files with 367 additions and 97 deletions.
diff --git a/src/protocol/libp2p/kademlia/mod.rs b/src/protocol/libp2p/kademlia/mod.rs
@@ -47,13 +47,12 @@ use tokio::sync::mpsc::{Receiver, Sender};
 
 use std::collections::{hash_map::Entry, HashMap};
 
+pub use self::handle::RecordsType;
 pub use config::{Config, ConfigBuilder};
 pub use handle::{KademliaEvent, KademliaHandle, Quorum, RoutingTableUpdateMode};
 pub use query::QueryId;
 pub use record::{Key as RecordKey, PeerRecord, Record};
 
-pub use self::handle::RecordsType;
-
 /// Logging target for the file.
 const LOG_TARGET: &str = "litep2p::ipfs::kademlia";