001 /**
002 * Copyright (c) 2010 Yahoo! Inc. All rights reserved.
003 * Licensed under the Apache License, Version 2.0 (the "License");
004 * you may not use this file except in compliance with the License.
005 * You may obtain a copy of the License at
006 *
007 * http://www.apache.org/licenses/LICENSE-2.0
008 *
009 * Unless required by applicable law or agreed to in writing, software
010 * distributed under the License is distributed on an "AS IS" BASIS,
011 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
012 * See the License for the specific language governing permissions and
013 * limitations under the License. See accompanying LICENSE file.
014 */
015 package org.apache.oozie.servlet;
016
017 import java.util.Arrays;
018 import java.util.Collections;
019 import java.util.List;
020
021 import javax.servlet.http.HttpServletRequest;
022 import javax.servlet.http.HttpServletResponse;
023
024 import org.apache.oozie.ErrorCode;
025 import org.apache.oozie.client.OozieClient.SYSTEM_MODE;
026 import org.apache.oozie.client.rest.JsonTags;
027 import org.apache.oozie.client.rest.RestConstants;
028 import org.apache.oozie.service.CallableQueueService;
029 import org.apache.oozie.service.Services;
030 import org.json.simple.JSONArray;
031 import org.json.simple.JSONObject;
032
033 public class V1AdminServlet extends BaseAdminServlet {
034
035 private static final long serialVersionUID = 1L;
036 private static final String INSTRUMENTATION_NAME = "v1admin";
037 private static final ResourceInfo RESOURCES_INFO[] = new ResourceInfo[7];
038
039 static {
040 RESOURCES_INFO[0] = new ResourceInfo(RestConstants.ADMIN_STATUS_RESOURCE, Arrays.asList("PUT", "GET"),
041 Arrays.asList(new ParameterInfo(RestConstants.ADMIN_SYSTEM_MODE_PARAM, String.class, true,
042 Arrays.asList("PUT"))));
043 RESOURCES_INFO[1] = new ResourceInfo(RestConstants.ADMIN_OS_ENV_RESOURCE, Arrays.asList("GET"),
044 Collections.EMPTY_LIST);
045 RESOURCES_INFO[2] = new ResourceInfo(RestConstants.ADMIN_JAVA_SYS_PROPS_RESOURCE, Arrays.asList("GET"),
046 Collections.EMPTY_LIST);
047 RESOURCES_INFO[3] = new ResourceInfo(RestConstants.ADMIN_CONFIG_RESOURCE, Arrays.asList("GET"),
048 Collections.EMPTY_LIST);
049 RESOURCES_INFO[4] = new ResourceInfo(RestConstants.ADMIN_INSTRUMENTATION_RESOURCE, Arrays.asList("GET"),
050 Collections.EMPTY_LIST);
051 RESOURCES_INFO[5] = new ResourceInfo(RestConstants.ADMIN_BUILD_VERSION_RESOURCE, Arrays.asList("GET"),
052 Collections.EMPTY_LIST);
053 RESOURCES_INFO[6] = new ResourceInfo(RestConstants.ADMIN_QUEUE_DUMP_RESOURCE, Arrays.asList("GET"),
054 Collections.EMPTY_LIST);
055 }
056
057 public V1AdminServlet() {
058 super(INSTRUMENTATION_NAME, RESOURCES_INFO);
059 modeTag = RestConstants.ADMIN_SYSTEM_MODE_PARAM;
060 }
061
062 /*
063 * (non-Javadoc)
064 *
065 * @see
066 * org.apache.oozie.servlet.BaseAdminServlet#populateOozieMode(org.json.
067 * simple.JSONObject)
068 */
069 @SuppressWarnings("unchecked")
070 @Override
071 protected void populateOozieMode(JSONObject json) {
072 json.put(JsonTags.OOZIE_SYSTEM_MODE, Services.get().getSystemMode().toString());
073 }
074
075 /*
076 * (non-Javadoc)
077 *
078 * @see
079 * org.apache.oozie.servlet.BaseAdminServlet#setOozieMode(javax.servlet.
080 * http.HttpServletRequest, javax.servlet.http.HttpServletResponse,
081 * java.lang.String)
082 */
083 @Override
084 protected void setOozieMode(HttpServletRequest request,
085 HttpServletResponse response, String resourceName)
086 throws XServletException {
087 if (resourceName.equals(RestConstants.ADMIN_STATUS_RESOURCE)) {
088 SYSTEM_MODE sysMode = SYSTEM_MODE.valueOf(request.getParameter(modeTag));
089 Services.get().setSystemMode(sysMode);
090 response.setStatus(HttpServletResponse.SC_OK);
091 }
092 else {
093 throw new XServletException(HttpServletResponse.SC_BAD_REQUEST,
094 ErrorCode.E0301, resourceName);
095 }
096 }
097
098 /**
099 * Get a json array of queue dump
100 *
101 * @param JSONObject the result json object that contains a JSONArray for the callable dump
102 *
103 * @see
104 * org.apache.oozie.servlet.BaseAdminServlet#getQueueDump(org.json.simple
105 * .JSONObject)
106 */
107 @SuppressWarnings("unchecked")
108 @Override
109 protected void getQueueDump(JSONObject json) throws XServletException {
110 List<String> list = Services.get().get(CallableQueueService.class).getQueueDump();
111 JSONArray array = new JSONArray();
112 for (String str: list) {
113 JSONObject jObject = new JSONObject();
114 jObject.put(JsonTags.CALLABLE_DUMP, str);
115 array.add(jObject);
116 }
117 json.put(JsonTags.QUEUE_DUMP, array);
118 }
119
120 }