1
0
mirror of https://github.com/ipfs/kubo.git synced 2025-07-02 03:28:25 +08:00

bitswap: network interface changed

Had to change the network interface from DialPeer(peer.ID) to
DialPeer(peer.PeerInfo), so that addresses of a provider are
handed to the network.

@maybebtc and I are discussing whether this should go all the
way down to the network, or whether the network _should always
work_ with just an ID (which means the network needs to be
able to resolve ID -> Addresses, using the routing system.
This latter point might mean that "routing" might need to
break down into subcomponents. It's a bit sketchy that
the Network would become smarter than just dial/listen and
I/O, but maybe there's a distinction between net.Network,
and something like a peernet.Network that has routing
built in...)
This commit is contained in:
Juan Batiz-Benet
2014-12-23 04:13:52 -08:00
parent a7eb52a7fa
commit d970f538e5
4 changed files with 14 additions and 11 deletions

View File

@ -176,14 +176,16 @@ func (bs *bitswap) sendWantListTo(ctx context.Context, peers <-chan peer.PeerInf
message.AddEntry(wanted.Key, wanted.Priority)
}
wg := sync.WaitGroup{}
for peerToQuery := range peers {
log.Event(ctx, "PeerToQuery", peerToQuery.ID)
for pi := range peers {
log.Debugf("bitswap.sendWantListTo: %s %s", pi.ID, pi.Addrs)
log.Event(ctx, "PeerToQuery", pi.ID)
wg.Add(1)
go func(p peer.ID) {
go func(pi peer.PeerInfo) {
defer wg.Done()
p := pi.ID
log.Event(ctx, "DialPeer", p)
err := bs.sender.DialPeer(ctx, p)
err := bs.sender.DialPeer(ctx, pi)
if err != nil {
log.Errorf("Error sender.DialPeer(%s): %s", p, err)
return
@ -198,7 +200,7 @@ func (bs *bitswap) sendWantListTo(ctx context.Context, peers <-chan peer.PeerInf
// communication fails. May require slightly different API to
// get better guarantees. May need shared sequence numbers.
bs.engine.MessageSent(p, message)
}(peerToQuery.ID)
}(pi)
}
wg.Wait()
return nil

View File

@ -12,7 +12,7 @@ import (
type BitSwapNetwork interface {
// DialPeer ensures there is a connection to peer.
DialPeer(context.Context, peer.ID) error
DialPeer(context.Context, peer.PeerInfo) error
// SendMessage sends a BitSwap message to a peer.
SendMessage(

View File

@ -53,8 +53,9 @@ func (bsnet *impl) handleNewStream(s inet.Stream) {
}
func (bsnet *impl) DialPeer(ctx context.Context, p peer.ID) error {
return bsnet.network.DialPeer(ctx, p)
func (bsnet *impl) DialPeer(ctx context.Context, p peer.PeerInfo) error {
bsnet.network.Peerstore().AddAddresses(p.ID, p.Addrs)
return bsnet.network.DialPeer(ctx, p.ID)
}
func (bsnet *impl) SendMessage(

View File

@ -165,10 +165,10 @@ func (nc *networkClient) SendRequest(
return nc.network.SendRequest(ctx, nc.local, to, message)
}
func (nc *networkClient) DialPeer(ctx context.Context, p peer.ID) error {
func (nc *networkClient) DialPeer(ctx context.Context, p peer.PeerInfo) error {
// no need to do anything because dialing isn't a thing in this test net.
if !nc.network.HasPeer(p) {
return fmt.Errorf("Peer not in network: %s", p)
if !nc.network.HasPeer(p.ID) {
return fmt.Errorf("Peer not in network: %s", p.ID)
}
return nil
}