50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
// +build balsam
|
|
|
|
package test
|
|
|
|
import (
|
|
"context"
|
|
"github.com/ipfs/go-datastore"
|
|
"github.com/ipfs/go-ipns"
|
|
"github.com/testground/sdk-go/runtime"
|
|
"github.com/libp2p/go-libp2p"
|
|
"github.com/libp2p/go-libp2p-core/host"
|
|
"github.com/libp2p/go-libp2p-core/peer"
|
|
kaddht "github.com/libp2p/go-libp2p-kad-dht"
|
|
dhtopts "github.com/libp2p/go-libp2p-kad-dht/opts"
|
|
)
|
|
|
|
func createDHT(ctx context.Context, h host.Host, ds datastore.Batching, opts *SetupOpts, info *DHTNodeInfo) (*kaddht.IpfsDHT, error) {
|
|
dhtOptions := []dhtopts.Option{
|
|
dhtopts.Protocols("/testground/kad/1.0.0"),
|
|
dhtopts.Datastore(ds),
|
|
dhtopts.BucketSize(opts.BucketSize),
|
|
dhtopts.RoutingTableRefreshQueryTimeout(opts.Timeout),
|
|
dhtopts.NamespacedValidator("ipns", ipns.Validator{KeyBook: h.Peerstore()}),
|
|
}
|
|
|
|
if !opts.AutoRefresh {
|
|
dhtOptions = append(dhtOptions, dhtopts.DisableAutoRefresh())
|
|
}
|
|
|
|
if info.Properties.Undialable && opts.ClientMode {
|
|
dhtOptions = append(dhtOptions, dhtopts.Client(true))
|
|
}
|
|
|
|
dht, err := kaddht.New(ctx, h, dhtOptions...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return dht, nil
|
|
}
|
|
|
|
func getTaggedLibp2pOpts(opts *SetupOpts, info *DHTNodeInfo) []libp2p.Option { return nil }
|
|
|
|
func getAllProvRecordsNum() int { return 1000 }
|
|
|
|
func specializedTraceQuery(ctx context.Context, runenv *runtime.RunEnv, tag string) context.Context {
|
|
return ctx
|
|
}
|
|
|
|
func TableHealth(dht *kaddht.IpfsDHT, peers map[peer.ID]*DHTNodeInfo, ri *DHTRunInfo) {}
|