[manila-csi-plugin] Add NodeGetVolumeStats (kubernetes#1737)

NodeGetVolumeStats is now forwarded to the partner plugin if it supports
GET_VOLUME_STATS Node Service capability.

Author: gman0

pkg/csi/manila/csiclient/client.go

@@ -37,6 +37,10 @@ func (c *NodeSvcClient) GetCapabilities(ctx context.Context) (*csi.NodeGetCapabi
 	return c.cl.NodeGetCapabilities(ctx, &csi.NodeGetCapabilitiesRequest{})
 }
 
+func (c *NodeSvcClient) GetVolumeStats(ctx context.Context, req *csi.NodeGetVolumeStatsRequest) (*csi.NodeGetVolumeStatsResponse, error) {
+	return c.cl.NodeGetVolumeStats(ctx, req)
+}
+
 func (c *NodeSvcClient) StageVolume(ctx context.Context, req *csi.NodeStageVolumeRequest) (*csi.NodeStageVolumeResponse, error) {
 	return c.cl.NodeStageVolume(ctx, req)
 }

pkg/csi/manila/csiclient/interface.go

@@ -25,6 +25,7 @@ import (
 
 type Node interface {
 	GetCapabilities(ctx context.Context) (*csi.NodeGetCapabilitiesResponse, error)
+	GetVolumeStats(ctx context.Context, req *csi.NodeGetVolumeStatsRequest) (*csi.NodeGetVolumeStatsResponse, error)
 
 	StageVolume(ctx context.Context, req *csi.NodeStageVolumeRequest) (*csi.NodeStageVolumeResponse, error)
 	UnstageVolume(ctx context.Context, req *csi.NodeUnstageVolumeRequest) (*csi.NodeUnstageVolumeResponse, error)

pkg/csi/manila/nodeserver.go

@@ -337,7 +337,13 @@ func (ns *nodeServer) NodeGetCapabilities(ctx context.Context, req *csi.NodeGetC
 }
 
 func (ns *nodeServer) NodeGetVolumeStats(ctx context.Context, req *csi.NodeGetVolumeStatsRequest) (*csi.NodeGetVolumeStatsResponse, error) {
-	return nil, status.Error(codes.Unimplemented, "")
+	csiConn, err := ns.d.csiClientBuilder.NewConnectionWithContext(ctx, ns.d.fwdEndpoint)
+	if err != nil {
+		return nil, status.Error(codes.Unavailable, fmtGrpcConnError(ns.d.fwdEndpoint, err))
+	}
+	defer csiConn.Close()
+
+	return ns.d.csiClientBuilder.NewNodeServiceClient(csiConn).GetVolumeStats(ctx, req)
 }
 
 func (ns *nodeServer) NodeExpandVolume(ctx context.Context, req *csi.NodeExpandVolumeRequest) (*csi.NodeExpandVolumeResponse, error) {

tests/sanity/manila/fakecsiclient.go

@@ -22,6 +22,8 @@ import (
 
 	"github.com/container-storage-interface/spec/lib/go/csi"
 	"google.golang.org/grpc"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
 	"k8s.io/cloud-provider-openstack/pkg/csi/manila/csiclient"
 )
 
@@ -62,6 +64,10 @@ func (c fakeNodeSvcClient) GetCapabilities(context.Context) (*csi.NodeGetCapabil
 	}, nil
 }
 
+func (c fakeNodeSvcClient) GetVolumeStats(ctx context.Context, req *csi.NodeGetVolumeStatsRequest) (*csi.NodeGetVolumeStatsResponse, error) {
+	return nil, status.Error(codes.Unimplemented, "")
+}
+
 func (c fakeNodeSvcClient) StageVolume(context.Context, *csi.NodeStageVolumeRequest) (*csi.NodeStageVolumeResponse, error) {
 	return &csi.NodeStageVolumeResponse{}, nil
 }