Return the IOPS of the candidate which is chosen in the storage distribution algorithm.

Signed-off-by: Aditya Dani <aditya@portworx.com>

Author: adityadani

azure/azure_storage_management_test.go

@@ -51,6 +51,7 @@ func TestAzureStorageDistribution(t *testing.T) {
 				DriveType:        "Premium_LRS",
 				InstancesPerZone: 2,
 				DriveCount:       1,
+				IOPS:             1100,
 			},
 		},
 	}
@@ -83,6 +84,7 @@ func TestAzureStorageDistribution(t *testing.T) {
 				DriveType:        "Premium_LRS",
 				InstancesPerZone: 1,
 				DriveCount:       1,
+				IOPS:             1100,
 			},
 		},
 	}
@@ -113,6 +115,7 @@ func TestAzureStorageDistribution(t *testing.T) {
 				DriveType:        "Premium_LRS",
 				InstancesPerZone: 3,
 				DriveCount:       1,
+				IOPS:             5000,
 			},
 		},
 	}
@@ -143,6 +146,7 @@ func TestAzureStorageDistribution(t *testing.T) {
 				DriveType:        "Premium_LRS",
 				InstancesPerZone: 2,
 				DriveCount:       1,
+				IOPS:             2300,
 			},
 		},
 	}
@@ -174,6 +178,7 @@ func TestAzureStorageDistribution(t *testing.T) {
 				DriveType:        "Premium_LRS",
 				InstancesPerZone: 2,
 				DriveCount:       1,
+				IOPS:             2300,
 			},
 		},
 	}
@@ -204,6 +209,7 @@ func TestAzureStorageDistribution(t *testing.T) {
 				DriveType:        "Premium_LRS",
 				InstancesPerZone: 1,
 				DriveCount:       1,
+				IOPS:             7500,
 			},
 		},
 	}
@@ -257,12 +263,14 @@ func TestAzureStorageDistribution(t *testing.T) {
 				DriveType:        "Premium_LRS",
 				InstancesPerZone: 1,
 				DriveCount:       1,
+				IOPS:             1100,
 			},
 			&cloudops.StoragePoolSpec{
 				DriveCapacityGiB: 1024,
 				DriveType:        "Premium_LRS",
 				InstancesPerZone: 3,
 				DriveCount:       1,
+				IOPS:             5000,
 			},
 		},
 	}

cloud_storage_management.go

@@ -93,6 +93,8 @@ type StoragePoolSpec struct {
 	DriveCount uint32 `json:"drive_count" yaml:"drive_count"`
 	// InstancesPerZone is the number of instances per zone
 	InstancesPerZone int `json:"instances_per_zone" yaml:"instances_per_zone"`
+	// IOPS is the IOPS of the drive
+	IOPS uint32 `json:"iops" yaml:"iops"`
 }
 
 // StorageDistributionResponse is the result returned the CloudStorage Decision Matrix

common/storage_distribution.go

@@ -64,6 +64,7 @@ func GetStorageDistribution(
 				DriveType:        candidate.DriveType,
 				InstancesPerZone: instancePerZone,
 				DriveCount:       candidate.InstanceMinDrives,
+				IOPS:             candidate.IOPS,
 			},
 		)