@Override public void dropFlowFiles(DropFlowFileRequest dropRequest, String requestor) { queue.dropFlowFiles(dropRequest, requestor); }
private synchronized boolean complete() { if (!queue.isEmpty()) { return false; } this.rebalanceTask = null; return true; }
@Override public FlowFileQueueContents packageForRebalance(String newPartitionName) { return queue.packageForRebalance(newPartitionName); }
polled = queue.poll(expiredRecords, -1, pollWaitMillis); } catch (final InterruptedException ie) { Thread.currentThread().interrupt(); toDistribute.add(polled); final List<FlowFileRecord> additionalRecords = queue.poll(999, expiredRecords, -1); toDistribute.addAll(additionalRecords); queue.acknowledge(toDistribute);
@Override public void putAll(final Collection<FlowFileRecord> flowFiles) { queue.putAll(flowFiles); }
@Override public void put(final FlowFileRecord flowFile) { queue.put(flowFile); }
@Override public SwapSummary recoverSwappedFlowFiles() { return this.queue.recoverSwappedFlowFiles(); }
public StandardRebalancingPartition(final FlowFileSwapManager swapManager, final int swapThreshold, final EventReporter eventReporter, final LoadBalancedFlowFileQueue flowFileQueue, final DropFlowFileAction dropAction) { this.queue = new BlockingSwappablePriorityQueue(swapManager, swapThreshold, eventReporter, flowFileQueue, dropAction, SWAP_PARTITION_NAME); this.queueIdentifier = flowFileQueue.getIdentifier(); this.flowFileQueue = flowFileQueue; this.description = "RebalancingPartition[queueId=" + queueIdentifier + "]"; }
@Override public void rebalance(final FlowFileQueueContents queueContents) { if (queueContents.getActiveFlowFiles().isEmpty() && queueContents.getSwapLocations().isEmpty()) { return; } queue.inheritQueueContents(queueContents); rebalanceFromQueue(); }
@Override public void rebalance(final Collection<FlowFileRecord> flowFiles) { queue.putAll(flowFiles); rebalanceFromQueue(); }