Merge pull request ipfs#170 from libp2p/bug/bootstrap_immediately

Bootstrap immediately after calls to BootstrapWithConfig
ariescodescream · Jul 24, 2018 · d4e1e7e · d4e1e7e
2 parents bb96b20 + 9d8818b
commit d4e1e7e
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/dht_bootstrap.go b/dht_bootstrap.go
@@ -77,7 +77,17 @@ func (dht *IpfsDHT) BootstrapWithConfig(cfg BootstrapConfig) (goprocess.Process,
 		return nil, fmt.Errorf("invalid number of queries: %d", cfg.Queries)
 	}
 
-	proc := periodicproc.Tick(cfg.Period, dht.bootstrapWorker(cfg))
+	proc := dht.Process().Go(func(p goprocess.Process) {
+		<-p.Go(dht.bootstrapWorker(cfg)).Closed()
+		for {
+			select {
+			case <-time.After(cfg.Period):
+				<-p.Go(dht.bootstrapWorker(cfg)).Closed()
+			case <-p.Closing():
+				return
+			}
+		}
+	})
 
 	return proc, nil
 }