public static void setFunctionRuntimeContext(Function function, RuntimeContext context){ if (function instanceof RichFunction) { RichFunction richFunction = (RichFunction) function; richFunction.setRuntimeContext(context); } }
@Override public void open(Configuration parameters) throws Exception { this.fun = InterpreterUtils.deserializeFunction(getRuntimeContext(), this.serFun); if (this.fun instanceof RichFunction) { try { final RichFunction rf = (RichFunction) this.fun; rf.setRuntimeContext(getRuntimeContext()); rf.open(parameters); } catch (PyException pe) { throw createAndLogException(pe); } } }
@Override public void open(Configuration parameters) throws Exception { this.fun = InterpreterUtils.deserializeFunction(getRuntimeContext(), this.serFun); if (this.fun instanceof RichFunction) { try { final RichFunction winFun = (RichFunction) this.fun; winFun.setRuntimeContext(getRuntimeContext()); winFun.open(parameters); } catch (PyException pe) { throw AbstractPythonUDF.createAndLogException(pe, LOG); } } }
private static <T extends Serializable> List<T> runRichSourceFunction(SourceFunction<T> sourceFunction) throws Exception { try (MockEnvironment environment = new MockEnvironmentBuilder() .setTaskName("MockTask") .setMemorySize(3 * 1024 * 1024) .setInputSplitProvider(new MockInputSplitProvider()) .setBufferSize(1024) .build()) { AbstractStreamOperator<?> operator = mock(AbstractStreamOperator.class); when(operator.getExecutionConfig()).thenReturn(new ExecutionConfig()); RuntimeContext runtimeContext = new StreamingRuntimeContext( operator, environment, new HashMap<>()); ((RichFunction) sourceFunction).setRuntimeContext(runtimeContext); ((RichFunction) sourceFunction).open(new Configuration()); return runNonRichSourceFunction(sourceFunction); } }
public static void setFunctionRuntimeContext(Function function, RuntimeContext context){ if (function instanceof RichFunction) { RichFunction richFunction = (RichFunction) function; richFunction.setRuntimeContext(context); } }
public static void setFunctionRuntimeContext(Function function, RuntimeContext context){ if (function instanceof RichFunction) { RichFunction richFunction = (RichFunction) function; richFunction.setRuntimeContext(context); } }
@Override public void setRuntimeContext(RuntimeContext t) { super.setRuntimeContext(t); for (FilterFunction<? super T> component : components) { if (component instanceof RichFunction) { ((RichFunction) component).setRuntimeContext(t); } } }
@Override public void setRuntimeContext(RuntimeContext t) { super.setRuntimeContext(t); for (FilterFunction<? super T> component : components) { if (component instanceof RichFunction) { ((RichFunction) component).setRuntimeContext(t); } } }