Examples with Column henplus.view.Column used on opensource projects

Example 6 with Column

use of henplus.view.Column in project henplus by neurolabs.

the class DescribeCommand method execute.

/**
     * execute the command given.
     */
@Override
public int execute(final SQLSession session, final String cmd, final String param) {
    // make use of properties for these properties?
    // (since the options just toggle, this may be convenient)
    boolean showDescriptions = true;
    boolean showIndex = "idescribe".equals(cmd);
    boolean showTime = true;
    final StringTokenizer st = new StringTokenizer(param);
    if (st.countTokens() < 1) {
        return SYNTAX_ERROR;
    }
    // this was a flag to ensure that all options come before the tablenames
    // can probably be removed...
    final boolean moreOptions = true;
    while (st.hasMoreTokens()) {
        String tabName = st.nextToken();
        if (moreOptions && tabName.startsWith("-")) {
            if (tabName.indexOf('i') > -1) {
                showIndex = !showIndex;
            }
            if (tabName.indexOf('v') > -1) {
                showDescriptions = !showDescriptions;
            }
            if (tabName.indexOf('t') > -1) {
                showTime = !showTime;
            }
        } else {
            // more_options = false; // options can stand at every position
            // --> toggle
            boolean correctName = true;
            if (tabName.startsWith("\"")) {
                tabName = stripQuotes(tabName);
                correctName = false;
            }
            // separate schama and table.
            String schema = null;
            final int schemaDelim = tabName.indexOf('.');
            if (schemaDelim > 0) {
                schema = tabName.substring(0, schemaDelim);
                tabName = tabName.substring(schemaDelim + 1);
            }
            // FIXME: provide correct name as well for schema!
            if (correctName) {
                final String alternative = _tableCompleter.correctTableName(tabName);
                if (alternative != null && !alternative.equals(tabName)) {
                    tabName = alternative;
                    HenPlus.out().println("describing table: '" + tabName + "' (corrected name)");
                }
            }
            ResultSet rset = null;
            final Set<String> doubleCheck = new HashSet<String>();
            try {
                _interrupted = false;
                SigIntHandler.getInstance().pushInterruptable(this);
                boolean anyLeftArrow = false;
                boolean anyRightArrow = false;
                final long startTime = System.currentTimeMillis();
                final String catalog = session.getConnection().getCatalog();
                String description = null;
                String tableType = null;
                if (_interrupted) {
                    return SUCCESS;
                }
                final DatabaseMetaData meta = session.getConnection().getMetaData();
                for (int i = 0; i < DESC_META.length; ++i) {
                    DESC_META[i].resetWidth();
                }
                rset = meta.getTables(catalog, schema, tabName, LIST_TABLES);
                if (rset != null && rset.next()) {
                    tableType = rset.getString(4);
                    // remark
                    description = rset.getString(5);
                }
                rset.close();
                /*
                     * get primary keys.
                     */
                if (_interrupted) {
                    return SUCCESS;
                }
                final Map<String, String> pks = new HashMap<String, String>();
                rset = meta.getPrimaryKeys(null, schema, tabName);
                if (rset != null) {
                    while (!_interrupted && rset.next()) {
                        final String col = rset.getString(4);
                        final int pkseq = rset.getInt(5);
                        final String pkname = rset.getString(6);
                        String desc = pkname != null ? pkname : "*";
                        if (pkseq > 1) {
                            desc = new StringBuilder().append(desc).append("{").append(pkseq).append("}").toString();
                        }
                        pks.put(col, desc);
                    }
                    rset.close();
                }
                /*
                     * get referenced primary keys.
                     */
                if (_interrupted) {
                    return SUCCESS;
                }
                rset = meta.getExportedKeys(null, schema, tabName);
                if (rset != null) {
                    while (!_interrupted && rset.next()) {
                        final String col = rset.getString(4);
                        String fktable = rset.getString(7);
                        final String fkcolumn = rset.getString(8);
                        fktable = new StringBuilder().append(fktable).append("(").append(fkcolumn).append(")").toString();
                        String desc = pks.get(col);
                        desc = desc == null ? new StringBuilder().append(" <- ").append(fktable).toString() : new StringBuilder().append(desc).append("\n <- ").append(fktable).toString();
                        anyLeftArrow = true;
                        pks.put(col, desc);
                    }
                    rset.close();
                }
                /*
                     * get foreign keys.
                     */
                if (_interrupted) {
                    return SUCCESS;
                }
                final Map<String, String> fks = new HashMap<String, String>();
                // with foreign keys...
                try {
                    rset = meta.getImportedKeys(null, schema, tabName);
                } catch (final NoSuchElementException e) {
                    Logger.debug("Database problem reading meta data: ", e);
                }
                if (rset != null) {
                    while (!_interrupted && rset.next()) {
                        String table = rset.getString(3);
                        final String pkcolumn = rset.getString(4);
                        table = table + "(" + pkcolumn + ")";
                        final String col = rset.getString(8);
                        final String fkname = rset.getString(12);
                        String desc = fkname != null ? new StringBuilder().append(fkname).append("\n -> ").toString() : " -> ";
                        desc += table;
                        anyRightArrow = true;
                        fks.put(col, desc);
                    }
                    rset.close();
                }
                HenPlus.out().println(("VIEW".equals(tableType) ? "View: " : "Table: ") + tabName);
                if (description != null) {
                    HenPlus.out().println(description);
                }
                if (catalog != null) {
                    HenPlus.msg().println("catalog: " + catalog);
                }
                if (anyLeftArrow) {
                    HenPlus.msg().println(" '<-' : referenced by");
                }
                if (anyRightArrow) {
                    HenPlus.msg().println(" '->' : referencing");
                }
                /*
                     * if all columns belong to the same table name, then don't
                     * report it. A different table name may only occur in rare
                     * circumstance like object oriented databases.
                     */
                boolean allSameTableName = true;
                /*
                     * build up actual describe table.
                     */
                if (_interrupted) {
                    return SUCCESS;
                }
                rset = meta.getColumns(catalog, schema, tabName, null);
                final List<Column[]> rows = new ArrayList<Column[]>();
                int colNum = 0;
                boolean anyDescription = false;
                if (rset != null) {
                    while (!_interrupted && rset.next()) {
                        final Column[] row = new Column[9];
                        row[0] = new Column(++colNum);
                        final String thisTabName = rset.getString(3);
                        row[1] = new Column(thisTabName);
                        allSameTableName &= tabName.equals(thisTabName);
                        final String colname = rset.getString(4);
                        if (doubleCheck.contains(colname)) {
                            continue;
                        }
                        doubleCheck.add(colname);
                        row[2] = new Column(colname);
                        String type = rset.getString(6);
                        final int colSize = rset.getInt(7);
                        final int colDp = rset.getInt(9);
                        if (colSize > 0) {
                            if (colDp == 0) {
                                type = type + "(" + colSize + ")";
                            } else {
                                type = type + "(" + colSize + "," + colDp + ")";
                            }
                        }
                        row[3] = new Column(type);
                        final String defaultVal = rset.getString(13);
                        row[4] = new Column(rset.getString(18));
                        // oracle appends newline to default values for some
                        // reason.
                        row[5] = new Column((defaultVal != null ? defaultVal.trim() : null));
                        final String pkdesc = pks.get(colname);
                        row[6] = new Column(pkdesc != null ? pkdesc : "");
                        final String fkdesc = fks.get(colname);
                        row[7] = new Column(fkdesc != null ? fkdesc : "");
                        final String colDesc = showDescriptions ? rset.getString(12) : null;
                        row[8] = new Column(colDesc);
                        anyDescription |= colDesc != null;
                        rows.add(row);
                    }
                }
                rset.close();
                /*
                     * we render the table now, since we only know now, whether
                     * we will show the first column and the description column
                     * or not.
                     */
                DESC_META[1].setDisplay(!allSameTableName);
                DESC_META[8].setDisplay(anyDescription);
                final TableRenderer table = new TableRenderer(DESC_META, HenPlus.out());
                final Iterator<Column[]> it = rows.iterator();
                while (it.hasNext()) {
                    table.addRow(it.next());
                }
                table.closeTable();
                if (_interrupted) {
                    return SUCCESS;
                }
                if (showIndex) {
                    showIndexInformation(tabName, schema, meta);
                }
                if (showTime) {
                    TimeRenderer.printTime(System.currentTimeMillis() - startTime, HenPlus.out());
                    HenPlus.out().println();
                }
            } catch (final Exception e) {
                final String ex = e.getMessage() != null ? e.getMessage().trim() : e.toString();
                Logger.error("Database problem reading meta data: ", ex);
                return EXEC_FAILED;
            } finally {
                if (rset != null) {
                    try {
                        rset.close();
                    } catch (final Exception e) {
                    }
                }
            }
        }
    }
    return SUCCESS;
}

Example 7 with Column

use of henplus.view.Column in project henplus by neurolabs.

the class AliasCommand method showAliases.

private void showAliases() {
    DRV_META[0].resetWidth();
    DRV_META[1].resetWidth();
    final TableRenderer table = new TableRenderer(DRV_META, HenPlus.out());
    for (Map.Entry<String, String> entry : _aliases.entrySet()) {
        final Column[] row = new Column[2];
        row[0] = new Column(entry.getKey());
        row[1] = new Column(entry.getValue());
        table.addRow(row);
    }
    table.closeTable();
}

Example 8 with Column

use of henplus.view.Column in project henplus by neurolabs.

the class ConnectCommand method showSessions.

private void showSessions() {
    HenPlus.msg().println("current session is marked with '*'");
    for (int i = 0; i < SESS_META.length; ++i) {
        SESS_META[i].resetWidth();
    }
    final TableRenderer table = new TableRenderer(SESS_META, HenPlus.out());
    for (String sessName : _sessionManager.getSessionNames()) {
        final SQLSession session = _sessionManager.getSessionByName(sessName);
        final String prepend = sessName.equals(_currentSessionName) ? " * " : "   ";
        final Column[] row = new Column[5];
        row[0] = new Column(prepend + sessName);
        row[1] = new Column(session.getUsername());
        row[2] = new Column(session.getURL());
        row[3] = new Column(TimeRenderer.renderTime(session.getUptime()));
        row[4] = new Column(session.getStatementCount());
        table.addRow(row);
    }
    table.closeTable();
}

Example 9 with Column

use of henplus.view.Column in project henplus by neurolabs.

the class DriverCommand method execute.

/**
     * execute the command given.
     */
@Override
public int execute(final SQLSession currentSession, final String cmd, final String param) {
    final StringTokenizer st = new StringTokenizer(param);
    final int argc = st.countTokens();
    if ("list-drivers".equals(cmd)) {
        if (argc == 0) {
            HenPlus.msg().println("loaded drivers are marked with '*' (otherwise not found in CLASSPATH)");
            DRV_META[0].resetWidth();
            DRV_META[1].resetWidth();
            DRV_META[2].resetWidth();
            DRV_META[3].resetWidth();
            final TableRenderer table = new TableRenderer(DRV_META, HenPlus.out());
            for (Entry<String, DriverDescription> entry : _drivers.entrySet()) {
                final Column[] row = new Column[4];
                final DriverDescription desc = entry.getValue();
                final String dbName = entry.getKey();
                row[0] = new Column((desc.isLoaded() ? "* " : "  ") + dbName);
                row[1] = new Column(desc.getClassName());
                row[2] = new Column(desc.getVersion());
                row[3] = new Column(desc.getSampleURL());
                table.addRow(row);
            }
            table.closeTable();
            return SUCCESS;
        } else {
            return SYNTAX_ERROR;
        }
    } else if ("register".equals(cmd)) {
        if (argc < 2 || argc > 3) {
            return SYNTAX_ERROR;
        }
        final String shortname = (String) st.nextElement();
        final String driverClass = (String) st.nextElement();
        String sampleURL = null;
        if (argc >= 3) {
            sampleURL = (String) st.nextElement();
        }
        DriverDescription drv;
        drv = new DriverDescription(driverClass, sampleURL);
        if (!drv.isLoaded()) {
            HenPlus.msg().println("cannot load driver class '" + driverClass + "'");
            return EXEC_FAILED;
        } else {
            _drivers.put(shortname, drv);
        }
    } else if ("unregister".equals(cmd)) {
        if (argc != 1) {
            return SYNTAX_ERROR;
        }
        final String shortname = (String) st.nextElement();
        if (!_drivers.containsKey(shortname)) {
            HenPlus.msg().println("unknown driver for '" + shortname + "'");
            return EXEC_FAILED;
        } else {
            _drivers.remove(shortname);
        }
    }
    return SUCCESS;
}

Example 10 with Column

use of henplus.view.Column in project henplus by neurolabs.

the class DumpCommand method printMetaDataInfo.

private void printMetaDataInfo(final MetaProperty[] prop) {
    HenPlus.out().println();
    META_HEADERS[0].resetWidth();
    META_HEADERS[1].resetWidth();
    final TableRenderer table = new TableRenderer(META_HEADERS, HenPlus.out());
    for (int i = 0; i < prop.length; ++i) {
        final Column[] row = new Column[3];
        row[0] = new Column(prop[i].getFieldName());
        row[1] = new Column(prop[i].getTypeName());
        row[2] = new Column(prop[i].getMaxLength());
        table.addRow(row);
    }
    table.closeTable();
}