/**
* Copyright (c) 2010 Yahoo! Inc. All rights reserved.
* 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. See accompanying LICENSE file.
*/
package org.apache.oozie.servlet;
import java.io.IOException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.hadoop.conf.Configuration;
import org.apache.oozie.DagEngine;
import org.apache.oozie.DagEngineException;
import org.apache.oozie.client.rest.JsonBean;
import org.apache.oozie.service.DagEngineService;
import org.apache.oozie.service.Services;
import org.json.simple.JSONObject;
import org.apache.oozie.ErrorCode;
@SuppressWarnings("serial")
public class V0JobServlet extends BaseJobServlet {
private static final String INSTRUMENTATION_NAME = "v0job";
public V0JobServlet() {
super(INSTRUMENTATION_NAME);
}
/*
* v0 service method to start a job
*/
@Override
protected void startJob(HttpServletRequest request, HttpServletResponse response) throws XServletException,
IOException {
DagEngine dagEngine = Services.get().get(DagEngineService.class).getDagEngine(getUser(request),
getAuthToken(request));
String jobId = getResourceName(request);
try {
dagEngine.start(jobId);
}
catch (DagEngineException ex) {
throw new XServletException(HttpServletResponse.SC_BAD_REQUEST, ex);
}
}
/*
* v0 service method to resume a job
*/
@Override
protected void resumeJob(HttpServletRequest request, HttpServletResponse response) throws XServletException,
IOException {
DagEngine dagEngine = Services.get().get(DagEngineService.class).getDagEngine(getUser(request),
getAuthToken(request));
String jobId = getResourceName(request);
try {
dagEngine.resume(jobId);
}
catch (DagEngineException ex) {
throw new XServletException(HttpServletResponse.SC_BAD_REQUEST, ex);
}
}
/*
* v0 service method to suspend a job
*/
@Override
protected void suspendJob(HttpServletRequest request, HttpServletResponse response) throws XServletException,
IOException {
DagEngine dagEngine = Services.get().get(DagEngineService.class).getDagEngine(getUser(request),
getAuthToken(request));
String jobId = getResourceName(request);
try {
dagEngine.suspend(jobId);
}
catch (DagEngineException ex) {
throw new XServletException(HttpServletResponse.SC_BAD_REQUEST, ex);
}
}
/*
* v0 service method to kill a job
*/
@Override
protected void killJob(HttpServletRequest request, HttpServletResponse response) throws XServletException,
IOException {
DagEngine dagEngine = Services.get().get(DagEngineService.class).getDagEngine(getUser(request),
getAuthToken(request));
String jobId = getResourceName(request);
try {
dagEngine.kill(jobId);
}
catch (DagEngineException ex) {
throw new XServletException(HttpServletResponse.SC_BAD_REQUEST, ex);
}
}
/*
* v0 service method to change a job
*/
protected void changeJob(HttpServletRequest request, HttpServletResponse response) throws XServletException,
IOException {
// This code should not be reached. But if it happens somehow, we throw
// bad request exception.
throw new XServletException(HttpServletResponse.SC_BAD_REQUEST, ErrorCode.E1014);
}
/*
* v0 service method to reRun a job
*/
@Override
protected JSONObject reRunJob(HttpServletRequest request, HttpServletResponse response, Configuration conf)
throws XServletException, IOException {
DagEngine dagEngine = Services.get().get(DagEngineService.class).getDagEngine(getUser(request),
getAuthToken(request));
String jobId = getResourceName(request);
try {
dagEngine.reRun(jobId, conf);
}
catch (DagEngineException ex) {
throw new XServletException(HttpServletResponse.SC_BAD_REQUEST, ex);
}
return null;
}
/*
* v0 service method to get a job in JsonBean representation
*/
@Override
protected JsonBean getJob(HttpServletRequest request, HttpServletResponse response) throws XServletException,
IOException {
DagEngine dagEngine = Services.get().get(DagEngineService.class).getDagEngine(getUser(request),
getAuthToken(request));
JsonBean jobBean = null;
String jobId = getResourceName(request);
try {
jobBean = (JsonBean) dagEngine.getJob(jobId);
}
catch (DagEngineException ex) {
throw new XServletException(HttpServletResponse.SC_BAD_REQUEST, ex);
}
return jobBean;
}
/*
* v0 service method to get a job definition in String format
*/
@Override
protected String getJobDefinition(HttpServletRequest request, HttpServletResponse response)
throws XServletException, IOException {
DagEngine dagEngine = Services.get().get(DagEngineService.class).getDagEngine(getUser(request),
getAuthToken(request));
String wfDefinition = null;
String jobId = getResourceName(request);
try {
wfDefinition = dagEngine.getDefinition(jobId);
}
catch (DagEngineException ex) {
throw new XServletException(HttpServletResponse.SC_BAD_REQUEST, ex);
}
return wfDefinition;
}
/*
* v0 service method to stream a job log into response object
*/
@Override
protected void streamJobLog(HttpServletRequest request, HttpServletResponse response) throws XServletException,
IOException {
DagEngine dagEngine = Services.get().get(DagEngineService.class).getDagEngine(getUser(request),
getAuthToken(request));
String jobId = getResourceName(request);
try {
dagEngine.streamLog(jobId, response.getWriter());
}
catch (DagEngineException ex) {
throw new XServletException(HttpServletResponse.SC_BAD_REQUEST, ex);
}
}
}