SlaOption.java

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>();

//		slaObj.put("type", type);
//		slaObj.put("info", info);
//		slaObj.put("actions", actions);
		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;
		}
	}


}