migrate hyperdisk/chd/storagepools to GCE v1 disk API

Author: amacaskill

pkg/gce-cloud-provider/compute/gce-compute.go

@@ -419,7 +419,7 @@ func convertV1CustomerEncryptionKeyToBeta(v1Key *computev1.CustomerEncryptionKey
 	}
 }
 
-func convertV1DiskToBetaDisk(v1Disk *computev1.Disk, provisionedThroughputOnCreate int64) *computebeta.Disk {
+func convertV1DiskToBetaDisk(v1Disk *computev1.Disk) *computebeta.Disk {
 	var dek *computebeta.CustomerEncryptionKey = nil
 
 	if v1Disk.DiskEncryptionKey != nil {
@@ -447,8 +447,8 @@ func convertV1DiskToBetaDisk(v1Disk *computev1.Disk, provisionedThroughputOnCrea
 	if v1Disk.ProvisionedIops > 0 {
 		betaDisk.ProvisionedIops = v1Disk.ProvisionedIops
 	}
-	if provisionedThroughputOnCreate > 0 {
-		betaDisk.ProvisionedThroughput = provisionedThroughputOnCreate
+	if v1Disk.ProvisionedThroughput > 0 {
+		betaDisk.ProvisionedThroughput = v1Disk.ProvisionedThroughput
 	}
 
 	return betaDisk
@@ -543,7 +543,7 @@ func (cloud *CloudProvider) insertRegionalDisk(
 
 	if gceAPIVersion == GCEAPIVersionBeta {
 		var insertOp *computebeta.Operation
-		betaDiskToCreate := convertV1DiskToBetaDisk(diskToCreate, 0)
+		betaDiskToCreate := convertV1DiskToBetaDisk(diskToCreate)
 		betaDiskToCreate.MultiWriter = multiWriter
 		insertOp, err = cloud.betaService.RegionDisks.Insert(project, volKey.Region, betaDiskToCreate).Context(ctx).Do()
 		if insertOp != nil {
@@ -619,13 +619,9 @@ func (cloud *CloudProvider) insertZonalDisk(
 		opName        string
 		gceAPIVersion = GCEAPIVersionV1
 	)
-	if multiWriter || containsBetaDiskType(hyperdiskTypes, params.DiskType) {
+	if multiWriter {
 		gceAPIVersion = GCEAPIVersionBeta
 	}
-	storagePoolsEnabled := params.StoragePools != nil
-	if storagePoolsEnabled {
-		gceAPIVersion = GCEAPIVersionAlpha
-	}
 
 	diskToCreate := &computev1.Disk{
 		Name:        volKey.Name,
@@ -638,6 +634,17 @@ func (cloud *CloudProvider) insertZonalDisk(
 	if params.ProvisionedIOPSOnCreate > 0 {
 		diskToCreate.ProvisionedIops = params.ProvisionedIOPSOnCreate
 	}
+	if params.ProvisionedThroughputOnCreate > 0 {
+		diskToCreate.ProvisionedThroughput = params.ProvisionedThroughputOnCreate
+	}
+
+	if params.StoragePools != nil {
+		sp := common.StoragePoolInZone(params.StoragePools, volKey.Zone)
+		if sp == nil {
+			return status.Errorf(codes.InvalidArgument, "cannot create disk in zone %q: no Storage Pools exist in zone", volKey.Zone)
+		}
+		diskToCreate.StoragePool = sp.ResourceName
+	}
 
 	if snapshotID != "" {
 		_, snapshotType, _, err := common.SnapshotIDToProjectKey(snapshotID)
@@ -662,20 +669,20 @@ func (cloud *CloudProvider) insertZonalDisk(
 			KmsKeyName: params.DiskEncryptionKMSKey,
 		}
 	}
+	diskToCreate.EnableConfidentialCompute = params.EnableConfidentialCompute
 
 	if gceAPIVersion == GCEAPIVersionBeta {
 		var insertOp *computebeta.Operation
-		betaDiskToCreate := convertV1DiskToBetaDisk(diskToCreate, params.ProvisionedThroughputOnCreate)
+		betaDiskToCreate := convertV1DiskToBetaDisk(diskToCreate)
 		betaDiskToCreate.MultiWriter = multiWriter
-		betaDiskToCreate.EnableConfidentialCompute = params.EnableConfidentialCompute
 		insertOp, err = cloud.betaService.Disks.Insert(project, volKey.Zone, betaDiskToCreate).Context(ctx).Do()
 		if insertOp != nil {
 			opName = insertOp.Name
 		}
 	} else if gceAPIVersion == GCEAPIVersionAlpha {
 		var insertOp *computealpha.Operation
 		var storagePool *common.StoragePool
-		if storagePoolsEnabled {
+		if params.StoragePools != nil {
 			storagePool = common.StoragePoolInZone(params.StoragePools, volKey.Zone)
 			if storagePool == nil {
 				return status.Errorf(codes.InvalidArgument, "cannot create disk in zone %q: no Storage Pools exist in zone", volKey.Zone)