Examples with FutureCompletingBlockingQueue org.apache.flink.connector.base.source.reader.synchronization.FutureCompletingBlockingQueue used on opensource projects

Example 11 with FutureCompletingBlockingQueue

use of org.apache.flink.connector.base.source.reader.synchronization.FutureCompletingBlockingQueue in project flink by apache.

the class MockBaseSource method createReader.

@Override
public SourceReader<Integer, MockSourceSplit> createReader(SourceReaderContext readerContext) {
    FutureCompletingBlockingQueue<RecordsWithSplitIds<int[]>> elementsQueue = new FutureCompletingBlockingQueue<>();
    Configuration config = new Configuration();
    config.setInteger(SourceReaderOptions.ELEMENT_QUEUE_CAPACITY, 1);
    config.setLong(SourceReaderOptions.SOURCE_READER_CLOSE_TIMEOUT, 30000L);
    MockSplitReader.Builder builder = MockSplitReader.newBuilder().setNumRecordsPerSplitPerFetch(2).setBlockingFetch(true);
    return new MockSourceReader(elementsQueue, builder::build, config, readerContext);
}