From 3508e135e6b6a54faa8f73e2fa3197eebe9cf402 Mon Sep 17 00:00:00 2001
From: Michael Crosby <crosbymichael@gmail.com>
Date: Mon, 26 Jun 2017 10:18:01 -0700
Subject: [PATCH] Add NewWithGRPCOpts to specify grpc dial opts

Signed-off-by: Michael Crosby <crosbymichael@gmail.com>
---
 client.go | 24 ++++++++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

diff --git a/client.go b/client.go
index c8c08295d..5f9a3f469 100644
--- a/client.go
+++ b/client.go
@@ -46,7 +46,8 @@ func init() {
 }
 
 type clientOpts struct {
-	defaultns string
+	defaultns   string
+	dialOptions []grpc.DialOption
 }
 
 type ClientOpt func(c *clientOpts) error
@@ -58,6 +59,14 @@ func WithDefaultNamespace(ns string) ClientOpt {
 	}
 }
 
+// WithDialOpts allows grpc.DialOptions to be set on the connection
+func WithDialOpts(opts []grpc.DialOption) ClientOpt {
+	return func(c *clientOpts) error {
+		c.dialOptions = opts
+		return nil
+	}
+}
+
 // New returns a new containerd client that is connected to the containerd
 // instance provided by address
 func New(address string, opts ...ClientOpt) (*Client, error) {
@@ -67,13 +76,15 @@ func New(address string, opts ...ClientOpt) (*Client, error) {
 			return nil, err
 		}
 	}
-
 	gopts := []grpc.DialOption{
 		grpc.WithBlock(),
 		grpc.WithInsecure(),
 		grpc.WithTimeout(100 * time.Second),
-		grpc.WithDialer(dialer),
 		grpc.FailOnNonTempDialError(true),
+		grpc.WithDialer(dialer),
+	}
+	if len(copts.dialOptions) > 0 {
+		gopts = copts.dialOptions
 	}
 	if copts.defaultns != "" {
 		unary, stream := newNSInterceptors(copts.defaultns)
@@ -82,11 +93,16 @@ func New(address string, opts ...ClientOpt) (*Client, error) {
 			grpc.WithStreamInterceptor(stream),
 		)
 	}
-
 	conn, err := grpc.Dial(dialAddress(address), gopts...)
 	if err != nil {
 		return nil, errors.Wrapf(err, "failed to dial %q", address)
 	}
+	return NewWithConn(conn, opts...)
+}
+
+// NewWithConn returns a new containerd client that is connected to the containerd
+// instance provided by the connection
+func NewWithConn(conn *grpc.ClientConn, opts ...ClientOpt) (*Client, error) {
 	return &Client{
 		conn:    conn,
 		runtime: fmt.Sprintf("%s.%s", plugin.RuntimePlugin, runtime.GOOS),