kad: Update local memory store only if the query is put_value

Signed-off-by: Alexandru Vasile <[email protected]>
paritytech · Apr 18, 2024 · 96e827b · 96e827b
1 parent 1f38a1b
commit 96e827b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/protocol/libp2p/kademlia/mod.rs b/src/protocol/libp2p/kademlia/mod.rs
@@ -741,7 +741,6 @@ impl Kademlia {
                         Some(KademliaCommand::PutRecord { record, query_id, peers }) => {
                             tracing::debug!(target: LOG_TARGET, ?query_id, key = ?record.key, "store record to DHT");
 
-                            self.store.put(record.clone());
                             let key = Key::new(record.key.clone());
 
 
@@ -761,6 +760,8 @@ impl Kademlia {
                                     peers
                                 );
                             } else {
+                                self.store.put(record.clone());
+
                                 self.engine.start_put_record(
                                     query_id,
                                     record,