• Home
• nl.topicus
• spanner-jdbc
• 1.0.8
• source code
• CloudSpannerResultSet.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

nl.topicus.jdbc.resultset.CloudSpannerResultSet Maven / Gradle / Ivy

package nl.topicus.jdbc.resultset;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.io.Reader;
import java.io.StringReader;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.net.MalformedURLException;
import java.net.URL;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.sql.Array;
import java.sql.Date;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.SQLWarning;
import java.sql.Statement;
import java.sql.Time;
import java.sql.Timestamp;
import java.util.Calendar;
import java.util.List;
import java.util.Map;

import nl.topicus.jdbc.shaded.com.google.cloud.spanner.SpannerException;
import nl.topicus.jdbc.shaded.com.google.cloud.spanner.Type;
import nl.topicus.jdbc.shaded.com.google.cloud.spanner.Type.Code;

import nl.topicus.jdbc.CloudSpannerArray;
import nl.topicus.jdbc.CloudSpannerDataType;
import nl.topicus.jdbc.exception.CloudSpannerSQLException;
import nl.topicus.jdbc.statement.CloudSpannerStatement;
import nl.topicus.jdbc.util.CloudSpannerConversionUtil;

public class CloudSpannerResultSet extends AbstractCloudSpannerResultSet
{
	private nl.topicus.jdbc.shaded.com.google.cloud.spanner.ResultSet resultSet;

	private boolean closed = false;

	private boolean wasNull = false;

	private boolean beforeFirst = true;

	private int currentRowIndex = -1;

	private boolean afterLast = false;

	private boolean nextCalledForMetaData = false;

	private boolean nextCalledForMetaDataResult = false;

	private final CloudSpannerStatement statement;

	private final String sql;

	CloudSpannerResultSet(CloudSpannerStatement statement, String sql)
	{
		this.statement = statement;
		this.sql = sql;
	}

	public CloudSpannerResultSet(CloudSpannerStatement statement, nl.topicus.jdbc.shaded.com.google.cloud.spanner.ResultSet resultSet,
			String sql)
	{
		this.statement = statement;
		this.resultSet = resultSet;
		this.sql = sql;
	}

	void setResultSet(nl.topicus.jdbc.shaded.com.google.cloud.spanner.ResultSet rs)
	{
		this.resultSet = rs;
	}

	@Override
	public boolean wasNull() throws SQLException
	{
		return wasNull;
	}

	private void ensureOpenAndInValidPosition() throws SQLException
	{
		ensureOpen();
		ensureAfterFirst();
		ensureBeforeLast();
	}

	private void ensureAfterFirst() throws SQLException
	{
		if (beforeFirst)
			throw new CloudSpannerSQLException("Before first record", nl.topicus.jdbc.shaded.com.google.rpc.Code.FAILED_PRECONDITION);
	}

	private void ensureBeforeLast() throws SQLException
	{
		if (afterLast)
			throw new CloudSpannerSQLException("After last record", nl.topicus.jdbc.shaded.com.google.rpc.Code.FAILED_PRECONDITION);
	}

	protected void ensureOpen() throws SQLException
	{
		if (closed)
			throw new CloudSpannerSQLException("Resultset is closed", nl.topicus.jdbc.shaded.com.google.rpc.Code.FAILED_PRECONDITION);
	}

	@Override
	public boolean next() throws SQLException
	{
		ensureOpen();
		if (!beforeFirst && nextCalledForMetaData)
		{
			nextCalledForMetaData = false;
			return nextCalledForMetaDataResult;
		}
		beforeFirst = false;
		currentRowIndex++;
		boolean res = false;
		try
		{
			res = resultSet.next();
		}
		catch (SpannerException e)
		{
			throw new CloudSpannerSQLException(e);
		}
		afterLast = !res;

		return res;
	}

	@Override
	public boolean isFirst() throws SQLException
	{
		return currentRowIndex == 0;
	}

	@Override
	public int getRow() throws SQLException
	{
		return currentRowIndex + 1;
	}

	@Override
	public void close() throws SQLException
	{
		if (resultSet != null)
			resultSet.close();
		closed = true;
	}

	@Override
	public String getString(int columnIndex) throws SQLException
	{
		return isNull(columnIndex) ? null : resultSet.getString(columnIndex - 1);
	}

	@Override
	public URL getURL(int columnIndex) throws SQLException
	{
		try
		{
			return isNull(columnIndex) ? null : new URL(resultSet.getString(columnIndex - 1));
		}
		catch (MalformedURLException e)
		{
			throw new CloudSpannerSQLException("Invalid URL: " + resultSet.getString(columnIndex - 1),
					nl.topicus.jdbc.shaded.com.google.rpc.Code.INVALID_ARGUMENT);
		}
	}

	@Override
	public boolean getBoolean(int columnIndex) throws SQLException
	{
		return isNull(columnIndex) ? false : resultSet.getBoolean(columnIndex - 1);
	}

	@Override
	public long getLong(int columnIndex) throws SQLException
	{
		return isNull(columnIndex) ? 0 : resultSet.getLong(columnIndex - 1);
	}

	@Override
	public double getDouble(int columnIndex) throws SQLException
	{
		return isNull(columnIndex) ? 0 : resultSet.getDouble(columnIndex - 1);
	}

	private BigDecimal toBigDecimal(double d)
	{
		return BigDecimal.valueOf(d);
	}

	private BigDecimal toBigDecimal(double d, int scale)
	{
		BigDecimal res = BigDecimal.valueOf(d);
		res = res.setScale(scale, RoundingMode.HALF_UP);
		return res;
	}

	@Override
	public BigDecimal getBigDecimal(int columnIndex, int scale) throws SQLException
	{
		return isNull(columnIndex) ? null : toBigDecimal(resultSet.getDouble(columnIndex - 1), scale);
	}

	@Override
	public byte[] getBytes(int columnIndex) throws SQLException
	{
		return isNull(columnIndex) ? null : resultSet.getBytes(columnIndex - 1).toByteArray();
	}

	@Override
	public Date getDate(int columnIndex) throws SQLException
	{
		return isNull(columnIndex) ? null : CloudSpannerConversionUtil.toSqlDate(resultSet.getDate(columnIndex - 1));
	}

	@Override
	public Time getTime(int columnIndex) throws SQLException
	{
		return isNull(columnIndex) ? null
				: CloudSpannerConversionUtil.toSqlTime(resultSet.getTimestamp(columnIndex - 1));
	}

	@Override
	public Timestamp getTimestamp(int columnIndex) throws SQLException
	{
		return isNull(columnIndex) ? null : resultSet.getTimestamp(columnIndex - 1).toSqlTimestamp();
	}

	@Override
	public String getString(String columnLabel) throws SQLException
	{
		return isNull(columnLabel) ? null : resultSet.getString(columnLabel);
	}

	@Override
	public URL getURL(String columnLabel) throws SQLException
	{
		try
		{
			return isNull(columnLabel) ? null : new URL(resultSet.getString(columnLabel));
		}
		catch (MalformedURLException e)
		{
			throw new CloudSpannerSQLException("Invalid URL: " + resultSet.getString(columnLabel),
					nl.topicus.jdbc.shaded.com.google.rpc.Code.INVALID_ARGUMENT);
		}
	}

	@Override
	public boolean getBoolean(String columnLabel) throws SQLException
	{
		return isNull(columnLabel) ? false : resultSet.getBoolean(columnLabel);
	}

	@Override
	public long getLong(String columnLabel) throws SQLException
	{
		return isNull(columnLabel) ? 0 : resultSet.getLong(columnLabel);
	}

	@Override
	public double getDouble(String columnLabel) throws SQLException
	{
		return isNull(columnLabel) ? 0 : resultSet.getDouble(columnLabel);
	}

	@Override
	public BigDecimal getBigDecimal(String columnLabel, int scale) throws SQLException
	{
		return isNull(columnLabel) ? null : toBigDecimal(resultSet.getDouble(columnLabel), scale);
	}

	@Override
	public byte[] getBytes(String columnLabel) throws SQLException
	{
		return isNull(columnLabel) ? null : resultSet.getBytes(columnLabel).toByteArray();
	}

	@Override
	public Date getDate(String columnLabel) throws SQLException
	{
		return isNull(columnLabel) ? null : CloudSpannerConversionUtil.toSqlDate(resultSet.getDate(columnLabel));
	}

	@Override
	public Time getTime(String columnLabel) throws SQLException
	{
		return isNull(columnLabel) ? null : CloudSpannerConversionUtil.toSqlTime(resultSet.getTimestamp(columnLabel));
	}

	@Override
	public Timestamp getTimestamp(String columnLabel) throws SQLException
	{
		return isNull(columnLabel) ? null : resultSet.getTimestamp(columnLabel).toSqlTimestamp();
	}

	@Override
	public CloudSpannerResultSetMetaData getMetaData() throws SQLException
	{
		ensureOpen();
		if (beforeFirst)
		{
			nextCalledForMetaDataResult = resultSet.next();
			afterLast = !nextCalledForMetaDataResult;
			beforeFirst = false;
			nextCalledForMetaData = true;
		}
		return new CloudSpannerResultSetMetaData(resultSet, statement, sql);
	}

	@Override
	public int findColumn(String columnLabel) throws SQLException
	{
		ensureOpen();
		try
		{
			return resultSet.getColumnIndex(columnLabel) + 1;
		}
		catch (IllegalArgumentException e)
		{
			throw new CloudSpannerSQLException("Column not found: " + columnLabel, nl.topicus.jdbc.shaded.com.google.rpc.Code.INVALID_ARGUMENT,
					e);
		}
	}

	private boolean isNull(int columnIndex) throws SQLException
	{
		ensureOpenAndInValidPosition();
		boolean res = resultSet.isNull(columnIndex - 1);
		wasNull = res;
		return res;
	}

	private boolean isNull(String columnName) throws SQLException
	{
		ensureOpenAndInValidPosition();
		boolean res = resultSet.isNull(columnName);
		wasNull = res;
		return res;
	}

	@Override
	public BigDecimal getBigDecimal(int columnIndex) throws SQLException
	{
		return isNull(columnIndex) ? null : toBigDecimal(resultSet.getDouble(columnIndex - 1));
	}

	@Override
	public BigDecimal getBigDecimal(String columnLabel) throws SQLException
	{
		return isNull(columnLabel) ? null : toBigDecimal(resultSet.getDouble(columnLabel));
	}

	@Override
	public Statement getStatement() throws SQLException
	{
		ensureOpen();
		return statement;
	}

	@Override
	public Date getDate(int columnIndex, Calendar cal) throws SQLException
	{
		return isNull(columnIndex) ? null
				: CloudSpannerConversionUtil.toSqlDate(resultSet.getDate(columnIndex - 1), cal);
	}

	@Override
	public Date getDate(String columnLabel, Calendar cal) throws SQLException
	{
		return isNull(columnLabel) ? null : CloudSpannerConversionUtil.toSqlDate(resultSet.getDate(columnLabel), cal);
	}

	@Override
	public Time getTime(int columnIndex, Calendar cal) throws SQLException
	{
		return isNull(columnIndex) ? null
				: CloudSpannerConversionUtil.toSqlTime(resultSet.getTimestamp(columnIndex - 1));
	}

	@Override
	public Time getTime(String columnLabel, Calendar cal) throws SQLException
	{
		return isNull(columnLabel) ? null : CloudSpannerConversionUtil.toSqlTime(resultSet.getTimestamp(columnLabel));
	}

	@Override
	public Timestamp getTimestamp(int columnIndex, Calendar cal) throws SQLException
	{
		return isNull(columnIndex) ? null : resultSet.getTimestamp(columnIndex - 1).toSqlTimestamp();
	}

	@Override
	public Timestamp getTimestamp(String columnLabel, Calendar cal) throws SQLException
	{
		return isNull(columnLabel) ? null : resultSet.getTimestamp(columnLabel).toSqlTimestamp();
	}

	@Override
	public boolean isClosed() throws SQLException
	{
		return closed;
	}

	@Override
	public byte getByte(int columnIndex) throws SQLException
	{
		return isNull(columnIndex) ? 0 : (byte) resultSet.getLong(columnIndex - 1);
	}

	@Override
	public short getShort(int columnIndex) throws SQLException
	{
		return isNull(columnIndex) ? 0 : (short) resultSet.getLong(columnIndex - 1);
	}

	@Override
	public int getInt(int columnIndex) throws SQLException
	{
		return isNull(columnIndex) ? 0 : (int) resultSet.getLong(columnIndex - 1);
	}

	@Override
	public float getFloat(int columnIndex) throws SQLException
	{
		return isNull(columnIndex) ? 0 : (float) resultSet.getDouble(columnIndex - 1);
	}

	@Override
	public byte getByte(String columnLabel) throws SQLException
	{
		return isNull(columnLabel) ? 0 : (byte) resultSet.getLong(columnLabel);
	}

	@Override
	public short getShort(String columnLabel) throws SQLException
	{
		return isNull(columnLabel) ? 0 : (short) resultSet.getLong(columnLabel);
	}

	@Override
	public int getInt(String columnLabel) throws SQLException
	{
		return isNull(columnLabel) ? 0 : (int) resultSet.getLong(columnLabel);
	}

	@Override
	public float getFloat(String columnLabel) throws SQLException
	{
		return isNull(columnLabel) ? 0 : (float) resultSet.getDouble(columnLabel);
	}

	@Override
	public Object getObject(String columnLabel) throws SQLException
	{
		Type type = resultSet.getColumnType(columnLabel);
		return isNull(columnLabel) ? null : getObject(type, columnLabel);
	}

	@Override
	public Object getObject(int columnIndex) throws SQLException
	{
		Type type = resultSet.getColumnType(columnIndex - 1);
		return isNull(columnIndex) ? null : getObject(type, columnIndex);
	}

	private Object getObject(Type type, String columnLabel) throws SQLException
	{
		return getObject(type, resultSet.getColumnIndex(columnLabel) + 1);
	}

	private Object getObject(Type type, int columnIndex) throws SQLException
	{
		if (type == Type.bool())
			return getBoolean(columnIndex);
		if (type == Type.bytes())
			return getBytes(columnIndex);
		if (type == Type.date())
			return getDate(columnIndex);
		if (type == Type.float64())
			return getDouble(columnIndex);
		if (type == Type.int64())
			return getLong(columnIndex);
		if (type == Type.string())
			return getString(columnIndex);
		if (type == Type.timestamp())
			return getTimestamp(columnIndex);
		if (type.getCode() == Code.ARRAY)
			return getArray(columnIndex);
		throw new CloudSpannerSQLException("Unknown type: " + type.toString(), nl.topicus.jdbc.shaded.com.google.rpc.Code.INVALID_ARGUMENT);
	}

	@Override
	public Array getArray(String columnLabel) throws SQLException
	{
		Type type = resultSet.getColumnType(columnLabel);
		if (type.getCode() != Code.ARRAY)
			throw new CloudSpannerSQLException("Column with label " + columnLabel + " does not contain an array",
					nl.topicus.jdbc.shaded.com.google.rpc.Code.INVALID_ARGUMENT);
		return getArray(resultSet.getColumnIndex(columnLabel) + 1);
	}

	@Override
	public Array getArray(int columnIndex) throws SQLException
	{
		if (isNull(columnIndex))
			return null;
		Type type = resultSet.getColumnType(columnIndex - 1);
		if (type.getCode() != Code.ARRAY)
			throw new CloudSpannerSQLException("Column with index " + columnIndex + " does not contain an array",
					nl.topicus.jdbc.shaded.com.google.rpc.Code.INVALID_ARGUMENT);
		CloudSpannerDataType dataType = CloudSpannerDataType.getType(type.getArrayElementType().getCode());
		List elements = dataType.getArrayElements(resultSet, columnIndex - 1);

		return CloudSpannerArray.createArray(dataType, elements);
	}

	@Override
	public SQLWarning getWarnings() throws SQLException
	{
		ensureOpen();
		return null;
	}

	@Override
	public void clearWarnings() throws SQLException
	{
		ensureOpen();
	}

	@Override
	public boolean isBeforeFirst() throws SQLException
	{
		ensureOpen();
		return beforeFirst;
	}

	@Override
	public boolean isAfterLast() throws SQLException
	{
		ensureOpen();
		return afterLast;
	}

	@Override
	public Reader getCharacterStream(int columnIndex) throws SQLException
	{
		String val = getString(columnIndex);
		return val == null ? null : new StringReader(val);
	}

	@Override
	public Reader getCharacterStream(String columnLabel) throws SQLException
	{
		String val = getString(columnLabel);
		return val == null ? null : new StringReader(val);
	}

	private InputStream getInputStream(String val, Charset charset)
	{
		if (val == null)
			return null;
		byte[] b = val.getBytes(charset);
		return new ByteArrayInputStream(b);
	}

	@Override
	public InputStream getAsciiStream(int columnIndex) throws SQLException
	{
		return getInputStream(getString(columnIndex), StandardCharsets.US_ASCII);
	}

	@Override
	public InputStream getUnicodeStream(int columnIndex) throws SQLException
	{
		return getInputStream(getString(columnIndex), StandardCharsets.UTF_16LE);
	}

	@Override
	public InputStream getBinaryStream(int columnIndex) throws SQLException
	{
		byte[] val = getBytes(columnIndex);
		return val == null ? null : new ByteArrayInputStream(val);
	}

	@Override
	public InputStream getAsciiStream(String columnLabel) throws SQLException
	{
		return getInputStream(getString(columnLabel), StandardCharsets.US_ASCII);
	}

	@Override
	public InputStream getUnicodeStream(String columnLabel) throws SQLException
	{
		return getInputStream(getString(columnLabel), StandardCharsets.UTF_16LE);
	}

	@Override
	public InputStream getBinaryStream(String columnLabel) throws SQLException
	{
		byte[] val = getBytes(columnLabel);
		return val == null ? null : new ByteArrayInputStream(val);
	}

	@Override
	public String getNString(int columnIndex) throws SQLException
	{
		return getString(columnIndex);
	}

	@Override
	public String getNString(String columnLabel) throws SQLException
	{
		return getString(columnLabel);
	}

	@Override
	public Reader getNCharacterStream(int columnIndex) throws SQLException
	{
		return getCharacterStream(columnIndex);
	}

	@Override
	public Reader getNCharacterStream(String columnLabel) throws SQLException
	{
		return getCharacterStream(columnLabel);
	}

	@Override
	public int getHoldability() throws SQLException
	{
		return ResultSet.HOLD_CURSORS_OVER_COMMIT;
	}

	@Override
	public  T getObject(int columnIndex, Class type) throws SQLException
	{
		return convertObject(getObject(columnIndex), type, resultSet.getColumnType(columnIndex - 1));
	}

	@Override
	public  T getObject(String columnLabel, Class type) throws SQLException
	{
		return convertObject(getObject(columnLabel), type, resultSet.getColumnType(columnLabel));
	}

	@Override
	public Object getObject(int columnIndex, Map> map) throws SQLException
	{
		return convertObject(getObject(columnIndex), map, resultSet.getColumnType(columnIndex - 1));
	}

	@Override
	public Object getObject(String columnLabel, Map> map) throws SQLException
	{
		return convertObject(getObject(columnLabel), map, resultSet.getColumnType(columnLabel));
	}

	@SuppressWarnings("unchecked")
	private  T convertObject(Object o, Class javaType, Type type) throws SQLException
	{
		return (T) CloudSpannerConversionUtil.convert(o, type, javaType);
	}

	private Object convertObject(Object o, Map> map, Type type) throws SQLException
	{
		if (map == null)
			throw new CloudSpannerSQLException("Map may not be null", nl.topicus.jdbc.shaded.com.google.rpc.Code.INVALID_ARGUMENT);
		if (o == null)
			return null;
		Class javaType = map.get(type.getCode().name());
		if (javaType == null)
			return o;
		return CloudSpannerConversionUtil.convert(o, type, javaType);
	}

}