Add VolumeStats for Windows

add support for getting volume stats for windows

Author: jingxu97

cmd/gce-pd-csi-driver/main.go

@@ -96,7 +96,10 @@ func handle() {
 			klog.Fatalf("Failed to get safe mounter: %v", err)
 		}
 		deviceUtils := mountmanager.NewDeviceUtils()
-		statter := mountmanager.NewStatter()
+		statter, err := mountmanager.NewStatter()
+		if err != nil {
+			klog.Fatalf("Failed to set up Statter: %v", err)
+		}
 		meta, err := metadataservice.NewMetadataService()
 		if err != nil {
 			klog.Fatalf("Failed to set up metadata service: %v", err)

pkg/gce-pd-csi-driver/node.go

@@ -385,7 +385,7 @@ func (ns *GCENodeServer) NodeGetVolumeStats(ctx context.Context, req *csi.NodeGe
 		return nil, status.Error(codes.InvalidArgument, "NodeGetVolumeStats volume path was empty")
 	}
 
-	_, err := os.Stat(req.VolumePath)
+	_, err := os.Lstat(req.VolumePath)
 	if err != nil {
 		if os.IsNotExist(err) {
 			return nil, status.Errorf(codes.NotFound, "path %s does not exist", req.VolumePath)
@@ -411,7 +411,6 @@ func (ns *GCENodeServer) NodeGetVolumeStats(ctx context.Context, req *csi.NodeGe
 			},
 		}, nil
 	}
-
 	available, capacity, used, inodesFree, inodes, inodesUsed, err := ns.VolumeStatter.StatFS(req.VolumePath)
 	if err != nil {
 		return nil, status.Errorf(codes.Internal, "failed to get fs info on path %s: %v", req.VolumePath, err)

pkg/gce-pd-csi-driver/utils_windows.go

@@ -17,6 +17,7 @@ package gceGCEDriver
 
 import (
 	"fmt"
+
 	"k8s.io/utils/mount"
 	"sigs.k8s.io/gcp-compute-persistent-disk-csi-driver/pkg/common"
 	mounter "sigs.k8s.io/gcp-compute-persistent-disk-csi-driver/pkg/mount-manager"

pkg/mount-manager/statter_linux.go

@@ -25,8 +25,8 @@ var _ Statter = realStatter{}
 type realStatter struct {
 }
 
-func NewStatter() realStatter {
-	return realStatter{}
+func NewStatter() (realStatter, error) {
+	return realStatter{}, nil
 }
 
 // IsBlock checks if the given path is a block device

pkg/mount-manager/statter_windows.go

@@ -16,27 +16,57 @@ limitations under the License.
 package mountmanager
 
 import (
-	"fmt"
+	"context"
+
+	volumeapi "github.com/kubernetes-csi/csi-proxy/client/api/volume/v1beta1"
+	volumeclient "github.com/kubernetes-csi/csi-proxy/client/groups/volume/v1beta1"
 )
 
 var _ Statter = realStatter{}
 
 type realStatter struct {
+	VolumeClient *volumeclient.Client
 }
 
-func NewStatter() realStatter {
-	return realStatter{}
+func NewStatter() (realStatter, error) {
+	volumeClient, err := volumeclient.NewClient()
+	if err != nil {
+		return realStatter{}, err
+	}
+	return realStatter{
+		VolumeClient: volumeClient,
+	}, nil
 }
 
 // IsBlock checks if the given path is a block device
 func (realStatter) IsBlockDevice(fullPath string) (bool, error) {
 	return false, nil
 }
 
-//TODO (jinxu): implement StatFS to get metrics
-func (realStatter) StatFS(path string) (available, capacity, used, inodesFree, inodes, inodesUsed int64, err error) {
+// StatFS returns volume usage information
+func (r realStatter) StatFS(path string) (available, capacity, used, inodesFree, inodes, inodesUsed int64, err error) {
 	zero := int64(0)
-	return zero, zero, zero, zero, zero, zero, fmt.Errorf("Not implemented")
+
+	idRequest := &volumeapi.VolumeIDFromMountRequest{
+		Mount: path,
+	}
+	idResponse, err := r.VolumeClient.GetVolumeIDFromMount(context.Background(), idRequest)
+	if err != nil {
+		return zero, zero, zero, zero, zero, zero, err
+	}
+	volumeId := idResponse.GetVolumeId()
+
+	request := &volumeapi.VolumeStatsRequest{
+		VolumeId: volumeId,
+	}
+	response, err := r.VolumeClient.VolumeStats(context.Background(), request)
+	if err != nil {
+		return zero, zero, zero, zero, zero, zero, err
+	}
+	capacity = response.GetVolumeSize()
+	used = response.GetVolumeUsedSize()
+	available = capacity - used
+	return available, capacity, used, zero, zero, zero, nil
 }
 
 type fakeStatter struct{}