public static void closeFunction(Function function) throws Exception{ if (function instanceof RichFunction) { RichFunction richFunction = (RichFunction) function; richFunction.close(); } }
@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); } } }
public static RuntimeContext getFunctionRuntimeContext(Function function, RuntimeContext defaultContext){ if (function instanceof RichFunction) { RichFunction richFunction = (RichFunction) function; return richFunction.getRuntimeContext(); } else { return defaultContext; } }
public static void setFunctionRuntimeContext(Function function, RuntimeContext context){ if (function instanceof RichFunction) { RichFunction richFunction = (RichFunction) function; richFunction.setRuntimeContext(context); } }
public static void openFunction(Function function, Configuration parameters) throws Exception{ if (function instanceof RichFunction) { RichFunction richFunction = (RichFunction) function; richFunction.open(parameters); } }
public static void setFunctionRuntimeContext(Function function, RuntimeContext context){ if (function instanceof RichFunction) { RichFunction richFunction = (RichFunction) function; richFunction.setRuntimeContext(context); } }
public static void openFunction(Function function, Configuration parameters) throws Exception{ if (function instanceof RichFunction) { RichFunction richFunction = (RichFunction) function; richFunction.open(parameters); } }
@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); } } }
@Override public void close() throws Exception { if (this.fun instanceof RichFunction) { try { ((RichFunction) this.fun).close(); } catch (PyException pe) { throw createAndLogException(pe); } } }
public static void setFunctionRuntimeContext(Function function, RuntimeContext context){ if (function instanceof RichFunction) { RichFunction richFunction = (RichFunction) function; richFunction.setRuntimeContext(context); } }
public static void openFunction(Function function, Configuration parameters) throws Exception{ if (function instanceof RichFunction) { RichFunction richFunction = (RichFunction) function; richFunction.open(parameters); } }
public static RuntimeContext getFunctionRuntimeContext(Function function, RuntimeContext defaultContext){ if (function instanceof RichFunction) { RichFunction richFunction = (RichFunction) function; return richFunction.getRuntimeContext(); } else { return defaultContext; } }
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); } }
@Override public void close() throws Exception { if (this.fun instanceof RichFunction) { try { ((RichFunction) this.fun).close(); } catch (PyException pe) { throw AbstractPythonUDF.createAndLogException(pe, LOG); } } }
@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 open(Configuration parameters) throws Exception { super.open(parameters); for (FilterFunction<? super T> component : components) { if (component instanceof RichFunction) { ((RichFunction) component).open(parameters); } } }
public static RuntimeContext getFunctionRuntimeContext(Function function, RuntimeContext defaultContext){ if (function instanceof RichFunction) { RichFunction richFunction = (RichFunction) function; return richFunction.getRuntimeContext(); } else { return defaultContext; } }
if (producer != null) { try { producer.close();
@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 open(Configuration parameters) throws Exception { super.open(parameters); for (FilterFunction<? super T> component : components) { if (component instanceof RichFunction) { ((RichFunction) component).open(parameters); } } }