protected List<String> getJobParents(
NodeTemplate nodeTemplate, String nodeName, Map<String, NodeTemplate> nodes) {
// Get Chronos parent job dependency
String parentJobCapabilityName = "parent_job";
Map<String, NodeTemplate> parentJobs =
toscaService.getAssociatedNodesByCapability(nodes, nodeTemplate, parentJobCapabilityName);
if (parentJobs.isEmpty()) {
return null;
} else {
// WARNING: cycle check is done later!
return Lists.newArrayList(parentJobs.keySet());
}
}
protected Job createJob(
Map<String, NodeTemplate> nodes,
String deploymentId,
String nodeName,
NodeTemplate nodeTemplate,
Map<String, Resource> resources) {
try {
Job chronosJob = new Job();
// Init job infos
// Get the generated UUID for the node (in DB resource ?)
// FIXME This is just for prototyping... Otherwise is madness!!
Resource resourceJob = resources.get(nodeName);
chronosJob.setName(resourceJob.getId());
// TODO Validation
chronosJob.setRetries(
Integer.parseInt(
(String)
toscaService.getNodePropertyValueByName(nodeTemplate, "retries").getValue()));
chronosJob.setCommand(
(String) toscaService.getNodePropertyValueByName(nodeTemplate, "command").getValue());
// TODO Enable epsilon setting in TOSCA tplt ?
chronosJob.setEpsilon("PT10S");
ListPropertyValue inputUris =
(ListPropertyValue) toscaService.getNodePropertyValueByName(nodeTemplate, "uris");
if (inputUris != null && !inputUris.getValue().isEmpty()) {
// Convert List<Object> to List<String>
chronosJob.setUris(
inputUris
.getValue()
.stream()
.map(e -> ((PropertyValue<?>) e).getValue().toString())
.collect(Collectors.toList()));
}
List<EnvironmentVariable> envs = new ArrayList<>();
ComplexPropertyValue inputEnvVars =
((ComplexPropertyValue)
toscaService.getNodePropertyValueByName(nodeTemplate, "environment_variables"));
if (inputEnvVars != null) {
for (Map.Entry<String, Object> var : inputEnvVars.getValue().entrySet()) {
EnvironmentVariable envVar = new EnvironmentVariable();
envVar.setName(var.getKey());
envVar.setValue(((PropertyValue<?>) var.getValue()).getValue().toString());
envs.add(envVar);
}
chronosJob.setEnvironmentVariables(envs);
}
// Docker image
// TODO Remove hard-coded?
String supportedType = "tosca.artifacts.Deployment.Image.Container.Docker";
DeploymentArtifact image;
// <image> artifact available
if (nodeTemplate.getArtifacts() == null
|| (image = nodeTemplate.getArtifacts().get("image")) == null) {
throw new IllegalArgumentException(
String.format(
"<image> artifact not found in node <%s> of type <%s>",
nodeName, nodeTemplate.getType()));
}
// <image> artifact type check
if (!image.getArtifactType().equals(supportedType)) {
throw new IllegalArgumentException(
String.format(
"Unsupported artifact type for <image> artifact in node <%s> of type <%s>. "
+ "Given <%s>, supported <%s>",
nodeName,
nodeTemplate.getType(),
nodeTemplate.getArtifacts().get("image").getArtifactType(),
supportedType));
}
// Requirements
// Get Docker host dependency
String dockerCapabilityName = "host";
Map<String, NodeTemplate> dockerRelationships =
toscaService.getAssociatedNodesByCapability(nodes, nodeTemplate, dockerCapabilityName);
Double dockerNumCpus = null;
Double dockerMemSize = null;
if (!dockerRelationships.isEmpty()) {
/*
* WARNING: The TOSCA validation should already check the limits (currently Alien4Cloud does
* not...)
*/
NodeTemplate dockerNode = dockerRelationships.values().iterator().next();
Capability dockerCapability = dockerNode.getCapabilities().get(dockerCapabilityName);
dockerNumCpus =
Double.parseDouble(
(String)
toscaService
.getCapabilityPropertyValueByName(dockerCapability, "num_cpus")
.getValue());
// Converting Memory Size (as TOSCA scalar-unit.size)
SizeType tmp = new SizeType();
String memSizeRaw =
(String)
toscaService
.getCapabilityPropertyValueByName(dockerCapability, "mem_size")
.getValue();
dockerMemSize = tmp.parse(memSizeRaw).convert("MB"); // Chronos wants MB
}
Container container = new Container();
container.setType("DOCKER");
// Run the container in Privileged Mode
Parameters param = new Parameters();
param.setKey("privileged");
param.setValue("true");
Collection<Parameters> parameters = new ArrayList<Parameters>();
parameters.add(param);
container.setParameters(parameters);
// FIXME ForcePullImage must be parametrizable by tosca template
container.setForcePullImage(true);
////////////////////////////////////////////////////////////////
container.setImage(
(String)
((PropertyValue<?>) nodeTemplate.getArtifacts().get("image").getFile()).getValue());
if (dockerNumCpus != null) {
chronosJob.setCpus(dockerNumCpus);
}
if (dockerMemSize != null) {
chronosJob.setMem(dockerMemSize);
}
chronosJob.setContainer(container);
return chronosJob;
} catch (Exception ex) {
throw new RuntimeException(
String.format(
"Failed to parse node <%s> of type <%s>: %s",
nodeName, nodeTemplate.getType(), ex.getMessage()),
ex);
}
}