098cbc9c67
all of the packages and tests work with both grpc and drpc. we'll probably need to do some jenkins pipelines to run the tests with drpc as well. most of the changes are really due to a bit of cleanup of the pkg/transport.Client api into an rpc.Dialer in the spirit of a net.Dialer. now that we don't need observers, we can pass around stateless configuration to everything rather than stateful things that issue observations. it also adds a DialAddressID for the case where we don't have a pb.Node, but we do have an address and want to assert some ID. this happened pretty frequently, and now there's no more weird contortions creating custom tls options, etc. a lot of the other changes are being consistent/using the abstractions in the rpc package to do rpc style things like finding peer information, or checking status codes. Change-Id: Ief62875e21d80a21b3c56a5a37f45887679f9412
115 lines
3.5 KiB
Go
115 lines
3.5 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package contact
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
|
|
"github.com/zeebo/errs"
|
|
"go.uber.org/zap"
|
|
|
|
"storj.io/storj/pkg/identity"
|
|
"storj.io/storj/pkg/pb"
|
|
"storj.io/storj/pkg/rpc/rpcstatus"
|
|
"storj.io/storj/pkg/storj"
|
|
"storj.io/storj/satellite/overlay"
|
|
)
|
|
|
|
// Endpoint implements the contact service Endpoints.
|
|
type Endpoint struct {
|
|
log *zap.Logger
|
|
service *Service
|
|
}
|
|
|
|
// NewEndpoint returns a new contact service endpoint.
|
|
func NewEndpoint(log *zap.Logger, service *Service) *Endpoint {
|
|
return &Endpoint{
|
|
log: log,
|
|
service: service,
|
|
}
|
|
}
|
|
|
|
// CheckIn is periodically called by storage nodes to keep the satellite informed of its existence,
|
|
// address, and operator information. In return, this satellite keeps the node informed of its
|
|
// reachability.
|
|
// When a node checks-in with the satellite, the satellite pings the node back to confirm they can
|
|
// successfully connect.
|
|
func (endpoint *Endpoint) CheckIn(ctx context.Context, req *pb.CheckInRequest) (_ *pb.CheckInResponse, err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
peerID, err := identity.PeerIdentityFromContext(ctx)
|
|
if err != nil {
|
|
return nil, rpcstatus.Error(rpcstatus.Internal, Error.Wrap(err).Error())
|
|
}
|
|
nodeID := peerID.ID
|
|
|
|
err = endpoint.service.peerIDs.Set(ctx, nodeID, peerID)
|
|
if err != nil {
|
|
return nil, rpcstatus.Error(rpcstatus.Internal, Error.Wrap(err).Error())
|
|
}
|
|
|
|
lastIP, err := overlay.GetNetwork(ctx, req.Address)
|
|
if err != nil {
|
|
return nil, rpcstatus.Error(rpcstatus.Internal, Error.Wrap(err).Error())
|
|
}
|
|
|
|
pingNodeSuccess, pingErrorMessage, err := endpoint.pingBack(ctx, req, nodeID)
|
|
if err != nil {
|
|
return nil, rpcstatus.Error(rpcstatus.Internal, Error.Wrap(err).Error())
|
|
}
|
|
nodeInfo := overlay.NodeCheckInInfo{
|
|
NodeID: peerID.ID,
|
|
Address: &pb.NodeAddress{
|
|
Address: req.Address,
|
|
Transport: pb.NodeTransport_TCP_TLS_GRPC,
|
|
},
|
|
LastIP: lastIP,
|
|
IsUp: pingNodeSuccess,
|
|
Capacity: req.Capacity,
|
|
Operator: req.Operator,
|
|
Version: req.Version,
|
|
}
|
|
err = endpoint.service.overlay.UpdateCheckIn(ctx, nodeInfo)
|
|
if err != nil {
|
|
return nil, rpcstatus.Error(rpcstatus.Internal, Error.Wrap(err).Error())
|
|
}
|
|
|
|
endpoint.log.Debug("checking in", zap.String("node addr", req.Address), zap.Bool("ping node succes", pingNodeSuccess))
|
|
return &pb.CheckInResponse{
|
|
PingNodeSuccess: pingNodeSuccess,
|
|
PingErrorMessage: pingErrorMessage,
|
|
}, nil
|
|
}
|
|
|
|
func (endpoint *Endpoint) pingBack(ctx context.Context, req *pb.CheckInRequest, peerID storj.NodeID) (bool, string, error) {
|
|
client, err := newClient(ctx, endpoint.service.dialer, req.Address, peerID)
|
|
if err != nil {
|
|
// if this is a network error, then return the error otherwise just report internal error
|
|
_, ok := err.(net.Error)
|
|
if ok {
|
|
return false, "", Error.New("failed to connect to %s: %v", req.Address, err)
|
|
}
|
|
endpoint.log.Info("pingBack internal error", zap.String("error", err.Error()))
|
|
return false, "", Error.New("couldn't connect to client at addr: %s due to internal error.", req.Address)
|
|
}
|
|
defer func() { err = errs.Combine(err, client.Close()) }()
|
|
|
|
pingNodeSuccess := true
|
|
var pingErrorMessage string
|
|
|
|
_, err = client.pingNode(ctx, &pb.ContactPingRequest{})
|
|
if err != nil {
|
|
pingNodeSuccess = false
|
|
pingErrorMessage = "erroring while trying to pingNode due to internal error"
|
|
_, ok := err.(net.Error)
|
|
if ok {
|
|
pingErrorMessage = fmt.Sprintf("network erroring while trying to pingNode: %v\n", err)
|
|
}
|
|
}
|
|
|
|
return pingNodeSuccess, pingErrorMessage, err
|
|
}
|