com.clican.pluto.orm.tool.TableMetadata.java Source code

Introduction

Here is the source code for com.clican.pluto.orm.tool.TableMetadata.java

Source

/**
 * The Clican-Pluto software suit is Copyright 2009, Clican Company
 * and individual contributors, and is licensed under the GNU LGPL.
 *
 * @author wezhang
 *
 */
package com.clican.pluto.orm.tool;

import java.sql.DatabaseMetaData;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Map;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * As there are too many unreasonable design in the
 * <code>org.hibernate.tool.hbm2ddl.TableMetadata</code> We have to recreate
 * this class to use.
 * 
 * @see org.hibernate.tool.hbm2ddl.TableMetadata
 * 
 * @author wezhang
 * 
 */
public class TableMetadata {

    private static final Log log = LogFactory.getLog(TableMetadata.class);

    private final String catalog;
    private final String schema;
    private final String name;
    private final Map<String, ColumnMetadata> columns = new HashMap<String, ColumnMetadata>();
    private final Map<String, ForeignKeyMetadata> foreignKeys = new HashMap<String, ForeignKeyMetadata>();
    private final Map<String, IndexMetadata> indexes = new HashMap<String, IndexMetadata>();

    public TableMetadata(ResultSet rs, DatabaseMetaData meta, boolean extras) throws SQLException {
        catalog = rs.getString("TABLE_CAT");
        schema = rs.getString("TABLE_SCHEM");
        name = rs.getString("TABLE_NAME");
        initColumns(meta);
        if (extras) {
            initForeignKeys(meta);
            initIndexes(meta);
        }
        String cat = catalog == null ? "" : catalog + '.';
        String schem = schema == null ? "" : schema + '.';
        log.info("table found: " + cat + schem + name);
        log.info("columns: " + columns.keySet());
        if (extras) {
            log.info("foreign keys: " + foreignKeys.keySet());
            log.info("indexes: " + indexes.keySet());
        }
    }

    public String getName() {
        return name;
    }

    public String getCatalog() {
        return catalog;
    }

    public String getSchema() {
        return schema;
    }

    public String toString() {
        return "TableMetadata(" + name + ')';
    }

    public ColumnMetadata getColumnMetadata(String columnName) {
        return (ColumnMetadata) columns.get(columnName.toLowerCase());
    }

    public ForeignKeyMetadata getForeignKeyMetadataByColumnNames(String[] columnNames) {
        for (ForeignKeyMetadata fkm : foreignKeys.values()) {
            if (columnNames.length != fkm.getColumns().length) {
                continue;
            }
            int count = 0;
            for (ColumnMetadata cm : fkm.getColumns()) {
                for (String columnName : columnNames) {
                    if (cm.getName().equalsIgnoreCase(columnName)) {
                        count++;
                    }
                }
            }
            if (count == columnNames.length) {
                return fkm;
            }
        }
        return null;
    }

    public ForeignKeyMetadata getForeignKeyMetadata(String keyName) {
        return (ForeignKeyMetadata) foreignKeys.get(keyName.toLowerCase());
    }

    public IndexMetadata getIndexMetadata(String indexName) {
        return (IndexMetadata) indexes.get(indexName.toLowerCase());
    }

    private void addForeignKey(ResultSet rs) throws SQLException {
        String fk = rs.getString("FK_NAME");

        if (fk == null)
            return;

        ForeignKeyMetadata info = getForeignKeyMetadata(fk);
        if (info == null) {
            info = new ForeignKeyMetadata(rs);
            foreignKeys.put(info.getName().toLowerCase(), info);
        }

        info.addColumn(getColumnMetadata(rs.getString("FKCOLUMN_NAME")));
    }

    private void addIndex(ResultSet rs) throws SQLException {
        String index = rs.getString("INDEX_NAME");

        if (index == null)
            return;

        IndexMetadata info = getIndexMetadata(index);
        if (info == null) {
            info = new IndexMetadata(rs);
            indexes.put(info.getName().toLowerCase(), info);
        }

        info.addColumn(getColumnMetadata(rs.getString("COLUMN_NAME")));
    }

    public void addColumn(ResultSet rs) throws SQLException {
        String column = rs.getString("COLUMN_NAME");

        if (column == null)
            return;

        if (getColumnMetadata(column) == null) {
            ColumnMetadata info = new ColumnMetadata(rs);
            columns.put(info.getName().toLowerCase(), info);
        }
    }

    private void initForeignKeys(DatabaseMetaData meta) throws SQLException {
        ResultSet rs = null;

        try {
            rs = meta.getImportedKeys(catalog, schema, name);
            while (rs.next())
                addForeignKey(rs);
        } finally {
            if (rs != null)
                rs.close();
        }
    }

    private void initIndexes(DatabaseMetaData meta) throws SQLException {
        ResultSet rs = null;

        try {
            rs = meta.getIndexInfo(catalog, schema, name, false, true);

            while (rs.next()) {
                if (rs.getShort("TYPE") == DatabaseMetaData.tableIndexStatistic)
                    continue;
                addIndex(rs);
            }
        } finally {
            if (rs != null)
                rs.close();
        }
    }

    private void initColumns(DatabaseMetaData meta) throws SQLException {
        ResultSet rs = null;

        try {
            rs = meta.getColumns(catalog, schema, name, "%");
            while (rs.next())
                addColumn(rs);
        } finally {
            if (rs != null)
                rs.close();
        }
    }

}

// $Id$