• Home
• io.snappydata
• snappydata-store-client
• 1.6.1
• source code
• ClientFetchColumnValue.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.

io.snappydata.thrift.internal.ClientFetchColumnValue Maven / Gradle / Ivy

/*
 * Copyright (c) 2010-2015 Pivotal Software, Inc. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you
 * may not use this file except in compliance with the License. You
 * may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * permissions and limitations under the License. See accompanying
 * LICENSE file.
 */
/*
 * Changes for SnappyData data platform.
 *
 * Portions Copyright (c) 2017 SnappyData, Inc. All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you
 * may not use this file except in compliance with the License. You
 * may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * permissions and limitations under the License. See accompanying
 * LICENSE file.
 */

package io.snappydata.thrift.internal;

import java.io.InputStream;
import java.io.Reader;
import java.math.BigDecimal;
import java.net.URL;
import java.sql.*;
import java.util.Calendar;
import java.util.Map;

import com.pivotal.gemfirexd.internal.shared.common.reference.SQLState;
import io.snappydata.thrift.Row;
import io.snappydata.thrift.RowSet;
import io.snappydata.thrift.common.ColumnValueConverter;
import io.snappydata.thrift.common.Converters;
import io.snappydata.thrift.common.ThriftExceptionUtil;
import io.snappydata.thrift.snappydataConstants;

/**
 * Common base class to fetch various types of column values from a Row like
 * those in {@link ResultSet}.
 */
abstract class ClientFetchColumnValue {

  protected final ClientService service;
  protected ClientFinalizer finalizer;
  protected boolean wasNull;

  protected ClientFetchColumnValue(ClientService service, byte entityType) {
    this.service = service;
    this.finalizer = entityType != snappydataConstants.INVALID_ID
        ? new ClientFinalizer(this, service, entityType) : null;
  }

  protected final HostConnection getLobSource(boolean throwOnFailure,
      String op) throws SQLException {
    return this.service.getLobSource(throwOnFailure, op);
  }

  protected final void setCurrentSource(byte entityType, long newId, RowSet rs) {
    ClientFinalizer finalizer = this.finalizer;
    if (newId != snappydataConstants.INVALID_ID) {
      final HostConnection currentSource = service.getCurrentHostConnection();
      HostConnection newSource = currentSource;
      if (rs != null && (currentSource.connId != rs.connId
          || currentSource.token != rs.token
          || currentSource.hostAddr != rs.source)) {
        newSource = new HostConnection(rs.source, rs.connId, rs.token);
      }
      if (finalizer == null) {
        // create a new finalizer
        this.finalizer = finalizer = new ClientFinalizer(this, this.service,
            entityType);
      }
      finalizer.updateReferentData(newId, newSource);
    } else if (finalizer != null) {
      // clear the finalizer
      finalizer.clearAll();
      this.finalizer = null;
    }
  }

  protected final void clearFinalizer() {
    final ClientFinalizer finalizer = this.finalizer;
    if (finalizer != null) {
      finalizer.clearAll();
      this.finalizer = null;
    }
  }

  protected void reset() {
    this.wasNull = false;
  }

  protected final String getString(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "String", false, columnIndex);
    String str = cvc.toString(row, columnIndex, this.service);
    this.wasNull = (str == null);
    return str;
  }

  protected final boolean getBoolean(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "boolean", false, columnIndex);
    boolean v = cvc.toBoolean(row, columnIndex);
    this.wasNull = cvc.isNull();
    return v;
  }

  protected final byte getByte(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "byte", false, columnIndex);
    byte v = cvc.toByte(row, columnIndex);
    this.wasNull = cvc.isNull();
    return v;
  }

  protected final short getShort(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "short", false, columnIndex);
    short v = cvc.toShort(row, columnIndex);
    this.wasNull = cvc.isNull();
    return v;
  }

  protected final int getInt(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "int", false, columnIndex);
    int v = cvc.toInteger(row, columnIndex);
    this.wasNull = cvc.isNull();
    return v;
  }

  protected final long getLong(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "long", false, columnIndex);
    long v = cvc.toLong(row, columnIndex);
    this.wasNull = cvc.isNull();
    return v;
  }

  protected final float getFloat(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "float", false, columnIndex);
    float v = cvc.toFloat(row, columnIndex);
    this.wasNull = cvc.isNull();
    return v;
  }

  protected final double getDouble(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "doube", false, columnIndex);
    double v = cvc.toDouble(row, columnIndex);
    this.wasNull = cvc.isNull();
    return v;
  }

  protected final BigDecimal getBigDecimal(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "BigDecimal", false, columnIndex);
    BigDecimal v = cvc.toBigDecimal(row, columnIndex);
    this.wasNull = (v == null);
    return v;
  }

  protected final BigDecimal getBigDecimal(final int columnIndex,
      final int scale, final int snappyTypeValue, final Row row)
      throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "BigDecimal", false, columnIndex);
    BigDecimal v = cvc.toBigDecimal(row, columnIndex);
    if (v != null) {
      // rounding as per server side EmbedResultSet20
      v = v.setScale(scale, BigDecimal.ROUND_HALF_DOWN);
      this.wasNull = false;
      return v;
    } else {
      this.wasNull = true;
      return null;
    }
  }

  protected final byte[] getBytes(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "byte[]", false, columnIndex);
    byte[] v = cvc.toBytes(row, columnIndex, this.service);
    this.wasNull = (v == null);
    return v;
  }

  protected final Date getDate(final int columnIndex, final Calendar cal,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "Date", false, columnIndex);
    Date v = cvc.toDate(row, columnIndex, cal);
    this.wasNull = (v == null);
    return v;
  }

  protected final Time getTime(final int columnIndex, final Calendar cal,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "Time", false, columnIndex);
    Time v = cvc.toTime(row, columnIndex, cal);
    this.wasNull = (v == null);
    return v;
  }

  protected final Timestamp getTimestamp(final int columnIndex,
      final Calendar cal, final int snappyTypeValue, final Row row)
      throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "Timestamp", false, columnIndex);
    Timestamp v = cvc.toTimestamp(row, columnIndex, cal);
    this.wasNull = (v == null);
    return v;
  }

  protected final Object getObject(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "Object", false, columnIndex);
    Object v = cvc.toObject(row, columnIndex, this.service);
    this.wasNull = (v == null);
    return v;
  }

  protected final Blob getBlob(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "Blob", false, columnIndex);
    Blob v = cvc.toBlob(row, columnIndex, this.service);
    this.wasNull = (v == null);
    return v;
  }

  protected final InputStream getBinaryStream(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "BinaryStream", false, columnIndex);
    InputStream v = cvc.toBinaryStream(row, columnIndex, this.service);
    this.wasNull = (v == null);
    return v;
  }

  protected final Clob getClob(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "Clob", false, columnIndex);
    Clob v = cvc.toClob(row, columnIndex, this.service);
    this.wasNull = (v == null);
    return v;
  }

  protected final Reader getCharacterStream(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "CharacterStream", false, columnIndex);
    Reader v = cvc.toCharacterStream(row, columnIndex, this.service);
    this.wasNull = (v == null);
    return v;
  }

  protected final InputStream getAsciiStream(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    ColumnValueConverter cvc = Converters.getConverter(
        snappyTypeValue, "AsciiStream", false, columnIndex);
    InputStream v = cvc.toAsciiStream(row, columnIndex, this.service);
    this.wasNull = (v == null);
    return v;
  }

  protected Object getObject(final int columnIndex,
      final Map> map, final int snappyTypeValue,
      final Row row) throws SQLException {
    if (map == null) {
      throw ThriftExceptionUtil.newSQLException(SQLState.INVALID_API_PARAMETER,
          null, null, "map", "FetchColumnValue.getObject(int,Map)");
    }
    if (map.isEmpty()) {
      // Map is empty call the normal getObject method.
      return getObject(columnIndex, snappyTypeValue, row);
    } else {
      throw ThriftExceptionUtil
          .notImplemented("FetchColumnValue.getObject(int,Map)");
    }
  }

  protected Ref getRef(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    throw ThriftExceptionUtil.notImplemented("FetchColumnValue.getRef");
  }

  protected Array getArray(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    throw ThriftExceptionUtil.notImplemented("FetchColumnValue.getArray");
  }

  protected URL getURL(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    throw ThriftExceptionUtil.notImplemented("FetchColumnValue.getURL");
  }

  protected RowId getRowId(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    throw ThriftExceptionUtil.notImplemented("FetchColumnValue.getRowId");
  }

  protected NClob getNClob(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    throw ThriftExceptionUtil.notImplemented("FetchColumnValue.getNClob");
  }

  protected SQLXML getSQLXML(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    throw ThriftExceptionUtil.notImplemented("FetchColumnValue.getSQLXML");
  }

  protected String getNString(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    throw ThriftExceptionUtil.notImplemented("FetchColumnValue.getNString");
  }

  protected Reader getNCharacterStream(final int columnIndex,
      final int snappyTypeValue, final Row row) throws SQLException {
    throw ThriftExceptionUtil
        .notImplemented("FetchColumnValue.getNCharacterStream");
  }

  protected  T getObject(final int columnIndex, final Class type,
      final int snappyTypeValue, final Row row) throws SQLException {
    throw ThriftExceptionUtil
        .notImplemented("FetchColumnValue.getObject(int,Class)");
  }
}