All Downloads are FREE. Search and download functionalities are using the official Maven repository.

acolyte.jdbc.RowList Maven / Gradle / Ivy

Go to download

jdbc-driver

There is a newer version: 1.2.9
Show newest version
package acolyte.jdbc;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.io.IOException;

import java.util.Collections;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.HashMap;
import java.util.TreeSet;
import java.util.List;
import java.util.Map;

import java.math.BigDecimal;

import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.SQLWarning;
import java.sql.Timestamp;
import java.sql.Array;
import java.sql.Date;
import java.sql.Time;

import javax.sql.rowset.serial.SerialBlob;

import org.apache.commons.lang3.builder.HashCodeBuilder;
import org.apache.commons.lang3.builder.EqualsBuilder;

import org.apache.commons.io.IOUtils;

/**
 * Type-safe list of row.
 *
 * @author Cedric Chantepie
 */
public abstract class RowList {

    /**
     * Returns unmodifiable rows list.
     * @return the list of row
     */
    public abstract List getRows();

    /**
     * Appends |row|.
     *
     * @param row the row to be appended
     * @return Updated row list
     * @deprecated Append operation with multiple column values provided by sub-classes.
     */
    @Deprecated
    protected abstract RowList append(R row);

    /**
     * Returns copy of row list with updated column names/labels.
     *
     * @param columnIndex Index of column (first index is 1)
     * @param label Column name/label
     * @return a row list with specified column label
     * @see Column#name
     */
    public abstract RowList withLabel(int columnIndex, String label);

    /**
     * Returns copy of row list with updated metadata, 
     * including whether specified column is |nullable|.
     *
     * @param columnIndex Index of column (first index is 1)
     * @param nullable Is column nullable?
     * @return a row list with given nullable flag
     * @see Column#nullable
     */
    public abstract RowList withNullable(int columnIndex, boolean nullable);

    public abstract RowList withCycling(boolean cycling);

    public abstract boolean isCycling();

    /**
     * Returns result set from these rows.
     *
     * @param maxRows Limit for the maximum number of rows. 
     * If <= 0 no limit will be set.
     * If the limit is set and exceeded, the excess rows are silently dropped.
     * @return ResultSet for this list of rows
     */
    public RowResultSet resultSet(int maxRows) {
        if (maxRows <= 0) {
            return new RowResultSet(getRows(), this.isCycling());
        } // end of if

        return new RowResultSet(getRows().subList(0, maxRows),
                                   this.isCycling());
    } // end of resultSet

    /**
     * Returns result set from these rows (without row limit).
     *
     * @see #resultSet(int)
     * @return ResultSet for this list of rows
     */
    public RowResultSet resultSet() {
        return resultSet(0);
    } // end of resultSet

    /**
     * Returns this row list wrapped as handler result.
     * @return Query result from this list of rows
     */
    public QueryResult asResult() {
        return new QueryResult.Default(this);
    } // end of asResult

    /**
     * Returns ordered classes of columns.
     * @return Classes of columns
     */
    public abstract List> getColumnClasses();

    /**
     * Gets column mappings, from index to label.
     *
     * @return Column mappings, or empty map (not null) if none
     */
    public abstract Map getColumnLabels();

    /**
     * Gets column mappings, from index to nullable flag.
     *
     * @return Column mappings, or empty map (not null) if none
     */
    public abstract Map getColumnNullables();

    // --- Shared ---

    /**
     * Nil row list instance
     */
    public static final NilRowList Nil = new NilRowList();

    // --- Inner classes ---

    /**
     * Nil row list.
     */
    private static final class NilRowList extends RowList {
        // --- Properties ---

        /**
         * Empty list of row
         */
        private final List rows = 
            new ArrayList(0);

        /**
         * Empty list of column classes
         */
        private final List> colClasses = new ArrayList>(0);

        /**
         * Empty labels
         */
        private final Map labels = 
            new HashMap(0);

        /**
         * Empty flags
         */
        private final Map nullables = 
            new HashMap(0);

        private final boolean cycling = false;

        // --- Constructors ---

        /**
         * No-arg constructor.
         */
        private NilRowList() { }

        // ---

        /**
         * Returns empty list of row
         */
        public List getRows() {
            return this.rows;
        } // end of getRows

        /**
         * Returns unchanged nil row list.
         */
        public NilRowList append(final Row.Nothing row) { return this; }

        /**
         * Returns unchanged nil row list.
         */
        public NilRowList withLabel(final int columnIndex, final String label) {
            return this;
        } // end of withLabel

        /**
         * Returns unchanged nil row list.
         */
        public NilRowList withNullable(final int columnIndex, 
                                       final boolean nullable) {

            return this;
        } // end of withNullable

        /**
         * Returns unchanged nil row list.
         */
        public NilRowList withCycling(final boolean cycling) {
            return this;
        } // end of withCycling

        public boolean isCycling() {
            return this.cycling;
        } // end of isCycling

        /**
         * Returns empty list of columns classes.
         */
        public List> getColumnClasses() {
            return this.colClasses;
        } // end of getColumnClasses

        /**
         * Returns empty list of column labels.
         */
        public Map getColumnLabels() {
            return this.labels;
        } // end of getColumnLabels

        /**
         * Returns empty list of nullable flags.
         */
        public Map getColumnNullables() {
            return this.nullables;
        } // end of getColumnNullables
    } // end of class Nil

    /**
     * Creates column definition.
     *
     * @param  the type of the column
     * @param columnClass the class of the column
     * @param name the column name
     * @return the column definition
     * @throws IllegalArgumentException if |columnClass| is null, 
     * or |name| is empty.
     */
    public static  Column Column(final Class columnClass,
                                       final String name) {
        
        return new Column(columnClass, name);
    } // end of column

    /**
     * Result set made from list of row.
     *
     * @param  the row type
     */
    public final class RowResultSet extends AbstractResultSet {
        final List> columnClasses;
        final Map columnLabels;
        final Map columnNullables;
        List rows;
        final AbstractStatement statement;
        final SQLWarning warning;
        private Object last;
        // TODO: tableName

        // --- Constructors ---

        /**
         * Constructor
         * @param rows the list of rows
         */
        protected RowResultSet(final List rows, final boolean cycling) {
            this(rows, null, null, null, cycling);
        } // end of 

        /**
         * Copy constructor.
         * 
         * @param rows the list of rows
         * @param last the cursor to last result
         * @param statement the associated statement
         * @param warning the SQL warning
         * @param cycling
         */
        private RowResultSet(final List rows,
                             final Object last,
                             final AbstractStatement statement,
                             final SQLWarning warning,
                             final boolean cycling) {

            this(getColumnClasses(), getColumnLabels(), getColumnNullables(),
                 rows, last, statement, warning, cycling);

        } // end of 

        /**
         * Copy constructor.
         * 
         * @param columnClasses the column classes
         * @param columnLabels the column labels
         * @param columnNullables the nullable flags for the columns
         * @param rows the list of rows
         * @param last the cursor to last result
         * @param statement the associated statement
         * @param warning the SQL warning
         * @param cycling
         */
        private RowResultSet(final List> columnClasses,
                             final Map columnLabels,
                             final Map columnNullables,
                             final List rows,
                             final Object last,
                             final AbstractStatement statement,
                             final SQLWarning warning,
                             final boolean cycling) {

            if (columnClasses == null || columnLabels == null ||
                columnNullables == null || rows == null) {
                // Impossible
                throw new IllegalArgumentException();
            } // end of if

            this.columnClasses = columnClasses;
            this.columnLabels = columnLabels;
            this.columnNullables = columnNullables;
            this.rows = Collections.unmodifiableList(rows);

            this.statement = statement;
            this.warning = warning;
            this.last = null;

            super.fetchSize = rows.size();
            super.cycling = cycling;

            if (this.statement != null && super.fetchSize > 0 &&
                "true".equals(this.statement.connection.getProperties().
                              get("acolyte.resultSet.initOnFirstRow"))) {

                // Initially move to first row, contrary to JDBC specs
                this.row = 1;
            }
        } // end of 

        // --- 

        /**
         * Returns updated resultset, attached with given |statement|.
         *
         * @param statement the associated statement
         * @return Result set associated with given statement
         */
        public RowResultSet withStatement(final AbstractStatement statement) {
            return new RowResultSet(this.rows, this.last, 
                                       statement, this.warning, this.cycling);

        } // end of withStatement

        /**
         * Returns updated resultset, with cycling set.
         */
        public RowResultSet withCycling(final boolean cycling) {
            return new RowResultSet(this.rows, this.last, 
                                       this.statement, this.warning, cycling);
        } // end of withCycling

        /**
         * Returns updated resultset, with given |warning|.
         *
         * @param warning the SQL warning
         * @return Result set associated with SQL warning
         */
        public RowResultSet withWarning(final SQLWarning warning) {
            return new RowResultSet(this.rows, this.last, 
                                       this.statement, warning, this.cycling);

        } // end of withWarning

        /**
         * Returns update resultset, with rows only including values 
         * for the specified column.
         *
         * @param columnNames the names of the columns
         * @return Result set with the projected rows
         */
        public RowResultSet withProjection(final String[] columnNames) {
            final ArrayList indexes = new ArrayList();

            final HashMap labels =
                new HashMap();

            final HashMap nullables =
                new HashMap();

            int i = 1;
            for (final String name : columnNames) {
                final Integer index = this.columnLabels.get(name);
                final Boolean nable = this.columnNullables.get(index);

                if (index != null) {
                    final Integer updIdx = new Integer(i++);

                    indexes.add(index);
                    labels.put(name, updIdx);
                    nullables.put(updIdx, nable);
                }
            }

            return withProjection(indexes, labels, nullables);
        } // end of withProjection

        public RowResultSet withProjection(final int[] columnIndexes) {
            final ArrayList indexes = new ArrayList();

            final HashMap labels =
                new HashMap();

            final HashMap nullables =
                new HashMap();

            int upd = 1;

            for (final int index : columnIndexes) {
                for (final String columnName : this.columnLabels.keySet()) {
                    final Integer i = this.columnLabels.get(columnName);

                    if (i.intValue() == index) {
                        final Boolean nable = this.columnNullables.get(i);
                        final Integer updIdx = new Integer(upd++);

                        indexes.add(i);
                        labels.put(columnName, updIdx);
                        nullables.put(updIdx, nable);
                    }
                }
            }

            return withProjection(indexes, labels, nullables);
        }

        private RowResultSet withProjection(final List columnIndexes, final Map labels, final Map nullables) {
            final ArrayList> classes =
                new ArrayList>(labels.size());

            for (final String name : labels.keySet()) {
                Integer index = labels.get(name);
                Class cls = this.columnClasses.get(index.intValue()-1);

                classes.add(cls);
            }

            // ---
                
            final ArrayList projected =
                new ArrayList(this.rows.size());

            for (final R row : this.rows) {
                final List orig = row.cells();

                final ArrayList cells =
                    new ArrayList(columnIndexes.size());

                for (final Integer index : columnIndexes) {
                    cells.add(orig.get(index.intValue()-1));
                }

                projected.add(new Row.Untyped(cells));
            }

            return new RowResultSet(classes, labels, nullables, projected,
                                    this.last, this.statement, this.warning,
                                    this.cycling);
            
        } // end of withProjection

        /**
         * {@inheritDoc}
         */
        public SQLWarning getWarnings() throws SQLException {
            return this.warning;
        } // end of getWarnings

        /**
         * {@inheritDoc}
         */
        public boolean equals(final Object o) {
            if (o == null || !(o instanceof RowResultSet)) {
                return false;
            } // end of if

            // ---

            @SuppressWarnings("unchecked")
            final RowResultSet other = (RowResultSet) o;

            return new EqualsBuilder().
                append(this.columnClasses, other.columnClasses).
                append(this.columnLabels, other.columnLabels).
                append(this.columnNullables, other.columnNullables).
                append(this.cycling, other.cycling).
                isEquals();

        } // end of equals

        /**
         * {@inheritDoc}
         */
        public int hashCode() {
            return new HashCodeBuilder(11, 9).
                append(this.columnClasses).
                append(this.columnLabels).
                append(this.columnNullables).
                append(this.cycling).
                toHashCode();

        } // end of hashCode

        // --- ResultSet implementation ---

        /**
         * {@inheritDoc}
         */
        public AbstractStatement getStatement() {
            return this.statement;
        } // end of getStatement
        
        /**
         * {@inheritDoc}
         */
        public void setFetchSize(final int maxRows) throws SQLException {
            checkClosed();

            synchronized(this) {
                if (maxRows > this.rows.size()) {
                    return;
                }
                
                this.rows = this.rows.subList(0, maxRows);
            }
        } // end of setFetchSize
        
        /**
         * {@inheritDoc}
         */
        public int getFetchSize() throws SQLException {
            checkClosed();

            return this.fetchSize;
        } // end of getFetchSize

        /**
         * {@inheritDoc}
         */
        public boolean wasNull() throws SQLException {
            checkClosed();

            return (this.last == null);
        } // end of wasNull

        /**
         * {@inheritDoc}
         */
        public Object getObject(final int columnIndex) throws SQLException {
            checkClosed();

            if (!isOn()) {
                throw new SQLException("Not on a row");
            } // end of if

            final int idx = columnIndex - 1;
            final List cells = this.rows.get(this.row-1).cells();

            if (idx < 0 || idx >= cells.size()) {
                throw new SQLException("Invalid column index: " + columnIndex);
            } // end of if

            // ---

            return this.last = cells.get(idx);
        } // end of getObject

        /**
         * {@inheritDoc}
         */
        public Object getObject(final String columnLabel) throws SQLException {
            checkClosed();

            if (!isOn()) {
                throw new SQLException("Not on a row");
            } // end of if

            if (columnLabel == null || 
                !this.columnLabels.containsKey(columnLabel)) {

                throw new SQLException("Invalid label: " + columnLabel);
            } // end of if

            // ---

            final int columnIndex = findColumn(columnLabel);
            final int idx = columnIndex - 1;
            final List cells = this.rows.get(this.row-1).cells();

            if (idx < 0 || idx >= cells.size()) {
                throw new SQLException("Invalid column index: " + columnIndex);
            } // end of if

            // ---

            return this.last = cells.get(idx);
        } // end of getObject            

        /**
         * {@inheritDoc}
         */
        public Object getObject(final int columnIndex, 
                                final Map> typemap) 
            throws SQLException {

            return getObject(columnIndex);
        } // end of getObject

        /**
         * {@inheritDoc}
         */
        public Object getObject(final String columnLabel, 
                                final Map> typemap) 
            throws SQLException {
            
            return getObject(columnLabel);
        } // end of getObject

        /**
         * {@inheritDoc}
         */
        public  T getObject(final int columnIndex, 
                                              final Class type) 
            throws SQLException {
            
            if (type == null) {
                throw new SQLException("Invalid type");
            } // end of if

            // ---

            final Object val = getObject(columnIndex);

            if (val == null) {
                return null;
            } // end of if

            // ---

            return convert(val, type);
        } // end of getObject

        /**
         * {@inheritDoc}
         */
        public  T getObject(final String columnLabel,
                                              final Class type) 
            throws SQLException {
            
            if (type == null) {
                throw new SQLException("Invalid type");
            } // end of if

            // ---

            final Object val = getObject(columnLabel);

            if (val == null) {
                return null;
            } // end of if

            return convert(val, type);
        } // end of getObject

        /**
         * {@inheritDoc}
         */
        public Array getArray(final int columnIndex) throws SQLException {
            final Object val = getObject(columnIndex);

            if (val == null) {
                return null;
            } // end of if

            // ---

            try {
                return convert(val, Array.class);
            } catch (SQLException e) {
                throw new SQLException("Not an Array: " + columnIndex);
            } // end of catch
        } // end of getArray
        
        /**
         * {@inheritDoc}
         */
        public Array getArray(final String columnLabel) throws SQLException {
            final Object val = getObject(columnLabel);

            if (val == null) {
                return null;
            } // end of if

            // ---

            try {
                return convert(val, Array.class);
            } catch (SQLException e) {
                throw new SQLException("Not an Array: " + columnLabel);
            } // end of catch
        } // end of getArray

        /**
         * {@inheritDoc}
         */
        public String getString(final int columnIndex) throws SQLException {
            final Object val = getObject(columnIndex);

            if (val == null) {
                return null;
            } // end of if

            if (val instanceof String) {
                return (String) val;
            } // end of if

            // ---

            return String.valueOf(val);
        } // end of getString

        /**
         * {@inheritDoc}
         */
        public String getString(final String columnLabel) throws SQLException {
            final Object val = getObject(columnLabel);

            if (val == null) {
                return null;
            } // end of if

            if (val instanceof String) {
                return (String) val;
            } // end of if

            // ---

            return String.valueOf(val);
        } // end of getString

        /**
         * {@inheritDoc}
         */
        public java.sql.Blob getBlob(final int columnIndex) 
            throws SQLException {

            final Object val = getObject(columnIndex);

            if (val == null) {
                return null;
            } // end of if

            return convert(val, java.sql.Blob.class);
        } // end of getBlob

        /**
         * {@inheritDoc}
         */
        public java.sql.Blob getBlob(final String columnLabel) 
            throws SQLException {

            final Object val = getObject(columnLabel);

            if (val == null) {
                return null;
            } // end of if

            return convert(val, Blob.class);
        } // end of getBlob

        /**
         * {@inheritDoc}
         */
        public boolean getBoolean(final int columnIndex) throws SQLException {
            final Object val = getObject(columnIndex);

            if (val == null) {
                return false;
            } // end of if

            if (val instanceof Boolean) {
                return (Boolean) val;
            } // end of if

            return (val.toString().charAt(0) != '0');
        } // end of getBoolean

        /**
         * {@inheritDoc}
         */
        public boolean getBoolean(final String columnLabel) 
            throws SQLException {

            final Object val = getObject(columnLabel);

            if (val == null) {
                return false;
            } // end of if

            if (val instanceof Boolean) {
                return (Boolean) val;
            } // end of if

            return (val.toString().charAt(0) != '0');
        } // end of getBoolean

        /**
         * {@inheritDoc}
         */
        public byte getByte(final int columnIndex) throws SQLException {
            final Object val = getObject(columnIndex);

            if (val == null) {
                return 0;
            } // end of if

            if (val instanceof Number) {
                return ((Number) val).byteValue();
            } // end of if

            return -1;
        } // end of getByte

        /**
         * {@inheritDoc}
         */
        public byte getByte(final String columnLabel) 
            throws SQLException {

            final Object val = getObject(columnLabel);

            if (val == null) {
                return 0;
            } // end of if

            if (val instanceof Number) {
                return ((Number) val).byteValue();
            } // end of if

            return -1;
        } // end of getByte

        /**
         * {@inheritDoc}
         */
        public byte[] getBytes(final int columnIndex) throws SQLException {
            final Object val = getObject(columnIndex);

            if (val == null) {
                return null;
            } // end of if

            return convert(val, byte[].class);
        } // end of getBytes

        /**
         * {@inheritDoc}
         */
        public byte[] getBytes(final String columnLabel) throws SQLException {
            final Object val = getObject(columnLabel);

            if (val == null) {
                return null;
            } // end of if

            return convert(val, byte[].class);
        } // end of getBytes

        /**
         * {@inheritDoc}
         */
        public short getShort(final int columnIndex) throws SQLException {
            final Object val = getObject(columnIndex);

            if (val == null) {
                return 0;
            } // end of if

            if (val instanceof Number) {
                return ((Number) val).shortValue();
            } // end of if

            return -1;
        } // end of getShort

        /**
         * {@inheritDoc}
         */
        public short getShort(final String columnLabel) 
            throws SQLException {

            final Object val = getObject(columnLabel);

            if (val == null) {
                return 0;
            } // end of if

            if (val instanceof Number) {
                return ((Number) val).shortValue();
            } // end of if

            return -1;
        } // end of getShort

        /**
         * {@inheritDoc}
         */
        public int getInt(final int columnIndex) throws SQLException {
            final Object val = getObject(columnIndex);

            if (val == null) {
                return 0;
            } // end of if

            if (val instanceof Number) {
                return ((Number) val).intValue();
            } // end of if

            return -1;
        } // end of getInt

        /**
         * {@inheritDoc}
         */
        public int getInt(final String columnLabel) 
            throws SQLException {

            final Object val = getObject(columnLabel);

            if (val == null) {
                return 0;
            } // end of if

            if (val instanceof Number) {
                return ((Number) val).intValue();
            } // end of if

            return -1;
        } // end of getInt

        /**
         * {@inheritDoc}
         */
        public long getLong(final int columnIndex) throws SQLException {
            final Object val = getObject(columnIndex);

            if (val == null) {
                return 0;
            } // end of if

            if (val instanceof Number) {
                return ((Number) val).longValue();
            } // end of if

            return -1;
        } // end of getLong

        /**
         * {@inheritDoc}
         */
        public long getLong(final String columnLabel) 
            throws SQLException {

            final Object val = getObject(columnLabel);

            if (val == null) {
                return 0;
            } // end of if

            if (val instanceof Number) {
                return ((Number) val).longValue();
            } // end of if

            return -1;
        } // end of getLong

        /**
         * {@inheritDoc}
         */
        public float getFloat(final int columnIndex) throws SQLException {
            final Object val = getObject(columnIndex);

            if (val == null) {
                return 0;
            } // end of if

            if (val instanceof Number) {
                return ((Number) val).floatValue();
            } // end of if

            return -1;
        } // end of getFloat

        /**
         * {@inheritDoc}
         */
        public float getFloat(final String columnLabel) 
            throws SQLException {

            final Object val = getObject(columnLabel);

            if (val == null) {
                return 0;
            } // end of if

            if (val instanceof Number) {
                return ((Number) val).floatValue();
            } // end of if

            return -1;
        } // end of getFloat

        /**
         * {@inheritDoc}
         */
        public double getDouble(final int columnIndex) throws SQLException {
            final Object val = getObject(columnIndex);

            if (val == null) {
                return 0;
            } // end of if

            if (val instanceof Number) {
                return ((Number) val).doubleValue();
            } // end of if

            return -1;
        } // end of getDouble

        /**
         * {@inheritDoc}
         */
        public double getDouble(final String columnLabel) 
            throws SQLException {

            final Object val = getObject(columnLabel);

            if (val == null) {
                return 0;
            } // end of if

            if (val instanceof Number) {
                return ((Number) val).doubleValue();
            } // end of if

            return -1;
        } // end of getDouble

        /**
         * {@inheritDoc}
         */
        public BigDecimal getBigDecimal(final int columnIndex) 
            throws SQLException {

            final Object val = getObject(columnIndex);

            if (val == null) {
                return null;
            } // end of if

            if (val instanceof BigDecimal) {
                return (BigDecimal) val;
            } // end of if

            if (val instanceof Number) {
                return new BigDecimal(val.toString());
            } // end of if

            throw new SQLException("Not a BigDecimal: " + columnIndex);
        } // end of getBigDecimal

        /**
         * {@inheritDoc}
         */
        public BigDecimal getBigDecimal(final String columnLabel) 
            throws SQLException {

            final Object val = getObject(columnLabel);

            if (val == null) {
                return null;
            } // end of if

            if (val instanceof BigDecimal) {
                return (BigDecimal) val;
            } // end of if

            if (val instanceof Number) {
                return new BigDecimal(val.toString());
            } // end of if

            throw new SQLException("Not a BigDecimal: " + columnLabel);
        } // end of getBigDecimal

        /**
         * {@inheritDoc}
         */
        public BigDecimal getBigDecimal(final int columnIndex,
                                        final int scale) 
            throws SQLException {

            final Object val = getObject(columnIndex);

            if (val == null) {
                return null;
            } // end of if

            final BigDecimal bd = (val instanceof BigDecimal) 
                ? (BigDecimal) val
                : (val instanceof Number) 
                ? new BigDecimal(val.toString())
                : null;

            if (bd != null) {
                return bd.setScale(scale, BigDecimal.ROUND_DOWN);
            } // end of if

            throw new SQLException("Not a BigDecimal: " + columnIndex);
        } // end of getBigDecimal

        /**
         * {@inheritDoc}
         */
        public BigDecimal getBigDecimal(final String columnLabel, 
                                        final int scale) 
            throws SQLException {

            final Object val = getObject(columnLabel);

            if (val == null) {
                return null;
            } // end of if

            final BigDecimal bd = (val instanceof BigDecimal) 
                ? (BigDecimal) val
                : (val instanceof Number) 
                ? new BigDecimal(val.toString())
                : null;

            if (bd != null) {
                return bd.setScale(scale, BigDecimal.ROUND_DOWN);
            } // end of if

            throw new SQLException("Not a BigDecimal: " + columnLabel);
        } // end of getBigDecimal

        /**
         * {@inheritDoc}
         */
        public InputStream getBinaryStream(final int columnIndex) 
            throws SQLException {

            final Object val = getObject(columnIndex);

            if (val == null) {
                return null;
            } // end of if

            // ---

            try {
                return convert(val, InputStream.class);
            } catch (SQLException e) {
                throw new SQLException("Not an BinaryStream: " + columnIndex);
            } // end of catch
        } // end of getBinaryStream
        
        /**
         * {@inheritDoc}
         */
        public InputStream getBinaryStream(final String columnLabel) 
            throws SQLException {

            final Object val = getObject(columnLabel);

            if (val == null) {
                return null;
            } // end of if

            // ---

            try {
                return convert(val, InputStream.class);
            } catch (SQLException e) {
                throw new SQLException("Not an BinaryStream: " + columnLabel);
            } // end of catch
        } // end of getBinaryStream

        /**
         * {@inheritDoc}
         */
        public Date getDate(final int columnIndex) throws SQLException {
            final Object val = getObject(columnIndex);

            if (val == null) {
                return null;
            } // end of if

            // ---

            if (val instanceof Date) {
                return (Date) val;
            } // end of if

            if (val instanceof java.util.Date) {
                return new Date(((java.util.Date) val).getTime());
            } // end of if

            throw new SQLException("Not a Date: " + columnIndex);
        } // end of getDate

        /**
         * {@inheritDoc}
         */
        public Date getDate(final String columnLabel) throws SQLException {
            final Object val = getObject(columnLabel);

            if (val == null) {
                return null;
            } // end of if

            // ---

            if (val instanceof Date) {
                return (Date) val;
            } // end of if

            if (val instanceof java.util.Date) {
                return new Date(((java.util.Date) val).getTime());
            } // end of if

            throw new SQLException("Not a Date: " + columnLabel);
        } // end of getDate

        /**
         * {@inheritDoc}
         */
        public Date getDate(final int columnIndex, 
                            final Calendar cal) throws SQLException {

            return getDate(columnIndex);
        } // end of getDate

        /**
         * {@inheritDoc}
         */
        public Date getDate(final String columnLabel, 
                            final Calendar cal) throws SQLException {

            return getDate(columnLabel);
        } // end of getDate

        /**
         * {@inheritDoc}
         */
        public Time getTime(final int columnIndex) throws SQLException {
            final Object val = getObject(columnIndex);

            if (val == null) {
                return null;
            } // end of if

            // ---

            if (val instanceof Time) {
                return (Time) val;
            } // end of if

            if (val instanceof java.util.Date) {
                return new Time(((java.util.Date) val).getTime());
            } // end of if

            throw new SQLException("Not a Time: " + columnIndex);
        } // end of getTime

        /**
         * {@inheritDoc}
         */
        public Time getTime(final String columnLabel) throws SQLException {
            final Object val = getObject(columnLabel);

            if (val == null) {
                return null;
            } // end of if

            // ---

            if (val instanceof Time) {
                return (Time) val;
            } // end of if

            if (val instanceof java.util.Date) {
                return new Time(((java.util.Date) val).getTime());
            } // end of if

            throw new SQLException("Not a Time: " + columnLabel);
        } // end of getTime

        /**
         * {@inheritDoc}
         */
        public Time getTime(final int columnIndex, 
                            final Calendar cal) throws SQLException {

            return getTime(columnIndex);
        } // end of getTime

        /**
         * {@inheritDoc}
         */
        public Time getTime(final String columnLabel, 
                            final Calendar cal) throws SQLException {

            return getTime(columnLabel);
        } // end of getTime

        /**
         * {@inheritDoc}
         */
        public Timestamp getTimestamp(final int columnIndex) 
            throws SQLException {

            final Object val = getObject(columnIndex);

            if (val == null) {
                return null;
            } // end of if

            // ---

            if (val instanceof Timestamp) {
                return (Timestamp) val;
            } // end of if

            if (val instanceof java.util.Date) {
                return new Timestamp(((java.util.Date) val).getTime());
            } // end of if

            throw new SQLException("Not a Timestamp: " + columnIndex);
        } // end of getTimestamp

        /**
         * {@inheritDoc}
         */
        public Timestamp getTimestamp(final String columnLabel) 
            throws SQLException {

            final Object val = getObject(columnLabel);

            if (val == null) {
                return null;
            } // end of if

            // ---

            if (val instanceof Timestamp) {
                return (Timestamp) val;
            } // end of if

            if (val instanceof java.util.Date) {
                return new Timestamp(((java.util.Date) val).getTime());
            } // end of if

            throw new SQLException("Not a Timestamp: " + columnLabel);
        } // end of getTimestamp

        /**
         * {@inheritDoc}
         */
        public Timestamp getTimestamp(final int columnIndex, 
                                      final Calendar cal) throws SQLException {
            
            return getTimestamp(columnIndex);
        } // end of getTimestamp

        /**
         * {@inheritDoc}
         */
        public Timestamp getTimestamp(final String columnLabel, 
                                      final Calendar cal) throws SQLException {
            
            return getTimestamp(columnLabel);
        } // end of getTimestamp

        /**
         * {@inheritDoc}
         */
        public int findColumn(final String columnLabel) throws SQLException {
            return this.columnLabels.get(columnLabel);
        } // end of findColumn

        /**
         * Convert not null value.
         */
        private  T convert(final Object val, 
                                             final Class type) 
            throws SQLException {

            final Class clazz = val.getClass();

            if (type.isAssignableFrom(clazz)) {
                return type.cast(val);
            } // end of if

            if (java.util.Date.class.isAssignableFrom(type) &&
                java.util.Date.class.isAssignableFrom(clazz)) {

                @SuppressWarnings("unchecked")
                final Class origType = 
                    (Class) clazz;
                
                final java.util.Date orig = origType.cast(val);

                if (Date.class.isAssignableFrom(type)) {
                    return type.cast(new Date(orig.getTime()));
                } else if (Time.class.isAssignableFrom(type)) {
                    return type.cast(new Time(orig.getTime()));
                } else if (Timestamp.class.isAssignableFrom(type)) {
                    return type.cast(new Timestamp(orig.getTime()));
                } // end of else if
                
                throw new SQLException("Fails to convert temporal type");
            } // end of if

            if (Number.class.isAssignableFrom(type) &&
                Number.class.isAssignableFrom(clazz)) {

                @SuppressWarnings("unchecked")
                final Class origType = 
                    (Class) clazz;
                
                final Number num = origType.cast(val);

                if (Byte.class.isAssignableFrom(type)) {
                    return type.cast(new Byte(num.toString()));
                } else if (Double.class.isAssignableFrom(type)) {
                    return type.cast(new Double(num.toString()));
                } else if (Float.class.isAssignableFrom(type)) {
                    return type.cast(new Float(num.toString()));
                } else if (Integer.class.isAssignableFrom(type)) {
                    return type.cast(new Integer(num.toString()));
                } else if (Long.class.isAssignableFrom(type)) {
                    return type.cast(new Long(num.toString()));
                } else if (Short.class.isAssignableFrom(type)) {
                    return type.cast(new Short(num.toString()));
                } // end of if

                throw new SQLException("Fails to convert numeric type");
            } // end of if

            if (Array.class.isAssignableFrom(type)) {
                if (val instanceof Array) {
                    return (T) val;
                } // end of if

                final Class c = val.getClass();

                if (c.isArray()) {
                    return (T) ImmutableArray.
                        getInstance(c.getComponentType(), (Object[]) val);
                    
                } // end of if

                throw new SQLException("Fails to convert array");
            } // end of if

            if (byte[].class.equals(type)) return (T) getBytes(val);

            if (java.sql.Blob.class.isAssignableFrom(type)) {
                return (T) getBlob(val);
            } // end of if

            if (InputStream.class.isAssignableFrom(type)) {
                return (T) getBinaryStream(val);
            } // end of if

            throw new SQLException("Incompatible type: " + type + ", " + clazz);
        } // end of convert

        /**
         * {@inheritDoc}
         */
        public ResultSetMetaData getMetaData() throws SQLException {
            return new RowListMetaData(this.columnClasses,
                                       this.columnLabels,
                                       this.columnNullables);

        } // end of getMetaData

        /**
         * Tries to get bytes from raw |value|.
         *
         * @param value the binary value
         * @return the created byte array
         * @throws SQLException if fails to create the byte array
         */
        private byte[] getBytes(final Object value) throws SQLException {
            if (value instanceof byte[]) return (byte[]) value;

            if (value instanceof InputStream) {
                try {
                    final InputStream in = (InputStream) value;
                    if (in.markSupported()) in.reset();

                    return IOUtils.toByteArray(in);
                } catch (IOException e) {
                    throw new SQLException("Fails to get stream bytes", e);
                } // end of catch
            } // end of if

            if (value instanceof Blob) {
                InputStream in = null;

                try {
                    in = ((java.sql.Blob) value).getBinaryStream();

                    return IOUtils.toByteArray(in);
                } catch (Exception e) {
                    throw new SQLException("Fails to read BLOB content", e);
                } finally {
                    IOUtils.closeQuietly(in);
                } // end of finally
            } // end of if

            throw new SQLException("Cannot get bytes: " + value);
        } // end of getBytes

        /**
         * Tries to get binary stream from raw |value|.
         *
         * @param value the binary value
         * @return the created input stream
         * @throws SQLException if fails to create the binary stream
         */
        private InputStream getBinaryStream(final Object value)
            throws SQLException {

            if (value instanceof InputStream) return (InputStream) value;

            if (value instanceof byte[]) {
                return new ByteArrayInputStream((byte[]) value);
            } // end of if

            if (value instanceof Blob) {
                return ((java.sql.Blob) value).getBinaryStream();
            } // end of if

            throw new SQLException("Cannot get binary stream: " + value);
        } // end of getBinaryStream

        /**
         * Tries to get BLOB from raw |value|.
         *
         * @param value the binary value
         * @return the created BLOB
         * @throws SQLException if fails to create the BLOB
         */
        public java.sql.Blob getBlob(final Object value) throws SQLException {
            if (value instanceof java.sql.Blob) return (java.sql.Blob) value;

            if (value instanceof byte[]) {
                return new SerialBlob((byte[]) value);
            } // end of if

            return new SerialBlob(getBytes(value));
        } // end of getBlob
    } // end of class RowResultSet

    /**
     * Result set metadata for RowList.
     */
    public final class RowListMetaData implements ResultSetMetaData {
        final List> columnClasses;
        final Map columnLabels;
        final Map columnNullables;

        // --- Constructors ---

        /**
         * No-arg constructor.
         */
        private RowListMetaData(final List> columnClasses,
                                final Map columnLabels,
                                final Map columnNullables) {

            this.columnClasses = columnClasses;
            this.columnLabels = columnLabels;
            this.columnNullables = columnNullables;
        } // end of 

        // ---

        /**
         * {@inheritDoc}
         */
        public String getCatalogName(final int column) throws SQLException {
            return "";
        } // end of getCatalogName

        /**
         * {@inheritDoc}
         */
        public String getSchemaName(final int column) throws SQLException {
            return "";
        } // end of getSchemaName

        /**
         * {@inheritDoc}
         */
        public String getTableName(final int column) throws SQLException {
            return "";
        } // end of getTableName

        /**
         * {@inheritDoc}
         */
        public int getColumnCount() throws SQLException {
            return this.columnClasses.size();
        } // end of getColumnCount

        /**
         * {@inheritDoc}
         */
        public String getColumnClassName(final int column) throws SQLException {
            return this.columnClasses.get(column-1).getName();
        } // end of getColumnClassName

        /**
         * {@inheritDoc}
         */
        public int getColumnDisplaySize(final int column) 
            throws SQLException {

            return Integer.MAX_VALUE;
        } // end of getColumnDisplaySize

        /**
         * {@inheritDoc}
         */
        public String getColumnName(final int column) throws SQLException {
            for (final Map.Entry vk : this.columnLabels.
                     entrySet()) {

                if (vk.getValue().intValue() == column) {
                    return vk.getKey();
                } // end of if
            } // end of for

            return null;
        } // end of getColumnName

        /**
         * {@inheritDoc}
         */
        public String getColumnLabel(final int column) throws SQLException {
            return getColumnName(column);
        } // end of getColumnLabel

        /**
         * {@inheritDoc}
         */
        public boolean isSigned(final int column) throws SQLException {
            final int type = getColumnType(column);

            if (type == -1) {
                return false;
            } // end of if

            final Boolean s = Defaults.jdbcTypeSigns.get(type);

            return (s == null) ? false : s;
        } // end of isSigned

        /**
         * {@inheritDoc}
         */
        public int isNullable(final int column) throws SQLException {
            final Boolean b = this.columnNullables.get(column);

            return (b == null) ? ResultSetMetaData.columnNullableUnknown :
                (Boolean.TRUE.equals(b))
                ? ResultSetMetaData.columnNullable 
                : ResultSetMetaData.columnNoNulls;

        } // end of isNullable

        /**
         * {@inheritDoc}
         */
        public boolean isCurrency(final int column) throws SQLException {
            return false;
        } // end of isCurrency

        /**
         * {@inheritDoc}
         */
        public int getPrecision(final int column) throws SQLException {
            final int type = getColumnType(column);
            
            if (type == -1) {
                return 0;
            } // end of if

            final Integer p = Defaults.jdbcTypePrecisions.get(type);

            return (p == null) ? 0 : p;
        } // end of getPrecision

        /**
         * {@inheritDoc}
         */
        public int getScale(final int column) throws SQLException {
            final int type = getColumnType(column);
            
            if (type == -1) {
                return 0;
            } // end of if

            final Integer s = Defaults.jdbcTypeScales.get(type);

            return (s == null) ? 0 : s;
        } // end of getScale

        /**
         * {@inheritDoc}
         */
        public int getColumnType(final int column) throws SQLException {
            final Class clazz = this.columnClasses.get(column-1);

            if (clazz == null) {
                return -1;
            } // end of if

            // ---

            String cn = null;

            if (clazz.isPrimitive()) {
                if (clazz.equals(Boolean.TYPE)) { 
                    cn = Boolean.class.getName();
                } // end of if

                if (clazz.equals(Character.TYPE)) {
                    cn = Character.class.getName();
                } // end of if

                if (clazz.equals(Byte.TYPE)) {
                    cn = Byte.class.getName();
                } // end of if

                if (clazz.equals(Short.TYPE)) {
                    cn = Short.class.getName();
                } // end of if

                if (clazz.equals(Integer.TYPE)) {
                    cn = Integer.class.getName();
                } // end of if

                if (clazz.equals(Long.TYPE)) {
                    cn = Long.class.getName();
                } // end of if

                if (clazz.equals(Float.TYPE)) {
                    cn = Float.class.getName();
                } // end of if

                if (clazz.equals(Double.TYPE)) {
                    cn = Double.class.getName();
                } // end of if
            } else {
                cn = clazz.getName();
            } // end of else

            final String className = cn;

            if (className == null) {
                return -1;
            } // end of if

            // ---

            for (final Map.Entry kv : Defaults.
                     jdbcTypeMappings.entrySet()) {

                if (kv.getValue().equals(className)) {
                    return kv.getKey();
                } // end of if
            } // end of for

            return -1;
        } // end of getColumnType
                    
        /**
         * {@inheritDoc}
         */
        public String getColumnTypeName(final int column) 
            throws SQLException {

            return Defaults.jdbcTypeNames.get(getColumnType(column));
        } // end of getColumnTypeName

        /**
         * {@inheritDoc}
         */
        public boolean isSearchable(final int column) throws SQLException {
            return true;
        } // end of isSearchable

        /**
         * {@inheritDoc}
         */
        public boolean isCaseSensitive(final int column) throws SQLException {
            return true;
        } // end of isCaseSensitive

        /**
         * {@inheritDoc}
         */
        public boolean isAutoIncrement(final int column) throws SQLException {
            return false;
        } // end of isAutoIncrement

        /**
         * {@inheritDoc}
         */
        public boolean isReadOnly(final int column) throws SQLException {
            return true;
        } // end of isReadOnly

        /**
         * {@inheritDoc}
         */
        public boolean isWritable(final int column) throws SQLException {
            return false;
        } // end of isWritable

        /**
         * {@inheritDoc}
         */
        public boolean isDefinitelyWritable(final int column) 
            throws SQLException {

            return false;
        } // end of isDefinitelyWritable

        /**
         * {@inheritDoc}
         */
        public boolean isWrapperFor(final Class iface) 
            throws SQLException {

            return iface.isAssignableFrom(this.getClass());
        } // end of isWrapperFor

        /**
         * {@inheritDoc}
         */
        public  T unwrap(final Class iface) throws SQLException {
            if (!isWrapperFor(iface)) {
                throw new SQLException();
            } // end of if

            @SuppressWarnings("unchecked")
                final T proxy = (T) this;

            return proxy;
        } // end of unwrap
    } // end of RowListMetaData
} // end of class RowList
    

    

    

            

    

            



    
    






    
© 2015 - 2024 Weber Informatics LLC | Privacy Policy