a2fa5c4c5a
* adds enums to nodetype * updating nodetype todos * ran pb updates * reorder nodetypes * adding checks * wip * wip * wip * bug in test-captplanet * wip * add values to storagenode, satellite, captplanet binaries * Cleanup * more cleanup * wip * lint * lint * wip * fixes bug * regenerate protos Change-Id: Id270212e8c7479e52641058042cf23b5317ab773 * limit node type changes to kademlia Change-Id: I9c1a6cc4a79e05086627f0fdeb5028c62ce754f4 * dpanic Change-Id: Id952a2ad13c807ebaea0ec0a875405e267d81c3e * review comments Change-Id: I7f9b77ef22779dd012fd490375b136014f51f834
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information
|
|
|
|
package node
|
|
|
|
import (
|
|
"context"
|
|
|
|
"storj.io/storj/pkg/dht"
|
|
"storj.io/storj/pkg/pb"
|
|
)
|
|
|
|
// Node is the storj definition for a node in the network
|
|
type Node struct {
|
|
dht dht.DHT
|
|
self pb.Node
|
|
pool *ConnectionPool
|
|
}
|
|
|
|
// Lookup queries nodes looking for a particular node in the network
|
|
func (node *Node) Lookup(ctx context.Context, to pb.Node, find pb.Node) ([]*pb.Node, error) {
|
|
to.Type.DPanicOnInvalid("node Lookup")
|
|
conn, err := node.pool.Dial(ctx, &to)
|
|
|
|
if err != nil {
|
|
return nil, NodeClientErr.Wrap(err)
|
|
}
|
|
resp, err := conn.Query(ctx, &pb.QueryRequest{
|
|
Limit: 20,
|
|
Sender: &node.self,
|
|
Target: &find,
|
|
Pingback: true,
|
|
})
|
|
if err != nil {
|
|
return nil, NodeClientErr.Wrap(err)
|
|
}
|
|
rt, err := node.dht.GetRoutingTable(ctx)
|
|
if err != nil {
|
|
return nil, NodeClientErr.Wrap(err)
|
|
}
|
|
if err := rt.ConnectionSuccess(&to); err != nil {
|
|
return nil, NodeClientErr.Wrap(err)
|
|
}
|
|
return resp.Response, nil
|
|
}
|
|
|
|
// Ping attempts to establish a connection with a node to verify it is alive
|
|
func (node *Node) Ping(ctx context.Context, to pb.Node) (bool, error) {
|
|
to.Type.DPanicOnInvalid("node ping")
|
|
conn, err := node.pool.Dial(ctx, &to)
|
|
if err != nil {
|
|
return false, NodeClientErr.Wrap(err)
|
|
}
|
|
_, err = conn.Ping(ctx, &pb.PingRequest{})
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return true, nil
|
|
}
|
|
|
|
// Disconnect closes all connections within the pool
|
|
func (node *Node) Disconnect() error {
|
|
return node.pool.DisconnectAll()
|
|
}
|