@Override
public Set<Resource> buildStack(
Stack stack,
String gateWayUserData,
String coreUserData,
Map<String, Object> setupProperties) {
LOGGER.info("Assembling launch request for stack: {}", stack);
CloudContext cloudContext = new CloudContext(stack);
CloudCredential cloudCredential = credentialConverter.convert(stack.getCredential());
CloudStack cloudStack = cloudStackConverter.convert(stack, coreUserData, gateWayUserData);
instanceMetadataService.saveInstanceRequests(stack, cloudStack.getGroups());
LaunchStackRequest launchRequest =
new LaunchStackRequest(cloudContext, cloudCredential, cloudStack);
LOGGER.info("Triggering event: {}", launchRequest);
eventBus.notify(launchRequest.selector(), Event.wrap(launchRequest));
try {
LaunchStackResult res = launchRequest.await();
LOGGER.info("Result: {}", res);
validateResourceResults(cloudContext, res);
List<CloudResourceStatus> results = res.getResults();
updateNodeCount(stack.getId(), cloudStack.getGroups(), results, true);
return transformResults(results, stack);
} catch (InterruptedException e) {
LOGGER.error("Error while launching stack", e);
throw new OperationException(
"Unexpected exception occurred during build stack: " + e.getMessage());
}
}
@Override
public List<CloudResourceStatus> update(
AuthenticatedContext authenticatedContext, CloudStack stack, List<CloudResource> resources) {
String stackName = authenticatedContext.getCloudContext().getName();
boolean existingNetwork = isExistingNetwork(stack);
boolean assignFloatingIp = assignFloatingIp(stack);
String existingSubnetCidr = getExistingSubnetCidr(authenticatedContext, stack);
String heatTemplate =
heatTemplateBuilder.build(
stackName,
stack.getGroups(),
stack.getSecurity(),
stack.getImage(),
existingNetwork,
existingSubnetCidr != null,
assignFloatingIp);
Map<String, String> parameters =
heatTemplateBuilder.buildParameters(
authenticatedContext,
stack.getNetwork(),
stack.getImage(),
existingNetwork,
existingSubnetCidr);
return updateHeatStack(authenticatedContext, resources, heatTemplate, parameters);
}
@SuppressWarnings("unchecked")
@Override
public List<CloudResourceStatus> launch(
AuthenticatedContext authenticatedContext,
CloudStack stack,
PersistenceNotifier notifier,
AdjustmentType adjustmentType,
Long threshold) {
String stackName = authenticatedContext.getCloudContext().getName();
boolean existingNetwork = isExistingNetwork(stack);
boolean assignFloatingIp = assignFloatingIp(stack);
String existingSubnetCidr = getExistingSubnetCidr(authenticatedContext, stack);
String heatTemplate =
heatTemplateBuilder.build(
stackName,
stack.getGroups(),
stack.getSecurity(),
stack.getImage(),
existingNetwork,
existingSubnetCidr != null,
assignFloatingIp);
Map<String, String> parameters =
heatTemplateBuilder.buildParameters(
authenticatedContext,
stack.getNetwork(),
stack.getImage(),
existingNetwork,
existingSubnetCidr);
OSClient client = openStackClient.createOSClient(authenticatedContext);
Stack heatStack =
client
.heat()
.stacks()
.create(
Builders.stack()
.name(stackName)
.template(heatTemplate)
.disableRollback(false)
.parameters(parameters)
.timeoutMins(OPERATION_TIMEOUT)
.build());
CloudResource cloudResource =
new CloudResource.Builder().type(ResourceType.HEAT_STACK).name(heatStack.getId()).build();
try {
notifier.notifyAllocation(cloudResource, authenticatedContext.getCloudContext());
} catch (Exception e) {
// Rollback
terminate(authenticatedContext, stack, Lists.newArrayList(cloudResource));
}
List<CloudResourceStatus> resources =
check(authenticatedContext, Lists.newArrayList(cloudResource));
LOGGER.debug("Launched resources: {}", resources);
return resources;
}
private CloudStack removeDeleteRequestedInstances(CloudStack stack) {
List<Group> groups = new ArrayList<>();
for (Group group : stack.getGroups()) {
List<CloudInstance> instances = new ArrayList<>(group.getInstances());
for (CloudInstance instance : group.getInstances()) {
if (InstanceStatus.DELETE_REQUESTED == instance.getTemplate().getStatus()) {
instances.remove(instance);
}
}
groups.add(new Group(group.getName(), group.getType(), instances));
}
return new CloudStack(
groups, stack.getNetwork(), stack.getSecurity(), stack.getImage(), stack.getParameters());
}
@Override
public Set<Resource> addInstances(
Stack stack,
String gateWayUserData,
String coreUserData,
Integer adjustment,
String instanceGroup) {
LOGGER.debug("Assembling upscale stack event for stack: {}", stack);
CloudContext cloudContext = new CloudContext(stack);
CloudCredential cloudCredential = credentialConverter.convert(stack.getCredential());
InstanceGroup group = stack.getInstanceGroupByInstanceGroupName(instanceGroup);
group.setNodeCount(group.getNodeCount() + adjustment);
CloudStack cloudStack = cloudStackConverter.convert(stack, coreUserData, gateWayUserData);
instanceMetadataService.saveInstanceRequests(stack, cloudStack.getGroups());
List<CloudResource> resources = cloudResourceConverter.convert(stack.getResources());
UpscaleStackRequest<UpscaleStackResult> upscaleRequest =
new UpscaleStackRequest<>(cloudContext, cloudCredential, cloudStack, resources);
LOGGER.info("Triggering upscale stack event: {}", upscaleRequest);
eventBus.notify(upscaleRequest.selector(), Event.wrap(upscaleRequest));
try {
UpscaleStackResult res = upscaleRequest.await();
LOGGER.info("Upscale stack result: {}", res);
List<CloudResourceStatus> results = res.getResults();
updateNodeCount(stack.getId(), cloudStack.getGroups(), results, false);
validateResourceResults(cloudContext, res);
Set<Resource> resourceSet = transformResults(results, stack);
if (resourceSet.isEmpty()) {
throw new OperationException(
"Failed to upscale the cluster since all create request failed: "
+ results.get(0).getStatusReason());
}
return resourceSet;
} catch (InterruptedException e) {
LOGGER.error("Error while upscaling the stack", e);
throw new OperationException(e);
}
}
