@Override public void onTrigger(ProcessContext context, ProcessSessionFactory sessionFactory) throws ProcessException { if (source instanceof PollableSource) { super.onTrigger(context, sessionFactory); } else if (source instanceof EventDrivenSource) { ProcessSessionFactory old = sessionFactoryRef.getAndSet(sessionFactory); if (old != sessionFactory) { if (runnerRef.get() != null) { stopped(); sessionFactoryRef.set(sessionFactory); } runnerRef.set(new EventDrivenSourceRunner()); eventDrivenSourceChannelRef.set(new NifiSessionFactoryChannel(sessionFactoryRef.get(), SUCCESS)); eventDrivenSourceChannelRef.get().start(); source.setChannelProcessor(new ChannelProcessor( new NifiChannelSelector(eventDrivenSourceChannelRef.get()))); runnerRef.get().setSource(source); runnerRef.get().start(); } } }
@Override public String toString() { return "EventDrivenSourceRunner: { source:" + getSource() + " }"; }
@OnStopped public void stopped() { if (source instanceof PollableSource) { source.stop(); } else { EventDrivenSourceRunner runner = runnerRef.get(); if (runner != null) { runner.stop(); runnerRef.compareAndSet(runner, null); } NifiSessionFactoryChannel eventDrivenSourceChannel = eventDrivenSourceChannelRef.get(); if (eventDrivenSourceChannel != null) { eventDrivenSourceChannel.stop(); eventDrivenSourceChannelRef.compareAndSet(eventDrivenSourceChannel, null); } } sessionFactoryRef.set(null); }
/** * Static factory method to instantiate a source runner implementation that * corresponds to the type of {@link Source} specified. * * @param source The source to run * @return A runner that can run the specified source * @throws IllegalArgumentException if the specified source does not implement * a supported derived interface of {@link SourceRunner}. */ public static SourceRunner forSource(Source source) { SourceRunner runner = null; if (source instanceof PollableSource) { runner = new PollableSourceRunner(); ((PollableSourceRunner) runner).setSource((PollableSource) source); } else if (source instanceof EventDrivenSource) { runner = new EventDrivenSourceRunner(); ((EventDrivenSourceRunner) runner).setSource((EventDrivenSource) source); } else { throw new IllegalArgumentException("No known runner type for source " + source); } return runner; }
/** * Static factory method to instantiate a source runner implementation that * corresponds to the type of {@link Source} specified. * * @param source The source to run * @return A runner that can run the specified source * @throws IllegalArgumentException if the specified source does not implement * a supported derived interface of {@link SourceRunner}. */ public static SourceRunner forSource(Source source) { SourceRunner runner = null; if (source instanceof PollableSource) { runner = new PollableSourceRunner(); ((PollableSourceRunner) runner).setSource((PollableSource) source); } else if (source instanceof EventDrivenSource) { runner = new EventDrivenSourceRunner(); ((EventDrivenSourceRunner) runner).setSource((EventDrivenSource) source); } else { throw new IllegalArgumentException("No known runner type for source " + source); } return runner; }
@Override public void onTrigger(ProcessContext context, ProcessSessionFactory sessionFactory) throws ProcessException { if (source instanceof PollableSource) { super.onTrigger(context, sessionFactory); } else if (source instanceof EventDrivenSource) { ProcessSessionFactory old = sessionFactoryRef.getAndSet(sessionFactory); if (old != sessionFactory) { if (runnerRef.get() != null) { stopped(); sessionFactoryRef.set(sessionFactory); } runnerRef.set(new EventDrivenSourceRunner()); eventDrivenSourceChannelRef.set(new NifiSessionFactoryChannel(sessionFactoryRef.get(), SUCCESS)); eventDrivenSourceChannelRef.get().start(); source.setChannelProcessor(new ChannelProcessor( new NifiChannelSelector(eventDrivenSourceChannelRef.get()))); runnerRef.get().setSource(source); runnerRef.get().start(); } } }
@Override public void stop() { Source source = getSource(); source.stop(); ChannelProcessor cp = source.getChannelProcessor(); cp.close(); lifecycleState = LifecycleState.STOP; }
@OnStopped public void stopped() { if (source instanceof PollableSource) { source.stop(); } else { EventDrivenSourceRunner runner = runnerRef.get(); if (runner != null) { runner.stop(); runnerRef.compareAndSet(runner, null); } NifiSessionFactoryChannel eventDrivenSourceChannel = eventDrivenSourceChannelRef.get(); if (eventDrivenSourceChannel != null) { eventDrivenSourceChannel.stop(); eventDrivenSourceChannelRef.compareAndSet(eventDrivenSourceChannel, null); } } sessionFactoryRef.set(null); }
@Override public void start() { Source source = getSource(); ChannelProcessor cp = source.getChannelProcessor(); cp.initialize(); source.start(); lifecycleState = LifecycleState.START; }
@Override public String toString() { return "EventDrivenSourceRunner: { source:" + getSource() + " }"; }
@Override public void start() { Source source = getSource(); ChannelProcessor cp = source.getChannelProcessor(); cp.initialize(); source.start(); lifecycleState = LifecycleState.START; }
@Override public void stop() { Source source = getSource(); source.stop(); ChannelProcessor cp = source.getChannelProcessor(); cp.close(); lifecycleState = LifecycleState.STOP; }