/*
* Copyright 2012 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.executor.mail;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import azkaban.executor.ExecutableFlow;
import azkaban.executor.ExecutionOptions;
import azkaban.executor.ExecutionOptions.FailureAction;
import azkaban.utils.EmailMessage;
import azkaban.utils.Emailer;
import azkaban.utils.Utils;
public class DefaultMailCreator implements MailCreator {
public static final String DEFAULT_MAIL_CREATOR = "default";
private static HashMap<String, MailCreator> registeredCreators = new HashMap<>();
private static MailCreator defaultCreator;
private static final DateFormat DATE_FORMATTER = new SimpleDateFormat(
"yyyy/MM/dd HH:mm:ss z");
public static void registerCreator(String name, MailCreator creator) {
registeredCreators.put(name, creator);
}
public static MailCreator getCreator(String name) {
MailCreator creator = registeredCreators.get(name);
if (creator == null) {
creator = defaultCreator;
}
return creator;
}
static {
defaultCreator = new DefaultMailCreator();
registerCreator(DEFAULT_MAIL_CREATOR, defaultCreator);
}
@Override
public boolean createFirstErrorMessage(ExecutableFlow flow,
EmailMessage message, String azkabanName, String scheme,
String clientHostname, String clientPortNumber, String... vars) {
ExecutionOptions option = flow.getExecutionOptions();
List<String> emailList = option.getFailureEmails();
int execId = flow.getExecutionId();
if (emailList != null && !emailList.isEmpty()) {
message.addAllToAddress(emailList);
message.setMimeType("text/html");
message.setSubject("Flow '" + flow.getFlowId() + "' has encountered a failure on "
+ azkabanName);
message.println("<h2 style=\"color:#FF0000\"> Execution '"
+ flow.getExecutionId() + "' of flow '" + flow.getFlowId() + "' of project '"
+ flow.getProjectName() + "' has encountered a failure on " + azkabanName + "</h2>");
if (option.getFailureAction() == FailureAction.CANCEL_ALL) {
message
.println("This flow is set to cancel all currently running jobs.");
} else if (option.getFailureAction() == FailureAction.FINISH_ALL_POSSIBLE) {
message
.println("This flow is set to complete all jobs that aren't blocked by the failure.");
} else {
message
.println("This flow is set to complete all currently running jobs before stopping.");
}
message.println("<table>");
message.println("<tr><td>Start Time</td><td>"
+ convertMSToString(flow.getStartTime()) + "</td></tr>");
message.println("<tr><td>End Time</td><td>"
+ convertMSToString(flow.getEndTime()) + "</td></tr>");
message.println("<tr><td>Duration</td><td>"
+ Utils.formatDuration(flow.getStartTime(), flow.getEndTime())
+ "</td></tr>");
message.println("<tr><td>Status</td><td>" + flow.getStatus() + "</td></tr>");
message.println("</table>");
message.println("");
String executionUrl =
scheme + "://" + clientHostname + ":" + clientPortNumber + "/"
+ "executor?" + "execid=" + execId;
message.println("<a href=\"" + executionUrl + "\">" + flow.getFlowId()
+ " Execution Link</a>");
message.println("");
message.println("<h3>Reason</h3>");
List<String> failedJobs = Emailer.findFailedJobs(flow);
message.println("<ul>");
for (String jobId : failedJobs) {
message.println("<li><a href=\"" + executionUrl + "&job=" + jobId
+ "\">Failed job '" + jobId + "' Link</a></li>");
}
message.println("</ul>");
return true;
}
return false;
}
@Override
public boolean createErrorEmail(ExecutableFlow flow, EmailMessage message,
String azkabanName, String scheme, String clientHostname,
String clientPortNumber, String... vars) {
ExecutionOptions option = flow.getExecutionOptions();
List<String> emailList = option.getFailureEmails();
int execId = flow.getExecutionId();
if (emailList != null && !emailList.isEmpty()) {
message.addAllToAddress(emailList);
message.setMimeType("text/html");
message.setSubject("Flow '" + flow.getFlowId() + "' has failed on "
+ azkabanName);
message.println("<h2 style=\"color:#FF0000\"> Execution '" + execId
+ "' of flow '" + flow.getFlowId() + "' of project '"
+ flow.getProjectName() + "' has failed on " + azkabanName + "</h2>");
message.println("<table>");
message.println("<tr><td>Start Time</td><td>"
+ convertMSToString(flow.getStartTime()) + "</td></tr>");
message.println("<tr><td>End Time</td><td>"
+ convertMSToString(flow.getEndTime()) + "</td></tr>");
message.println("<tr><td>Duration</td><td>"
+ Utils.formatDuration(flow.getStartTime(), flow.getEndTime())
+ "</td></tr>");
message.println("<tr><td>Status</td><td>" + flow.getStatus() + "</td></tr>");
message.println("</table>");
message.println("");
String executionUrl =
scheme + "://" + clientHostname + ":" + clientPortNumber + "/"
+ "executor?" + "execid=" + execId;
message.println("<a href=\"" + executionUrl + "\">" + flow.getFlowId()
+ " Execution Link</a>");
message.println("");
message.println("<h3>Reason</h3>");
List<String> failedJobs = Emailer.findFailedJobs(flow);
message.println("<ul>");
for (String jobId : failedJobs) {
message.println("<li><a href=\"" + executionUrl + "&job=" + jobId
+ "\">Failed job '" + jobId + "' Link</a></li>");
}
for (String reasons : vars) {
message.println("<li>" + reasons + "</li>");
}
message.println("</ul>");
return true;
}
return false;
}
@Override
public boolean createSuccessEmail(ExecutableFlow flow, EmailMessage message,
String azkabanName, String scheme, String clientHostname,
String clientPortNumber, String... vars) {
ExecutionOptions option = flow.getExecutionOptions();
List<String> emailList = option.getSuccessEmails();
int execId = flow.getExecutionId();
if (emailList != null && !emailList.isEmpty()) {
message.addAllToAddress(emailList);
message.setMimeType("text/html");
message.setSubject("Flow '" + flow.getFlowId() + "' has succeeded on "
+ azkabanName);
message.println("<h2> Execution '" + flow.getExecutionId()
+ "' of flow '" + flow.getFlowId() + "' of project '"
+ flow.getProjectName() + "' has succeeded on " + azkabanName + "</h2>");
message.println("<table>");
message.println("<tr><td>Start Time</td><td>"
+ convertMSToString(flow.getStartTime()) + "</td></tr>");
message.println("<tr><td>End Time</td><td>"
+ convertMSToString(flow.getEndTime()) + "</td></tr>");
message.println("<tr><td>Duration</td><td>"
+ Utils.formatDuration(flow.getStartTime(), flow.getEndTime())
+ "</td></tr>");
message.println("<tr><td>Status</td><td>" + flow.getStatus() + "</td></tr>");
message.println("</table>");
message.println("");
String executionUrl =
scheme + "://" + clientHostname + ":" + clientPortNumber + "/"
+ "executor?" + "execid=" + execId;
message.println("<a href=\"" + executionUrl + "\">" + flow.getFlowId()
+ " Execution Link</a>");
return true;
}
return false;
}
private static String convertMSToString(long timeInMS) {
if (timeInMS < 0) {
return "N/A";
} else {
return DATE_FORMATTER.format(new Date(timeInMS));
}
}
}