com.mapr.ojai.store.impl

Class SingleRowListener

java.lang.Object

com.mapr.ojai.store.impl.SingleRowListener

All Implemented Interfaces:

org.apache.drill.exec.rpc.user.UserResultsListener

public abstract class SingleRowListener
extends java.lang.Object
implements org.apache.drill.exec.rpc.user.UserResultsListener

Convenience class to help with implementing a specialized Drill result listener for cases where the result is known to only be a single row. TODO https://maprdrill.atlassian.net/browse/MAPRDB-6 This was copied from the Drill source because that instance isn't public. If it gets published, delete this and use the public one. https://issues.apache.org/jira/browse/DRILL-5041

Constructor Summary

Constructors

Constructor and Description
SingleRowListener()

Method Summary

Modifier and Type	Method and Description
void	cleanup() Clean up any resources used.
void	dataArrived(org.apache.drill.exec.rpc.user.QueryDataBatch result, org.apache.drill.exec.rpc.ConnectionThrottle throttle) A QueryData message was received
java.util.List<org.apache.drill.exec.proto.UserBitShared.DrillPBError>	getErrorList() Get an immutable copy of the list of all errors received so far.
org.apache.drill.exec.proto.UserBitShared.QueryResult.QueryState	getQueryState() Get the last known QueryState.
void	queryCompleted(org.apache.drill.exec.proto.UserBitShared.QueryResult.QueryState state) The query has completed (successsful completion or cancellation).
void	queryIdArrived(org.apache.drill.exec.proto.UserBitShared.QueryId queryId) QueryId is available.
protected abstract void	rowArrived(org.apache.drill.exec.rpc.user.QueryDataBatch queryDataBatch) A record has arrived and is ready for access.
void	submissionFailed(org.apache.drill.common.exceptions.UserException ex) The query has failed.
void	waitForCompletion() Wait for the completion of this query; receiving a record or an error will both cause the query to be considered complete

Methods inherited from class java.lang.Object

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

Constructor Detail

SingleRowListener

public SingleRowListener()

Method Detail

queryIdArrived

public void queryIdArrived(org.apache.drill.exec.proto.UserBitShared.QueryId queryId)

Description copied from interface: org.apache.drill.exec.rpc.user.UserResultsListener

QueryId is available. Called when a query is successfully submitted to the server.

Specified by:

queryIdArrived in interface org.apache.drill.exec.rpc.user.UserResultsListener

Parameters:

queryId - sent by the server along Acks.OK

submissionFailed

public void submissionFailed(org.apache.drill.common.exceptions.UserException ex)

Description copied from interface: org.apache.drill.exec.rpc.user.UserResultsListener

The query has failed. Most likely called when the server returns a FAILED query state. Can also be called if dataArrived() throws an exception

Specified by:

submissionFailed in interface org.apache.drill.exec.rpc.user.UserResultsListener

Parameters:

ex - exception describing the cause of the failure

queryCompleted

public void queryCompleted(org.apache.drill.exec.proto.UserBitShared.QueryResult.QueryState state)

Description copied from interface: org.apache.drill.exec.rpc.user.UserResultsListener

The query has completed (successsful completion or cancellation). The listener will not receive any other data or result message. Called when the server returns a terminal-non failing- state (COMPLETED or CANCELLED)

Specified by:

queryCompleted in interface org.apache.drill.exec.rpc.user.UserResultsListener

dataArrived

public void dataArrived(org.apache.drill.exec.rpc.user.QueryDataBatch result,
                        org.apache.drill.exec.rpc.ConnectionThrottle throttle)

Description copied from interface: org.apache.drill.exec.rpc.user.UserResultsListener

A QueryData message was received

Specified by:

dataArrived in interface org.apache.drill.exec.rpc.user.UserResultsListener

Parameters:

result - data batch received

throttle - connection throttle

getQueryState

public org.apache.drill.exec.proto.UserBitShared.QueryResult.QueryState getQueryState()

Get the last known QueryState.

Returns:

the query state; may be null if no query state has been received

getErrorList

public java.util.List<org.apache.drill.exec.proto.UserBitShared.DrillPBError> getErrorList()

Get an immutable copy of the list of all errors received so far.

Returns:

list of errors received

rowArrived

protected abstract void rowArrived(org.apache.drill.exec.rpc.user.QueryDataBatch queryDataBatch)

A record has arrived and is ready for access.

Derived classes provide whatever implementation they require here to access the record's data.

Parameters:

queryDataBatch - result batch holding the row

waitForCompletion

public void waitForCompletion()
                       throws java.lang.Exception

Wait for the completion of this query; receiving a record or an error will both cause the query to be considered complete

Throws:

java.lang.Exception - if there was any kind of problem

cleanup

public void cleanup()

Clean up any resources used.

Derived classes may use this to free things like allocators or files that were used to record data received in resultArrived().