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.
com.github.adejanovski.cassandra.jdbc.CassandraResultSet Maven / Gradle / Ivy
/*
*
* 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.
*/
package com.github.adejanovski.cassandra.jdbc;
import static com.github.adejanovski.cassandra.jdbc.Utils.*;
import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.net.URL;
import java.nio.ByteBuffer;
import java.nio.charset.CharacterCodingException;
import java.sql.*;
import java.sql.Date;
import java.util.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.datastax.driver.core.ColumnDefinitions;
import com.datastax.driver.core.ColumnDefinitions.Definition;
import com.datastax.driver.core.DataType;
import com.datastax.driver.core.Row;
import com.datastax.driver.core.UDTValue;
import com.datastax.driver.core.exceptions.InvalidTypeException;
import com.google.common.collect.Iterators;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import com.google.common.collect.Sets;
/**
*
* The Supported Data types in CQL are as follows:
*
*
*
* type
* java type
* description
*
*
* ascii
* String
* ASCII character string
*
*
* bigint
* Long
* 64-bit signed long
*
*
* blob
* ByteBuffer
* Arbitrary bytes (no validation)
*
*
* boolean
* Boolean
* true or false
*
*
* counter
* Long
* Counter column (64-bit long)
*
*
* decimal
* BigDecimal
* Variable-precision decimal
*
*
* double
* Double
* 64-bit IEEE-754 floating point
*
*
* float
* Float
* 32-bit IEEE-754 floating point
*
*
* int
* Integer
* 32-bit signed int
*
*
* text
* String
* UTF8 encoded string
*
*
* timestamp
* Date
* A timestamp
*
*
* uuid
* UUID
* Type 1 or type 4 UUID
*
*
* varchar
* String
* UTF8 encoded string
*
*
* varint
* BigInteger
* Arbitrary-precision integer
*
*
*
*/
class CassandraResultSet extends AbstractResultSet implements CassandraResultSetExtras
{
private static final Logger logger = LoggerFactory.getLogger(CassandraResultSet.class);
public static final int DEFAULT_TYPE = ResultSet.TYPE_FORWARD_ONLY;
public static final int DEFAULT_CONCURRENCY = ResultSet.CONCUR_READ_ONLY;
public static final int DEFAULT_HOLDABILITY = ResultSet.HOLD_CURSORS_OVER_COMMIT;
private Row currentRow;
/**
* The rows iterator.
*/
private int currentIteratorIndex=0;
private Iterator rowsIterator;
private ArrayList> rowsIterators;
int rowNumber = 0;
// the current row key when iterating through results.
private byte[] curRowKey = null;
/**
* The values.
*/
// private List values = new ArrayList();
/**
* The index map.
*/
// private Map indexMap = new HashMap();
private final CResultSetMetaData meta;
private final CassandraStatement statement;
private int resultSetType;
private int fetchDirection;
private int fetchSize;
private boolean wasNull;
private com.datastax.driver.core.ResultSet driverResultSet;
//private ArrayList severalDriverResultSet;
//private CqlMetadata schema;
/**
* no argument constructor.
*/
CassandraResultSet()
{
statement = null;
meta = new CResultSetMetaData();
}
/**
* Instantiates a new cassandra result set from a com.datastax.driver.core.ResultSet.
*/
CassandraResultSet(CassandraStatement statement, com.datastax.driver.core.ResultSet resultSet) throws SQLException
{
this.statement = statement;
this.resultSetType = statement.getResultSetType();
this.fetchDirection = statement.getFetchDirection();
this.fetchSize = statement.getFetchSize();
this.driverResultSet = resultSet;
this.rowsIterators = Lists.newArrayList();
this.currentIteratorIndex = 0;
//this.schema = resultSet.schema;
// Initialize meta-data from schema
populateMetaData();
//rowsIterators.add(resultSet.iterator());
rowsIterator = resultSet.iterator();
//colDefinitions = resultSet.getColumnDefinitions();
if (hasMoreRows())
{
populateColumns();
}
meta = new CResultSetMetaData();
}
/**
* Instantiates a new cassandra result set from a com.datastax.driver.core.ResultSet.
*/
CassandraResultSet(CassandraStatement statement, ArrayList resultSets) throws SQLException
{
this.statement = statement;
this.resultSetType = statement.getResultSetType();
this.fetchDirection = statement.getFetchDirection();
this.fetchSize = statement.getFetchSize();
//this.rowsIterators = Lists.newArrayList();
// We have several result sets, but we will use only the first one for metadata needs
this.driverResultSet = resultSets.get(0);
// Now we concatenate iterators of the different result sets into a single one and voilà !! ;)
currentIteratorIndex=0;
rowsIterator = driverResultSet.iterator();
for(int i=1;i currentRow.getColumnDefinitions().asList().size()) throw new SQLSyntaxErrorException(String.format(MUST_BE_POSITIVE, String.valueOf(index)) + " " + currentRow.getColumnDefinitions().asList().size());
}
}else if(driverResultSet!=null){
if(driverResultSet.getColumnDefinitions()!=null){
if (index < 1 || index > driverResultSet.getColumnDefinitions().asList().size()) throw new SQLSyntaxErrorException(String.format(MUST_BE_POSITIVE, String.valueOf(index)) + " " + driverResultSet.getColumnDefinitions().asList().size());
}
}
}
private final void checkName(String name) throws SQLException
{
//if (indexMap.get(name) == null) throw new SQLSyntaxErrorException(String.format(VALID_LABELS, name));
if(currentRow!=null){
if(currentRow.isNull(name))
wasNull=true;
else
wasNull=false;
if (!currentRow.getColumnDefinitions().contains(name)) throw new SQLSyntaxErrorException(String.format(VALID_LABELS, name));
}else if(driverResultSet!=null){
if(driverResultSet.getColumnDefinitions()!=null){
if (!driverResultSet.getColumnDefinitions().contains(name)) throw new SQLSyntaxErrorException(String.format(VALID_LABELS, name));
}
}
}
private final void checkNotClosed() throws SQLException
{
if (isClosed()) throw new SQLRecoverableException(WAS_CLOSED_RSLT);
}
public void clearWarnings() throws SQLException
{
// This implementation does not support the collection of warnings so clearing is a no-op
// but it is still an exception to call this on a closed connection.
checkNotClosed();
}
public void close() throws SQLException
{
/* indexMap = null;
values = null; */
if(!isClosed()){
this.statement.close();
}
}
public int findColumn(String name) throws SQLException
{
checkNotClosed();
checkName(name);
return currentRow.getColumnDefinitions().getIndexOf(name);
}
public boolean first() throws SQLException
{
throw new SQLFeatureNotSupportedException(NOT_SUPPORTED);
}
public BigDecimal getBigDecimal(int index) throws SQLException
{
checkIndex(index);
return currentRow.getDecimal(index - 1);
}
/** @deprecated */
public BigDecimal getBigDecimal(int index, int scale) throws SQLException
{
checkIndex(index);
return currentRow.getDecimal(index - 1).setScale(scale);
}
public BigDecimal getBigDecimal(String name) throws SQLException
{
checkName(name);
return currentRow.getDecimal(name);
}
/** @deprecated */
public BigDecimal getBigDecimal(String name, int scale) throws SQLException
{
checkName(name);
return currentRow.getDecimal(name).setScale(scale);
}
public BigInteger getBigInteger(int index) throws SQLException
{
checkIndex(index);
return currentRow.getVarint(index - 1);
}
public BigInteger getBigInteger(String name) throws SQLException
{
checkName(name);
return currentRow.getVarint(name);
}
public boolean getBoolean(int index) throws SQLException
{
checkIndex(index);
return currentRow.getBool(index-1);
}
public boolean getBoolean(String name) throws SQLException
{
checkName(name);
return currentRow.getBool(name);
}
public byte getByte(int index) throws SQLException
{
checkIndex(index);
return currentRow.getBytes(index - 1).get();
}
public byte getByte(String name) throws SQLException
{
checkName(name);
return currentRow.getBytes(name).get();
}
public byte[] getBytes(int index) throws SQLException
{
return currentRow.getBytes(index-1).array();
}
public byte[] getBytes(String name) throws SQLException
{
return currentRow.getBytes(name).array();
}
/* public TypedColumn getColumn(int index) throws SQLException
{
checkIndex(index);
checkNotClosed();
return values.get(index - 1);
}
public TypedColumn getColumn(String name) throws SQLException
{
checkName(name);
checkNotClosed();
return values.get(indexMap.get(name).intValue());
}
*/
public int getConcurrency() throws SQLException
{
checkNotClosed();
return statement.getResultSetConcurrency();
}
public Date getDate(int index) throws SQLException
{
checkIndex(index);
if(currentRow.getDate(index - 1)==null){
return null;
}
return new java.sql.Date(currentRow.getDate(index - 1).getTime());
}
public Date getDate(int index, Calendar calendar) throws SQLException
{
checkIndex(index);
// silently ignore the Calendar argument; its a hint we do not need
return getDate(index-1);
}
public Date getDate(String name) throws SQLException
{
checkName(name);
if(currentRow.getDate(name)==null){
return null;
}
return new java.sql.Date(currentRow.getDate(name).getTime());
}
public Date getDate(String name, Calendar calendar) throws SQLException
{
checkName(name);
// silently ignore the Calendar argument; its a hint we do not need
return getDate(name);
}
public double getDouble(int index) throws SQLException
{
checkIndex(index);
try{
if(currentRow.getColumnDefinitions().getType(index-1).getName().toString().equals("float")){
return currentRow.getFloat(index-1);
}
return currentRow.getDouble(index-1);
}catch(InvalidTypeException e){
throw new SQLNonTransientException(e);
}
}
@SuppressWarnings("cast")
public double getDouble(String name) throws SQLException
{
checkName(name);
try{
if(currentRow.getColumnDefinitions().getType(name).getName().toString().equals("float")){
return (double)currentRow.getFloat(name);
}
return currentRow.getDouble(name);
}catch(InvalidTypeException e){
throw new SQLNonTransientException(e);
}
}
public int getFetchDirection() throws SQLException
{
checkNotClosed();
return fetchDirection;
}
public int getFetchSize() throws SQLException
{
checkNotClosed();
return fetchSize;
}
public float getFloat(int index) throws SQLException
{
checkIndex(index);
return currentRow.getFloat(index-1);
}
public float getFloat(String name) throws SQLException
{
checkName(name);
return currentRow.getFloat(name);
}
public int getHoldability() throws SQLException
{
checkNotClosed();
return statement.getResultSetHoldability();
}
public int getInt(int index) throws SQLException
{
checkIndex(index);
return currentRow.getInt(index-1);
}
public int getInt(String name) throws SQLException
{
checkName(name);
return currentRow.getInt(name);
}
public byte[] getKey() throws SQLException
{
return curRowKey;
}
public List getList(int index) throws SQLException
{
checkIndex(index);
if(currentRow.getColumnDefinitions().getType(index-1).isCollection()){
try {
return Lists.newArrayList(currentRow.getList(index - 1,Class.forName(currentRow.getColumnDefinitions().getType(index-1).getTypeArguments().get(0).asJavaClass().getCanonicalName())));
} catch (ClassNotFoundException e) {
logger.warn("Error while executing getList()", e);
}
}
return currentRow.getList(index - 1,String.class);
}
public List getList(String name) throws SQLException
{
checkName(name);
if(currentRow.getColumnDefinitions().getType(name).isCollection()){
try {
return Lists.newArrayList(currentRow.getList(name,Class.forName(currentRow.getColumnDefinitions().getType(name).getTypeArguments().get(0).asJavaClass().getCanonicalName())));
} catch (ClassNotFoundException e) {
logger.warn("Error while executing getList()", e);
}
}
return currentRow.getList(name,String.class);
}
@SuppressWarnings("cast")
public long getLong(int index) throws SQLException
{
checkIndex(index);
try{
if(currentRow.getColumnDefinitions().getType(index-1).getName().toString().equals("int")){
return (long)currentRow.getInt(index-1);
}else if(currentRow.getColumnDefinitions().getType(index-1).getName().toString().equals("varint")){
return currentRow.getVarint(index-1).longValue();
}else{
return currentRow.getLong(index-1);
}
}catch(InvalidTypeException e){
throw new SQLNonTransientException(e);
}
}
@SuppressWarnings("cast")
public long getLong(String name) throws SQLException
{
checkName(name);
try{
if(currentRow.getColumnDefinitions().getType(name).getName().toString().equals("int")){
return (long)currentRow.getInt(name);
}else if(currentRow.getColumnDefinitions().getType(name).getName().toString().equals("varint")){
return currentRow.getVarint(name).longValue();
}else{
return currentRow.getLong(name);
}
}catch(InvalidTypeException e){
throw new SQLNonTransientException(e);
}
}
public Map getMap(int index) throws SQLException
{
checkIndex(index);
if(currentRow.isNull(index-1))
wasNull=true;
else
wasNull=false;
return currentRow.getMap(index-1,String.class,String.class); // TODO: replace with a real type check
}
public Map getMap(String name) throws SQLException
{
checkName(name);
return currentRow.getMap(name,String.class,String.class); // TODO: replace with a real type check
}
public ResultSetMetaData getMetaData() throws SQLException
{
checkNotClosed();
return meta;
}
@SuppressWarnings({ "cast", "boxing" })
public Object getObject(int index) throws SQLException
{
checkIndex(index);
List datatypes=null;
if(currentRow.getColumnDefinitions().getType(index-1).getName().toString().equals("udt")){
return (Object) currentRow.getUDTValue(index-1);
}
if(currentRow.getColumnDefinitions().getType(index-1).getName().toString().equals("tuple")){
return (Object) currentRow.getTupleValue(index-1);
}
if(currentRow.getColumnDefinitions().getType(index-1).isCollection()){
datatypes = currentRow.getColumnDefinitions().getType(index-1).getTypeArguments();
if(currentRow.getColumnDefinitions().getType(index-1).getName().toString().equals("set")){
if( datatypes.get(0).getName().toString().equals("udt")){
return (Object) Sets.newLinkedHashSet(currentRow.getSet(index-1,TypesMap.getTypeForComparator("udt").getType()));
}else if( datatypes.get(0).getName().toString().equals("tuple")){
if( datatypes.get(0).getName().toString().equals("udt")){
return (Object) Lists.newArrayList(currentRow.getList(index-1,TypesMap.getTypeForComparator("udt").getType()));
}else if( datatypes.get(0).getName().toString().equals("tuple")){
return (Object) Lists.newArrayList(currentRow.getList(index-1,TypesMap.getTypeForComparator("tuple").getType()));
}
else{
return (Object) Lists.newArrayList(currentRow.getList(index-1,TypesMap.getTypeForComparator(datatypes.get(0).toString()).getType()));
}
}
else{
return (Object) Sets.newLinkedHashSet(currentRow.getSet(index-1,TypesMap.getTypeForComparator(datatypes.get(0).toString()).getType()));
}
}
if(currentRow.getColumnDefinitions().getType(index-1).getName().toString().equals("list")){
return (Object) Lists.newArrayList(currentRow.getList(index-1,TypesMap.getTypeForComparator(datatypes.get(0).toString()).getType()));
}
if(currentRow.getColumnDefinitions().getType(index-1).getName().toString().equals("map")){
Class keyType = TypesMap.getTypeForComparator(datatypes.get(0).toString()).getType();
if( datatypes.get(0).getName().toString().equals("udt")){
keyType = TypesMap.getTypeForComparator("udt").getType();
}else if( datatypes.get(0).getName().toString().equals("tuple")){
keyType = TypesMap.getTypeForComparator("tuple").getType();
}
Class valueType = TypesMap.getTypeForComparator(datatypes.get(1).toString()).getType();
if( datatypes.get(1).getName().toString().equals("udt")){
valueType = TypesMap.getTypeForComparator("udt").getType();
}else if( datatypes.get(1).getName().toString().equals("tuple")){
valueType = TypesMap.getTypeForComparator("tuple").getType();
}
return (Object) Maps.newHashMap(currentRow.getMap(index-1,keyType,valueType));
}
}else{
String typeName = currentRow.getColumnDefinitions().getType(index-1).getName().toString();
if (typeName.equals("varchar")){
return (Object) currentRow.getString(index-1);
}else if (typeName.equals("ascii")){
return (Object) currentRow.getString(index-1);
}else if (typeName.equals("integer")){
return (Object) currentRow.getInt(index-1);
}else if (typeName.equals("bigint")){
return (Object) currentRow.getLong(index-1);
}else if (typeName.equals("blob")){
return (Object) currentRow.getBytes(index-1);
}else if (typeName.equals("boolean")){
return (Object) currentRow.getBool(index-1);
}else if (typeName.equals("counter")){
return (Object) currentRow.getLong(index-1);
}else if (typeName.equals("decimal")){
return (Object) currentRow.getDecimal(index-1);
}else if (typeName.equals("double")){
return (Object) currentRow.getDouble(index-1);
}else if (typeName.equals("float")){
return (Object) currentRow.getFloat(index-1);
}else if (typeName.equals("inet")){
return (Object) currentRow.getInet(index-1);
}else if (typeName.equals("int")){
return (Object) currentRow.getInt(index-1);
}else if (typeName.equals("text")){
return (Object) currentRow.getString(index-1);
}else if (typeName.equals("timestamp")){
return (Object) new Timestamp((currentRow.getDate(index-1)).getTime());
}else if (typeName.equals("uuid")){
return (Object) currentRow.getUUID(index-1);
}else if (typeName.equals("timeuuid")){
return (Object) currentRow.getUUID(index-1);
}else if (typeName.equals("varint")){
return (Object) currentRow.getInt(index-1);
}
}
return null;
}
@SuppressWarnings({ "cast", "boxing" })
public Object getObject(String name) throws SQLException
{
checkName(name);
List datatypes=null;
if(currentRow.getColumnDefinitions().getType(name).getName().toString().equals("udt")){
return (Object) currentRow.getUDTValue(name);
}
if(currentRow.getColumnDefinitions().getType(name).getName().toString().equals("tuple")){
return (Object) currentRow.getTupleValue(name);
}
if(currentRow.getColumnDefinitions().getType(name).isCollection()){
datatypes = currentRow.getColumnDefinitions().getType(name).getTypeArguments();
if(currentRow.getColumnDefinitions().getType(name).getName().toString().equals("set")){
if( datatypes.get(0).getName().toString().equals("udt")){
return (Object) Sets.newLinkedHashSet(currentRow.getSet(name,TypesMap.getTypeForComparator("udt").getType()));
}else if( datatypes.get(0).getName().toString().equals("tuple")){
return (Object) Sets.newLinkedHashSet(currentRow.getSet(name,TypesMap.getTypeForComparator("tuple").getType()));
}
else{
return (Object) Sets.newLinkedHashSet(currentRow.getSet(name,TypesMap.getTypeForComparator(datatypes.get(0).toString()).getType()));
}
}
if(currentRow.getColumnDefinitions().getType(name).getName().toString().equals("list")){
if( datatypes.get(0).getName().toString().equals("udt")){
return (Object) Lists.newArrayList(currentRow.getList(name,TypesMap.getTypeForComparator("udt").getType()));
}else if( datatypes.get(0).getName().toString().equals("tuple")){
return (Object) Lists.newArrayList(currentRow.getList(name,TypesMap.getTypeForComparator("tuple").getType()));
}
else{
return (Object) Lists.newArrayList(currentRow.getList(name,TypesMap.getTypeForComparator(datatypes.get(0).toString()).getType()));
}
}
if(currentRow.getColumnDefinitions().getType(name).getName().toString().equals("map")){
Class keyType = TypesMap.getTypeForComparator(datatypes.get(0).toString()).getType();
if( datatypes.get(0).getName().toString().equals("udt")){
keyType = TypesMap.getTypeForComparator("udt").getType();
}else if( datatypes.get(0).getName().toString().equals("tuple")){
keyType = TypesMap.getTypeForComparator("tuple").getType();
}
Class valueType = TypesMap.getTypeForComparator(datatypes.get(1).toString()).getType();
if( datatypes.get(1).getName().toString().equals("udt")){
valueType = TypesMap.getTypeForComparator("udt").getType();
}else if( datatypes.get(1).getName().toString().equals("tuple")){
valueType = TypesMap.getTypeForComparator("tuple").getType();
}
return (Object) Maps.newHashMap(currentRow.getMap(name,keyType,valueType));
}
}else{
String typeName = currentRow.getColumnDefinitions().getType(name).getName().toString();
if (typeName.equals("varchar")){
return (Object) currentRow.getString(name);
}else if (typeName.equals("ascii")){
return (Object) currentRow.getString(name);
}else if (typeName.equals("integer")){
return (Object) currentRow.getInt(name);
}else if (typeName.equals("bigint")){
return (Object) currentRow.getLong(name);
}else if (typeName.equals("blob")){
return (Object) currentRow.getBytes(name);
}else if (typeName.equals("boolean")){
return (Object) currentRow.getBool(name);
}else if (typeName.equals("counter")){
return (Object) currentRow.getLong(name);
}else if (typeName.equals("decimal")){
return (Object) currentRow.getDecimal(name);
}else if (typeName.equals("double")){
return (Object) currentRow.getDouble(name);
}else if (typeName.equals("float")){
return (Object) currentRow.getFloat(name);
}else if (typeName.equals("inet")){
return (Object) currentRow.getInet(name);
}else if (typeName.equals("int")){
return (Object) currentRow.getInt(name);
}else if (typeName.equals("text")){
return (Object) currentRow.getString(name);
}else if (typeName.equals("timestamp")){
return (Object) new Timestamp((currentRow.getDate(name)).getTime());
}else if (typeName.equals("uuid")){
return (Object) currentRow.getUUID(name);
}else if (typeName.equals("timeuuid")){
return (Object) currentRow.getUUID(name);
}else if (typeName.equals("varint")){
return (Object) currentRow.getInt(name);
}
}
return null;
}
public int getRow() throws SQLException
{
checkNotClosed();
return rowNumber;
}
public short getShort(int index) throws SQLException
{
checkIndex(index);
return (short) currentRow.getInt(index - 1);
}
public Set getSet(int index) throws SQLException
{
checkIndex(index);
try {
return currentRow.getSet(index-1,Class.forName(currentRow.getColumnDefinitions().getType(index-1).getTypeArguments().get(0).asJavaClass().getCanonicalName()));
} catch (ClassNotFoundException e) {
logger.warn("Error while performing getSet()", e);
}
return null;
}
public Set getSet(String name) throws SQLException
{
checkName(name);
try {
return currentRow.getSet(name,Class.forName(currentRow.getColumnDefinitions().getType(name).getTypeArguments().get(0).asJavaClass().getCanonicalName()));
} catch (ClassNotFoundException e) {
logger.warn("Error while performing getSet()", e);
}
return null;
}
public short getShort(String name) throws SQLException
{
checkName(name);
return (short) currentRow.getInt(name);
}
public Statement getStatement() throws SQLException
{
checkNotClosed();
return statement;
}
public String getString(int index) throws SQLException
{
checkIndex(index);
try{
if(currentRow.getColumnDefinitions().getType(index-1).isCollection()){
return getObject(index).toString();
}
return currentRow.getString(index-1);
}catch(Exception e){
return getObject(index).toString();
}
}
public String getString(String name) throws SQLException
{
checkName(name);
try{
if(currentRow.getColumnDefinitions().getType(name).isCollection()){
return getObject(name).toString();
}
return currentRow.getString(name);
}catch(Exception e){
return getObject(name).toString();
}
}
public Time getTime(int index) throws SQLException
{
checkIndex(index);
java.util.Date date = currentRow.getDate(index-1);
if(date==null)
return null;
return new Time(currentRow.getDate(index - 1).getTime());
}
public Time getTime(int index, Calendar calendar) throws SQLException
{
checkIndex(index);
// silently ignore the Calendar argument; its a hint we do not need
java.util.Date date = currentRow.getDate(index-1);
if(date==null)
return null;
return getTime(index-1);
}
public Time getTime(String name) throws SQLException
{
checkName(name);
java.util.Date date = currentRow.getDate(name);
if(date==null)
return null;
return new Time(currentRow.getDate(name).getTime());
}
public Time getTime(String name, Calendar calendar) throws SQLException
{
checkName(name);
// silently ignore the Calendar argument; its a hint we do not need
return getTime(name);
}
public Timestamp getTimestamp(int index) throws SQLException
{
checkIndex(index);
java.util.Date date = currentRow.getDate(index-1);
if(date==null)
return null;
return new Timestamp(currentRow.getDate(index - 1).getTime());
}
public Timestamp getTimestamp(int index, Calendar calendar) throws SQLException
{
checkIndex(index);
// silently ignore the Calendar argument; its a hint we do not need
java.util.Date date = currentRow.getDate(index-1);
if(date==null)
return null;
return getTimestamp(index-1);
}
public Timestamp getTimestamp(String name) throws SQLException
{
checkName(name);
java.util.Date date = currentRow.getDate(name);
if(date==null)
return null;
return new Timestamp(currentRow.getDate(name).getTime());
}
public Timestamp getTimestamp(String name, Calendar calendar) throws SQLException
{
checkName(name);
// silently ignore the Calendar argument; its a hint we do not need
return getTimestamp(name);
}
public int getType() throws SQLException
{
checkNotClosed();
return resultSetType;
}
// URL (awaiting some clarifications as to how it is stored in C* ... just a validated Sting in URL format?
public URL getURL(int arg0) throws SQLException
{
throw new SQLFeatureNotSupportedException(NOT_SUPPORTED);
}
public URL getURL(String arg0) throws SQLException
{
throw new SQLFeatureNotSupportedException(NOT_SUPPORTED);
}
// These Methods are planned to be implemented soon; but not right now...
// Each set of methods has a more detailed set of issues that should be considered fully...
public SQLWarning getWarnings() throws SQLException
{
checkNotClosed();
// the rationale is there are no warnings to return in this implementation...
return null;
}
public boolean isAfterLast() throws SQLException
{
checkNotClosed();
return rowNumber == Integer.MAX_VALUE;
}
public boolean isBeforeFirst() throws SQLException
{
checkNotClosed();
return rowNumber == 0;
}
public boolean isClosed() throws SQLException
{
if(this.statement==null){
return true;
}
return this.statement.isClosed();
}
public boolean isFirst() throws SQLException
{
checkNotClosed();
return rowNumber == 1;
}
public boolean isLast() throws SQLException
{
checkNotClosed();
return !rowsIterator.hasNext();
}
public boolean isWrapperFor(Class iface) throws SQLException
{
return CassandraResultSetExtras.class.isAssignableFrom(iface);
}
// Navigation between rows within the returned set of rows
// Need to use a list iterator so next() needs completely re-thought
public boolean last() throws SQLException
{
throw new SQLFeatureNotSupportedException(NOT_SUPPORTED);
}
public synchronized boolean next() throws SQLException
{
if (hasMoreRows())
{
// populateColumns is called upon init to set up the metadata fields; so skip first call
if (rowNumber != 0) populateColumns();
//else rowsIterator.next();
// populateColumns();
rowNumber++;
return true;
}
rowNumber = Integer.MAX_VALUE;
return false;
}
public boolean previous() throws SQLException
{
throw new SQLFeatureNotSupportedException(NOT_SUPPORTED);
}
public boolean relative(int arg0) throws SQLException
{
throw new SQLFeatureNotSupportedException(NOT_SUPPORTED);
}
@SuppressWarnings("boxing")
public void setFetchDirection(int direction) throws SQLException
{
checkNotClosed();
if (direction == FETCH_FORWARD || direction == FETCH_REVERSE || direction == FETCH_UNKNOWN)
{
if ((getType() == TYPE_FORWARD_ONLY) && (direction != FETCH_FORWARD)) throw new SQLSyntaxErrorException("attempt to set an illegal direction : " + direction);
fetchDirection = direction;
}
throw new SQLSyntaxErrorException(String.format(BAD_FETCH_DIR, direction));
}
@SuppressWarnings("boxing")
public void setFetchSize(int size) throws SQLException
{
checkNotClosed();
if (size < 0) throw new SQLException(String.format(BAD_FETCH_SIZE, size));
fetchSize = size;
}
@SuppressWarnings("unchecked")
public T unwrap(Class iface) throws SQLException
{
if (iface.equals(CassandraResultSetExtras.class)) return (T) this;
throw new SQLFeatureNotSupportedException(String.format(NO_INTERFACE, iface.getSimpleName()));
}
public boolean wasNull() throws SQLException
{
return wasNull;
}
/**
* RSMD implementation. The metadata returned refers to the column
* values, not the column names.
*/
class CResultSetMetaData implements ResultSetMetaData
{
/**
* return the Cassandra Cluster Name as the Catalog
*/
public String getCatalogName(int column) throws SQLException
{
//checkIndex(column);
return statement.connection.getCatalog();
}
public String getColumnClassName(int column) throws SQLException
{/*
checkIndex(column);
return values.get(column - 1).getValueType().getType().getName();*/
if(currentRow!=null){
return currentRow.getColumnDefinitions().getType(column-1).asJavaClass().getCanonicalName();
}
return driverResultSet.getColumnDefinitions().asList().get(column-1).getType().asJavaClass().getCanonicalName();
}
public int getColumnCount() throws SQLException
{
if(currentRow!=null){
return currentRow.getColumnDefinitions().size();
}
return driverResultSet.getColumnDefinitions().size();
}
@SuppressWarnings("rawtypes")
public int getColumnDisplaySize(int column) throws SQLException
{
//checkIndex(column);
Definition col = null;
if(currentRow!=null){
col = currentRow.getColumnDefinitions().asList().get(column-1);
}else{
col = driverResultSet.getColumnDefinitions().asList().get(column-1);
}
try{
int length = -1;
AbstractJdbcType jtype = TypesMap.getTypeForComparator(col.getType().toString());
if (jtype instanceof JdbcBytes) length = Integer.MAX_VALUE / 2;
if (jtype instanceof JdbcAscii || jtype instanceof JdbcUTF8) length = Integer.MAX_VALUE;
if (jtype instanceof JdbcUUID) length = 36;
if (jtype instanceof JdbcInt32) length = 4;
if (jtype instanceof JdbcLong) length = 8;
// String stringValue = getObject(column).toString();
//return (stringValue == null ? -1 : stringValue.length());
return length;
}catch(Exception e){
return -1;
}
//return -1;
}
public String getColumnLabel(int column) throws SQLException
{
//checkIndex(column);
return getColumnName(column);
}
public String getColumnName(int column) throws SQLException
{
//checkIndex(column);
if(currentRow!=null){
return currentRow.getColumnDefinitions().getName(column-1);
}
return driverResultSet.getColumnDefinitions().asList().get(column-1).getName();
}
public int getColumnType(int column) throws SQLException
{
DataType type = null;
if(currentRow!=null){
type = currentRow.getColumnDefinitions().getType(column-1);
}else{
type = driverResultSet.getColumnDefinitions().asList().get(column-1).getType();
}
return TypesMap.getTypeForComparator(type.toString()).getJdbcType();
}
/**
* Spec says "database specific type name"; for Cassandra this means the AbstractType.
*/
public String getColumnTypeName(int column) throws SQLException
{
//checkIndex(column);
DataType type = null;
if(currentRow!=null){
type = currentRow.getColumnDefinitions().getType(column-1);
}else{
type = driverResultSet.getColumnDefinitions().getType(column-1);
}
return type.toString();
}
public int getPrecision(int column) throws SQLException
{
// checkIndex(column);
// TypedColumn col = values.get(column - 1);
// return col.getValueType().getPrecision(col.getValue());
return 0;
}
public int getScale(int column) throws SQLException
{
// checkIndex(column);
// TypedColumn tc = values.get(column - 1);
// return tc.getValueType().getScale(tc.getValue());
return 0;
}
/**
* return the DEFAULT current Keyspace as the Schema Name
*/
public String getSchemaName(int column) throws SQLException
{
//checkIndex(column);
return statement.connection.getSchema();
}
/* public String getTableName(int column) throws SQLException
{
throw new SQLFeatureNotSupportedException();
}*/
public boolean isAutoIncrement(int column) throws SQLException
{
return true;
// checkIndex(column);
// return values.get(column - 1).getValueType() instanceof JdbcCounterColumn; // todo: check Value is correct.
}
public boolean isCaseSensitive(int column) throws SQLException
{
// checkIndex(column);
// TypedColumn tc = values.get(column - 1);
// return tc.getValueType().isCaseSensitive();
return true;
}
public boolean isCurrency(int column) throws SQLException
{
// checkIndex(column);
// TypedColumn tc = values.get(column - 1);
// return tc.getValueType().isCurrency();
return false;
}
public boolean isDefinitelyWritable(int column) throws SQLException
{
//checkIndex(column);
return isWritable(column);
}
/**
* absence is the equivalent of null in Cassandra
*/
public int isNullable(int column) throws SQLException
{
//checkIndex(column);
return ResultSetMetaData.columnNullable;
}
public boolean isReadOnly(int column) throws SQLException
{
//checkIndex(column);
return column == 0;
}
public boolean isSearchable(int column) throws SQLException
{
//checkIndex(column);
return false;
}
public boolean isSigned(int column) throws SQLException
{
// checkIndex(column);
// TypedColumn tc = values.get(column - 1);
// return tc.getValueType().isSigned();
return false;
}
public boolean isWrapperFor(Class iface) throws SQLException
{
return false;
}
public boolean isWritable(int column) throws SQLException
{
//checkIndex(column);
return column > 0;
}
public T unwrap(Class iface) throws SQLException
{
throw new SQLFeatureNotSupportedException(String.format(NO_INTERFACE, iface.getSimpleName()));
}
@Override
public String getTableName(int column) throws SQLException {
String tableName="";
if(currentRow!=null){
tableName = currentRow.getColumnDefinitions().getTable(column-1);
}else{
tableName = driverResultSet.getColumnDefinitions().getTable(column-1);
}
return tableName;
}
}
@Override
public RowId getRowId(int columnIndex) throws SQLException {
// TODO Auto-generated method stub
return null;
}
@Override
public RowId getRowId(String columnLabel) throws SQLException {
// TODO Auto-generated method stub
return null;
}
@Override
public InputStream getBinaryStream(int columnIndex) throws SQLException {
checkIndex(columnIndex);
byte[] bytes = new byte[currentRow.getBytes(columnIndex-1).remaining()];
currentRow.getBytes(columnIndex-1).get(bytes, 0, bytes.length);
return new ByteArrayInputStream(bytes);
}
@Override
public InputStream getBinaryStream(String columnLabel) throws SQLException {
// TODO Auto-generated method stub
checkName(columnLabel);
byte[] bytes = new byte[currentRow.getBytes(columnLabel).remaining()];
currentRow.getBytes(columnLabel).get(bytes, 0, bytes.length);
return new ByteArrayInputStream(bytes);
}
private boolean hasNextIterator(){
return currentIteratorIndex
