Merge pull request #719 from erikreppel/close-client-connection

Add gRPC conn as prop of the ApiClient struct so it can be closed at a later time

src/client/client.go

@@ -29,6 +29,7 @@ type APIClient struct {
 	PfsAPIClient
 	PpsAPIClient
 	BlockAPIClient
+	gRPCConn *grpc.ClientConn
 }
 
 // NewFromAddress constructs a new APIClient for the server at pachAddr.
@@ -42,6 +43,7 @@ func NewFromAddress(pachAddr string) (*APIClient, error) {
 		pfs.NewAPIClient(clientConn),
 		pps.NewAPIClient(clientConn),
 		pfs.NewBlockAPIClient(clientConn),
+		clientConn,
 	}, nil
 }
 
@@ -58,6 +60,11 @@ func NewInCluster() (*APIClient, error) {
 	return NewFromAddress(fmt.Sprintf("%v:650", pachAddr))
 }
 
+// Close the connection to gRPC
+func (c *APIClient) Close() error {
+	return c.gRPCConn.Close()
+}
+
 // DeleteAll deletes everything in the cluster.
 // Use with caution, there is no undo.
 func (c APIClient) DeleteAll() error {