mirror of
https://github.com/ipfs/kubo.git
synced 2025-09-09 19:32:24 +08:00
ca32a83394
@whyrusleeping @jbenet this is a WIP with the DHT. wip License: MIT Signed-off-by: Brian Tiger Chow <brian@perfmode.com> Conflicts: epictest/addcat_test.go exchange/bitswap/testnet/peernet.go exchange/bitswap/testutils.go routing/mock/centralized_server.go routing/mock/centralized_test.go routing/mock/interface.go fix(routing/mock) fill in function definition
45 lines
1.2 KiB
Go
45 lines
1.2 KiB
Go
package bitswap
|
|
|
|
import (
|
|
context "github.com/jbenet/go-ipfs/Godeps/_workspace/src/code.google.com/p/go.net/context"
|
|
ds "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-datastore"
|
|
bsnet "github.com/jbenet/go-ipfs/exchange/bitswap/network"
|
|
mockpeernet "github.com/jbenet/go-ipfs/net/mock"
|
|
peer "github.com/jbenet/go-ipfs/peer"
|
|
mockrouting "github.com/jbenet/go-ipfs/routing/mock"
|
|
testutil "github.com/jbenet/go-ipfs/util/testutil"
|
|
)
|
|
|
|
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.Peer) bsnet.BitSwapNetwork {
|
|
peers := pn.Mocknet.Peers()
|
|
client, err := pn.Mocknet.AddPeer(p.PrivateKey(), p.Address())
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
for _, other := range peers {
|
|
pn.Mocknet.LinkPeers(p.ID(), other)
|
|
}
|
|
routing := pn.routingserver.ClientWithDatastore(context.TODO(), p, ds.NewMapDatastore())
|
|
return bsnet.NewFromIpfsNetwork(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{}
|