com.netflix.conductor.core.execution.tasks.Event.java Source code

Introduction

Here is the source code for com.netflix.conductor.core.execution.tasks.Event.java

Source

/**
 * Copyright 2016 Netflix, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
/**
 * 
 */
package com.netflix.conductor.core.execution.tasks;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.netflix.conductor.common.metadata.tasks.Task;
import com.netflix.conductor.common.metadata.tasks.Task.Status;
import com.netflix.conductor.common.run.Workflow;
import com.netflix.conductor.core.events.EventQueues;
import com.netflix.conductor.core.events.queue.Message;
import com.netflix.conductor.core.events.queue.ObservableQueue;
import com.netflix.conductor.core.execution.WorkflowExecutor;

/**
 * @author Viren
 *
 */
public class Event extends WorkflowSystemTask {

    private static Logger logger = LoggerFactory.getLogger(Event.class);

    private ObjectMapper om = new ObjectMapper();

    private enum Sink {
        conductor, sqs
    }

    public Event() {
        super("EVENT");
    }

    @Override
    public void start(Workflow workflow, Task task, WorkflowExecutor provider) throws Exception {

        Map<String, Object> payload = new HashMap<>();
        payload.putAll(task.getInputData());
        payload.put("workflowInstanceId", workflow.getWorkflowId());
        payload.put("workflowType", workflow.getWorkflowType());
        payload.put("workflowVersion", workflow.getVersion());
        payload.put("correlationId", workflow.getCorrelationId());

        String payloadJson = om.writeValueAsString(payload);
        Message message = new Message(task.getTaskId(), payloadJson, task.getTaskId());
        ObservableQueue queue = getQueue(workflow, task);
        if (queue != null) {
            queue.publish(Arrays.asList(message));
            task.setStatus(Status.COMPLETED);
        }
    }

    @Override
    public boolean execute(Workflow workflow, Task task, WorkflowExecutor provider) throws Exception {

        if (task.getStatus().equals(Status.SCHEDULED)) {
            long timeSince = System.currentTimeMillis() - task.getScheduledTime();
            if (timeSince > 600_000) {
                start(workflow, task, provider);
                return true;
            } else {
                return false;
            }
        }

        return false;
    }

    @Override
    public void cancel(Workflow workflow, Task task, WorkflowExecutor provider) throws Exception {
        Message message = new Message(task.getTaskId(), null, task.getTaskId());
        getQueue(workflow, task).ack(Arrays.asList(message));
    }

    private ObservableQueue getQueue(Workflow workflow, Task task) {
        String sinkValue = "" + task.getInputData().get("sink");
        int indx = sinkValue.indexOf(':');
        if (indx != -1) {
            sinkValue = sinkValue.substring(0, indx);
        }
        Sink sink = null;
        try {
            sink = Sink.valueOf(sinkValue);
        } catch (Exception e) {
            logger.error(e.getMessage(), e);
        }
        if (sink == null) {
            task.setStatus(Status.FAILED);
            task.setReasonForIncompletion("Invalid sink specified: " + sinkValue);
            return null;
        }

        String event = null;
        if (sink == Sink.conductor) {
            String cq = workflow.getWorkflowType() + ":" + task.getReferenceTaskName();
            event = "conductor:" + cq;
        } else if (sink == Sink.sqs) {
            event = "" + task.getInputData().get("sink");
        }

        task.getOutputData().put("event_produced", event);

        try {
            return EventQueues.getQueue(event, true);
        } catch (Exception e) {
            logger.error(e.getMessage(), e);
            task.setStatus(Status.FAILED);
            task.setReasonForIncompletion("Error when trying to access the specified queue/topic: " + sinkValue
                    + ", error: " + e.getMessage());
            return null;
        }

    }
}