fluencelabs/js-libp2p
1
0
Fork 0
mirror of https://github.com/fluencelabs/js-libp2p synced 2025-06-16 10:41:23 +00:00
Code Issues Projects Releases Wiki Activity

feat: discovery modules from transports should be added (#510)

Browse Source 
* feat: discovery modules from transports should be added

* chore: apply suggestions from code review

Co-Authored-By: Jacob Heun <jacobheun@gmail.com>

* chore: address review

* chore: apply suggestions from code review

Co-Authored-By: Jacob Heun <jacobheun@gmail.com>
This commit is contained in:
Vasco Santos
2019-12-11 17:06:40 +01:00
committed by Jacob Heun
parent f540112835
commit af96dcc499
5 changed files with 149 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/index.js
View File

@ -48,7 +48,7 @@ class Libp2p extends EventEmitter {
this._modules = this._options.modules
this._config = this._options.config
this._transport = [] // Transport instances/references
this._discovery = [] // Discovery service instances/references
this._discovery = new Map() // Discovery service instances/references
this.peerStore = new PeerStore()
@ -437,7 +437,7 @@ class Libp2p extends EventEmitter {
* @returns {Promise<void>}
*/
_setupPeerDiscovery () {
for (const DiscoveryService of this._modules.peerDiscovery || []) {
const setupService = (DiscoveryService) => {
let config = {
enabled: true // on by default
}
@ -448,7 +448,8 @@ class Libp2p extends EventEmitter {
config = { ...config, ...this._config.peerDiscovery[DiscoveryService.tag] }
}
if (config.enabled) {
if (config.enabled &&
!this._discovery.has(DiscoveryService.tag)) { // not already added
let discoveryService
if (typeof DiscoveryService === 'function') {
@ -458,11 +459,23 @@ class Libp2p extends EventEmitter {
}
discoveryService.on('peer', this._onDiscoveryPeer)
this._discovery.push(discoveryService)
this._discovery.set(DiscoveryService.tag, discoveryService)
}
}
return this._discovery.map(d => d.start())
// Discovery modules
for (const DiscoveryService of this._modules.peerDiscovery || []) {
setupService(DiscoveryService)
}
// Transport modules with discovery
for (const Transport of this.transportManager.getTransports()) {
if (Transport.discovery) {
setupService(Transport.discovery)
}
}
return Promise.all(Array.from(this._discovery.values(), d => d.start()))
}
}