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 */
018 package org.apache.oozie.client.rest;
019
020 import java.util.HashSet;
021 import java.util.List;
022 import java.util.Set;
023
024 import org.apache.oozie.BundleJobBean;
025 import org.apache.oozie.CoordinatorActionBean;
026 import org.apache.oozie.CoordinatorJobBean;
027 import org.apache.oozie.client.BulkResponse;
028 import org.json.simple.JSONArray;
029 import org.json.simple.JSONObject;
030
031 /**
032 * Server-side implementation class of the client interface BulkResponse
033 * Declares all the bulk request specific user parameters and handling as JSON object
034 */
035 public class BulkResponseImpl implements BulkResponse, JsonBean {
036 private BundleJobBean bundle;
037 private CoordinatorJobBean coordinator;
038 private CoordinatorActionBean action;
039
040 public static final String BULK_FILTER_BUNDLE_NAME = "bundle";
041 public static final String BULK_FILTER_COORD_NAME = "coordinators";
042 public static final String BULK_FILTER_LEVEL = "filterlevel";
043 public static final String BULK_FILTER_STATUS = "actionstatus";
044 public static final String BULK_FILTER_START_CREATED_EPOCH = "startcreatedtime";
045 public static final String BULK_FILTER_END_CREATED_EPOCH = "endcreatedtime";
046 public static final String BULK_FILTER_START_NOMINAL_EPOCH = "startscheduledtime";
047 public static final String BULK_FILTER_END_NOMINAL_EPOCH = "endscheduledtime";
048 public static final String BULK_DATE_FORMAT = "yyyy-MM-dd'T'HH:mm:SS'Z'";
049
050 public static final Set<String> BULK_FILTER_NAMES = new HashSet<String>();
051
052 static {
053
054 BULK_FILTER_NAMES.add(BulkResponseImpl.BULK_FILTER_BUNDLE_NAME);
055 BULK_FILTER_NAMES.add(BulkResponseImpl.BULK_FILTER_COORD_NAME);
056 BULK_FILTER_NAMES.add(BulkResponseImpl.BULK_FILTER_LEVEL);
057 BULK_FILTER_NAMES.add(BulkResponseImpl.BULK_FILTER_STATUS);
058 BULK_FILTER_NAMES.add(BulkResponseImpl.BULK_FILTER_START_CREATED_EPOCH);
059 BULK_FILTER_NAMES.add(BulkResponseImpl.BULK_FILTER_END_CREATED_EPOCH);
060 BULK_FILTER_NAMES.add(BulkResponseImpl.BULK_FILTER_START_NOMINAL_EPOCH);
061 BULK_FILTER_NAMES.add(BulkResponseImpl.BULK_FILTER_END_NOMINAL_EPOCH);
062
063 }
064
065 /**
066 * Construct JSON object using the bulk request object and the associated tags
067 */
068 public JSONObject toJSONObject() {
069 return toJSONObject("GMT");
070 }
071
072 /**
073 * Construct JSON object using the bulk request object and the associated tags
074 */
075 @SuppressWarnings("unchecked")
076 public JSONObject toJSONObject(String timeZoneId) {
077 JSONObject json = new JSONObject();
078
079 json.put(JsonTags.BULK_RESPONSE_BUNDLE, bundle.toJSONObject());
080 json.put(JsonTags.BULK_RESPONSE_COORDINATOR, coordinator.toJSONObject());
081 json.put(JsonTags.BULK_RESPONSE_ACTION, action.toJSONObject());
082
083 return json;
084 }
085
086 /* (non-Javadoc)
087 * @see org.apache.oozie.client.BulkResponse#getBundle()
088 */
089 @Override
090 public BundleJobBean getBundle() {
091 return bundle;
092 }
093
094 /* (non-Javadoc)
095 * @see org.apache.oozie.client.BulkResponse#getCoordinator()
096 */
097 @Override
098 public CoordinatorJobBean getCoordinator() {
099 return coordinator;
100 }
101
102 /* (non-Javadoc)
103 * @see org.apache.oozie.client.BulkResponse#getAction()
104 */
105 @Override
106 public CoordinatorActionBean getAction() {
107 return action;
108 }
109
110 /**
111 * Sets the bundle comprising this bulk response object
112 * @param BundleJobBean
113 */
114 public void setBundle(BundleJobBean bj) {
115 this.bundle = bj;
116 }
117
118 /**
119 * Sets the coordinator comprising this bulk response object
120 * @param CoordinatorJobBean
121 */
122 public void setCoordinator(CoordinatorJobBean cj) {
123 this.coordinator = cj;
124 }
125
126 /**
127 * Sets the coord action comprising this bulk response object
128 * @param CoordinatorActionBean
129 */
130 public void setAction(CoordinatorActionBean ca) {
131 this.action = ca;
132 }
133
134 /**
135 * Convert a nodes list into a JSONArray.
136 *
137 * @param actions nodes list.
138 * @param timeZoneId time zone to use for dates in the JSON array.
139 * @return the corresponding JSON array.
140 */
141 @SuppressWarnings("unchecked")
142 public static JSONArray toJSONArray(List<? extends BulkResponseImpl> responses, String timeZoneId) {
143 JSONArray array = new JSONArray();
144 for (BulkResponseImpl response : responses) {
145 array.add(response.toJSONObject(timeZoneId));
146 }
147 return array;
148 }
149 }