org.springframework.yarn.am.grid.support.ProjectionData Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of spring-yarn-core Show documentation
Spring Yarn Core
The newest version!
/*
 * Copyright 2014 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.springframework.yarn.am.grid.support;

import java.util.HashMap;
import java.util.Map;

/**
 * Data object which works between a configuration and projections.
 *
 * @author Janne Valkealahti
 *
 */
public class ProjectionData {

	private String type;
	private Integer priority;
	private Integer memory;
	private Integer virtualCores;
	private Boolean locality;
	private Integer any = 0;
	private Map hosts = new HashMap();
	private Map racks = new HashMap();
	private Map properties = new HashMap();

	/**
	 * Instantiates a new projection data.
	 */
	public ProjectionData() {
	}

	/**
	 * Instantiates a new projection data.
	 *
	 * @param any the any count
	 */
	public ProjectionData(Integer any) {
		this(any, null, null);
	}

	/**
	 * Instantiates a new projection data.
	 *
	 * @param any the any count
	 * @param hosts the host count map
	 * @param racks the rack count map
	 */
	public ProjectionData(Integer any, Map hosts, Map racks) {
		this(any, hosts, racks, null, null);
	}

	/**
	 * Instantiates a new projection data.
	 *
	 * @param any the any count
	 * @param hosts the host count map
	 * @param racks the rack count map
	 * @param properties the extra properties
	 */
	public ProjectionData(Integer any, Map hosts, Map racks,
			Map properties) {
		this(any, hosts, racks, properties, null, null);
	}

	/**
	 * Instantiates a new projection data.
	 *
	 * @param any the any count
	 * @param hosts the host count map
	 * @param racks the rack count map
	 * @param type the type
	 * @param priority the priority
	 */
	public ProjectionData(Integer any, Map hosts, Map racks, String type,
			Integer priority) {
		this(any, hosts, racks, null, type, priority);
	}

	/**
	 * Instantiates a new projection data.
	 *
	 * @param any the any count
	 * @param hosts the host count map
	 * @param racks the rack count map
	 * @param properties the extra properties
	 * @param type the type
	 * @param priority the priority
	 */
	public ProjectionData(Integer any, Map hosts, Map racks,
			Map properties, String type, Integer priority) {
		this.any = any;
		if (hosts != null) {
			this.hosts = hosts;
		}
		if (racks != null) {
			this.racks = racks;
		}
		if (properties != null) {
			this.properties = properties;
		}
		this.type = type;
		this.priority = priority;
	}

	public String getType() {
		return type;
	}

	public void setType(String type) {
		this.type = type;
	}

	public Integer getPriority() {
		return priority;
	}

	public void setPriority(Integer priority) {
		if (priority != null) {
			this.priority = priority;
		}
	}

	public void setVirtualCores(Integer virtualCores) {
		this.virtualCores = virtualCores;
	}

	public Integer getVirtualCores() {
		return virtualCores;
	}

	public void setMemory(Integer memory) {
		this.memory = memory;
	}

	public void setLocality(Boolean locality) {
		this.locality = locality;
	}

	public Boolean getLocality() {
		return locality;
	}

	public Integer getMemory() {
		return memory;
	}

	public Integer getAny() {
		return any;
	}

	public void setAny(int any) {
		this.any = any;
	}

	public Map getHosts() {
		return hosts;
	}

	public void setHosts(Map hosts) {
		if (hosts != null) {
			this.hosts = hosts;
		}
	}

	public Map getRacks() {
		return racks;
	}

	public void setRacks(Map racks) {
		if (racks != null) {
			this.racks = racks;
		}
	}

	public void setHost(String host, Integer count) {
		hosts.put(host, count);
	}

	public void setRack(String rack, Integer count) {
		racks.put(rack, count);
	}

	public Map getProperties() {
		return properties;
	}

	public void setProperties(Map properties) {
		this.properties = properties;
	}

	/**
	 * Merge projection data to this instance by setting fields
	 * from other if set. This method returns a new
	 * instance of {@link ProjectionData}.
	 *
	 * @param other the projection to merge
	 * @return the new merged projection data
	 */
	public ProjectionData merge(ProjectionData other) {
		ProjectionData data = new ProjectionData();
		data.type = other.type != null ? other.type : this.type;
		data.priority = other.priority != null ? other.priority : this.priority;
		data.memory = other.memory != null ? other.memory : this.memory;
		data.virtualCores = other.virtualCores != null ? other.virtualCores : this.virtualCores;
		data.locality = other.locality != null ? other.locality : this.locality;
		data.any = (other.any != null && other.any > 0) ? other.any : this.any;
		if (other.hosts != null) {
			data.hosts.putAll(other.hosts);
		}
		if (other.racks != null) {
			data.racks.putAll(other.racks);
		}
		if (other.properties != null) {
			data.properties.putAll(other.properties);
		}
		return data;
	}

	@Override
	public String toString() {
		return "ProjectionData [type=" + type + ", priority=" + priority + ", memory=" + memory + ", virtualCores="
				+ virtualCores + ", locality=" + locality + ", any=" + any + ", hosts=" + hosts + ", racks=" + racks
				+ ", properties=" + properties + "]";
	}

}