8d779d3d3e
* disconnect from nodeclient * cleanup connections in tests * kademlia disconnects from nodeclient * updating disconnect method for mocks * creates separate disconnect and removeAll methods for tests * adds init to connection pool * fix folder cleanup and disconnect * creates and cleans up test db files and disconnects kad * removes db/.keep * includes disconnect within cleanup methods * creates public init method on connection pool to handle mutex copy issues * remove all after disconnect * pair creation and destruction * checks disconnect error * remove ctx * fixes mock kad
37 lines
881 B
Go
37 lines
881 B
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information
|
|
|
|
package node
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/zeebo/errs"
|
|
"storj.io/storj/pkg/dht"
|
|
"storj.io/storj/pkg/pb"
|
|
"storj.io/storj/pkg/provider"
|
|
"storj.io/storj/pkg/transport"
|
|
)
|
|
|
|
//NodeClientErr is the class for all errors pertaining to node client operations
|
|
var NodeClientErr = errs.Class("node client error")
|
|
|
|
// NewNodeClient instantiates a node client
|
|
func NewNodeClient(identity *provider.FullIdentity, self pb.Node, dht dht.DHT) (Client, error) {
|
|
client := transport.NewClient(identity)
|
|
node := &Node{
|
|
dht: dht,
|
|
self: self,
|
|
tc: client,
|
|
cache: NewConnectionPool(),
|
|
}
|
|
node.cache.Init()
|
|
return node, nil
|
|
}
|
|
|
|
// Client is the Node client communication interface
|
|
type Client interface {
|
|
Lookup(ctx context.Context, to pb.Node, find pb.Node) ([]*pb.Node, error)
|
|
Disconnect() error
|
|
}
|