nl.topicus.jdbc.metadata.AbstractCloudSpannerWrapper 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
There is a newer version: 1.1.6
package nl.topicus.jdbc.metadata;

import java.sql.Date;
import java.sql.SQLException;
import java.sql.Timestamp;
import java.sql.Types;
import java.sql.Wrapper;
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.exception.CloudSpannerSQLException;

public abstract class AbstractCloudSpannerWrapper implements Wrapper {

  public static int extractColumnType(Type type) {
    if (type.equals(Type.bool()))
      return Types.BOOLEAN;
    if (type.equals(Type.bytes()))
      return Types.BINARY;
    if (type.equals(Type.date()))
      return Types.DATE;
    if (type.equals(Type.float64()))
      return Types.DOUBLE;
    if (type.equals(Type.int64()))
      return Types.BIGINT;
    if (type.equals(Type.string()))
      return Types.NVARCHAR;
    if (type.equals(Type.timestamp()))
      return Types.TIMESTAMP;
    if (type.getCode() == Code.ARRAY)
      return Types.ARRAY;
    return Types.OTHER;
  }

  public static String getGoogleTypeName(int sqlType) {
    if (sqlType == Types.BOOLEAN)
      return Type.bool().getCode().name();
    if (sqlType == Types.BINARY)
      return Type.bytes().getCode().name();
    if (sqlType == Types.DATE)
      return Type.date().getCode().name();
    if (sqlType == Types.DOUBLE || sqlType == Types.FLOAT || sqlType == Types.DECIMAL)
      return Type.float64().getCode().name();
    if (sqlType == Types.BIGINT || sqlType == Types.INTEGER || sqlType == Types.TINYINT)
      return Type.int64().getCode().name();
    if (sqlType == Types.NVARCHAR)
      return Type.string().getCode().name();
    if (sqlType == Types.TIMESTAMP)
      return Type.timestamp().getCode().name();
    if (sqlType == Types.ARRAY)
      return Code.ARRAY.name();

    return "Other";
  }

  public static String getClassName(int sqlType) {
    if (sqlType == Types.BOOLEAN)
      return Boolean.class.getName();
    if (sqlType == Types.BINARY)
      return Byte[].class.getName();
    if (sqlType == Types.DATE)
      return Date.class.getName();
    if (sqlType == Types.DOUBLE || sqlType == Types.FLOAT || sqlType == Types.DECIMAL)
      return Double.class.getName();
    if (sqlType == Types.BIGINT || sqlType == Types.INTEGER || sqlType == Types.TINYINT)
      return Long.class.getName();
    if (sqlType == Types.NVARCHAR)
      return String.class.getName();
    if (sqlType == Types.TIMESTAMP)
      return Timestamp.class.getName();
    if (sqlType == Types.ARRAY)
      return Object.class.getName();

    return null;
  }

  public static String getClassName(Type type) {
    if (type == Type.bool())
      return Boolean.class.getName();
    if (type == Type.bytes())
      return Byte[].class.getName();
    if (type == Type.date())
      return Date.class.getName();
    if (type == Type.float64())
      return Double.class.getName();
    if (type == Type.int64())
      return Long.class.getName();
    if (type == Type.string())
      return String.class.getName();
    if (type == Type.timestamp())
      return Timestamp.class.getName();
    if (type.getCode() == Code.ARRAY) {
      if (type.getArrayElementType() == Type.bool())
        return Boolean[].class.getName();
      if (type.getArrayElementType() == Type.bytes())
        return Byte[][].class.getName();
      if (type.getArrayElementType() == Type.date())
        return Date[].class.getName();
      if (type.getArrayElementType() == Type.float64())
        return Double[].class.getName();
      if (type.getArrayElementType() == Type.int64())
        return Long[].class.getName();
      if (type.getArrayElementType() == Type.string())
        return String[].class.getName();
      if (type.getArrayElementType() == Type.timestamp())
        return Timestamp[].class.getName();
    }
    return null;
  }

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

  @Override
  public  T unwrap(Class iface) throws SQLException {
    if (isWrapperFor(getClass())) {
      return iface.cast(this);
    }
    throw new CloudSpannerSQLException("Cannot unwrap to " + iface.getName(),
        nl.topicus.jdbc.shaded.com.google.rpc.Code.INVALID_ARGUMENT);
  }

}