diff --git a/dht.go b/dht.go index feff527..11764fa 100644 --- a/dht.go +++ b/dht.go @@ -100,7 +100,7 @@ func (dht *IpfsDHT) Connect(ctx context.Context, npeer peer.Peer) (peer.Peer, er // // /ip4/10.20.30.40/tcp/1234/ipfs/Qxhxxchxzcncxnzcnxzcxzm // - err := dht.dialer.DialPeer(npeer) + err := dht.dialer.DialPeer(ctx, npeer) if err != nil { return nil, err } @@ -311,7 +311,7 @@ func (dht *IpfsDHT) getFromPeerList(ctx context.Context, key u.Key, peerlist []*pb.Message_Peer, level int) ([]byte, error) { for _, pinfo := range peerlist { - p, err := dht.ensureConnectedToPeer(pinfo) + p, err := dht.ensureConnectedToPeer(ctx, pinfo) if err != nil { log.Errorf("getFromPeers error: %s", err) continue @@ -496,14 +496,14 @@ func (dht *IpfsDHT) peerFromInfo(pbp *pb.Message_Peer) (peer.Peer, error) { return p, nil } -func (dht *IpfsDHT) ensureConnectedToPeer(pbp *pb.Message_Peer) (peer.Peer, error) { +func (dht *IpfsDHT) ensureConnectedToPeer(ctx context.Context, pbp *pb.Message_Peer) (peer.Peer, error) { p, err := dht.peerFromInfo(pbp) if err != nil { return nil, err } // dial connection - err = dht.dialer.DialPeer(p) + err = dht.dialer.DialPeer(ctx, p) return p, err } @@ -556,7 +556,7 @@ func (dht *IpfsDHT) Bootstrap(ctx context.Context) { if err != nil { log.Error("Bootstrap peer error: %s", err) } - err = dht.dialer.DialPeer(p) + err = dht.dialer.DialPeer(ctx, p) if err != nil { log.Errorf("Bootstrap peer error: %s", err) } diff --git a/ext_test.go b/ext_test.go index a6d1d93..1dabb5b 100644 --- a/ext_test.go +++ b/ext_test.go @@ -4,7 +4,6 @@ import ( "testing" crand "crypto/rand" - context "github.com/jbenet/go-ipfs/Godeps/_workspace/src/code.google.com/p/go.net/context" "github.com/jbenet/go-ipfs/Godeps/_workspace/src/code.google.com/p/goprotobuf/proto" @@ -82,7 +81,7 @@ type fauxNet struct { } // DialPeer attempts to establish a connection to a given peer -func (f *fauxNet) DialPeer(peer.Peer) error { +func (f *fauxNet) DialPeer(context.Context, peer.Peer) error { return nil } diff --git a/query.go b/query.go index 557af09..f0478ff 100644 --- a/query.go +++ b/query.go @@ -230,7 +230,7 @@ func (r *dhtQueryRunner) queryPeer(p peer.Peer) { // make sure we're connected to the peer. // (Incidentally, this will add it to the peerstore too) - err := r.query.dialer.DialPeer(p) + err := r.query.dialer.DialPeer(r.ctx, p) if err != nil { log.Debugf("ERROR worker for: %v -- err connecting: %v", p, err) r.Lock() diff --git a/routing.go b/routing.go index da400fe..c0d86e1 100644 --- a/routing.go +++ b/routing.go @@ -145,7 +145,7 @@ func (dht *IpfsDHT) FindProvidersAsync(ctx context.Context, key u.Key, count int log.Error(err) return } - dht.addPeerListAsync(key, pmes.GetProviderPeers(), ps, count, peerOut) + dht.addPeerListAsync(ctx, key, pmes.GetProviderPeers(), ps, count, peerOut) }(pp) } wg.Wait() @@ -154,13 +154,13 @@ func (dht *IpfsDHT) FindProvidersAsync(ctx context.Context, key u.Key, count int return peerOut } -func (dht *IpfsDHT) addPeerListAsync(k u.Key, peers []*pb.Message_Peer, ps *peerSet, count int, out chan peer.Peer) { +func (dht *IpfsDHT) addPeerListAsync(ctx context.Context, k u.Key, peers []*pb.Message_Peer, ps *peerSet, count int, out chan peer.Peer) { done := make(chan struct{}) for _, pbp := range peers { go func(mp *pb.Message_Peer) { defer func() { done <- struct{}{} }() // construct new peer - p, err := dht.ensureConnectedToPeer(mp) + p, err := dht.ensureConnectedToPeer(ctx, mp) if err != nil { log.Error("%s", err) return