jp.co.tis.gsp.tools.db.beans.Entity Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of gsp-dba-maven-plugin Show documentation
To automate the routine work of the DBA, it is a tool to be able to concentrate on the data modeling work.
There is a newer version: 5.0.0
/*
 * Copyright (C) 2015 coastland
 *
 * 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 jp.co.tis.gsp.tools.db.beans;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.xml.bind.annotation.XmlAttribute;
import javax.xml.bind.annotation.XmlElementRef;
import javax.xml.bind.annotation.XmlRootElement;

@XmlRootElement(name="ENTITY")
public class Entity {
	private Integer id;

	private String label;

	private String name;

	private List columnList;

	private List indexList;

	private List foreignKeyList;

	private Integer showType;

	private Erd erd;
	
	private boolean havePrimaryKey = false;
	
	private String schema;
	
	
	@XmlAttribute(name="SCHEMA")
	public String getSchema() {
        return schema;
    }

    public void setSchema(String schema) {
        this.schema = schema;
    }

    @XmlAttribute(name="ID")
	public Integer getId() {
		return id;
	}

	public void setId(Integer id) {
		this.id = id;
	}

	@XmlAttribute(name="L-NAME")
	public String getLabel() {
		return label;
	}

	public void setLabel(String label) {
		this.label = label;
	}

	@XmlAttribute(name="P-NAME")
	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}

	@XmlAttribute(name="SHOWTYPE")
	public Integer getShowType() {
		return showType;
	}

	public void setShowType(Integer showType) {
		this.showType = showType;
	}

	@XmlElementRef(name="ATTR")
	public List getColumnList() {
		return columnList;
	}

	public void setColumnList(List columnList) {
		this.columnList = columnList;
	}

	@XmlElementRef(name="INDEX")
	public List getIndexList() {
		return indexList;
	}

	public void setIndexList(List indexList) {
		this.indexList = indexList;
	}

	public void setErd(Erd erd) {
		this.erd = erd;
	}
	

	public Column getColumn(Integer id) {
		for (Column column : columnList) {
			if ((int)column.getId() == (int)id) {
				return column;
			}
		}
		return null;
	}

	public List getForeignKeyList() {
		return foreignKeyList;
	}

	public void init() {
		Map foreignKeyMap = new HashMap();
		foreignKeyList = new ArrayList();
		if (columnList == null) return;
		for(Column column : columnList) {
            column.setEntity(this);
            if (column.isPrimaryKey()) {
				havePrimaryKey = true;
			}
			if (!column.hasForeignKeyColumn())
				continue;
			
			List fkColumnList = column.getForeignKeyColumnList();
			for(ForeignKeyColumn fkColumn : fkColumnList){
				Relation relation = erd.getRelation(fkColumn.getRelationId());
				if (relation.getShowType() != 0) {
					continue;
				}

				Entity parentEntity = erd.getEntity(relation.getParentEntityId());

				ForeignKey foreignKey;
				if (foreignKeyMap.containsKey(relation.getId())) {
					foreignKey = foreignKeyMap.get(relation.getId());
				} else {
					foreignKey = new ForeignKey();
					foreignKey.setReferenceEntity(parentEntity);
					foreignKeyMap.put(relation.getId(), foreignKey);
				}
				foreignKey.addColumn(column);
				foreignKey.addReferenceColumn(parentEntity.getColumn(fkColumn.getColumnId()));
			}
		}

		foreignKeyList.addAll(foreignKeyMap.values());

        for(Index index : indexList) {
            index.setEntity(this);
        }

    }

	@Override
	public boolean equals(Object obj) {
		if(obj == null || !(obj instanceof Entity))
			return false;
		Entity entity = (Entity)obj;
		return this.id.equals(entity.getId());
	}
	
	public boolean havePrimaryKey() {
		return this.havePrimaryKey;
	}
}