@Setup public void setup() { invoker = DoFnInvokers.invokerFor(splittableFn); invoker.invokeSetup(); }
@Override public void open(Configuration parameters) throws Exception { doFnInvoker = DoFnInvokers.invokerFor(doFn); doFnInvoker.invokeSetup(); }
@Setup public void setup() throws Exception { invoker = DoFnInvokers.invokerFor(fn); invoker.invokeSetup(); }
@Override public void open(Configuration parameters) throws Exception { doFnInvoker = DoFnInvokers.invokerFor(dofn); doFnInvoker.invokeSetup(); }
@Override public void open(Configuration parameters) throws Exception { doFnInvoker = DoFnInvokers.invokerFor(dofn); doFnInvoker.invokeSetup(); }
@Setup public void setup() { DoFnInvokers.invokerFor(underlyingDoFn).invokeSetup(); }
@Setup public void setup() { this.invoker = DoFnInvokers.invokerFor(fn); invoker.invokeSetup(); }
@Override public void open(Configuration parameters) throws Exception { doFnInvoker = DoFnInvokers.invokerFor(dofn); doFnInvoker.invokeSetup(); }
@Override public void open(Configuration parameters) throws Exception { doFnInvoker = DoFnInvokers.invokerFor(doFn); doFnInvoker.invokeSetup(); }
@Setup public void setup() { invoker = DoFnInvokers.invokerFor(fn); invoker.invokeSetup(); }
@Override public void open(Configuration parameters) throws Exception { doFnInvoker = DoFnInvokers.invokerFor(doFn); doFnInvoker.invokeSetup(); }
@Override public DoFn<?, ?> load(Thread key) throws Exception { DoFn<?, ?> fn = (DoFn<?, ?>) SerializableUtils.deserializeFromByteArray( original, "DoFn Copy in thread " + key.getName()); DoFnInvokers.invokerFor(fn).invokeSetup(); return fn; } }
@Override public void setup() { sideInputReader = new SideInputHandler(sideInputs, InMemoryStateInternals.<Void>forKey(null)); doFnInvoker = DoFnInvokers.invokerFor(doFn); doFnInvoker.invokeSetup(); doFnRunner = doFnRunnerFactory.createRunner(sideInputReader); pushedBackValues = new ArrayList<>(); outputManager.setup(mainOutput, sideOutputs); }
@SuppressWarnings("unchecked") private void initializeState() throws Exception { checkState(state == State.UNINITIALIZED, "Already initialized"); checkState(fn == null, "Uninitialized but fn != null"); if (cloningBehavior.equals(CloningBehavior.DO_NOT_CLONE)) { fn = origFn; } else { fn = (DoFn<InputT, OutputT>) SerializableUtils.deserializeFromByteArray( SerializableUtils.serializeToByteArray(origFn), origFn.toString()); } fnInvoker = DoFnInvokers.invokerFor(fn); fnInvoker.invokeSetup(); }
doFnInvoker.invokeSetup();
@Test public void testDoFnWithStartBundleSetupTeardown() throws Exception { class MockFn extends DoFn<String, String> { @ProcessElement public void processElement(ProcessContext c) {} @StartBundle public void startBundle(StartBundleContext c) {} @FinishBundle public void finishBundle(FinishBundleContext c) {} @Setup public void before() {} @Teardown public void after() {} } MockFn fn = mock(MockFn.class); DoFnInvoker<String, String> invoker = DoFnInvokers.invokerFor(fn); invoker.invokeSetup(); invoker.invokeStartBundle(mockStartBundleContext); invoker.invokeFinishBundle(mockFinishBundleContext); invoker.invokeTeardown(); verify(fn).before(); verify(fn).startBundle(mockStartBundleContext); verify(fn).finishBundle(mockFinishBundleContext); verify(fn).after(); }
throws Exception { if (!wasSetupCalled) { DoFnInvokers.invokerFor(doFn).invokeSetup(); wasSetupCalled = true;
doFnInvoker.invokeSetup();
doFnInvoker.invokeSetup();
doFnInvoker.invokeSetup();