@ProcessElement public void processElement(final ProcessContext c) { final InputT element = c.element().getKey(); invoker.invokeSplitRestriction( element, c.element().getValue(), new OutputReceiver<RestrictionT>() { @Override public void output(RestrictionT part) { c.output(KV.of(element, part)); } @Override public void outputWithTimestamp(RestrictionT part, Instant timestamp) { throw new UnsupportedOperationException(); } }); }
assertEquals(restriction, invoker.invokeGetInitialRestriction("blah")); final List<SomeRestriction> outputs = new ArrayList<>(); invoker.invokeSplitRestriction( "blah", restriction,
invoker.<RestrictionWithDefaultTracker>invokeGetRestrictionCoder(coderRegistry), instanceOf(CoderForDefaultTracker.class)); invoker.invokeSplitRestriction( "blah", "foo",