/*
* 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.webapp.servlet;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.joda.time.format.DateTimeFormat;
import azkaban.executor.ExecutableFlow;
import azkaban.executor.ExecutorManagerAdapter;
import azkaban.executor.ExecutorManagerException;
import azkaban.project.Project;
import azkaban.project.ProjectManager;
import azkaban.webapp.AzkabanWebServer;
import azkaban.webapp.session.Session;
public class HistoryServlet extends LoginAbstractAzkabanServlet {
private static final long serialVersionUID = 1L;
private ExecutorManagerAdapter executorManager;
private ProjectManager projectManager;
private ExecutorVMHelper vmHelper;
@Override
public void init(ServletConfig config) throws ServletException {
super.init(config);
AzkabanWebServer server = (AzkabanWebServer)getApplication();
executorManager = server.getExecutorManager();
projectManager = server.getProjectManager();
vmHelper = new ExecutorVMHelper();
}
@Override
protected void handleGet(HttpServletRequest req, HttpServletResponse resp,
Session session) throws ServletException, IOException {
if (hasParam(req, "ajax")) {
handleAJAXAction(req, resp, session);
}
else if (hasParam(req, "days")) {
handleHistoryDayPage(req, resp, session);
}
else if (hasParam(req, "timeline")) {
handleHistoryTimelinePage(req, resp, session);
}
else {
handleHistoryPage(req, resp, session);
}
}
private void handleAJAXAction(HttpServletRequest req, HttpServletResponse resp, Session session) throws ServletException, IOException {
HashMap<String, Object> ret = new HashMap<String, Object>();
String ajaxName = getParam(req, "ajax");
if (ajaxName.equals("fetch")) {
fetchHistoryData(req, resp, ret);
}
if (ret != null) {
this.writeJSON(resp, ret);
}
}
private void fetchHistoryData(HttpServletRequest req, HttpServletResponse resp, HashMap<String, Object> ret) throws ServletException {
// long start = getLongParam(req, "start");
// long end = getLongParam(req, "end");
//
// ret.put("start", start);
// ret.put("end", end);
//
// List<ExecutionReference> refs = executorManager.getFlowHistory(start, end);
// ArrayList<Object> refList = new ArrayList<Object>();
// for (ExecutionReference ref: refs) {
//
// HashMap<String,Object> refObj = new HashMap<String,Object>();
// refObj.put("execId", ref.getExecId());
// refObj.put("start", ref.getStartTime());
// refObj.put("end", ref.getEndTime());
// refObj.put("status", ref.getStatus().toString());
//
// refList.add(refObj);
// }
//
// ret.put("data", refList);
}
private void handleHistoryPage(HttpServletRequest req, HttpServletResponse resp, Session session) throws ServletException {
Page page = newPage(req, resp, session, "azkaban/webapp/servlet/velocity/historypage.vm");
int pageNum = getIntParam(req, "page", 1);
int pageSize = getIntParam(req, "size", 16);
page.add("vmutils", vmHelper);
if (pageNum < 0) {
pageNum = 1;
}
List<ExecutableFlow> history = null;
if(hasParam(req, "advfilter")) {
String projContain = getParam(req, "projcontain");
String flowContain = getParam(req, "flowcontain");
String userContain = getParam(req, "usercontain");
int status = getIntParam(req, "status");
String begin = getParam(req, "begin");
long beginTime = begin == "" ? -1 : DateTimeFormat.forPattern("MM/dd/yyyy-HH:mm").parseDateTime(begin).getMillis();
String end = getParam(req, "end");
long endTime = end == "" ? -1 : DateTimeFormat.forPattern("MM/dd/yyyy-HH:mm").parseDateTime(end).getMillis();
try {
history = executorManager.getExecutableFlows(projContain, flowContain, userContain, status, beginTime, endTime, (pageNum - 1)*pageSize, pageSize);
} catch (ExecutorManagerException e) {
page.add("error", e.getMessage());
}
}
else if(hasParam(req, "search")) {
String searchTerm = getParam(req, "searchterm");
try {
history = executorManager.getExecutableFlows(searchTerm, (pageNum - 1)*pageSize, pageSize);
} catch (ExecutorManagerException e) {
// TODO Auto-generated catch block
page.add("error", e.getMessage());
}
}
else {
try {
history = executorManager.getExecutableFlows((pageNum - 1)*pageSize, pageSize);
} catch (ExecutorManagerException e) {
e.printStackTrace();
}
}
page.add("flowHistory", history);
page.add("size", pageSize);
page.add("page", pageNum);
//keep the search terms so that we can navigate to later pages
if(hasParam(req, "searchterm") && !getParam(req, "searchterm").equals("")) {
page.add("search", "true");
page.add("search_term", getParam(req, "searchterm"));
}
if(hasParam(req, "advfilter")) {
page.add("advfilter", "true");
page.add("projcontain", getParam(req, "projcontain"));
page.add("flowcontain", getParam(req, "flowcontain"));
page.add("usercontain", getParam(req, "usercontain"));
page.add("status", getIntParam(req, "status"));
page.add("begin", getParam(req, "begin"));
page.add("end", getParam(req, "end"));
}
// else {
// page.add("search_term", "");
// }
if (pageNum == 1) {
page.add("previous", new PageSelection(1, pageSize, true, false));
}
else {
page.add("previous", new PageSelection(pageNum-1, pageSize, false, false));
}
page.add("next", new PageSelection(pageNum + 1, pageSize, false, false));
// Now for the 5 other values.
int pageStartValue = 1;
if (pageNum > 3) {
pageStartValue = pageNum - 2;
}
page.add("page1", new PageSelection(pageStartValue, pageSize, false, pageStartValue == pageNum));
pageStartValue++;
page.add("page2", new PageSelection(pageStartValue, pageSize, false, pageStartValue == pageNum));
pageStartValue++;
page.add("page3", new PageSelection(pageStartValue, pageSize, false, pageStartValue == pageNum));
pageStartValue++;
page.add("page4", new PageSelection(pageStartValue, pageSize, false, pageStartValue == pageNum));
pageStartValue++;
page.add("page5", new PageSelection(pageStartValue, pageSize, false, pageStartValue == pageNum));
pageStartValue++;
page.render();
}
private void handleHistoryTimelinePage(HttpServletRequest req, HttpServletResponse resp, Session session) {
// Page page = newPage(req, resp, session, "azkaban/webapp/servlet/velocity/historytimelinepage.vm");
// long currentTime = System.currentTimeMillis();
// long begin = getLongParam(req, "begin", currentTime - 86400000);
// long end = getLongParam(req, "end", currentTime);
//
// page.add("begin", begin);
// page.add("end", end);
//
// List<ExecutionReference> refs = executorManager.getFlowHistory(begin, end);
// ArrayList<Object> refList = new ArrayList<Object>();
// for (ExecutionReference ref: refs) {
//
// HashMap<String,Object> refObj = new HashMap<String,Object>();
// refObj.put("execId", ref.getExecId());
// refObj.put("start", ref.getStartTime());
// refObj.put("end", ref.getEndTime());
// refObj.put("status", ref.getStatus().toString());
//
// refList.add(refObj);
// }
//
// page.add("data", JSONUtils.toJSON(refList));
// page.render();
}
private void handleHistoryDayPage(HttpServletRequest req, HttpServletResponse resp, Session session) {
// Page page = newPage(req, resp, session, "azkaban/webapp/servlet/velocity/historydaypage.vm");
// long currentTime = System.currentTimeMillis();
// long begin = getLongParam(req, "begin", currentTime - 86400000);
// long end = getLongParam(req, "end", currentTime);
//
// page.add("begin", begin);
// page.add("end", end);
//
// List<ExecutionReference> refs = executorManager.getFlowHistory(begin, end);
// ArrayList<Object> refList = new ArrayList<Object>();
// for (ExecutionReference ref: refs) {
//
// HashMap<String,Object> refObj = new HashMap<String,Object>();
// refObj.put("execId", ref.getExecId());
// refObj.put("start", ref.getStartTime());
// refObj.put("end", ref.getEndTime());
// refObj.put("status", ref.getStatus().toString());
//
// refList.add(refObj);
// }
//
// page.add("data", JSONUtils.toJSON(refList));
// page.render();
}
public class PageSelection {
private int page;
private int size;
private boolean disabled;
private boolean selected;
public PageSelection(int page, int size, boolean disabled, boolean selected) {
this.page = page;
this.size = size;
this.disabled = disabled;
this.setSelected(selected);
}
public int getPage() {
return page;
}
public int getSize() {
return size;
}
public boolean getDisabled() {
return disabled;
}
public boolean isSelected() {
return selected;
}
public void setSelected(boolean selected) {
this.selected = selected;
}
}
@Override
protected void handlePost(HttpServletRequest req, HttpServletResponse resp,
Session session) throws ServletException, IOException {
// TODO Auto-generated method stub
}
public class ExecutorVMHelper {
public String getProjectName(int id) {
Project project = projectManager.getProject(id);
if (project == null) {
return String.valueOf(id);
}
return project.getName();
}
}
}