org.apache.cassandra.cql3.UntypedResultSet Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of cassandra-all Show documentation
Show all versions of cassandra-all Show documentation
The Apache Cassandra Project develops a highly scalable second-generation distributed database, bringing together Dynamo's fully distributed design and Bigtable's ColumnFamily-based data model.
/*
* 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 org.apache.cassandra.cql3;
import java.net.InetAddress;
import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.UUID;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;
import com.google.common.annotations.VisibleForTesting;
import com.datastax.driver.core.CodecUtils;
import org.apache.cassandra.cql3.functions.types.LocalDate;
import org.apache.cassandra.cql3.statements.SelectStatement;
import org.apache.cassandra.db.Clustering;
import org.apache.cassandra.db.ConsistencyLevel;
import org.apache.cassandra.db.DecoratedKey;
import org.apache.cassandra.db.ReadExecutionController;
import org.apache.cassandra.db.marshal.*;
import org.apache.cassandra.db.partitions.PartitionIterator;
import org.apache.cassandra.db.rows.Cell;
import org.apache.cassandra.db.rows.ComplexColumnData;
import org.apache.cassandra.schema.ColumnMetadata;
import org.apache.cassandra.schema.TableMetadata;
import org.apache.cassandra.service.ClientState;
import org.apache.cassandra.service.pager.QueryPager;
import org.apache.cassandra.utils.AbstractIterator;
import org.apache.cassandra.utils.FBUtilities;
import org.apache.cassandra.utils.TimeUUID;
import static org.apache.cassandra.utils.Clock.Global.nanoTime;
/** a utility for doing internal cql-based queries */
public abstract class UntypedResultSet implements Iterable
{
public static UntypedResultSet create(ResultSet rs)
{
return new FromResultSet(rs);
}
public static UntypedResultSet create(List> results)
{
return new FromResultList(results);
}
public static UntypedResultSet create(SelectStatement select, QueryPager pager, int pageSize)
{
return new FromPager(select, pager, pageSize);
}
/**
* This method is intended for testing purposes, since it executes query on cluster
* and not on the local node only.
*/
@VisibleForTesting
public static UntypedResultSet create(SelectStatement select,
ConsistencyLevel cl,
ClientState clientState,
QueryPager pager,
int pageSize)
{
return new FromDistributedPager(select, cl, clientState, pager, pageSize);
}
public boolean isEmpty()
{
return size() == 0;
}
public Stream stream()
{
return StreamSupport.stream(spliterator(), false);
}
public abstract int size();
public abstract Row one();
// No implemented by all subclasses, but we use it when we know it's there (for tests)
public abstract List metadata();
private static class FromResultSet extends UntypedResultSet
{
private final ResultSet cqlRows;
private FromResultSet(ResultSet cqlRows)
{
this.cqlRows = cqlRows;
}
public int size()
{
return cqlRows.size();
}
public Row one()
{
if (cqlRows.size() != 1)
throw new IllegalStateException("One row required, " + cqlRows.size() + " found");
return new Row(cqlRows.metadata.requestNames(), cqlRows.rows.get(0));
}
public Iterator iterator()
{
return new AbstractIterator()
{
final Iterator> iter = cqlRows.rows.iterator();
protected Row computeNext()
{
if (!iter.hasNext())
return endOfData();
return new Row(cqlRows.metadata.requestNames(), iter.next());
}
};
}
public List metadata()
{
return cqlRows.metadata.requestNames();
}
}
private static class FromResultList extends UntypedResultSet
{
private final List> cqlRows;
private FromResultList(List> cqlRows)
{
this.cqlRows = cqlRows;
}
public int size()
{
return cqlRows.size();
}
public Row one()
{
if (cqlRows.size() != 1)
throw new IllegalStateException("One row required, " + cqlRows.size() + " found");
return new Row(cqlRows.get(0));
}
public Iterator iterator()
{
return new AbstractIterator()
{
final Iterator> iter = cqlRows.iterator();
protected Row computeNext()
{
if (!iter.hasNext())
return endOfData();
return new Row(iter.next());
}
};
}
public List metadata()
{
throw new UnsupportedOperationException();
}
}
private static class FromPager extends UntypedResultSet
{
private final SelectStatement select;
private final QueryPager pager;
private final int pageSize;
private final List metadata;
private FromPager(SelectStatement select, QueryPager pager, int pageSize)
{
this.select = select;
this.pager = pager;
this.pageSize = pageSize;
this.metadata = select.getResultMetadata().requestNames();
}
public int size()
{
throw new UnsupportedOperationException();
}
public Row one()
{
throw new UnsupportedOperationException();
}
public Iterator iterator()
{
return new AbstractIterator()
{
private Iterator> currentPage;
protected Row computeNext()
{
long nowInSec = FBUtilities.nowInSeconds();
while (currentPage == null || !currentPage.hasNext())
{
if (pager.isExhausted())
return endOfData();
try (ReadExecutionController executionController = pager.executionController();
PartitionIterator iter = pager.fetchPageInternal(pageSize, executionController))
{
currentPage = select.process(iter, nowInSec, true).rows.iterator();
}
}
return new Row(metadata, currentPage.next());
}
};
}
public List metadata()
{
return metadata;
}
}
/**
* Pager that calls `execute` rather than `executeInternal`
*/
private static class FromDistributedPager extends UntypedResultSet
{
private final SelectStatement select;
private final ConsistencyLevel cl;
private final ClientState clientState;
private final QueryPager pager;
private final int pageSize;
private final List metadata;
private FromDistributedPager(SelectStatement select,
ConsistencyLevel cl,
ClientState clientState,
QueryPager pager, int pageSize)
{
this.select = select;
this.cl = cl;
this.clientState = clientState;
this.pager = pager;
this.pageSize = pageSize;
this.metadata = select.getResultMetadata().requestNames();
}
public int size()
{
throw new UnsupportedOperationException();
}
public Row one()
{
throw new UnsupportedOperationException();
}
public Iterator iterator()
{
return new AbstractIterator()
{
private Iterator> currentPage;
protected Row computeNext()
{
long nowInSec = FBUtilities.nowInSeconds();
while (currentPage == null || !currentPage.hasNext())
{
if (pager.isExhausted())
return endOfData();
try (PartitionIterator iter = pager.fetchPage(pageSize, cl, clientState, nanoTime()))
{
currentPage = select.process(iter, nowInSec, true).rows.iterator();
}
}
return new Row(metadata, currentPage.next());
}
};
}
public List metadata()
{
return metadata;
}
}
public static class Row
{
private final Map data = new HashMap<>();
private final List columns = new ArrayList<>();
public Row(Map data)
{
this.data.putAll(data);
}
public Row(List names, List columns)
{
this.columns.addAll(names);
for (int i = 0; i < names.size(); i++)
data.put(names.get(i).name.toString(), columns.get(i));
}
public static Row fromInternalRow(TableMetadata metadata, DecoratedKey key, org.apache.cassandra.db.rows.Row row)
{
Map data = new HashMap<>();
ByteBuffer[] keyComponents = SelectStatement.getComponents(metadata, key);
for (ColumnMetadata def : metadata.partitionKeyColumns())
data.put(def.name.toString(), keyComponents[def.position()]);
Clustering clustering = row.clustering();
for (ColumnMetadata def : metadata.clusteringColumns())
data.put(def.name.toString(), clustering.bufferAt(def.position()));
for (ColumnMetadata def : metadata.regularAndStaticColumns())
{
if (def.isSimple())
{
Cell cell = row.getCell(def);
if (cell != null)
data.put(def.name.toString(), cell.buffer());
}
else
{
ComplexColumnData complexData = row.getComplexColumnData(def);
if (complexData != null)
data.put(def.name.toString(), ((CollectionType) def.type).serializeForNativeProtocol(complexData.iterator()));
}
}
return new Row(data);
}
public boolean has(String column)
{
// Note that containsKey won't work because we may have null values
return data.get(column) != null;
}
public ByteBuffer getBlob(String column)
{
return data.get(column);
}
public String getString(String column)
{
return UTF8Type.instance.compose(data.get(column));
}
public boolean getBoolean(String column)
{
return BooleanType.instance.compose(data.get(column));
}
public byte getByte(String column)
{
return ByteType.instance.compose(data.get(column));
}
public short getShort(String column)
{
return ShortType.instance.compose(data.get(column));
}
public int getInt(String column)
{
return Int32Type.instance.compose(data.get(column));
}
public int getInt(String column, int ifNull)
{
ByteBuffer bytes = data.get(column);
return bytes == null ? ifNull : Int32Type.instance.compose(bytes);
}
public double getDouble(String column)
{
return DoubleType.instance.compose(data.get(column));
}
public ByteBuffer getBytes(String column)
{
return data.get(column);
}
public byte[] getByteArray(String column)
{
ByteBuffer buf = data.get(column);
byte[] arr = new byte[buf.remaining()];
for (int i = 0; i < arr.length; i++)
arr[i] = buf.get(buf.position() + i);
return arr;
}
public InetAddress getInetAddress(String column)
{
return InetAddressType.instance.compose(data.get(column));
}
public UUID getUUID(String column)
{
return UUIDType.instance.compose(data.get(column));
}
public UUID getUUID(String column, UUID ifNull)
{
ByteBuffer bytes = data.get(column);
return bytes == null ? ifNull : UUIDType.instance.compose(bytes);
}
public TimeUUID getTimeUUID(String column)
{
return TimeUUID.deserialize(data.get(column));
}
public Date getTimestamp(String column)
{
return TimestampType.instance.compose(data.get(column));
}
public LocalDate getDate(String column) { return LocalDate.fromDaysSinceEpoch(CodecUtils.fromUnsignedToSignedInt(data.get(column).getInt()));}
public long getLong(String column)
{
return LongType.instance.compose(data.get(column));
}
public Set getSet(String column, AbstractType type)
{
ByteBuffer raw = data.get(column);
return raw == null ? null : SetType.getInstance(type, true).compose(raw);
}
public List getList(String column, AbstractType type)
{
ByteBuffer raw = data.get(column);
return raw == null ? null : ListType.getInstance(type, true).compose(raw);
}
public Map getMap(String column, AbstractType keyType, AbstractType valueType)
{
ByteBuffer raw = data.get(column);
return raw == null ? null : MapType.getInstance(keyType, valueType, true).compose(raw);
}
public Set getFrozenSet(String column, AbstractType type)
{
ByteBuffer raw = data.get(column);
return raw == null ? null : SetType.getInstance(type, false).compose(raw);
}
public List getFrozenList(String column, AbstractType type)
{
ByteBuffer raw = data.get(column);
return raw == null ? null : ListType.getInstance(type, false).compose(raw);
}
public Map getFrozenMap(String column, AbstractType keyType, AbstractType valueType)
{
ByteBuffer raw = data.get(column);
return raw == null ? null : MapType.getInstance(keyType, valueType, false).compose(raw);
}
public Map getFrozenTextMap(String column)
{
return getFrozenMap(column, UTF8Type.instance, UTF8Type.instance);
}
public List getVector(String column, AbstractType elementType, int dimension)
{
ByteBuffer raw = data.get(column);
return raw == null ? null : VectorType.getInstance(elementType, dimension).compose(raw);
}
public List getColumns()
{
return columns;
}
@Override
public String toString()
{
return data.toString();
}
}
}