fluencelabs/go-libp2p-kad-dht
1
0
Fork 0
mirror of https://github.com/fluencelabs/go-libp2p-kad-dht synced 2025-07-30 19:42:01 +00:00
Code Issues Projects Releases Wiki Activity

refactor common woker code for BootstrapWithConfig and BootstrapOnSignal

Browse Source
This commit is contained in:
vyzo
2016-12-08 19:37:07 +02:00
parent 7a658d1e77
commit 7a4eb0f079
1 changed files with 9 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
dht_bootstrap.go
View File

@@ -78,16 +78,7 @@ func (dht *IpfsDHT) BootstrapWithConfig(cfg BootstrapConfig) (goprocess.Process,
return nil, fmt.Errorf("invalid number of queries: %d", cfg.Queries)
}
proc := periodicproc.Tick(cfg.Period, func(worker goprocess.Process) {
// it would be useful to be able to send out signals of when we bootstrap, too...
// maybe this is a good case for whole module event pub/sub?
ctx := dht.Context()
if err := dht.runBootstrap(ctx, cfg); err != nil {
log.Warning(err)
// A bootstrapping error is important to notice but not fatal.
}
})
proc := periodicproc.Tick(cfg.Period, dht.bootstrapWorker(cfg))
return proc, nil
}
@@ -107,7 +98,13 @@ func (dht *IpfsDHT) BootstrapOnSignal(cfg BootstrapConfig, signal <-chan time.Ti
return nil, fmt.Errorf("invalid signal: %v", signal)
}
proc := periodicproc.Ticker(signal, func(worker goprocess.Process) {
proc := periodicproc.Ticker(signal, dht.bootstrapWorker(cfg))
return proc, nil
}
func (dht *IpfsDHT) bootstrapWorker(cfg BootstrapConfig) func(worker goprocess.Process) {
return func(worker goprocess.Process) {
// it would be useful to be able to send out signals of when we bootstrap, too...
// maybe this is a good case for whole module event pub/sub?
@@ -116,9 +113,7 @@ func (dht *IpfsDHT) BootstrapOnSignal(cfg BootstrapConfig, signal <-chan time.Ti
log.Warning(err)
// A bootstrapping error is important to notice but not fatal.
}
})
return proc, nil
}
}
// runBootstrap builds up list of peers by requesting random peer IDs