@Override public List<Boolean> forkDataRecordImpl(WorkUnitState workUnitState, D input) { return this.embeddedForkOperator.forkDataRecord(workUnitState, input); }
List<Boolean> forkedRecords = forkOperator.forkDataRecord(this.taskState, convertedRecord); if (forkedRecords.size() != branches) { throw new ForkBranchMismatchException(String
@Test public void testForkMethods() { ForkOperator<String, String> dummyForkOperator = new IdentityForkOperator<String, String>(); WorkUnitState workUnitState = new WorkUnitState(); workUnitState.setProp(ConfigurationKeys.FORK_BRANCHES_KEY, 2); List<Boolean> schemas = dummyForkOperator.forkSchema(workUnitState, ""); Assert.assertEquals(schemas, Arrays.asList(true, true)); List<Boolean> records = dummyForkOperator.forkDataRecord(workUnitState, ""); Assert.assertEquals(records, Arrays.asList(true, true)); Assert.assertEquals(dummyForkOperator.getBranches(workUnitState), 2); workUnitState.setProp(ConfigurationKeys.FORK_BRANCHES_KEY, 0); schemas = dummyForkOperator.forkSchema(workUnitState, ""); Assert.assertTrue(schemas.isEmpty()); records = dummyForkOperator.forkDataRecord(workUnitState, ""); Assert.assertTrue(records.isEmpty()); Assert.assertEquals(dummyForkOperator.getBranches(workUnitState), 0); } }
if (r instanceof RecordEnvelope) { RecordEnvelope<D> recordEnvelope = (RecordEnvelope<D>) r; return new RecordWithForkMap<>(recordEnvelope, forkOperator.forkDataRecord(workUnitState, recordEnvelope.getRecord())); } else if (r instanceof ControlMessage) { return new RecordWithForkMap<D>((ControlMessage<D>) r, branches);
@Override public List<Boolean> forkDataRecordImpl(WorkUnitState workUnitState, D input) { return this.embeddedForkOperator.forkDataRecord(workUnitState, input); }
List<Boolean> forkedRecords = forkOperator.forkDataRecord(this.taskState, convertedRecord); if (forkedRecords.size() != branches) { throw new ForkBranchMismatchException(String
if (r instanceof RecordEnvelope) { RecordEnvelope<D> recordEnvelope = (RecordEnvelope<D>) r; return new RecordWithForkMap<>(recordEnvelope, forkOperator.forkDataRecord(workUnitState, recordEnvelope.getRecord())); } else if (r instanceof ControlMessage) { return new RecordWithForkMap<D>((ControlMessage<D>) r, branches);