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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spanner-jdbc Show documentation
JDBC Driver for Google Cloud Spanner
The newest version!
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 nextCalledForInternalReasons = false;

  private boolean nextCalledForInternalReasonsResult = 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) throws SQLException {
    this.statement = statement;
    this.resultSet = resultSet;
    this.sql = sql;
    callNextForInternalReasons();
  }

  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);
  }

  /**
   * Advances the underlying {@link nl.topicus.jdbc.shaded.com.google.cloud.spanner.ResultSet} to the first position. This
   * forces the query to be executed, and ensures that any invalid queries are reported as quickly
   * as possible.
   * 
   * @throws CloudSpannerSQLException
   */
  private void callNextForInternalReasons() throws CloudSpannerSQLException {
    if (!nextCalledForInternalReasons) {
      try {
        nextCalledForInternalReasonsResult = resultSet.next();
      } catch (SpannerException e) {
        throw new CloudSpannerSQLException(e);
      }
      nextCalledForInternalReasons = true;
    }
  }

  @Override
  public boolean next() throws SQLException {
    ensureOpen();
    if (beforeFirst && nextCalledForInternalReasons) {
      currentRowIndex++;
      nextCalledForInternalReasons = false;
      beforeFirst = false;
      afterLast = !nextCalledForInternalReasonsResult;
      return nextCalledForInternalReasonsResult;
    }
    boolean res = false;
    try {
      res = resultSet.next();
    } catch (SpannerException e) {
      throw new CloudSpannerSQLException(e);
    }
    currentRowIndex++;
    beforeFirst = false;
    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();
    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);
  }

}