columndata.java

eq跨平台查询工具源码 eq跨平台查询工具源码

/*
 * ColumnData.java
 *
 * Copyright (C) 2002, 2003, 2004, 2005, 2006 Takis Diakoumis
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 *
 */


package org.executequery.gui.browser;

import java.sql.Types;
import java.util.Vector;
import java.io.Serializable;

/* ----------------------------------------------------------
 * CVS NOTE: Changes to the CVS repository prior to the 
 *           release of version 3.0.0beta1 has meant a 
 *           resetting of CVS revision numbers.
 * ----------------------------------------------------------
 */

/** 
 * This class represents a single table
 * column definition. This includes data types
 * sizes, scales and key referencing meta data.
 *
 * @author   Takis Diakoumis
 * @version  $Revision: 1.7 $
 * @date     $Date: 2006/07/16 06:47:41 $
 */
public class ColumnData implements Serializable {
    
    static final long serialVersionUID = -4937385038396757064L;
    
    public static final int VALUE_REQUIRED = 0;
    
    public static final int VALUE_NOT_REQUIRED = 1;
    
    /** the catalog for this column */
    private String catalog;

    /** the schema for this column */
    private String schema;
    
    /** The table this column belongs to */
    private String tableName;
    
    /** The name of this column */
    private String columnName;
    
    /** The name of the SQL type of this column */
    private String columnType;
    
    /** The key of this column - if any
     *  (ie primary, foreign etc) */
    private String keyType;
    
    /** Whether this column is a primary key */
    private boolean primaryKey;
    
    /** Whether this column is a foreign key */
    private boolean foreignKey;
    
    /** The data size of this column */
    private int columnSize;
    
    /** The data scale of this column */
    private int columnScale;
    
    /** Whether this column is required ie. NOT NULL */
    private int columnRequired;
    
    /** The mapped SQL type */
    private int sqlType;
    
    /** the column's default value */
    private String defaultValue;
    
    /** This column's constraints as a <code>Vector</code>
     *  of <code>ColumnConstraint</code> objects */
    private Vector<ColumnConstraint> columnConstraints;
    
    /** Whether this column is a new column in the table */
    private boolean newColumn;
    
    /** Whether this column is marked as to be deleted */
    private boolean markedDeleted;
    
    public ColumnData() {
        primaryKey = false;
        foreignKey = false;
        newColumn = false;
        keyType = null;
    }
    
    public ColumnData(String columnName) {
        this();
        this.columnName = columnName;
    }
    
    public ColumnData(boolean newColumn) {
        this();
        this.newColumn = newColumn;
    }
    
    public ColumnConstraint[] getColumnConstraintsArray() {
        int v_size = columnConstraints.size();
        ColumnConstraint[] cca = new ColumnConstraint[v_size];
        
        for (int i = 0; i < v_size; i++) {
            cca[i] = columnConstraints.get(i);
        }
        
        return cca;
    }
    
    public Vector<ColumnConstraint> getColumnConstraintsVector() {
        return columnConstraints;
    }
    
    public void addConstraint(ColumnConstraint cc) {
        if (columnConstraints == null) {
            columnConstraints = new Vector<ColumnConstraint>();
        }
        columnConstraints.add(cc);        
    }
    
    public void resetConstraints() {
        if (columnConstraints != null) {
            columnConstraints.clear();
        }
    }
    
    public void removeConstraint(ColumnConstraint cc) {
        columnConstraints.remove(cc);
    }
    
    public boolean isNewColumn() {
        return newColumn;
    }
    
    public boolean isValid() {
        return (columnName != null && columnName.length() > 0) &&
        (tableName != null && tableName.length() > 0) &&
        (columnType != null && columnType.length() > 0);
    }
    
    public void setNewColumn(boolean newColumn) {
        this.newColumn = newColumn;
    }
    
    public int getColumnScale() {
        return columnScale;
    }
    
    public void setColumnScale(int columnScale) {
        this.columnScale = columnScale;
    }
    
    public void setNamesToUpper() {
        if (tableName != null) {
            tableName = tableName.toUpperCase();
        }        
        if (columnName != null) {
            columnName = columnName.toUpperCase();
        }        
    }
    
    public void setValues(ColumnData cd) {
        tableName = cd.getTableName();
        columnName = cd.getColumnName();
        columnType = cd.getColumnType();
        keyType = cd.getKeyType();
        primaryKey = cd.isPrimaryKey();
        foreignKey = cd.isForeignKey();
        columnSize = cd.getColumnSize();
        columnRequired = cd.getColumnRequired();
        sqlType = cd.getSQLType();
        
        Vector constraints = cd.getColumnConstraintsVector();
        if (constraints != null) {
            columnConstraints = (Vector)constraints.clone();
        }        
    }
    
    public boolean isPrimaryKey() {
        return primaryKey;
    }
    
    public boolean isForeignKey() {
        return foreignKey;
    }
    
    public boolean isKey() {
        return primaryKey || foreignKey;
    }
    
    public void setPrimaryKey(boolean primaryKey) {
        this.primaryKey = primaryKey;
    }
    
    public void setForeignKey(boolean foreignKey) {
        this.foreignKey = foreignKey;
    }
    
    public void setColumnRequired(int columnRequired) {
        this.columnRequired = columnRequired;
    }

    public int getColumnRequired() {
        return columnRequired;
    }
    
    /**
     * Returns whether this is a required column determined by
     * whether the column allows null values.
     *
     * @return true | false
     */
    public boolean isRequired() {
        return columnRequired == VALUE_REQUIRED;
    }
    
    public void setKeyType(String keyType) {
        this.keyType = keyType;
    }
    
    /**
     * Returns whether this column is a date type or 
     * extension of.
     *
     * ie. Types.DATE, Types.TIME, Types.TIMESTAMP.
     *
     * @return true | false
     */
    public boolean isDateDataType() {
        return sqlType == Types.DATE ||
                sqlType == Types.TIME ||
                sqlType == Types.TIMESTAMP;
    }
    
    public String getKeyType() {
        return keyType;
    }
    
    public void setColumnType(String columnType) {
        this.columnType = columnType;
    }
    
    public String getColumnType() {
        return columnType;
    }
    
    public void setTableName(String tableName) {
        this.tableName = tableName;
    }
    
    public void setSQLType(int sqlType) {
        this.sqlType = sqlType;
    }
    
    public int getSQLType() {
        return sqlType;
    }
    
    public String getTableName() {
        return tableName;
    }
    
    public void setColumnName(String columnName) {
        this.columnName = columnName;
    }
    
    public String getColumnName() {
        return columnName;
    }
    
    public void setColumnSize(int columnSize) {
        this.columnSize = columnSize;
    }
    
    public int getColumnSize() {
        return columnSize;
    }
    
    public String toString() {
        return columnName == null ? ColumnConstraint.EMPTY : columnName;
    }

    public String getCatalog() {
        return catalog;
    }

    public void setCatalog(String catalog) {
        this.catalog = catalog;
    }

    public String getSchema() {
        return schema;
    }

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

    public boolean isMarkedDeleted() {
        return markedDeleted;
    }

    public void setMarkedDeleted(boolean markedDeleted) {
        this.markedDeleted = markedDeleted;
    }

    public String getDefaultValue() {
        return defaultValue;
    }

    public void setDefaultValue(String defaultValue) {
        this.defaultValue = defaultValue;
    }
    
    /**
     * Returns a formatted string representation of the
     * column's data type and size - eg. VARCHAR(10).
     *
     * @return the formatted type string
     */
    public String getFormattedDataType() {
        String typeString = getColumnType();
        StringBuffer buffer = new StringBuffer(typeString);

        // if the type doesn't end with a digit or it
        // is a char type then add the size - attempt
        // here to avoid int4, int8 etc. type values

        int type = getSQLType();
        if (!typeString.matches("\\b\\D+\\d+\\b") ||
                (type == Types.CHAR ||
                 type == Types.VARCHAR ||
                 type == Types.LONGVARCHAR)) {

            if (getColumnSize() > 0 && !isDateDataType()) {
                buffer.append("(");
                buffer.append(getColumnSize());

                if (getColumnScale() > 0) {
                    buffer.append(",");
                    buffer.append(getColumnScale());
                }
                buffer.append(")");
            }
        }
        return buffer.toString();
    }

    
}