org.apache.drill.test

Class QueryBatchIterator

java.lang.Object

org.apache.drill.test.QueryBatchIterator

All Implemented Interfaces:

AutoCloseable, PullResultSetReaderImpl.UpstreamSource, PushResultSetReaderImpl.UpstreamSource

public class QueryBatchIterator
extends Object
implements PullResultSetReaderImpl.UpstreamSource, AutoCloseable

Iterator over batches returned from a query. Uses a listener to obtain the serialized batches, then decodes them into value vectors held in a vector container - the same structure as used in the query executor. This format allows us to use the "row set" classes on the output of the query.

Constructor Summary

Constructors

Constructor and Description
QueryBatchIterator(BufferAllocator allocator, BufferingQueryEventListener listener)

Method Summary

Modifier and Type	Method and Description
VectorContainer	batch()
int	batchCount()
void	close()
UserBitShared.QueryResult.QueryState	finalState()
boolean	next()
UserBitShared.QueryId	queryId()
String	queryIdString()
void	release()
void	retainData()
int	rowCount()
int	schemaVersion()
SelectionVector2	sv2()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

QueryBatchIterator

public QueryBatchIterator(BufferAllocator allocator,
                          BufferingQueryEventListener listener)

Method Detail

queryId

public UserBitShared.QueryId queryId()

queryIdString

public String queryIdString()

finalState

public UserBitShared.QueryResult.QueryState finalState()

batchCount

public int batchCount()

rowCount

public int rowCount()

next

public boolean next()

Specified by:

next in interface PullResultSetReaderImpl.UpstreamSource

schemaVersion

public int schemaVersion()

Specified by:

schemaVersion in interface PushResultSetReaderImpl.UpstreamSource

batch

public VectorContainer batch()

Specified by:

batch in interface PushResultSetReaderImpl.UpstreamSource

sv2

public SelectionVector2 sv2()

Specified by:

sv2 in interface PushResultSetReaderImpl.UpstreamSource

release

public void release()

Specified by:

release in interface PullResultSetReaderImpl.UpstreamSource

retainData

public void retainData()

close

public void close()

Specified by:

close in interface AutoCloseable