com.openshift.internal.restclient.model.Container Maven / Gradle / Ivy
The newest version!
/*******************************************************************************
* Copyright (c) 2015 Red Hat, Inc.
* Distributed under license by Red Hat, Inc. All rights reserved.
* This program is made available under the terms of the
* Eclipse Public License v1.0 which accompanies this distribution,
* and is available at http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* Red Hat, Inc. - initial API and implementation
******************************************************************************/
package com.openshift.internal.restclient.model;
import static com.openshift.internal.util.JBossDmrExtentions.asList;
import static com.openshift.internal.util.JBossDmrExtentions.asString;
import static com.openshift.internal.util.JBossDmrExtentions.get;
import static com.openshift.internal.util.JBossDmrExtentions.set;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import org.apache.commons.lang.StringUtils;
import org.jboss.dmr.ModelNode;
import org.jboss.dmr.ModelType;
import com.openshift.internal.restclient.model.probe.Probe;
import com.openshift.internal.restclient.model.properties.ResourcePropertyKeys;
import com.openshift.internal.restclient.model.volume.EmptyDirVolume;
import com.openshift.internal.restclient.model.volume.VolumeMount;
import com.openshift.restclient.images.DockerImageURI;
import com.openshift.restclient.model.IContainer;
import com.openshift.restclient.model.ILifecycle;
import com.openshift.restclient.model.IPort;
import com.openshift.restclient.model.probe.IProbe;
import com.openshift.restclient.model.volume.IVolume;
import com.openshift.restclient.model.volume.IVolumeMount;
public class Container extends ModelNodeAdapter implements IContainer, ResourcePropertyKeys {
private static final String IMAGE = "image";
private static final String ENV = "env";
private static final String IMAGE_PULL_POLICY = "imagePullPolicy";
private static final String COMMAND = "command";
private static final String COMMANDARGS = "args";
private static final String LIFECYCLE = "lifecycle";
private static final String VOLUMEMOUNTS = "volumeMounts";
private static final String PROPERTY_REQUESTS_MEMORY = "resources.requests.memory";
private static final String PROPERTY_REQUESTS_CPU = "resources.requests.cpu";
private static final String PROPERTY_LIMITS_MEMORY = "resources.limits.memory";
private static final String PROPERTY_LIMITS_CPU = "resources.limits.cpu";
private static final String LIVENESSPROBE = "livenessProbe";
private static final String READINESSPROBE = "readinessProbe";
private ModelNode node;
private Map propertyKeys;
public Container(ModelNode node) {
this(node, Collections.emptyMap());
}
/**
*
* @param node the node
* @param propertyKeys
* the override paths from the defaults
*/
public Container(ModelNode node, Map propertyKeys) {
super(node, propertyKeys);
this.node = node;
this.propertyKeys = propertyKeys;
}
@Override
public void setName(String name) {
set(node, propertyKeys, NAME, name);
}
@Override
public String getName() {
return asString(node, propertyKeys, NAME);
}
@Override
public void setImage(DockerImageURI tag) {
set(node, propertyKeys, IMAGE, tag.getAbsoluteUri());
}
@Override
public DockerImageURI getImage() {
return new DockerImageURI(asString(node, propertyKeys, IMAGE));
}
@Override
public void setEnvVars(Map vars) {
if (!vars.isEmpty()) {
ModelNode env = get(node, propertyKeys, ENV);
env.clear();
for (Entry var : vars.entrySet()) {
addEnvVar(var.getKey(), var.getValue());
}
}
}
@Override
public Map getEnvVars() {
HashMap hashMap = new HashMap<>();
ModelNode env = get(node, propertyKeys, ENV);
if (env.isDefined()) {
for (ModelNode var : env.asList()) {
hashMap.put(asString(var, propertyKeys, NAME), asString(var, propertyKeys, VALUE));
}
}
return hashMap;
}
@Override
public void addEnvVar(String key, String value) {
ModelNode env = get(node, propertyKeys, ENV);
ModelNode varNode = new ModelNode();
varNode.get(NAME).set(key);
varNode.get(VALUE).set(value);
env.add(varNode);
}
@Override
public void setPorts(Set ports) {
ModelNode nodePorts = get(node, propertyKeys, PORTS);
nodePorts.clear();
for (IPort port : ports) {
ModelNode portNode = nodePorts.add();
new Port(portNode, port);
}
}
@Override
public Set getPorts() {
ModelNode nodePorts = get(node, propertyKeys, PORTS);
Set ports = new HashSet<>();
if (nodePorts.isDefined()) {
for (ModelNode port : nodePorts.asList()) {
ports.add(new Port(port));
}
}
return ports;
}
@Override
public void setImagePullPolicy(String policy) {
set(node, propertyKeys, IMAGE_PULL_POLICY, policy);
}
@Override
public String getImagePullPolicy() {
return asString(node, propertyKeys, IMAGE_PULL_POLICY);
}
@Override
public void setCommand(List command) {
set(node, propertyKeys, COMMAND, command.toArray(new String[0]));
}
@Override
public List getCommand() {
return asList(node, propertyKeys, COMMAND, ModelType.STRING);
}
@Override
public void setCommandArgs(List args) {
set(node, propertyKeys, COMMANDARGS, args.toArray(new String[0]));
}
@Override
public List getCommandArgs() {
return asList(node, propertyKeys, COMMANDARGS, ModelType.STRING);
}
@Override
public void setLifecycle(ILifecycle lifecycle) {
ModelNode lifecycleNode = ModelNode.fromJSONString(lifecycle.toJson());
get(node, propertyKeys, LIFECYCLE).set(lifecycleNode);
}
@Override
public ILifecycle getLifecycle() {
if (node.has(LIFECYCLE)) {
return Lifecycle.fromJson(get(node, propertyKeys, LIFECYCLE));
} else {
return new Lifecycle.Builder().build();
}
}
@Override
public void setVolumes(Set volumes) {
ModelNode mounts = get(node, propertyKeys, VOLUMEMOUNTS);
mounts.clear();
for (IVolume volume : volumes) {
new EmptyDirVolume(mounts.add(), volume);
}
}
@Override
public Set getVolumes() {
Set volumes = new HashSet<>();
ModelNode mounts = get(node, propertyKeys, VOLUMEMOUNTS);
if (mounts.isDefined()) {
for (ModelNode node : mounts.asList()) {
volumes.add(new VolumeMount(node));
}
}
return volumes;
}
@Override
public void setVolumeMounts(Set volumes) {
ModelNode mounts = get(node, propertyKeys, VOLUMEMOUNTS);
mounts.clear();
for (IVolumeMount volume : volumes) {
new VolumeMount(mounts.add(), volume);
}
}
@Override
public Set getVolumeMounts() {
Set volumes = new HashSet<>();
ModelNode mounts = get(node, propertyKeys, VOLUMEMOUNTS);
if (mounts.isDefined()) {
for (ModelNode node : mounts.asList()) {
volumes.add(new VolumeMount(node));
}
}
return volumes;
}
@Override
public IVolumeMount addVolumeMount(String name) {
ModelNode mounts = get(node, propertyKeys, VOLUMEMOUNTS);
VolumeMount volume = new VolumeMount(mounts.add());
volume.setName(name);
return volume;
}
@Override
public String getRequestsMemory() {
return asString(node, propertyKeys, PROPERTY_REQUESTS_MEMORY);
}
@Override
public void setRequestsMemory(String requestsMemory) {
ModelNode child = get(getNode(), propertyKeys, PROPERTY_REQUESTS_MEMORY);
if (StringUtils.isBlank(requestsMemory)) {
child.clear();
} else {
child.set(requestsMemory);
}
}
@Override
public String getRequestsCPU() {
return asString(node, propertyKeys, PROPERTY_REQUESTS_CPU);
}
@Override
public void setRequestsCPU(String requestsCPU) {
ModelNode child = get(getNode(), propertyKeys, PROPERTY_REQUESTS_CPU);
if (StringUtils.isBlank(requestsCPU)) {
child.clear();
} else {
child.set(requestsCPU);
}
}
@Override
public String getLimitsMemory() {
return asString(node, propertyKeys, PROPERTY_LIMITS_MEMORY);
}
@Override
public void setLimitsMemory(String limitsMemory) {
ModelNode child = get(getNode(), propertyKeys, PROPERTY_LIMITS_MEMORY);
if (StringUtils.isBlank(limitsMemory)) {
child.clear();
} else {
child.set(limitsMemory);
}
}
@Override
public String getLimitsCPU() {
return asString(node, propertyKeys, PROPERTY_LIMITS_CPU);
}
@Override
public void setLimitsCPU(String limitsCPU) {
ModelNode child = get(getNode(), propertyKeys, PROPERTY_LIMITS_CPU);
if (StringUtils.isBlank(limitsCPU)) {
child.clear();
} else {
child.set(limitsCPU);
}
}
@Override
public IProbe getReadinessProbe() {
ModelNode readinessProbeNode = get(getNode(), propertyKeys, READINESSPROBE);
if (!readinessProbeNode.isDefined()) {
return null;
}
return new Probe(readinessProbeNode);
}
@Override
public IProbe getLivenessProbe() {
ModelNode lifenessProbeNode = get(getNode(), propertyKeys, LIVENESSPROBE);
if (!lifenessProbeNode.isDefined()) {
return null;
}
return new Probe(lifenessProbeNode);
}
@Override
public String toJson(){
return super.toJson(false);
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy