@SuppressWarnings("ObjectEquality") @Override public CompletableFuture<ConnectorSplitBatch> getNextBatch(ConnectorPartitionHandle partitionHandle, int maxSize) { if (!partitionHandle.equals(NOT_PARTITIONED)) { throw new IllegalArgumentException("partitionHandle must be NOT_PARTITIONED"); } int remainingSplits = splits.size() - offset; int size = Math.min(remainingSplits, maxSize); List<ConnectorSplit> results = splits.subList(offset, offset + size); offset += size; return completedFuture(new ConnectorSplitBatch(results, isFinished())); }
@Override public CompletableFuture<ConnectorSplitBatch> getNextBatch(ConnectorPartitionHandle partitionHandle, int maxSize) { checkArgument(partitionHandle.equals(NOT_PARTITIONED), "partitionHandle must be NOT_PARTITIONED"); return notEmptyFuture .thenApply(x -> getBatch(maxSize)) .thenApply(splits -> new ConnectorSplitBatch(splits, isFinished())); }
@SuppressWarnings("ObjectEquality") @Override public CompletableFuture<ConnectorSplitBatch> getNextBatch(ConnectorPartitionHandle partitionHandle, int maxSize) { if (!partitionHandle.equals(NOT_PARTITIONED)) { throw new IllegalArgumentException("partitionHandle must be NOT_PARTITIONED"); } int remainingSplits = splits.size() - offset; int size = Math.min(remainingSplits, maxSize); List<ConnectorSplit> results = splits.subList(offset, offset + size); offset += size; return completedFuture(new ConnectorSplitBatch(results, isFinished())); }