Make getProviders also query the local provider store

Peergos · Jul 11, 2023 · 4ed0f57 · 4ed0f57
1 parent 325d3fb
commit 4ed0f57
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/main/java/org/peergos/protocol/dht/Kademlia.java b/src/main/java/org/peergos/protocol/dht/Kademlia.java
@@ -183,6 +183,7 @@ public CompletableFuture<List<PeerAddresses>> findProviders(Multihash block, Hos
         byte[] key = block.bareMultihash().toBytes();
         Id keyId = Id.create(key, 256);
         List<PeerAddresses> providers = new ArrayList<>();
+        providers.addAll(engine.getProviders(block));
 
         SortedSet<RoutingEntry> toQuery = new TreeSet<>((a, b) -> b.key.getSharedPrefixLength(keyId) - a.key.getSharedPrefixLength(keyId));
         toQuery.addAll(engine.getKClosestPeers(key).stream()

diff --git a/src/main/java/org/peergos/protocol/dht/KademliaEngine.java b/src/main/java/org/peergos/protocol/dht/KademliaEngine.java
@@ -43,6 +43,10 @@ public synchronized void addIncomingConnection(PeerId peer, Multiaddr addr) {
         addressBook.addAddrs(peer, 0, addr);
     }
 
+    public Set<PeerAddresses> getProviders(Multihash h) {
+        return providersStore.getProviders(h);
+    }
+
     public List<PeerAddresses> getKClosestPeers(byte[] key) {
         int k = 20;
         List<Node> nodes;