Package io.trino.plugin.hive.util

Class AsyncQueue<T>

java.lang.Object

io.trino.plugin.hive.util.AsyncQueue<T>

Direct Known Subclasses:

ThrottledAsyncQueue

@ThreadSafe
public class AsyncQueue<T>
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final class	AsyncQueue.BorrowResult<T,R>

Field Summary

Fields

Modifier and Type	Field	Description
protected final Executor	executor

Constructor Summary

Constructors

Constructor	Description
AsyncQueue(int targetQueueSize, Executor executor)

Method Summary

Modifier and Type	Method	Description
<O> com.google.common.util.concurrent.ListenableFuture<O>	borrowBatchAsync(int maxSize, Function<List<T>,AsyncQueue.BorrowResult<T,O>> function)	Invoke function with up to maxSize elements removed from the head of the queue, and insert elements in the return value to the tail of the queue.
void	finish()
com.google.common.util.concurrent.ListenableFuture<List<T>>	getBatchAsync(int maxSize)
protected com.google.common.util.concurrent.SettableFuture<Void>	getNotEmptySignal()
boolean	isFinished()	Returns true if all future attempts to retrieve elements from this queue are guaranteed to return empty.
com.google.common.util.concurrent.ListenableFuture<Void>	offer(T element)
int	size()

Methods inherited from class java.lang.Object

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

Field Details

executor

protected final Executor executor

Constructor Details

AsyncQueue

public AsyncQueue(int targetQueueSize,
 Executor executor)

Method Details

isFinished

public boolean isFinished()

Returns true if all future attempts to retrieve elements from this queue are guaranteed to return empty.

finish

public void finish()

offer

public com.google.common.util.concurrent.ListenableFuture<Void> offer(T element)

size

public int size()

getBatchAsync

public com.google.common.util.concurrent.ListenableFuture<List<T>> getBatchAsync(int maxSize)

getNotEmptySignal

protected com.google.common.util.concurrent.SettableFuture<Void> getNotEmptySignal()

borrowBatchAsync

public <O>
com.google.common.util.concurrent.ListenableFuture<O> borrowBatchAsync(int maxSize,
 Function<List<T>,AsyncQueue.BorrowResult<T,O>> function)

Invoke function with up to maxSize elements removed from the head of the queue, and insert elements in the return value to the tail of the queue.

If no element is currently available, invocation of function will be deferred until some element is available, or no more elements will be. Spurious invocation of function is possible.

Insertion through return value of function will be effective even if finish() has been invoked. When borrow (of a non-empty list) is ongoing, isFinished() will return false. If an empty list is supplied to function, it must not return a result indicating intention to insert elements into the queue.