UpdatedNodeElement.java

package org.insightech.er.editor.model.tracking;

import java.io.Serializable;
import java.util.Collection;
import java.util.HashSet;
import java.util.Set;

import org.insightech.er.editor.model.diagram_contents.element.node.NodeElement;
import org.insightech.er.editor.model.diagram_contents.element.node.table.column.Column;
import org.insightech.er.editor.model.diagram_contents.element.node.table.column.NormalColumn;

public class UpdatedNodeElement implements Serializable {

	private static final long serialVersionUID = -1547406607441505291L;

	private NodeElement nodeElement;

	private Set<Column> addedColumns;

	private Set<Column> updatedColumns;

	private Set<Column> removedColumns;

	public UpdatedNodeElement(NodeElement nodeElement) {
		this.nodeElement = nodeElement;

		this.addedColumns = new HashSet<Column>();
		this.updatedColumns = new HashSet<Column>();
		this.removedColumns = new HashSet<Column>();
	}

	public NodeElement getNodeElement() {
		return nodeElement;
	}

	public void setAddedColumns(Collection<NormalColumn> columns) {
		this.addedColumns.clear();
		this.addedColumns.addAll(columns);
	}

	public void setUpdatedColumns(Collection<NormalColumn> columns) {
		this.updatedColumns.clear();
		this.updatedColumns.addAll(columns);
	}

	public void setRemovedColumns(Collection<NormalColumn> columns) {
		this.removedColumns.clear();
		this.removedColumns.addAll(columns);
	}

	public boolean isAdded(Column column) {
		if (this.addedColumns.contains(column)) {
			return true;
		}

		return false;
	}

	public boolean isUpdated(Column column) {
		if (this.updatedColumns.contains(column)) {
			return true;
		}

		return false;
	}

	public Set<Column> getRemovedColumns() {
		return removedColumns;
	}

}