use of org.guvnor.ala.pipeline.events.OnErrorPipelineExecutionEvent in project kie-wb-common by kiegroup.
the class PipelineExecutionListenerTest method testOnPipelineError.
@Test
public void testOnPipelineError() {
listener.onPipelineError(new OnErrorPipelineExecutionEvent(EXECUTION_ID, pipeline, stage, throwable));
verify(pipelineStatusChangeEvent, times(1)).fire(new PipelineStatusChangeEvent(traceKey, PipelineStatus.ERROR));
}
use of org.guvnor.ala.pipeline.events.OnErrorPipelineExecutionEvent in project kie-wb-common by kiegroup.
the class PipelineExecutor method continuePipeline.
private void continuePipeline(final PipelineContext context, final PipelineEventListener... eventListeners) {
while (!context.isFinished()) {
final Stage<Object, ?> stage = getCurrentStage(context);
final Object newInput = pollOutput(context);
try {
propagateEvent(new BeforeStageExecutionEvent(context.getExecutionId(), context.getPipeline(), stage), eventListeners);
stage.execute(newInput, output -> {
final ConfigExecutor executor = resolve(output.getClass());
if (output instanceof ContextAware) {
((ContextAware) output).setContext(Collections.unmodifiableMap(context.getValues()));
}
final Object newOutput = interpolate(context.getValues(), output);
if (executor == null) {
throw new RuntimeException("Fail to resolve ConfigExecutor for: " + output.getClass());
}
context.getValues().put(executor.inputId(), newOutput);
if (executor instanceof BiFunctionConfigExecutor) {
final Optional result = (Optional) ((BiFunctionConfigExecutor) executor).apply(newInput, newOutput);
context.pushOutput(executor.outputId(), result.get());
} else if (executor instanceof FunctionConfigExecutor) {
final Optional result = (Optional) ((FunctionConfigExecutor) executor).apply(newOutput);
context.pushOutput(executor.outputId(), result.get());
}
propagateEvent(new AfterStageExecutionEvent(context.getExecutionId(), context.getPipeline(), stage), eventListeners);
});
} catch (final Throwable t) {
t.printStackTrace();
final RuntimeException exception = new RuntimeException("An error occurred while executing the " + (stage == null ? "null" : stage.getName()) + " stage.", t);
propagateEvent(new OnErrorStageExecutionEvent(context.getExecutionId(), context.getPipeline(), stage, exception), eventListeners);
propagateEvent(new OnErrorPipelineExecutionEvent(context.getExecutionId(), context.getPipeline(), stage, exception), eventListeners);
throw exception;
}
}
final Object output = pollOutput(context);
while (context.hasCallbacks()) {
context.applyCallbackAndPop(output);
}
}
use of org.guvnor.ala.pipeline.events.OnErrorPipelineExecutionEvent in project kie-wb-common by kiegroup.
the class PipelineExecutorTaskManagerImplEventsTest method testOnPipelineErrorExecutionEvent.
private void testOnPipelineErrorExecutionEvent(boolean async) {
when(taskEntry.isAsync()).thenReturn(async);
OnErrorPipelineExecutionEvent event = new OnErrorPipelineExecutionEvent(TASK_ID, pipeline, stage, error);
taskManager.localListener.onPipelineError(event);
verify(task, times(1)).setPipelineStatus(PipelineExecutorTask.Status.ERROR);
verify(task, times(1)).setPipelineError(pipelineExecutorErrorCaptor.capture());
assertEquals(new PipelineExecutorError(ERROR_MESSAGE, error), pipelineExecutorErrorCaptor.getValue());
verifyExecutorRegistryUpdated(async);
verifyExternalListenersNotified(event);
}
Aggregations