/**
* Creates host documents in cloudstore.
*
* @param cloudstore CloudStore test environment to create documents in.
* @param numHosts The number of host documents to create.
* @param hostConfigurations A map from {@link HostConfiguration} to the probability that this
* host configuration is used in the deployment. The sum of all the values of this map must be
* 1.
* @param numDatastores The number of datastores.
* @param numDatastoresDistribution Distribution for number of datastores on each host. This
* distribution is expected to generate samples in the range [0, numDatastores].
* @throws Throwable
*/
public static void loadHosts(
TestEnvironment cloudstore,
int numHosts,
Map<HostConfiguration, Double> hostConfigurations,
int numDatastores,
IntegerDistribution numDatastoresDistribution)
throws Throwable {
int[] indices = new int[hostConfigurations.size()];
HostConfiguration[] configs = new HostConfiguration[hostConfigurations.size()];
double[] probabilities = new double[hostConfigurations.size()];
int i = 0;
for (Map.Entry<HostConfiguration, Double> entry : hostConfigurations.entrySet()) {
indices[i] = i;
configs[i] = entry.getKey();
probabilities[i] = entry.getValue();
i++;
}
EnumeratedIntegerDistribution configDistribution =
new EnumeratedIntegerDistribution(indices, probabilities);
for (i = 0; i < numHosts; i++) {
HostService.State host = new HostService.State();
host.hostAddress = "host" + i;
host.state = HostState.READY;
host.userName = "******";
host.password = "******";
host.reportedDatastores = new HashSet<>();
int numDatastoresPerHost = numDatastoresDistribution.sample();
assertThat(numDatastoresPerHost >= 0, is(true));
assertThat(numDatastoresPerHost <= numDatastores, is(true));
while (host.reportedDatastores.size() < numDatastoresPerHost) {
int randomInt = random.nextInt(numDatastores);
host.reportedDatastores.add(new UUID(0, randomInt).toString());
}
host.reportedNetworks = new HashSet<>();
host.usageTags = new HashSet<>(Arrays.asList(UsageTag.CLOUD.name()));
int configIndex = configDistribution.sample();
host.cpuCount = configs[configIndex].numCpus;
host.memoryMb = configs[configIndex].memoryMb;
host.documentSelfLink = new UUID(0, i).toString();
// TODO(mmutsuzaki) Support availability zones.
Operation result = cloudstore.sendPostAndWait(HostServiceFactory.SELF_LINK, host);
assertThat(result.getStatusCode(), is(200));
logger.debug("Created a host document: {}", Utils.toJson(host));
}
}
