com.github.cassandra.jdbc.provider.datastax.CassandraResultSet Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cassandra-jdbc-driver Show documentation
Type 4 JDBC driver for Apache Cassandra built on top of existing great libs like java driver from DataStax. It supports Cassandra 2.x and above with improved SQL compatibility.
There is a newer version: 0.6.4
Show newest version
/*
 *
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 *
 */
package com.github.cassandra.jdbc.provider.datastax;

import com.datastax.driver.core.ColumnDefinitions.Definition;
import com.datastax.driver.core.ResultSet;
import com.datastax.driver.core.Row;
import com.github.cassandra.jdbc.*;
import org.pmw.tinylog.Logger;

import java.sql.SQLException;

/**
 * This is a result set implementation built on top of DataStax Java driver.
 *
 * @author Zhichun Wu
 */
public class CassandraResultSet extends BaseCassandraResultSet {
    private Row _currentRow;
    private ResultSet _resultSet;

    protected CassandraResultSet(BaseCassandraStatement statement, CassandraCqlStatement parsedStmt, ResultSet rs) {
        super(statement, parsedStmt);

        if (rs != null) {
            for (Definition def : rs.getColumnDefinitions()) {
                CassandraColumnDefinition d = new CassandraColumnDefinition(
                        def.getKeyspace(), def.getTable(), def.getName(), def
                        .getType().getName().toString(), false);
                metadata.addColumnDefinition(d);
            }
        }

        _resultSet = rs;
    }

    @SuppressWarnings("unchecked")
    @Override
    protected  T getValue(int columnIndex, Class clazz)
            throws SQLException {
        Logger.trace("Trying to get value with inputs: line={}, column={}, type={}", getRow(), columnIndex, clazz);

        Object rawValue = null;
        T result = null;
        if (_currentRow != null) {
            String typeName = metadata.getColumnTypeName(columnIndex);

            if (clazz == String.class
                    && (CassandraDataType.TIME.getTypeName().equals(typeName)
                    || CassandraDataType.TIMESTAMP.getTypeName().equals(typeName))) {
                rawValue = _currentRow.getString(columnIndex - 1);
            } else {
                rawValue = _currentRow.getObject(columnIndex - 1);
            }

            Logger.trace("Got raw value [{}] from line #{}", rawValue, getRow());

            wasNull = rawValue == null;

            try {
                result = getDataTypeConverters().convert(rawValue, clazz, true);
            } catch (ClassCastException e) {
                Logger.warn(e, "Not able to convert [{}] to {}", rawValue, clazz);

                if (!quiet) {
                    throw new SQLException(e);
                }
            }
        }

        Logger.trace("Return value: raw={}, converted={}", rawValue, result);

        return result;
    }

    @Override
    protected boolean hasMore() {
        return _resultSet != null && !_resultSet.isExhausted();
    }

    @Override
    protected  void setValue(int columnIndex, T value) throws SQLException {
        throw CassandraErrors.notSupportedException();
    }

    @Override
    protected SQLException tryClose() {
        if (_resultSet != null) {
            _resultSet = null;
            _currentRow = null;
        }

        return null;
    }

    @Override
    protected boolean tryIterate() throws SQLException {
        boolean result = false;
        if (_resultSet != null) {
            try {
                _currentRow = _resultSet.one();
            } catch (Exception e) {
                throw new SQLException(e);
            }

            result = _currentRow != null;
        }

        return result;
    }

    @Override
    protected boolean tryMoveTo(int rows, boolean relativeIndex)
            throws SQLException {
        throw CassandraErrors.notSupportedException();
    }

    @Override
    protected Object unwrap() {
        return _resultSet;
    }
}