org.apache.drill.exec.work.batch

Class BaseRawBatchBuffer<T>

java.lang.Object

org.apache.drill.exec.work.batch.BaseRawBatchBuffer<T>

All Implemented Interfaces:

AutoCloseable, RawFragmentBatchProvider, RawBatchBuffer

Direct Known Subclasses:

SpoolingRawBatchBuffer, UnlimitedRawBatchBuffer

public abstract class BaseRawBatchBuffer<T>
extends Object
implements RawBatchBuffer

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
protected static interface	BaseRawBatchBuffer.BufferQueue<T>

Field Summary

Fields

Modifier and Type	Field and Description
protected BaseRawBatchBuffer.BufferQueue<T>	bufferQueue
protected int	bufferSizePerSocket
protected FragmentContext	context
protected AtomicBoolean	outOfMemory

Constructor Summary

Constructors

Constructor and Description
BaseRawBatchBuffer(FragmentContext context, int fragmentCount)

Method Summary

Methods

Modifier and Type	Method and Description
void	close()
void	enqueue(RawFragmentBatch batch) Add the next new raw fragment batch to the buffer.
protected abstract void	enqueueInner(RawFragmentBatch batch) implementation specific method to enqueue batch
protected int	getFragmentCount() Return the fragment count from construction time.
RawFragmentBatch	getNext()
protected boolean	isTerminated()
void	kill(FragmentContext context)
protected abstract void	upkeep(RawFragmentBatch batch) Handle miscellaneous tasks after batch retrieval

Methods inherited from class java.lang.Object

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

Field Detail

bufferQueue

protected BaseRawBatchBuffer.BufferQueue<T> bufferQueue

bufferSizePerSocket

protected final int bufferSizePerSocket

outOfMemory

protected final AtomicBoolean outOfMemory

context

protected final FragmentContext context

Constructor Detail

BaseRawBatchBuffer

public BaseRawBatchBuffer(FragmentContext context,
                  int fragmentCount)

Method Detail

getFragmentCount

protected int getFragmentCount()

Return the fragment count from construction time.

Returns:

the fragment count

enqueue

public void enqueue(RawFragmentBatch batch)
             throws IOException

Description copied from interface: RawBatchBuffer

Add the next new raw fragment batch to the buffer.

Specified by:

enqueue in interface RawBatchBuffer

Parameters:

batch - Batch to enqueue

Throws:

IOException

enqueueInner

protected abstract void enqueueInner(RawFragmentBatch batch)
                              throws IOException

implementation specific method to enqueue batch

Parameters:

batch -

Throws:

IOException

close

public void close()

Specified by:

close in interface AutoCloseable

Specified by:

close in interface RawFragmentBatchProvider

kill

public void kill(FragmentContext context)

Specified by:

kill in interface RawFragmentBatchProvider

getNext

public RawFragmentBatch getNext()
                         throws IOException

Specified by:

getNext in interface RawFragmentBatchProvider

Throws:

IOException

upkeep

protected abstract void upkeep(RawFragmentBatch batch)

Handle miscellaneous tasks after batch retrieval

isTerminated

protected boolean isTerminated()