/** * Finish bundle without checking for conditions. */ final void forceFinishBundle() { if (!bundleFinished) { bundleFinished = true; if (pushBackRunner == null) { doFnRunner.finishBundle(); } else { pushBackRunner.finishBundle(); } } }
/** * Checks whether it is time to finish the bundle and finish it. */ final void checkAndFinishBundle() { if (!bundleFinished) { if (currBundleCount >= bundleSize || System.currentTimeMillis() - prevBundleStartTime >= bundleMillis) { bundleFinished = true; if (pushBackRunner == null) { doFnRunner.finishBundle(); } else { pushBackRunner.finishBundle(); } } } }
protected void invokeFinishBundle() { if (bundleStarted) { pushbackDoFnRunner.finishBundle(); bundleStarted = false; elementCount = 0L; lastFinishBundleTime = getProcessingTimeService().getCurrentProcessingTime(); } }
private void invokeFinishBundle() { if (bundleStarted) { pushbackDoFnRunner.finishBundle(); bundleStarted = false; elementCount = 0L; lastFinishBundleTime = getProcessingTimeService().getCurrentProcessingTime(); } }
protected void invokeFinishBundle() { if (bundleStarted) { pushbackDoFnRunner.finishBundle(); bundleStarted = false; elementCount = 0L; lastFinishBundleTime = getProcessingTimeService().getCurrentProcessingTime(); } }
@Test public void startFinishBundleDelegates() { PushbackSideInputDoFnRunner runner = createRunner(ImmutableList.of(singletonView)); assertThat(underlying.started, is(true)); assertThat(underlying.finished, is(false)); runner.finishBundle(); assertThat(underlying.finished, is(true)); }
Iterables.addAll(pushedBackValues, nextPushedBackValues); doFnRunner.finishBundle();
pushbackDoFnRunner.finishBundle(); return pushedBack; } catch (UserCodeException ue) {
@Override public void fireTimer(Object key, Collection<TimerData> timerDataSet) { pushbackDoFnRunner.startBundle(); @SuppressWarnings("unchecked") Coder<Object> keyCoder = (Coder) currentKeyStateInternals.getKeyCoder(); ((StateInternalsProxy) currentKeyStateInternals).setKey(key); currentKeyTimerInternals.setContext( key, keyCoder, new Instant(this.currentInputWatermark), new Instant(this.currentOutputWatermark)); for (TimerData timerData : timerDataSet) { StateNamespace namespace = timerData.getNamespace(); checkArgument(namespace instanceof WindowNamespace); BoundedWindow window = ((WindowNamespace<?>) namespace).getWindow(); pushbackDoFnRunner.onTimer( timerData.getTimerId(), window, timerData.getTimestamp(), timerData.getDomain()); } pushbackDoFnRunner.finishBundle(); }