Fix some leaks and add notes about close handling (#2334)

cmd/statreceiver/graphite.go

@@ -43,6 +43,7 @@ func (d *GraphiteDest) Metric(application, instance string, key []byte, val floa
 		if err != nil {
 			return err
 		}
+		// TODO(leak): free connection
 		d.conn = conn
 		d.buf = bufio.NewWriter(conn)
 	}

cmd/storagenode/dashboard.go

@@ -17,6 +17,7 @@ import (
 	"github.com/golang/protobuf/ptypes"
 	"github.com/spf13/cobra"
 	"go.uber.org/zap"
+	"google.golang.org/grpc"
 
 	"storj.io/storj/internal/memory"
 	"storj.io/storj/internal/version"
@@ -29,13 +30,10 @@ const contactWindow = time.Minute * 10
 
 type dashboardClient struct {
 	client pb.PieceStoreInspectorClient
+	conn   *grpc.ClientConn
 }
 
-func (dash *dashboardClient) dashboard(ctx context.Context) (*pb.DashboardResponse, error) {
+func dialDashboardClient(ctx context.Context, address string) (*dashboardClient, error) {
-	return dash.client.Dashboard(ctx, &pb.DashboardRequest{})
-}
-
-func newDashboardClient(ctx context.Context, address string) (*dashboardClient, error) {
 	conn, err := transport.DialAddressInsecure(ctx, address)
 	if err != nil {
 		return &dashboardClient{}, err
@@ -43,9 +41,18 @@ func newDashboardClient(ctx context.Context, address string) (*dashboardClient,
 
 	return &dashboardClient{
 		client: pb.NewPieceStoreInspectorClient(conn),
+		conn:   conn,
 	}, nil
 }
 
+func (dash *dashboardClient) dashboard(ctx context.Context) (*pb.DashboardResponse, error) {
+	return dash.client.Dashboard(ctx, &pb.DashboardRequest{})
+}
+
+func (dash *dashboardClient) close() error {
+	return dash.conn.Close()
+}
+
 func cmdDashboard(cmd *cobra.Command, args []string) (err error) {
 	ctx := process.Ctx(cmd)
 
@@ -56,10 +63,15 @@ func cmdDashboard(cmd *cobra.Command, args []string) (err error) {
 		zap.S().Info("Node ID: ", ident.ID)
 	}
 
-	client, err := newDashboardClient(ctx, dashboardCfg.Address)
+	client, err := dialDashboardClient(ctx, dashboardCfg.Address)
 	if err != nil {
 		return err
 	}
+	defer func() {
+		if err := client.close(); err != nil {
+			zap.S().Debug("closing dashboard client failed", err)
+		}
+	}()
 
 	for {
 		data, err := client.dashboard(ctx)

internal/testplanet/uplink.go

@@ -162,6 +162,7 @@ func (uplink *Uplink) DialPiecestore(ctx context.Context, destination Peer) (*pi
 
 	signer := signing.SignerFromFullIdentity(uplink.Transport.Identity())
 
+	// TODO(leak): unclear ownership semantics
 	return piecestore.NewClient(uplink.Log.Named("uplink>piecestore"), signer, conn, piecestore.DefaultConfig), nil
 }
 

pkg/audit/verifier.go

@@ -476,6 +476,7 @@ func (verifier *Verifier) GetShare(ctx context.Context, limit *pb.AddressedOrder
 	if err != nil {
 		return Share{}, err
 	}
+	// TODO(leak): unclear ownership semantics
 	ps := piecestore.NewClient(
 		verifier.log.Named(storageNodeID.String()),
 		signing.SignerFromFullIdentity(verifier.transport.Identity()),

pkg/peertls/tlsopts/options_test.go

@@ -139,6 +139,8 @@ func TestOptions_ServerOption_Peer_CA_Whitelist(t *testing.T) {
 		conn, err := transportClient.DialNode(ctx, &target.Node, dialOption)
 		assert.NotNil(t, conn)
 		assert.NoError(t, err)
+
+		assert.NoError(t, conn.Close())
 	})
 }
 

pkg/storage/ec/client.go

@@ -62,6 +62,7 @@ func (ec *ecClient) newPSClient(ctx context.Context, n *pb.Node) (*piecestore.Cl
 	if err != nil {
 		return nil, err
 	}
+	// TODO(leak): unclear ownership semantics
 	return piecestore.NewClient(
 		zap.L().Named(n.Id.String()),
 		signing.SignerFromFullIdentity(ec.transport.Identity()),

satellite/vouchers/vouchers_test.go

@@ -55,6 +55,7 @@ func TestVouchers(t *testing.T) {
 
 			conn, err := tt.node.Transport.DialNode(ctx, &satellite)
 			require.NoError(t, err)
+			defer ctx.Check(conn.Close)
 
 			client := pb.NewVouchersClient(conn)
 

storagenode/piecestore/endpoint_test.go

@@ -99,6 +99,7 @@ func TestUpload(t *testing.T) {
 
 	client, err := planet.Uplinks[0].DialPiecestore(ctx, planet.StorageNodes[0])
 	require.NoError(t, err)
+	defer ctx.Check(client.Close)
 
 	for _, tt := range []struct {
 		pieceID       storj.PieceID
@@ -183,6 +184,7 @@ func TestDownload(t *testing.T) {
 	// upload test piece
 	client, err := planet.Uplinks[0].DialPiecestore(ctx, planet.StorageNodes[0])
 	require.NoError(t, err)
+	defer ctx.Check(client.Close)
 
 	var serialNumber storj.SerialNumber
 	_, _ = rand.Read(serialNumber[:])
@@ -289,6 +291,7 @@ func TestDelete(t *testing.T) {
 	// upload test piece
 	client, err := planet.Uplinks[0].DialPiecestore(ctx, planet.StorageNodes[0])
 	require.NoError(t, err)
+	defer ctx.Check(client.Close)
 
 	var serialNumber storj.SerialNumber
 	_, _ = rand.Read(serialNumber[:])

storagenode/piecestore/verification_test.go

@@ -129,6 +129,7 @@ func TestOrderLimitPutValidation(t *testing.T) {
 
 		client, err := planet.Uplinks[0].DialPiecestore(ctx, planet.StorageNodes[0])
 		require.NoError(t, err)
+		defer ctx.Check(client.Close)
 
 		signer := signing.SignerFromFullIdentity(planet.Satellites[0].Identity)
 		satellite := planet.Satellites[0].Identity
@@ -199,6 +200,7 @@ func TestOrderLimitGetValidation(t *testing.T) {
 	{ // upload test piece
 		client, err := planet.Uplinks[0].DialPiecestore(ctx, planet.StorageNodes[0])
 		require.NoError(t, err)
+		defer ctx.Check(client.Close)
 
 		signer := signing.SignerFromFullIdentity(planet.Satellites[0].Identity)
 		satellite := planet.Satellites[0].Identity
@@ -253,6 +255,7 @@ func TestOrderLimitGetValidation(t *testing.T) {
 	} {
 		client, err := planet.Uplinks[0].DialPiecestore(ctx, planet.StorageNodes[0])
 		require.NoError(t, err)
+		defer ctx.Check(client.Close)
 
 		signer := signing.SignerFromFullIdentity(planet.Satellites[0].Identity)
 		satellite := planet.Satellites[0].Identity

uplink/piecestore/client.go

@@ -51,6 +51,7 @@ func NewClient(log *zap.Logger, signer signing.Signer, conn *grpc.ClientConn, co
 	return &Client{
 		log:    log,
 		signer: signer,
+		// TODO(leak): unclear ownership semantics
 		conn:   conn,
 		client: pb.NewPiecestoreClient(conn),
 		config: config,