jp.co.tis.gsp.tools.db.beans.Column.java Source code

Introduction

Here is the source code for jp.co.tis.gsp.tools.db.beans.Column.java

Source

/*
 * 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.Arrays;
import java.util.List;

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

import org.apache.commons.lang.StringUtils;

@XmlRootElement(name = "ATTR")
public class Column {
    private Integer id;

    private String label;

    private String name;

    private String dataType;

    private Integer length;

    private Integer scale;

    private String defaultValue;

    private Short pkFlg;

    private Boolean isArray = false;

    private Entity entity;

    public Boolean isPrimaryKey() {
        return pkFlg == 1;
    }

    private Boolean isForeignKey() {
        for (ForeignKey foreignKey : entity.getForeignKeyList()) {
            if (foreignKey.getColumnList().contains(this)) {
                return true;
            }
        }
        return false;
    }

    public Boolean isSingularPrimaryKey() {
        if (!isPrimaryKey() || isForeignKey())
            return false;
        for (Index index : entity.getIndexList()) {
            if (index.getColumnList().contains(this) && index.getColumnList().size() == 1
                    && StringUtils.endsWithIgnoreCase(getName(), "_id"))
                return true;
        }
        return false;
    }

    public Boolean isAutoIncrement() {
        return (isSingularPrimaryKey() && isNumericDataType());
    }

    private Boolean isNumericDataType() {
        // ????????????????
        // ????????????
        List<String> numericDataTypeList = Arrays.asList(
                // Oracle
                "INTEGER", "SHORTINTEGER", "LONGINTEGER", "DECIMAL", "SHORTDECIMAL", "NUMBER",
                // Postgres
                "smallint", "integer", "bigint", "decimal", "numeric", "real", "double precision", "smallserial",
                "serial", "bigserial",
                // DB2
                "BIGINT", "SMALLINT", "INTEGER", "DOUBLE", "NUMERIC", "REAL",
                // H2Database
                "INT", "INTEGER", "MEDIUMINT", "INT4", "SIGNED", "TINYINT", "SMALLINT", "INT2", "YEAR", "BIGINT",
                "INT8", "IDENTITY", "DECIMAL", "NUMBER", "DEC", "NUMERIC", "DOUBLE", "DOUBLE PRECISION", "FLOAT",
                "FLOAT8", "REAL", "FLOAT4",
                // SQL Server
                "bigint", "bit", "decimal", "int", "money", "numeric", "smallint", "smallmoney", "tinyint", "float",
                "real",
                // MySQL
                "INTEGER", "SMALLINT", "DECIMAL", "NUMERIC", "FLOAT", "REAL", "DOUBLE PRECISION", "INT", "DEC",
                "FIXED", "DOUBLE", "BIT");
        return numericDataTypeList.contains(dataType);
    }

    public String getGeneratorKeyName() {
        return name + "_SEQ";
    }

    public Boolean isArray() {
        return isArray;
    }

    private Short required;

    public Boolean isNullable() {
        return required != 1;
    }

    private List<ForeignKeyColumn> foreignKeyColumnList;

    @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 = "DATATYPE")
    public String getDataType() {
        return dataType;
    }

    public void setDataType(String dataType) {
        if (StringUtils.equalsIgnoreCase(dataType, "ARRAY") || StringUtils.endsWithIgnoreCase(dataType, "_ARRAY")) {
            dataType = StringUtils.replace(dataType, "_ARRAY", "");
            if (StringUtils.isEmpty(dataType)) {
                dataType = "VARCHAR";
            }
            isArray = true;
        }
        this.dataType = dataType;
    }

    @XmlAttribute(name = "LENGTH")
    public Integer getLength() {
        return length;
    }

    public void setLength(Integer length) {
        this.length = length;
    }

    @XmlAttribute(name = "SCALE")
    public Integer getScale() {
        return scale;
    }

    public void setScale(Integer scale) {
        this.scale = scale;
    }

    @XmlAttribute(name = "PK")
    public Short getPkFlg() {
        return pkFlg;
    }

    public void setPkFlg(Short pkFlg) {
        this.pkFlg = pkFlg;
    }

    @XmlAttribute(name = "DEF")
    public String getDefaultValue() {
        if (StringUtils.equalsIgnoreCase(defaultValue, "AUTO_INCREMENT")) {
            return null;
        } else {
            return defaultValue;
        }
    }

    public void setDefaultValue(String defaultValue) {
        this.defaultValue = defaultValue;
    }

    @XmlAttribute(name = "NULL")
    public Short getRequired() {
        return required;
    }

    public void setRequired(Short required) {
        this.required = required;
    }

    @XmlElementRef(name = "FK")
    public List<ForeignKeyColumn> getForeignKeyColumnList() {
        return foreignKeyColumnList;
    }

    public void setForeignKeyColumnList(List<ForeignKeyColumn> foreignKeyColumnList) {
        this.foreignKeyColumnList = foreignKeyColumnList;
    }

    public boolean hasForeignKeyColumn() {
        return foreignKeyColumnList != null && foreignKeyColumnList.size() > 0;
    }

    @XmlTransient
    public Entity getEntity() {
        return entity;
    }

    public void setEntity(Entity entity) {
        this.entity = entity;
    }
}