dee2c137c8
* slowly but surely * hardcode ID for tests so we can get predictable results * skipping bad test * removing tests of bkad * wip * new algorithm for worker * clean up * remove skipped test * changes * uncomment * fixed conflicts * maybe done ? * cleanup * boot bkad * wip * cleanup * undo change * fixes * wip * wip * moving nodeID around * wip * wip * fixes * fixes after merge * added TODO * fixed tests post identity * linter fixes * wip * PR review comments * wip * fixing tests * fix tests * force db directory * bad test * fixes race condition * small cleanups * adding db folder * testing * wip * cleanup * cleanup * linters * export Restrict * add timeout * testing * linters * forgot one * moar fixes from master merge * PR comments * moar PR comments * removed stun flag * remove duplicate declaration * remove old tests * remove timeout * fix tests * missed one * changed StringToID >> IDFromString * PR comments * stupid linter * moevd overlay mock * fixed merge conflicts * fixes * linter
66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information
|
|
|
|
package node
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
|
|
"google.golang.org/grpc"
|
|
|
|
"storj.io/storj/pkg/dht"
|
|
"storj.io/storj/pkg/pb"
|
|
"storj.io/storj/pkg/pool"
|
|
"storj.io/storj/pkg/transport"
|
|
)
|
|
|
|
// Node is the storj definition for a node in the network
|
|
type Node struct {
|
|
dht dht.DHT
|
|
self pb.Node
|
|
tc transport.Client
|
|
cache pool.Pool
|
|
}
|
|
|
|
// Lookup queries nodes looking for a particular node in the network
|
|
func (n *Node) Lookup(ctx context.Context, to pb.Node, find pb.Node) ([]*pb.Node, error) {
|
|
v, err := n.cache.Get(ctx, to.GetId())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var conn *grpc.ClientConn
|
|
if c, ok := v.(*grpc.ClientConn); ok {
|
|
conn = c
|
|
} else {
|
|
c, err := n.tc.DialNode(ctx, &to)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := n.cache.Add(ctx, to.GetId(), c); err != nil {
|
|
log.Printf("Error %s occurred adding %s to cache", err, to.GetId())
|
|
}
|
|
conn = c
|
|
}
|
|
|
|
c := pb.NewNodesClient(conn)
|
|
resp, err := c.Query(ctx, &pb.QueryRequest{Limit: 20, Sender: &n.self, Target: &find, Pingback: true})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
rt, err := n.dht.GetRoutingTable(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := rt.ConnectionSuccess(&to); err != nil {
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return resp.Response, nil
|
|
}
|