make access to router synchronized

Peergos · Jul 11, 2023 · 40ac96b · 40ac96b
1 parent 4a7ae3e
commit 40ac96b
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/main/java/org/peergos/protocol/dht/KademliaEngine.java b/src/main/java/org/peergos/protocol/dht/KademliaEngine.java
@@ -33,19 +33,22 @@ public void setAddressBook(AddressBook addrs) {
         this.addressBook = addrs;
     }
 
-    public void addOutgoingConnection(PeerId peer, Multiaddr addr) {
+    public synchronized void addOutgoingConnection(PeerId peer, Multiaddr addr) {
         router.touch(Instant.now(), new Node(Id.create(Hash.sha256(peer.getBytes()), 256), peer.toString()));
         addressBook.addAddrs(peer, 0, addr);
     }
 
-    public void addIncomingConnection(PeerId peer, Multiaddr addr) {
+    public synchronized void addIncomingConnection(PeerId peer, Multiaddr addr) {
         router.touch(Instant.now(), new Node(Id.create(Hash.sha256(peer.getBytes()), 256), peer.toString()));
         addressBook.addAddrs(peer, 0, addr);
     }
 
     public List<PeerAddresses> getKClosestPeers(byte[] key) {
         int k = 20;
-        List<Node> nodes = router.find(Id.create(Hash.sha256(key), 256), k, false);
+        List<Node> nodes;
+        synchronized (this) {
+            nodes = router.find(Id.create(Hash.sha256(key), 256), k, false);
+        }
         System.out.println("Nodes: " + nodes.size());
         return nodes.stream()
                 .map(n -> {