Skip to content
This repository has been archived by the owner on Jun 26, 2023. It is now read-only.

Commit

Permalink
chore: remove peer-info usage on topology
Browse files Browse the repository at this point in the history
BREAKING CHANGE: topology api now uses peer-id instead of peer-info
  • Loading branch information
vasco-santos committed Apr 16, 2020
1 parent de64a49 commit 3daa016
Show file tree
Hide file tree
Showing 7 changed files with 65 additions and 52 deletions.
28 changes: 14 additions & 14 deletions src/topology/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -56,8 +56,8 @@ const toplogy = new MulticodecTopology({
max: 50,
multicodecs: ['/echo/1.0.0'],
handlers: {
onConnect: (peerInfo, conn) => {},
onDisconnect: (peerInfo) => {}
onConnect: (peerId, conn) => {},
onDisconnect: (peerId) => {}
}
})
```
Expand All @@ -69,8 +69,8 @@ The `MulticodecTopology` extends the `Topology`, which makes the `Topology` API
### Topology

- `Topology`
- `peers<Map<string, PeerInfo>>`: A Map of peers belonging to the topology.
- `disconnect<function(PeerInfo)>`: Called when a peer has been disconnected
- `peers<Map<string, PeerId>>`: A Map of peers belonging to the topology.
- `disconnect<function(PeerId)>`: Called when a peer has been disconnected

#### Constructor

Expand All @@ -79,8 +79,8 @@ const toplogy = new Topology({
min: 0,
max: 50,
handlers: {
onConnect: (peerInfo, conn) => {},
onDisconnect: (peerInfo) => {}
onConnect: (peerId, conn) => {},
onDisconnect: (peerId) => {}
}
})
```
Expand All @@ -95,27 +95,27 @@ const toplogy = new Topology({

#### Set a peer

- `topology.peers.set(id, peerInfo)`
- `topology.peers.set(id, peerId)`

Add a peer to the topology.

**Parameters**
- `id` is the `string` that identifies the peer to add.
- `peerInfo` is the [PeerInfo][peer-info] of the peer to add.
- `peerId` is the [PeerId][peer-id] of the peer to add.

#### Notify about a peer disconnected event

- `topology.disconnect(peerInfo)`
- `topology.disconnect(peerId)`

**Parameters**
- `peerInfo` is the [PeerInfo][peer-info] of the peer disconnected.
- `peerId` is the [PeerIid][peer-id] of the peer disconnected.

### Multicodec Topology

- `MulticodecTopology`
- `registrar<Registrar>`: The `Registrar` of the topology. This is set by the `Registrar` during registration.
- `peers<Map<string, PeerInfo>>`: The Map of peers that belong to the topology
- `disconnect<function(PeerInfo)>`: Disconnects a peer from the topology.
- `peers<Map<string, PeerId>>`: The Map of peers that belong to the topology
- `disconnect<function(PeerId)>`: Disconnects a peer from the topology.

#### Constructor

Expand All @@ -125,8 +125,8 @@ const toplogy = new MulticodecTopology({
max: 50,
multicodecs: ['/echo/1.0.0'],
handlers: {
onConnect: (peerInfo, conn) => {},
onDisconnect: (peerInfo) => {}
onConnect: (peerId, conn) => {},
onDisconnect: (peerId) => {}
}
})
```
Expand Down
7 changes: 4 additions & 3 deletions src/topology/index.js
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ class Topology {
this._onConnect = handlers.onConnect || noop
this._onDisconnect = handlers.onDisconnect || noop

// TODO: can this be a set?
this.peers = new Map()
}

Expand All @@ -35,11 +36,11 @@ class Topology {

/**
* Notify about peer disconnected event.
* @param {PeerInfo} peerInfo
* @param {PeerId} peerId
* @returns {void}
*/
disconnect (peerInfo) {
this._onDisconnect(peerInfo)
disconnect (peerId) {
this._onDisconnect(peerId)
}
}

Expand Down
27 changes: 14 additions & 13 deletions src/topology/multicodec-topology.js
Original file line number Diff line number Diff line change
Expand Up @@ -55,43 +55,44 @@ class MulticodecTopology extends Topology {

/**
* Update topology.
* @param {Array<PeerInfo>} peerInfoIterable
* @param {Array<PeerData>} peerDataIterable
* @returns {void}
*/
_updatePeers (peerInfoIterable) {
for (const peerInfo of peerInfoIterable) {
if (this.multicodecs.filter(multicodec => peerInfo.protocols.has(multicodec)).length) {
_updatePeers (peerDataIterable) {
for (const peerData of peerDataIterable) {
if (this.multicodecs.filter(multicodec => peerData.protocols.includes(multicodec)).length) {
// Add the peer regardless of whether or not there is currently a connection
this.peers.set(peerInfo.id.toB58String(), peerInfo)
this.peers.set(peerData.id.toB58String(), peerData.id)
// If there is a connection, call _onConnect
const connection = this._registrar.getConnection(peerInfo)
connection && this._onConnect(peerInfo, connection)
const connection = this._registrar.getConnection(peerData.id)
connection && this._onConnect(peerData.id, connection)
} else {
// Remove any peers we might be tracking that are no longer of value to us
this.peers.delete(peerInfo.id.toB58String())
this.peers.delete(peerData.id.toB58String())
}
}
}

/**
* Check if a new peer support the multicodecs for this topology.
* @param {Object} props
* @param {PeerInfo} props.peerInfo
* @param {PeerId} props.peerId
* @param {Array<string>} props.protocols
*/
_onProtocolChange ({ peerInfo, protocols }) {
const existingPeer = this.peers.get(peerInfo.id.toB58String())
_onProtocolChange ({ peerId, protocols }) {
const existingPeer = this.peers.get(peerId.toB58String())
const hasProtocol = protocols.filter(protocol => this.multicodecs.includes(protocol))

// Not supporting the protocol anymore?
if (existingPeer && hasProtocol.length === 0) {
this._onDisconnect(peerInfo)
this._onDisconnect(peerId)
}

// New to protocol support
for (const protocol of protocols) {
if (this.multicodecs.includes(protocol)) {
this._updatePeers([peerInfo])
const peerData = this._registrar.peerStore.get(peerId)
this._updatePeers([peerData])
return
}
}
Expand Down
40 changes: 24 additions & 16 deletions src/topology/tests/multicodec-topology.js
Original file line number Diff line number Diff line change
Expand Up @@ -8,19 +8,18 @@ chai.use(require('dirty-chai'))
const sinon = require('sinon')

const PeerId = require('peer-id')
const PeerInfo = require('peer-info')

const peers = require('../../utils/peers')

module.exports = (test) => {
describe('multicodec topology', () => {
let topology, peer
let topology, id

beforeEach(async () => {
topology = await test.setup()
if (!topology) throw new Error('missing multicodec topology')

const id = await PeerId.createFromJSON(peers[0])
peer = await PeerInfo.create(id)
id = await PeerId.createFromJSON(peers[0])
})

afterEach(async () => {
Expand All @@ -38,7 +37,7 @@ module.exports = (test) => {

it('should trigger "onDisconnect" on peer disconnected', () => {
sinon.spy(topology, '_onDisconnect')
topology.disconnect(peer)
topology.disconnect(id)

expect(topology._onDisconnect.callCount).to.equal(1)
})
Expand All @@ -47,13 +46,16 @@ module.exports = (test) => {
sinon.spy(topology, '_updatePeers')
expect(topology.peers.size).to.eql(0)

const peerStore = topology._registrar.peerStore

const id2 = await PeerId.createFromJSON(peers[1])
const peer2 = await PeerInfo.create(id2)
topology.multicodecs.forEach((m) => peer2.protocols.add(m))
peerStore.peers.set(id2.toB58String(), {
id: id2,
protocols: Array.from(topology.multicodecs)
})

const peerStore = topology._registrar.peerStore
peerStore.emit('change:protocols', {
peerInfo: peer2,
peerId: id2,
protocols: Array.from(topology.multicodecs)
})

Expand All @@ -65,28 +67,34 @@ module.exports = (test) => {
sinon.spy(topology, '_onDisconnect')
expect(topology.peers.size).to.eql(0)

const peerStore = topology._registrar.peerStore

const id2 = await PeerId.createFromJSON(peers[1])
const peer2 = await PeerInfo.create(id2)
topology.multicodecs.forEach((m) => peer2.protocols.add(m))
peerStore.peers.set(id2.toB58String(), {
id: id2,
protocols: Array.from(topology.multicodecs)
})

const peerStore = topology._registrar.peerStore
peerStore.emit('change:protocols', {
peerInfo: peer2,
peerId: id2,
protocols: Array.from(topology.multicodecs)
})

expect(topology.peers.size).to.eql(1)

topology.multicodecs.forEach((m) => peer2.protocols.delete(m))
peerStore.peers.set(id2.toB58String(), {
id: id2,
protocols: []
})
// Peer does not support the protocol anymore
peerStore.emit('change:protocols', {
peerInfo: peer2,
peerId: id2,
protocols: []
})

expect(topology.peers.size).to.eql(1)
expect(topology._onDisconnect.callCount).to.equal(1)
expect(topology._onDisconnect.calledWith(peer2)).to.equal(true)
expect(topology._onDisconnect.calledWith(id2)).to.equal(true)
})
})
}
8 changes: 3 additions & 5 deletions src/topology/tests/topology.js
Original file line number Diff line number Diff line change
Expand Up @@ -8,19 +8,17 @@ chai.use(require('dirty-chai'))
const sinon = require('sinon')

const PeerId = require('peer-id')
const PeerInfo = require('peer-info')
const peers = require('../../utils/peers')

module.exports = (test) => {
describe('topology', () => {
let topology, peer
let topology, id

beforeEach(async () => {
topology = await test.setup()
if (!topology) throw new Error('missing multicodec topology')

const id = await PeerId.createFromJSON(peers[0])
peer = await PeerInfo.create(id)
id = await PeerId.createFromJSON(peers[0])
})

afterEach(async () => {
Expand All @@ -38,7 +36,7 @@ module.exports = (test) => {

it('should trigger "onDisconnect" on peer disconnected', () => {
sinon.spy(topology, '_onDisconnect')
topology.disconnect(peer)
topology.disconnect(id)

expect(topology._onDisconnect.callCount).to.equal(1)
})
Expand Down
4 changes: 4 additions & 0 deletions test/topology/mock-peer-store.js
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,10 @@ class MockPeerStore extends EventEmitter {
super()
this.peers = peers
}

get (peerId) {
return this.peers.get(peerId.toB58String())
}
}

module.exports = MockPeerStore
3 changes: 2 additions & 1 deletion test/topology/multicodec-topology.spec.js
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,8 @@ describe('multicodec topology compliance tests', () => {
})

if (!registrar) {
const peerStore = new MockPeerStore([])
const peers = new Map()
const peerStore = new MockPeerStore(peers)

registrar = {
peerStore,
Expand Down

0 comments on commit 3daa016

Please sign in to comment.