use of org.apache.flink.table.runtime.generated.GeneratedCollectorWrapper in project flink by apache.
the class LookupJoinHarnessTest method createHarness.
// ---------------------------------------------------------------------------------
@SuppressWarnings("unchecked")
private OneInputStreamOperatorTestHarness<RowData, RowData> createHarness(JoinType joinType, FilterOnTable filterOnTable) throws Exception {
boolean isLeftJoin = joinType == JoinType.LEFT_JOIN;
ProcessFunction<RowData, RowData> joinRunner;
if (filterOnTable == FilterOnTable.WITHOUT_FILTER) {
joinRunner = new LookupJoinRunner(new GeneratedFunctionWrapper<>(new TestingFetcherFunction()), new GeneratedCollectorWrapper<>(new TestingFetcherCollector()), isLeftJoin, 2);
} else {
joinRunner = new LookupJoinWithCalcRunner(new GeneratedFunctionWrapper<>(new TestingFetcherFunction()), new GeneratedFunctionWrapper<>(new CalculateOnTemporalTable()), new GeneratedCollectorWrapper<>(new TestingFetcherCollector()), isLeftJoin, 2);
}
ProcessOperator<RowData, RowData> operator = new ProcessOperator<>(joinRunner);
return new OneInputStreamOperatorTestHarness<>(operator, inSerializer);
}
Aggregations