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
63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package node
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.uber.org/zap"
|
|
|
|
"storj.io/storj/pkg/dht"
|
|
"storj.io/storj/pkg/pb"
|
|
)
|
|
|
|
// Server implements the grpc Node Server
|
|
type Server struct {
|
|
dht dht.DHT
|
|
logger *zap.Logger
|
|
}
|
|
|
|
// NewServer returns a newly instantiated Node Server
|
|
func NewServer(dht dht.DHT) *Server {
|
|
return &Server{
|
|
dht: dht,
|
|
logger: zap.L(),
|
|
}
|
|
}
|
|
|
|
// Query is a node to node communication query
|
|
func (s *Server) Query(ctx context.Context, req *pb.QueryRequest) (*pb.QueryResponse, error) {
|
|
if s.logger == nil {
|
|
s.logger = zap.L()
|
|
}
|
|
rt, err := s.dht.GetRoutingTable(ctx)
|
|
if err != nil {
|
|
return &pb.QueryResponse{}, NodeClientErr.New("could not get routing table %s", err)
|
|
}
|
|
|
|
if req.GetPingback() {
|
|
_, err = s.dht.Ping(ctx, *req.Sender)
|
|
if err != nil {
|
|
err = rt.ConnectionFailed(req.Sender)
|
|
if err != nil {
|
|
s.logger.Error("could not respond to connection failed", zap.Error(err))
|
|
}
|
|
s.logger.Error("connection to node failed", zap.Error(err), zap.String("nodeID", req.Sender.Id))
|
|
}
|
|
|
|
err = rt.ConnectionSuccess(req.Sender)
|
|
if err != nil {
|
|
s.logger.Error("could not respond to connection success", zap.Error(err))
|
|
}
|
|
}
|
|
|
|
id := IDFromString(req.Target.Id)
|
|
nodes, err := rt.FindNear(id, int(req.Limit))
|
|
if err != nil {
|
|
return &pb.QueryResponse{}, NodeClientErr.New("could not find near %s", err)
|
|
}
|
|
|
|
return &pb.QueryResponse{Sender: req.Sender, Response: nodes}, nil
|
|
}
|