Examples with StopMessage org.pentaho.di.engine.api.remote.StopMessage used on opensource projects

Example 1 with StopMessage

use of org.pentaho.di.engine.api.remote.StopMessage in project pentaho-kettle by pentaho.

the class TransWebSocketEngineAdapter method stopAll.

@Override
public void stopAll() {
    try {
        getDaemonEndpoint().sendMessage(new StopMessage(getErrors() == 0 ? "User Request" : "Error reported"));
        if (getErrors() == 0) {
            waitUntilFinished();
            finishProcess(true);
        }
    } catch (KettleException e) {
        getLogChannel().logDebug(e.getMessage());
    }
}

Example 2 with StopMessage

use of org.pentaho.di.engine.api.remote.StopMessage in project pentaho-kettle by pentaho.

the class TransWebSocketEngineAdapter method wireStatusToTransListeners.

private void wireStatusToTransListeners() throws KettleException {
    messageEventService.addHandler(Util.getTransformationStatusEvent(), new MessageEventHandler() {

        @Override
        public void execute(Message message) throws MessageEventHandlerExecutionException {
            PDIEvent<RemoteSource, Status> transStatusEvent = (PDIEvent<RemoteSource, Status>) message;
            addStepPerformanceSnapShot();
            getTransListeners().forEach(l -> {
                try {
                    switch(transStatusEvent.getData()) {
                        case RUNNING:
                            l.transStarted(TransWebSocketEngineAdapter.this);
                            l.transActive(TransWebSocketEngineAdapter.this);
                            break;
                        case PAUSED:
                            break;
                        case STOPPED:
                            break;
                        case FAILED:
                        case FINISHED:
                            l.transFinished(TransWebSocketEngineAdapter.this);
                            setFinished(true);
                            break;
                    }
                } catch (KettleException e) {
                    throw new RuntimeException(e);
                }
            });
        }

        @Override
        public String getIdentifier() {
            return TRANSFORMATION_STATUS;
        }
    });
    messageEventService.addHandler(Util.getTransformationErrorEvent(), new MessageEventHandler() {

        @Override
        public void execute(Message message) throws MessageEventHandlerExecutionException {
            Throwable throwable = ((PDIEvent<RemoteSource, LogEntry>) message).getData().getThrowable();
            getLogChannel().logError("Error Executing Transformation", throwable);
            errors.incrementAndGet();
            finishProcess(true);
        }

        @Override
        public String getIdentifier() {
            return TRANSFORMATION_ERROR;
        }
    });
    messageEventService.addHandler(Util.getStopMessage(), new MessageEventHandler() {

        @Override
        public void execute(Message message) throws MessageEventHandlerExecutionException {
            StopMessage stopMessage = (StopMessage) message;
            if (stopMessage.sessionWasKilled() || stopMessage.operationFailed()) {
                getLogChannel().logError("Finalizing execution: " + stopMessage.getReasonPhrase());
            } else {
                getLogChannel().logBasic("Finalizing execution: " + stopMessage.getReasonPhrase());
            }
            finishProcess(false);
            try {
                getDaemonEndpoint().close(stopMessage.getReasonPhrase());
            } catch (KettleException e) {
                getLogChannel().logError("Error finalizing", e);
            }
            // let's shutdown the session monitor thread
            closeSessionMonitor();
            // Signal for the the waitUntilFinished blocker...
            transFinishedSignal.countDown();
        }

        @Override
        public String getIdentifier() {
            return TRANSFORMATION_STOP;
        }
    });
}

Example 3 with StopMessage

use of org.pentaho.di.engine.api.remote.StopMessage in project pentaho-kettle by pentaho.

the class MessageEventServiceTest method testStopMessageFireEvent.

@Test
public void testStopMessageFireEvent() throws KettleException {
    addHandlers(new StopMessage(""), messageEventHandler, messageEventHandler2);
    StopMessage msg = new StopMessage("User request");
    messageEventService.fireEvent(msg);
    verify(messageEventHandler).execute(msg);
    verify(messageEventHandler2).execute(msg);
}