chore: do not use peerstore.put on get-peer-info

libp2p · Apr 1, 2020 · c4df833 · c4df833
1 parent d3e4ad2
commit c4df833
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/get-peer-info.js b/src/get-peer-info.js
@@ -38,7 +38,12 @@ function getPeerInfo (peer, peerStore) {
 
   addr && peer.multiaddrs.add(addr)
 
-  return peerStore ? peerStore.put(peer, { replace: false }) : peer
+  if (peerStore) {
+    peerStore.addressBook.set(peer.id, peer.multiaddrs.toArray(), { replace: false })
+    peerStore.protoBook.set(peer.id, Array.from(peer.protocols), { replace: false })
+  }
+
+  return peer
 }
 
 /**

diff --git a/test/dialing/direct.node.js b/test/dialing/direct.node.js
@@ -123,7 +123,7 @@ describe('Dialing (direct, TCP)', () => {
     const peerId = await PeerId.createFromJSON(Peers[0])
     const peerInfo = new PeerInfo(peerId)
     peerInfo.multiaddrs.add(remoteAddr)
-    peerStore.put(peerInfo)
+    peerStore.addressBook.set(peerInfo.id, peerInfo.multiaddrs.toArray())
 
     const connection = await dialer.connectToPeer(peerInfo)
     expect(connection).to.exist()