Add FetchAvailableDiskTypesInZone method.

Author: ulladz

cluster-autoscaler/cloudprovider/gce/autoscaling_gce_client.go

@@ -131,6 +131,7 @@ type AutoscalingGceClient interface {
 	FetchZones(region string) ([]string, error)
 	FetchAvailableCpuPlatforms() (map[string][]string, error)
 	FetchAvailableDiskTypes(region string) (map[string][]string, error)
+	FetchAvailableDiskTypesInZone(zone string) ([]string, error)
 	FetchReservations() ([]*gce.Reservation, error)
 	FetchReservationsInProject(projectId string) ([]*gce.Reservation, error)
 	FetchListManagedInstancesResults(migRef GceRef) (string, error)
@@ -782,6 +783,23 @@ func (client *autoscalingGceClientV1) FetchAvailableDiskTypes(region string) (ma
 	return availableDiskTypes, nil
 }
 
+// FetchAvailableDiskTypesInZone returns a list of available disk types in a given zone.
+func (client *autoscalingGceClientV1) FetchAvailableDiskTypesInZone(zone string) ([]string, error) {
+	availableDiskTypes := []string{}
+
+	req := client.gceService.DiskTypes.List(client.projectId, zone)
+	if err := req.Pages(context.TODO(), func(page *gce.DiskTypeList) error {
+		for _, diskType := range page.Items {
+			availableDiskTypes = append(availableDiskTypes, diskType.Name)
+		}
+		return nil
+	}); err != nil {
+		return nil, err
+	}
+
+	return availableDiskTypes, nil
+}
+
 func (client *autoscalingGceClientV1) FetchMigTemplateName(migRef GceRef) (InstanceTemplateName, error) {
 	registerRequest("instance_group_managers", "get")
 	ctx, cancel := context.WithTimeout(context.Background(), client.operationPerCallTimeout)

cluster-autoscaler/cloudprovider/gce/mig_info_provider_test.go

@@ -176,6 +176,10 @@ func (client *mockAutoscalingGceClient) FetchAvailableDiskTypes(_ string) (map[s
 	return nil, nil
 }
 
+func (client *mockAutoscalingGceClient) FetchAvailableDiskTypesInZone(_ string) ([]string, error) {
+	return nil, nil
+}
+
 func (client *mockAutoscalingGceClient) FetchReservations() ([]*gce.Reservation, error) {
 	return nil, nil
 }