Merge pull request #40 from mtougeron/fix-attribute-path

Fix the path to the VolumeHandle field
mtougeron · Feb 9, 2022 · cfb1345 · cfb1345
2 parents c54c9bc + 7bb4406
commit cfb1345
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/kubernetes.go b/kubernetes.go
@@ -50,7 +50,7 @@ const (
 	regexpAWSVolumeID = `^aws:\/\/\w{2}-\w{4,9}-\d\w\/(vol-\w+)$`
 )
 
-func buildClient(kubeconfig string, kubeContext string) (*kubernetes.Clientset, error) {
+func BuildClient(kubeconfig string, kubeContext string) (*kubernetes.Clientset, error) {
 	config, err := rest.InClusterConfig()
 	if err != nil {
 		if kubeconfig == "" {
@@ -258,7 +258,7 @@ func processPersistentVolumeClaim(pvc *corev1.PersistentVolumeClaim) (string, ma
 		log.Errorf("cannot get volume.beta.kubernetes.io/storage-provisioner annotation")
 		return "", nil, errors.New("cannot get volume.beta.kubernetes.io/storage-provisioner annotation")
 	} else if provisionedBy == "ebs.csi.aws.com" {
-		volumeID = pv.Spec.CSI.VolumeHandle
+		volumeID = pv.Spec.PersistentVolumeSource.CSI.VolumeHandle
 	} else if provisionedBy == "kubernetes.io/aws-ebs" {
 		volumeID = parseAWSVolumeID(pv.Spec.PersistentVolumeSource.AWSElasticBlockStore.VolumeID)
 	}

diff --git a/main.go b/main.go
@@ -161,7 +161,7 @@ func main() {
 		os.Exit(1)
 	}
 
-	k8sClient, err = buildClient(kubeconfig, kubeContext)
+	k8sClient, err = BuildClient(kubeconfig, kubeContext)
 	if err != nil {
 		log.Fatalln("Unable to create kubernetes client", err)
 		os.Exit(1)