public void run(SourceContext<Object> ctx) throws Exception { try { this.fun.run(ctx); } catch (PyException pe) { throw createAndLogException(pe); } }
private static <T extends Serializable> List<T> runNonRichSourceFunction(SourceFunction<T> sourceFunction) { final List<T> outputs = new ArrayList<>(); try { SourceFunction.SourceContext<T> ctx = new CollectingSourceContext<T>(new Object(), outputs); sourceFunction.run(ctx); } catch (Exception e) { throw new RuntimeException("Cannot invoke source.", e); } return outputs; } }
userFunction.run(ctx);
userFunction.run(ctx);
userFunction.run(ctx);
userFunction.run(ctx);