mirror of
https://github.com/ipfs/kubo.git
synced 2026-03-03 23:38:07 +08:00
54d87b4ce0
This extracts the routing package to its own repository (https://github.com/ipfs/go-ipfs-routing). History has been preserved. The new module has been gx'ed and published. Imports have been rewritten and re-ordered accordingly. An internal dependency to go-ipfs/repo has been removed by substituting it with the go-datastore.Batching interface. License: MIT Signed-off-by: Hector Sanjuan <hector@protocol.ai>
43 lines
1.2 KiB
Go
43 lines
1.2 KiB
Go
package bitswap
|
|
|
|
import (
|
|
"context"
|
|
|
|
bsnet "github.com/ipfs/go-ipfs/exchange/bitswap/network"
|
|
|
|
mockpeernet "gx/ipfs/QmNh1kGFFdsPu79KNSaL4NUKUPb4Eiz4KHdMtFY6664RDp/go-libp2p/p2p/net/mock"
|
|
ds "gx/ipfs/QmPpegoMqhAEqjncrzArm7KVWAkCm78rqL2DPuNjhPrshg/go-datastore"
|
|
testutil "gx/ipfs/QmVvkK7s5imCiq3JVbL3pGfnhcCnf3LrFJPF4GE2sAoGZf/go-testutil"
|
|
mockrouting "gx/ipfs/QmZRcGYvxdauCd7hHnMYLYqcZRaDjv24c7eUNyJojAcdBb/go-ipfs-routing/mock"
|
|
peer "gx/ipfs/QmZoWKhxUmZ2seW4BzX6fJkNR8hh9PsGModr7q171yq2SS/go-libp2p-peer"
|
|
)
|
|
|
|
type peernet struct {
|
|
mockpeernet.Mocknet
|
|
routingserver mockrouting.Server
|
|
}
|
|
|
|
func StreamNet(ctx context.Context, net mockpeernet.Mocknet, rs mockrouting.Server) (Network, error) {
|
|
return &peernet{net, rs}, nil
|
|
}
|
|
|
|
func (pn *peernet) Adapter(p testutil.Identity) bsnet.BitSwapNetwork {
|
|
client, err := pn.Mocknet.AddPeer(p.PrivateKey(), p.Address())
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
routing := pn.routingserver.ClientWithDatastore(context.TODO(), p, ds.NewMapDatastore())
|
|
return bsnet.NewFromIpfsHost(client, routing)
|
|
}
|
|
|
|
func (pn *peernet) HasPeer(p peer.ID) bool {
|
|
for _, member := range pn.Mocknet.Peers() {
|
|
if p == member {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
var _ Network = (*peernet)(nil)
|