001/**
002 * Licensed to the Apache Software Foundation (ASF) under one
003 * or more contributor license agreements.  See the NOTICE file
004 * distributed with this work for additional information
005 * regarding copyright ownership.  The ASF licenses this file
006 * to you under the Apache License, Version 2.0 (the
007 * "License"); you may not use this file except in compliance
008 * with the License.  You may obtain a copy of the License at
009 *
010 *      http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018package org.apache.oozie.servlet;
019
020import java.io.IOException;
021import java.io.UnsupportedEncodingException;
022import java.net.URLDecoder;
023import java.text.ParseException;
024import java.util.Arrays;
025import java.util.HashSet;
026import java.util.List;
027import java.util.Map;
028import java.util.Set;
029
030import javax.servlet.ServletException;
031import javax.servlet.http.HttpServletRequest;
032import javax.servlet.http.HttpServletResponse;
033import org.apache.oozie.ErrorCode;
034import org.apache.oozie.XException;
035import org.apache.oozie.client.OozieClient;
036import org.apache.oozie.client.rest.RestConstants;
037import org.apache.oozie.command.CommandException;
038import org.apache.oozie.executor.jpa.sla.SLASummaryGetForFilterJPAExecutor;
039import org.apache.oozie.executor.jpa.sla.SLASummaryGetForFilterJPAExecutor.SLASummaryFilter;
040import org.apache.oozie.service.JPAService;
041import org.apache.oozie.service.Services;
042import org.apache.oozie.sla.SLASummaryBean;
043import org.apache.oozie.util.DateUtils;
044import org.apache.oozie.util.XLog;
045import org.json.simple.JSONObject;
046
047@SuppressWarnings("serial")
048public class V2SLAServlet extends SLAServlet {
049
050    private static final String INSTRUMENTATION_NAME = "v2sla";
051    private static final JsonRestServlet.ResourceInfo RESOURCES_INFO[] = new JsonRestServlet.ResourceInfo[1];
052    private static final Set<String> SLA_FILTER_NAMES = new HashSet<String>();
053
054    static {
055        SLA_FILTER_NAMES.add(OozieClient.FILTER_SLA_ID);
056        SLA_FILTER_NAMES.add(OozieClient.FILTER_SLA_PARENT_ID);
057        SLA_FILTER_NAMES.add(OozieClient.FILTER_SLA_APPNAME);
058        SLA_FILTER_NAMES.add(OozieClient.FILTER_SLA_NOMINAL_START);
059        SLA_FILTER_NAMES.add(OozieClient.FILTER_SLA_NOMINAL_END);
060    }
061
062    static {
063        RESOURCES_INFO[0] = new JsonRestServlet.ResourceInfo("", Arrays.asList("GET"),
064                Arrays.asList(new JsonRestServlet.ParameterInfo(RestConstants.JOBS_FILTER_PARAM, String.class, false,
065                        Arrays.asList("GET"))));
066    }
067
068    public V2SLAServlet() {
069        super(INSTRUMENTATION_NAME, RESOURCES_INFO);
070    }
071
072    @Override
073    public void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
074
075        XLog.getLog(getClass()).debug("Got SLA GET request:" + request.getQueryString());
076        try {
077            stopCron();
078            JSONObject json = getSLASummaryList(request, response);
079            startCron();
080            if (json == null) {
081                response.setStatus(HttpServletResponse.SC_OK);
082            }
083            else {
084                sendJsonResponse(response, HttpServletResponse.SC_OK, json);
085            }
086        }
087        catch (CommandException ce) {
088            XLog.getLog(getClass()).error("Command exception ", ce);
089            throw new XServletException(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, ce);
090        }
091        catch (RuntimeException re) {
092            XLog.getLog(getClass()).error("Runtime error ", re);
093            throw new XServletException(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, ErrorCode.E0307, re.getMessage());
094        }
095    }
096
097    private JSONObject getSLASummaryList(HttpServletRequest request, HttpServletResponse response)
098            throws ServletException, CommandException {
099        String timeZoneId = request.getParameter(RestConstants.TIME_ZONE_PARAM) == null ? null : request
100                .getParameter(RestConstants.TIME_ZONE_PARAM);
101        String filterString = request.getParameter(RestConstants.JOBS_FILTER_PARAM);
102        String maxResults = request.getParameter(RestConstants.LEN_PARAM);
103        int numMaxResults = 1000; // Default
104
105        if (maxResults != null) {
106            numMaxResults = Integer.parseInt(maxResults);
107        }
108
109        if (filterString == null || filterString.equals("")) {
110            throw new XServletException(HttpServletResponse.SC_BAD_REQUEST, ErrorCode.E0305,
111                    RestConstants.JOBS_FILTER_PARAM);
112        }
113
114        try {
115            Map<String, List<String>> filterList = parseFilter(URLDecoder.decode(filterString, "UTF-8"), SLA_FILTER_NAMES);
116            SLASummaryFilter filter = new SLASummaryFilter();
117            if (filterList.containsKey(OozieClient.FILTER_SLA_ID)) {
118                filter.setJobId(filterList.get(OozieClient.FILTER_SLA_ID).get(0));
119            }
120            if (filterList.containsKey(OozieClient.FILTER_SLA_PARENT_ID)) {
121                filter.setParentId(filterList.get(OozieClient.FILTER_SLA_PARENT_ID).get(0));
122            }
123            if (filterList.containsKey(OozieClient.FILTER_SLA_APPNAME)) {
124                filter.setAppName(filterList.get(OozieClient.FILTER_SLA_APPNAME).get(0));
125            }
126            if (filterList.containsKey(OozieClient.FILTER_SLA_NOMINAL_START)) {
127                filter.setNominalStart(DateUtils.parseDateUTC(filterList.get(OozieClient.FILTER_SLA_NOMINAL_START).get(0)));
128            }
129            if (filterList.containsKey(OozieClient.FILTER_SLA_NOMINAL_END)) {
130                filter.setNominalEnd(DateUtils.parseDateUTC(filterList.get(OozieClient.FILTER_SLA_NOMINAL_END).get(0)));
131            }
132
133            if (filter.getAppName() == null && filter.getJobId() == null && filter.getParentId() == null) {
134                throw new XServletException(HttpServletResponse.SC_BAD_REQUEST, ErrorCode.E0305,
135                        "At least one of the filter parameters - " + OozieClient.FILTER_SLA_ID + ","
136                                + OozieClient.FILTER_SLA_PARENT_ID + " or " + OozieClient.FILTER_SLA_APPNAME
137                                + " should be specified in the filter query parameter");
138            }
139
140            JPAService jpaService = Services.get().get(JPAService.class);
141            List<SLASummaryBean> slaSummaryList = null;
142            if (jpaService != null) {
143                slaSummaryList = jpaService.execute(new SLASummaryGetForFilterJPAExecutor(filter, numMaxResults));
144            }
145            else {
146                XLog.getLog(getClass()).error(ErrorCode.E0610);
147            }
148            return SLASummaryBean.toJSONObject(slaSummaryList, timeZoneId);
149        }
150        catch (XException ex) {
151            throw new CommandException(ex);
152        }
153        catch (UnsupportedEncodingException e) {
154            throw new XServletException(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, ErrorCode.E0307,
155                    "Unsupported Encoding", e);
156        }
157        catch (ParseException e) {
158            throw new XServletException(HttpServletResponse.SC_BAD_REQUEST, ErrorCode.E0303,
159                    filterString, e);
160        }
161
162    }
163
164}