org.sqlite.driver.RowsMeta Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sqlite-javacpp Show documentation
JDBC Driver for SQLite using JavaCPP
The newest version!
/*
 * The author disclaims copyright to this source code.  In place of
 * a legal notice, here is a blessing:
 *
 *    May you do good and not evil.
 *    May you find forgiveness for yourself and forgive others.
 *    May you share freely, never taking more than you give.
 */
package org.sqlite.driver;

import java.sql.ResultSetMetaData;
import java.sql.SQLException;

import org.sqlite.ColAffinities;
import org.sqlite.ErrCodes;
import org.sqlite.Stmt;
import org.sqlite.StmtException;

class RowsMeta implements ResultSetMetaData {
	private final Stmt stmt;

	RowsMeta(Stmt stmt) {
		this.stmt = stmt;
	}

	private static int fixCol(int columnIndex) {
		return columnIndex - 1;
	}

	@Override
	public int getColumnCount() { // Used by Hibernate
		return stmt.getColumnCount();
	}

	@Override
	public boolean isAutoIncrement(int column) throws SQLException {
		return stmt.getMetadata(fixCol(column))[2];
	}

	@Override
	public boolean isCaseSensitive(int column) throws SQLException {
		switch (stmt.getColumnAffinity(fixCol(column))) {
			case ColAffinities.INTEGER:
			case ColAffinities.NUMERIC:
			case ColAffinities.REAL:
				return false;
		}
		return true; // FIXME Collation 'NOCASE'
	}

	@Override
	public boolean isSearchable(int column) throws SQLException {
		checkColumn(column);
		return true;
	}

	@Override
	public boolean isCurrency(int column) throws SQLException {
		checkColumn(column);
		return false;
	}

	@Override
	public int isNullable(int column) throws SQLException {
		return stmt.getMetadata(fixCol(column))[0] ? columnNoNulls : columnNullable;
	}

	@Override
	public boolean isSigned(int column) throws SQLException {
		switch (stmt.getColumnAffinity(fixCol(column))) {
			case ColAffinities.INTEGER:
			case ColAffinities.NUMERIC:
			case ColAffinities.REAL:
				return true;
		}
		return false;
	}

	@Override
	public int getColumnDisplaySize(int column) throws SQLException {
		switch (stmt.getColumnAffinity(fixCol(column))) {
			case ColAffinities.INTEGER:
				return 20;
			case ColAffinities.REAL:
				return 25;
		}
		return 10; // Like in SQLite shell with column mode TODO
	}

	@Override
	public String getColumnLabel(int column) throws SQLException {
		return stmt.getColumnName(fixCol(column));
	}

	@Override
	public String getColumnName(int column) throws SQLException {
		final String name = stmt.getColumnOriginName(fixCol(column));
		if (name == null) {
			return getColumnLabel(column);
		}
		return name;
	}

	@Override
	public String getSchemaName(int column) {
		return "";
	}

	@Override
	public int getPrecision(int column) throws SQLException {
		switch (stmt.getColumnAffinity(fixCol(column))) {
			case ColAffinities.INTEGER:
				return 19;
			case ColAffinities.NUMERIC:
			case ColAffinities.REAL:
				return 15;
		}
		return 0;
	}

	@Override
	public int getScale(int column) throws SQLException {
		switch (stmt.getColumnAffinity(fixCol(column))) {
			case ColAffinities.INTEGER:
				return 0;
			case ColAffinities.NUMERIC:
			case ColAffinities.REAL:
				return 15;
		}
		return 0;
	}

	@Override
	public String getTableName(int column) throws SQLException {
		return nullToEmpty(stmt.getColumnTableName(fixCol(column)));
	}

	@Override
	public String getCatalogName(int column) throws SQLException {
		return nullToEmpty(stmt.getColumnDatabaseName(fixCol(column)));
	}

	@Override
	public int getColumnType(int column) throws SQLException {
		final int affinity = stmt.getColumnAffinity(fixCol(column));
		return DbMeta.getJavaType(affinity);
	}

	@Override
	public String getColumnTypeName(int column) throws SQLException {
		return stmt.getColumnDeclType(fixCol(column));
	}

	@Override
	public boolean isReadOnly(int column) throws SQLException {
		checkColumn(column);
		return false;
	}

	@Override
	public boolean isWritable(int column) throws SQLException {
		checkColumn(column);
		return true;
	}

	@Override
	public boolean isDefinitelyWritable(int column) {
		return true;
	}

	@Override
	public String getColumnClassName(int column) throws SQLException {
		final int affinity = stmt.getColumnAffinity(fixCol(column));
		switch (affinity) {
			case ColAffinities.TEXT:
				return "java.lang.String";
			case ColAffinities.INTEGER:
				return "java.lang.Long";
			case ColAffinities.REAL:
				return "java.lang.Double";
			case ColAffinities.NONE:
				return "[B";
			default:
				return "java.lang.Number";
		}
	}

	@Override
	public  T unwrap(Class iface) throws SQLException {
		if (iface.isAssignableFrom(getClass())) {
			return iface.cast(this);
		}
		throw new SQLException("Cannot unwrap to " + iface.getName());
	}

	@Override
	public boolean isWrapperFor(Class iface) {
		return iface.isAssignableFrom(getClass());
	}

	private void checkColumn(int column) throws SQLException {
		if (column > getColumnCount()) {
			throw new StmtException(stmt, String.format("column index (%d) > column count (%d)", column, getColumnCount()), ErrCodes.WRAPPER_SPECIFIC);
		}
	}

	private static String nullToEmpty(String s) {
		return s == null ? "" : s;
	}
}