be.ordina.msdashboard.nodes.model.Node Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2012-2016 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 be.ordina.msdashboard.nodes.model;

import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonProperty;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import static com.fasterxml.jackson.annotation.JsonInclude.Include.NON_EMPTY;

/**
 * General representation of an item presented in the graph on the UI.
 * Nodes can have links to other nodes and contain a set of details.
 *
 * @author Andreas Evers
 */
@JsonIgnoreProperties(ignoreUnknown = true)
@JsonInclude(NON_EMPTY)
public class Node {

	public static final String ID = "id";
	public static final String STATUS = "status";
	public static final String TYPE = "type";
	public static final String DETAILS = "details";
	public static final String LANE = "lane";
	// TODO: Implement this field in the aggregators
	public static final String VERSION = "version";

	@JsonProperty(ID)
	private String id;

	@JsonProperty(DETAILS)
	private Map details = new HashMap<>();

	@JsonProperty(LANE)
	private Integer lane;

	private Set linkedToNodeIds = new HashSet<>();

	private Set linkedFromNodeIds = new HashSet<>();

	Node() {
		// For Jackson
	}

	public Node(String id) {
		this.id = id;
		details = new HashMap<>();
		linkedToNodeIds = new HashSet<>();
		linkedFromNodeIds = new HashSet<>();
	}

	public void setLane(Integer lane) {
		this.lane = lane;
	}

	public Integer getLane() {
		return lane;
	}

	public void setId(java.lang.String id) {
		this.id = id;
	}

	public String getId() {
		return id;
	}

	public Set getLinkedToNodeIds() {
		if (linkedToNodeIds == null) {
			return new HashSet<>();
		}
		return linkedToNodeIds;
	}

	public void setLinkedToNodeIds(Set linkedToNodeIds) {
		this.linkedToNodeIds = linkedToNodeIds;
	}

	public Set getLinkedFromNodeIds() {
		if (linkedFromNodeIds == null) {
			return new HashSet<>();
		}
		return linkedFromNodeIds;
	}

	public void setLinkedFromNodeIds(Set linkedFromNodeIds) {
		this.linkedFromNodeIds = linkedFromNodeIds;
	}

	public Map getDetails() {
		if (details == null) {
			return new HashMap<>();
		}
		return details;
	}

	public void setDetails(Map details) {
		this.details = details;
	}

	public void addDetail(String key, Object value) {
		details.put(key, value);
	}

	public void mergeWith(Node node) {
		if (lane == null) {
			lane = node.getLane();
		}
		if (linkedToNodeIds == null) {
			linkedToNodeIds = node.getLinkedToNodeIds();
		} else {
			linkedToNodeIds.addAll(node.getLinkedToNodeIds());
		}
		if (linkedFromNodeIds == null) {
			linkedFromNodeIds = node.getLinkedFromNodeIds();
		} else {
			linkedFromNodeIds.addAll(node.getLinkedFromNodeIds());
		}
		if (details == null) {
			details = node.getDetails();
		} else {
			details.forEach((k, v) -> {
				if (v != null) {
					if (k.equals(STATUS)) {
						if (node.getDetails().get(k) != null) {
							details.merge(k, node.getDetails().get(k), (v1, v2) -> {
								if ("DOWN".equals(v1) || "DOWN".equals(v2)) {
									return "DOWN";
								} else if ("UP".equals(v1) || "UP".equals(v2)) {
									return "UP";
								} else {
									return "UNKNOWN";
								}
							});
						}
					} else {
						if (node.getDetails().get(k) != null) {
							details.merge(k, node.getDetails().get(k), (v1, v2) -> v1);
						}
					}
				}
			});
		}
	}

	@Override
	public boolean equals(Object o) {
		if (this == o) return true;
		if (o == null || getClass() != o.getClass()) return false;

		Node node = (Node) o;

		if (!id.equals(node.id)) return false;
		if (details != null ? !details.equals(node.details) : node.details != null) return false;
		if (lane != null ? !lane.equals(node.lane) : node.lane != null) return false;
		if (linkedToNodeIds != null ? !linkedToNodeIds.equals(node.linkedToNodeIds) : node.linkedToNodeIds != null)
			return false;
		return !(linkedFromNodeIds != null ? !linkedFromNodeIds.equals(node.linkedFromNodeIds) : node.linkedFromNodeIds != null);

	}

	@Override
	public int hashCode() {
		return id.hashCode();
	}

	@Override
	public String toString() {
		return "Node{" +
				"id='" + id + '\'' +
				", details=" + details +
				", lane=" + lane +
				", linkedToNodeIds=" + linkedToNodeIds +
				", linkedFromNodeIds=" + linkedFromNodeIds +
				'}';
	}
}