From 6f417cf0f160126b1ba146f3c272cb66a7e4d39a Mon Sep 17 00:00:00 2001
From: NickrenREN <renyuquan@caicloud.io>
Date: Tue, 13 Mar 2018 20:51:18 +0800
Subject: [PATCH] PluginCapability update for external-provisioner

---
 pkg/controller/controller.go | 36 +++++++++++++++++++++++++++++++++++-
 1 file changed, 35 insertions(+), 1 deletion(-)

diff --git a/pkg/controller/controller.go b/pkg/controller/controller.go
index e48afb0aa3..6af7b68b09 100644
--- a/pkg/controller/controller.go
+++ b/pkg/controller/controller.go
@@ -131,6 +131,33 @@ func getDriverName(conn *grpc.ClientConn, timeout time.Duration) (string, error)
 	return name, nil
 }
 
+func supportsPluginControllerService(conn *grpc.ClientConn, timeout time.Duration) (bool, error) {
+	ctx, cancel := context.WithTimeout(context.Background(), timeout)
+	defer cancel()
+
+	client := csi.NewIdentityClient(conn)
+	req := csi.GetPluginCapabilitiesRequest{}
+
+	rsp, err := client.GetPluginCapabilities(ctx, &req)
+	if err != nil {
+		return false, err
+	}
+	caps := rsp.GetCapabilities()
+	for _, cap := range caps {
+		if cap == nil {
+			continue
+		}
+		service := cap.GetService()
+		if service == nil {
+			continue
+		}
+		if service.GetType() == csi.PluginCapability_Service_CONTROLLER_SERVICE {
+			return true, nil
+		}
+	}
+	return false, nil
+}
+
 func supportsControllerCreateVolume(conn *grpc.ClientConn, timeout time.Duration) (bool, error) {
 	ctx, cancel := context.WithTimeout(context.Background(), timeout)
 	defer cancel()
@@ -163,7 +190,14 @@ func NewCSIProvisioner(client kubernetes.Interface, csiEndpoint string, connecti
 	if err != nil || grpcClient == nil {
 		glog.Fatalf("failed to connect to csi endpoint :%v", err)
 	}
-	ok, err := supportsControllerCreateVolume(grpcClient, connectionTimeout)
+	ok, err := supportsPluginControllerService(grpcClient, connectionTimeout)
+	if err != nil {
+		glog.Fatalf("failed to get support info :%v", err)
+	}
+	if !ok {
+		glog.Fatalf("no plugin controller service support detected")
+	}
+	ok, err = supportsControllerCreateVolume(grpcClient, connectionTimeout)
 	if err != nil {
 		glog.Fatalf("failed to get support info :%v", err)
 	}