storj/satellite/contact/client.go

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.

package contact

import (
	"context"

	"storj.io/common/pb"
	"storj.io/common/rpc"
	"storj.io/common/storj"
)

type client struct {
	conn   *rpc.Conn
	client pb.DRPCContactClient
}

// dialNodeURL dials the target contact endpoint.
func dialNodeURL(ctx context.Context, dialer rpc.Dialer, nodeurl storj.NodeURL) (*client, error) {
	conn, err := dialer.DialNodeURL(ctx, nodeurl)
	if err != nil {
		return nil, err
	}

	return &client{
		conn:   conn,
		client: pb.NewDRPCContactClient(conn),
	}, nil
}

// pingNode pings a node.
func (client *client) pingNode(ctx context.Context, req *pb.ContactPingRequest) (*pb.ContactPingResponse, error) {
	return client.client.PingNode(ctx, req)
}

// Close closes the connection.
func (client *client) Close() error {
	return client.conn.Close()
}