use of won.bot.framework.eventbot.event.impl.factory.StartFactoryAtomCreationEvent in project webofneeds by researchstudio-sat.
the class InitFactoryAction method doRun.
@Override
protected void doRun(Event event, EventListener executingListener) throws Exception {
if (!(event instanceof InitializeEvent) || !(getEventListenerContext().getBotContextWrapper() instanceof FactoryBotContextWrapper)) {
logger.error("InitFactoryAction can only handle InitializeEvent with FactoryBotContextWrapper");
return;
}
EventListenerContext ctx = getEventListenerContext();
EventBus bus = ctx.getEventBus();
FactoryBotContextWrapper botContextWrapper = (FactoryBotContextWrapper) ctx.getBotContextWrapper();
// create a targeted counter that will publish an event when the target is
// reached
// in this case, 0 unfinished atom creations means that all atoms were created
final TargetCounterDecorator creationUnfinishedCounter = new TargetCounterDecorator(ctx, new CounterImpl("creationUnfinished"), 0);
BotTrigger createFactoryAtomTrigger = new BotTrigger(ctx, Duration.ofMillis(FACTORYATOMCREATION_DURATION_INMILLIS));
createFactoryAtomTrigger.activate();
bus.subscribe(StartFactoryAtomCreationEvent.class, new ActionOnFirstEventListener(ctx, new PublishEventAction(ctx, new StartBotTriggerCommandEvent(createFactoryAtomTrigger))));
bus.subscribe(BotTriggerEvent.class, new ActionOnTriggerEventListener(ctx, createFactoryAtomTrigger, new BaseEventBotAction(ctx) {
@Override
protected void doRun(Event event, EventListener executingListener) throws Exception {
if (isTooManyMessagesInFlight(messagesInFlightCounter)) {
return;
}
adjustTriggerInterval(createFactoryAtomTrigger, messagesInFlightCounter);
// defined via spring
AtomProducer atomProducer = ctx.getAtomProducer();
Dataset dataset = atomProducer.create();
if (dataset == null && atomProducer.isExhausted()) {
bus.publish(new AtomProducerExhaustedEvent());
bus.unsubscribe(executingListener);
return;
}
URI atomUriFromProducer = null;
Resource atomResource = WonRdfUtils.AtomUtils.getAtomResource(dataset);
if (atomResource.isURIResource()) {
atomUriFromProducer = URI.create(atomResource.getURI());
}
if (atomUriFromProducer != null) {
URI atomURI = botContextWrapper.getURIFromInternal(atomUriFromProducer);
if (atomURI != null) {
bus.publish(new FactoryAtomCreationSkippedEvent());
} else {
bus.publish(new CreateAtomCommandEvent(dataset, botContextWrapper.getFactoryListName()));
}
}
}
}));
bus.subscribe(CreateAtomCommandSuccessEvent.class, // decrease the
new DecrementCounterAction(ctx, creationUnfinishedCounter), // count a successful atom
new IncrementCounterAction(ctx, atomCreationSuccessfulCounter), // creation
new BaseEventBotAction(ctx) {
@Override
protected void doRun(Event event, EventListener executingListener) throws Exception {
if (event instanceof CreateAtomCommandSuccessEvent) {
CreateAtomCommandSuccessEvent atomCreatedEvent = (CreateAtomCommandSuccessEvent) event;
botContextWrapper.addInternalIdToUriReference(atomCreatedEvent.getAtomUriBeforeCreation(), atomCreatedEvent.getAtomURI());
}
}
});
bus.subscribe(CreateAtomCommandEvent.class, // execute the atom creation for the atom in the event
new ExecuteCreateAtomCommandAction(ctx), // increase the
new IncrementCounterAction(ctx, atomCreationStartedCounter), // increase the
new IncrementCounterAction(ctx, creationUnfinishedCounter));
// if an atom is already created we skip the recreation of it and increase the
// atomCreationSkippedCounter
bus.subscribe(FactoryAtomCreationSkippedEvent.class, new IncrementCounterAction(ctx, atomCreationSkippedCounter));
// if a creation failed, we don't want to keep us from keeping the correct count
bus.subscribe(CreateAtomCommandFailureEvent.class, // decrease the
new DecrementCounterAction(ctx, creationUnfinishedCounter), // count an unsuccessful atom
new IncrementCounterAction(ctx, atomCreationFailedCounter));
// when the atomproducer is exhausted, we stop the creator (trigger) and we have
// to wait until all unfinished atom creations finish
// when they do, the InitFactoryFinishedEvent is published
bus.subscribe(AtomProducerExhaustedEvent.class, new ActionOnFirstEventListener(ctx, new MultipleActions(ctx, new PublishEventAction(ctx, new StopBotTriggerCommandEvent(createFactoryAtomTrigger)), new BaseEventBotAction(ctx) {
@Override
protected void doRun(Event event, EventListener executingListener) throws Exception {
// when we're called, there probably are atom creations
// unfinished, but there may not be
// a)
// first, prepare for the case when there are unfinished
// atom creations:
// we register a listener, waiting for the unfinished
// counter to reach 0
EventListener waitForUnfinishedAtomsListener = bus.subscribe(TargetCountReachedEvent.class, new ActionOnFirstEventListener(ctx, new TargetCounterFilter(creationUnfinishedCounter), new PublishEventAction(ctx, new InitFactoryFinishedEvent())));
// now, we can check if we've already reached the target
if (creationUnfinishedCounter.getCount() <= 0) {
// ok, turned out we didn't need that listener
bus.unsubscribe(waitForUnfinishedAtomsListener);
bus.publish(new InitFactoryFinishedEvent());
}
}
})));
bus.subscribe(InitFactoryFinishedEvent.class, new ActionOnFirstEventListener(ctx, "factoryCreateStatsLogger", new BaseEventBotAction(ctx) {
@Override
protected void doRun(Event event, EventListener executingListener) throws Exception {
logger.info("FactoryAtomCreation finished: total:{}, successful: {}, failed: {}, skipped: {}", new Object[] { atomCreationStartedCounter.getCount(), atomCreationSuccessfulCounter.getCount(), atomCreationFailedCounter.getCount(), atomCreationSkippedCounter.getCount() });
}
}));
// MessageInFlight counter handling *************************
bus.subscribe(MessageCommandEvent.class, new IncrementCounterAction(ctx, messagesInFlightCounter));
bus.subscribe(MessageCommandResultEvent.class, new DecrementCounterAction(ctx, messagesInFlightCounter));
// if we receive a
bus.subscribe(MessageCommandFailureEvent.class, new LogMessageCommandFailureAction(ctx));
// message command
// failure, log it
// Start the atom creation stuff
bus.publish(new StartFactoryAtomCreationEvent());
}
Aggregations