From cc846de1f587d6635417b8200def87ff1b24b303 Mon Sep 17 00:00:00 2001
From: boqiu <82121246@qq.com>
Date: Thu, 24 Oct 2024 18:38:41 +0800
Subject: [PATCH] do not propagate FindFile to whole network

---
 node/router/src/libp2p_event_handler.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/node/router/src/libp2p_event_handler.rs b/node/router/src/libp2p_event_handler.rs
index 439dafa..60270b5 100644
--- a/node/router/src/libp2p_event_handler.rs
+++ b/node/router/src/libp2p_event_handler.rs
@@ -343,7 +343,7 @@ impl Libp2pEventHandler {
             PubsubMessage::ExampleMessage(_) => MessageAcceptance::Ignore,
             PubsubMessage::NewFile(msg) => {
                 metrics::LIBP2P_HANDLE_PUBSUB_NEW_FILE.mark(1);
-                self.on_new_file(propagation_source, msg).await
+                self.on_new_file(source, msg).await
             }
             PubsubMessage::FindFile(msg) => {
                 metrics::LIBP2P_HANDLE_PUBSUB_FIND_FILE.mark(1);
@@ -616,6 +616,10 @@ impl Libp2pEventHandler {
             };
         }
 
+        // update peer shard config in cache
+        self.file_location_cache
+            .insert_peer_config(peer_id, announced_shard_config);
+
         // check if we have it
         if matches!(self.store.check_tx_completed(tx_id.seq).await, Ok(true)) {
             if let Ok(Some(tx)) = self.store.get_tx_by_seq_number(tx_id.seq).await {