opensource/kubo
1
0
Fork 0
mirror of https://github.com/ipfs/kubo.git synced 2025-06-30 09:59:13 +08:00
Code Issues Projects Releases Wiki Activity

feat(bitswap/network) expose peerstore

Browse Source
This commit is contained in:
Brian Tiger Chow
2014-12-23 07:07:58 -05:00
committed by Juan Batiz-Benet
parent a225568ff6
commit 60f3a874c2
3 changed files with 15 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
exchange/bitswap/network/interface.go
View File

@ -26,6 +26,8 @@ type BitSwapNetwork interface {
peer.ID, peer.ID,
bsmsg.BitSwapMessage) (incoming bsmsg.BitSwapMessage, err error) bsmsg.BitSwapMessage) (incoming bsmsg.BitSwapMessage, err error)
Peerstore() peer.Peerstore
// SetDelegate registers the Reciver to handle messages received from the // SetDelegate registers the Reciver to handle messages received from the
// network. // network.
SetDelegate(Receiver) SetDelegate(Receiver)

4
exchange/bitswap/network/ipfs_impl.go
View File

@ -70,6 +70,10 @@ func (bsnet *impl) SetDelegate(r Receiver) {
bsnet.receiver = r bsnet.receiver = r
} }
func (bsnet *impl) Peerstore() peer.Peerstore {
return bsnet.Peerstore()
}
// handleNewStream receives a new stream from the network. // handleNewStream receives a new stream from the network.
func (bsnet *impl) handleNewStream(s inet.Stream) { func (bsnet *impl) handleNewStream(s inet.Stream) {

12
exchange/bitswap/testnet/network.go
View File

@ -47,8 +47,9 @@ type network struct {
func (n *network) Adapter(p peer.ID) bsnet.BitSwapNetwork { func (n *network) Adapter(p peer.ID) bsnet.BitSwapNetwork {
client := &networkClient{ client := &networkClient{
local: p, local: p,
network: n, network: n,
peerstore: peer.NewPeerstore(),
} }
n.clients[p] = client n.clients[p] = client
return client return client
@ -148,7 +149,8 @@ func (n *network) SendRequest(
type networkClient struct { type networkClient struct {
local peer.ID local peer.ID
bsnet.Receiver bsnet.Receiver
network Network network Network
peerstore peer.Peerstore
} }
func (nc *networkClient) SendMessage( func (nc *networkClient) SendMessage(
@ -176,3 +178,7 @@ func (nc *networkClient) DialPeer(ctx context.Context, p peer.ID) error {
func (nc *networkClient) SetDelegate(r bsnet.Receiver) { func (nc *networkClient) SetDelegate(r bsnet.Receiver) {
nc.Receiver = r nc.Receiver = r
} }
func (nc *networkClient) Peerstore() peer.Peerstore {
return nc.peerstore
}