azkaban.sla.SlaOption.java Source code

Introduction

Here is the source code for azkaban.sla.SlaOption.java

Source

/*
 * Copyright 2014 LinkedIn Corp.
 *
 * 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 azkaban.sla;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.joda.time.DateTime;
import org.joda.time.format.DateTimeFormat;
import org.joda.time.format.DateTimeFormatter;

import azkaban.executor.ExecutableFlow;

public class SlaOption {

    public static final String TYPE_FLOW_FINISH = "FlowFinish";
    public static final String TYPE_FLOW_SUCCEED = "FlowSucceed";
    public static final String TYPE_FLOW_PROGRESS = "FlowProgress";

    public static final String TYPE_JOB_FINISH = "JobFinish";
    public static final String TYPE_JOB_SUCCEED = "JobSucceed";
    public static final String TYPE_JOB_PROGRESS = "JobProgress";

    public static final String INFO_DURATION = "Duration";
    public static final String INFO_FLOW_NAME = "FlowName";
    public static final String INFO_JOB_NAME = "JobName";
    public static final String INFO_PROGRESS_PERCENT = "ProgressPercent";
    public static final String INFO_EMAIL_LIST = "EmailList";

    // always alert
    public static final String ALERT_TYPE = "SlaAlertType";
    public static final String ACTION_CANCEL_FLOW = "SlaCancelFlow";
    public static final String ACTION_ALERT = "SlaAlert";

    private String type;
    private Map<String, Object> info;
    private List<String> actions;

    private static DateTimeFormatter fmt = DateTimeFormat.forPattern("MM/dd, YYYY HH:mm");

    public SlaOption(String type, List<String> actions, Map<String, Object> info) {
        this.type = type;
        this.info = info;
        this.actions = actions;
    }

    public String getType() {
        return type;
    }

    public void setType(String type) {
        this.type = type;
    }

    public Map<String, Object> getInfo() {
        return info;
    }

    public void setInfo(Map<String, Object> info) {
        this.info = info;
    }

    public List<String> getActions() {
        return actions;
    }

    public void setActions(List<String> actions) {
        this.actions = actions;
    }

    public Map<String, Object> toObject() {
        HashMap<String, Object> slaObj = new HashMap<String, Object>();

        slaObj.put("type", type);
        slaObj.put("info", info);
        slaObj.put("actions", actions);

        return slaObj;
    }

    @SuppressWarnings("unchecked")
    public static SlaOption fromObject(Object object) {

        HashMap<String, Object> slaObj = (HashMap<String, Object>) object;

        String type = (String) slaObj.get("type");
        List<String> actions = (List<String>) slaObj.get("actions");
        Map<String, Object> info = (Map<String, Object>) slaObj.get("info");

        return new SlaOption(type, actions, info);
    }

    public Object toWebObject() {
        HashMap<String, Object> slaObj = new HashMap<String, Object>();

        if (type.equals(TYPE_FLOW_FINISH) || type.equals(TYPE_FLOW_SUCCEED)) {
            slaObj.put("id", "");
        } else {
            slaObj.put("id", info.get(INFO_JOB_NAME));
        }
        slaObj.put("duration", info.get(INFO_DURATION));
        if (type.equals(TYPE_FLOW_FINISH) || type.equals(TYPE_JOB_FINISH)) {
            slaObj.put("rule", "FINISH");
        } else {
            slaObj.put("rule", "SUCCESS");
        }
        List<String> actionsObj = new ArrayList<String>();
        for (String act : actions) {
            if (act.equals(ACTION_ALERT)) {
                actionsObj.add("EMAIL");
            } else {
                actionsObj.add("KILL");
            }
        }
        slaObj.put("actions", actionsObj);

        return slaObj;
    }

    @Override
    public String toString() {
        return "Sla of " + getType() + getInfo() + getActions();
    }

    public static String createSlaMessage(SlaOption slaOption, ExecutableFlow flow) {
        String type = slaOption.getType();
        int execId = flow.getExecutionId();
        if (type.equals(SlaOption.TYPE_FLOW_FINISH)) {
            String flowName = (String) slaOption.getInfo().get(SlaOption.INFO_FLOW_NAME);
            String duration = (String) slaOption.getInfo().get(SlaOption.INFO_DURATION);
            String basicinfo = "SLA Alert: Your flow " + flowName + " failed to FINISH within " + duration
                    + "</br>";
            String expected = "Here is details : </br>" + "Flow " + flowName + " in execution " + execId
                    + " is expected to FINISH within " + duration + " from "
                    + fmt.print(new DateTime(flow.getStartTime())) + "</br>";
            String actual = "Actual flow status is " + flow.getStatus();
            return basicinfo + expected + actual;
        } else if (type.equals(SlaOption.TYPE_FLOW_SUCCEED)) {
            String flowName = (String) slaOption.getInfo().get(SlaOption.INFO_FLOW_NAME);
            String duration = (String) slaOption.getInfo().get(SlaOption.INFO_DURATION);
            String basicinfo = "SLA Alert: Your flow " + flowName + " failed to SUCCEED within " + duration
                    + "</br>";
            String expected = "Here is details : </br>" + "Flow " + flowName + " in execution " + execId
                    + " expected to FINISH within " + duration + " from "
                    + fmt.print(new DateTime(flow.getStartTime())) + "</br>";
            String actual = "Actual flow status is " + flow.getStatus();
            return basicinfo + expected + actual;
        } else if (type.equals(SlaOption.TYPE_JOB_FINISH)) {
            String jobName = (String) slaOption.getInfo().get(SlaOption.INFO_JOB_NAME);
            String duration = (String) slaOption.getInfo().get(SlaOption.INFO_DURATION);
            return "SLA Alert: Your job " + jobName + " failed to FINISH within " + duration + " in execution "
                    + execId;
        } else if (type.equals(SlaOption.TYPE_JOB_SUCCEED)) {
            String jobName = (String) slaOption.getInfo().get(SlaOption.INFO_JOB_NAME);
            String duration = (String) slaOption.getInfo().get(SlaOption.INFO_DURATION);
            return "SLA Alert: Your job " + jobName + " failed to SUCCEED within " + duration + " in execution "
                    + execId;
        } else {
            return "Unrecognized SLA type " + type;
        }
    }

}