author | hb |
Fri, 05 Jan 2018 13:42:53 +0530 | |
branch | jmx-rest-api |
changeset 56007 | d6cbabcaf518 |
parent 56006 | 352a4f213fc6 |
permissions | -rw-r--r-- |
55997 | 1 |
/* |
2 |
* Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved. |
|
3 |
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
|
4 |
* |
|
5 |
* This code is free software; you can redistribute it and/or modify it |
|
6 |
* under the terms of the GNU General Public License version 2 only, as |
|
7 |
* published by the Free Software Foundation. Oracle designates this |
|
8 |
* particular file as subject to the "Classpath" exception as provided |
|
9 |
* by Oracle in the LICENSE file that accompanied this code. |
|
10 |
* |
|
11 |
* This code is distributed in the hope that it will be useful, but WITHOUT |
|
12 |
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
|
13 |
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
|
14 |
* version 2 for more details (a copy is included in the LICENSE file that |
|
15 |
* accompanied this code). |
|
16 |
* |
|
17 |
* You should have received a copy of the GNU General Public License version |
|
18 |
* 2 along with this work; if not, write to the Free Software Foundation, |
|
19 |
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
|
20 |
* |
|
21 |
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
|
22 |
* or visit www.oracle.com if you need additional information or have any |
|
23 |
* questions. |
|
24 |
*/ |
|
25 |
||
55994 | 26 |
package com.oracle.jmx.remote.rest.http; |
27 |
||
56006 | 28 |
import com.oracle.jmx.remote.rest.json.JSONArray; |
55994 | 29 |
import com.oracle.jmx.remote.rest.json.JSONElement; |
30 |
import com.oracle.jmx.remote.rest.json.JSONObject; |
|
56006 | 31 |
import com.oracle.jmx.remote.rest.json.JSONPrimitive; |
55996 | 32 |
import com.oracle.jmx.remote.rest.json.parser.JSONParser; |
33 |
import com.oracle.jmx.remote.rest.json.parser.ParseException; |
|
55994 | 34 |
import com.oracle.jmx.remote.rest.mapper.JSONMapper; |
35 |
import com.oracle.jmx.remote.rest.mapper.JSONMappingException; |
|
36 |
import com.oracle.jmx.remote.rest.mapper.JSONMappingFactory; |
|
37 |
import com.sun.net.httpserver.HttpExchange; |
|
38 |
||
39 |
import javax.management.*; |
|
40 |
import javax.management.remote.rest.PlatformRestAdapter; |
|
41 |
import java.io.IOException; |
|
42 |
import java.io.UnsupportedEncodingException; |
|
56006 | 43 |
import java.net.URLDecoder; |
44 |
import java.nio.charset.StandardCharsets; |
|
55994 | 45 |
import java.util.*; |
46 |
import java.util.concurrent.ConcurrentHashMap; |
|
47 |
import java.util.concurrent.CopyOnWriteArrayList; |
|
48 |
import java.util.regex.Matcher; |
|
49 |
import java.util.regex.Pattern; |
|
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
50 |
import java.util.stream.Collectors; |
55994 | 51 |
|
52 |
public class MBeanCollectionResource implements RestResource, NotificationListener { |
|
53 |
||
54 |
private List<ObjectName> allowedMbeans; |
|
55 |
private final MBeanServer mBeanServer; |
|
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
56 |
private final Map<ObjectName, MBeanResource> mBeanResourceMap = new ConcurrentHashMap<>(); |
55994 | 57 |
private static final int pageSize = 10; |
55996 | 58 |
private static final String pathPrefix = "^/?jmx/servers/[a-zA-Z0-9\\-\\.]+/mbeans"; |
55994 | 59 |
|
56000 | 60 |
// Only MXBean or any other MBean that uses types |
61 |
// that have a valid mapper functions |
|
55994 | 62 |
private boolean isMBeanAllowed(ObjectName objName) { |
63 |
try { |
|
64 |
MBeanInfo mInfo = mBeanServer.getMBeanInfo(objName); |
|
56000 | 65 |
|
66 |
// Return true for MXbean |
|
67 |
Descriptor desc = mInfo.getDescriptor(); |
|
68 |
String isMxBean = (String) desc.getFieldValue("mxbean"); |
|
56001 | 69 |
if (isMxBean != null && isMxBean.equalsIgnoreCase("true")) |
56000 | 70 |
return true; |
71 |
||
72 |
// Check attribute types |
|
55994 | 73 |
MBeanAttributeInfo[] attrsInfo = mInfo.getAttributes(); |
74 |
for (MBeanAttributeInfo attrInfo : attrsInfo) { |
|
75 |
String type = attrInfo.getType(); |
|
76 |
if (!JSONMappingFactory.INSTANCE.isTypeMapped(type)) { |
|
77 |
return false; |
|
78 |
} |
|
79 |
} |
|
56000 | 80 |
|
81 |
// Check operation parameters and return types |
|
55994 | 82 |
MBeanOperationInfo[] operations = mInfo.getOperations(); |
83 |
for (MBeanOperationInfo opInfo : operations) { |
|
84 |
MBeanParameterInfo[] signature = opInfo.getSignature(); |
|
85 |
for (MBeanParameterInfo sig : signature) { |
|
86 |
if (!JSONMappingFactory.INSTANCE.isTypeMapped(sig.getType())) { |
|
87 |
return false; |
|
88 |
} |
|
89 |
} |
|
90 |
if (!JSONMappingFactory.INSTANCE.isTypeMapped(opInfo.getReturnType())) { |
|
91 |
return false; |
|
92 |
} |
|
93 |
} |
|
94 |
return true; |
|
56000 | 95 |
} catch (InstanceNotFoundException | IntrospectionException | |
96 |
ReflectionException | ClassNotFoundException ex) { |
|
55994 | 97 |
ex.printStackTrace(); |
98 |
return false; |
|
99 |
} |
|
100 |
} |
|
101 |
||
102 |
private void introspectMBeanTypes(MBeanServer server) { |
|
103 |
if (allowedMbeans.isEmpty()) { |
|
104 |
Set<ObjectInstance> allMBeans = server.queryMBeans(null, null); // get all Mbeans |
|
105 |
allMBeans.stream().filter((objIns) -> (isMBeanAllowed(objIns.getObjectName()))) |
|
106 |
.forEachOrdered(objIns -> allowedMbeans.add(objIns.getObjectName())); |
|
107 |
} |
|
108 |
} |
|
109 |
||
110 |
@Override |
|
111 |
public void handleNotification(Notification notification, Object handback) { |
|
112 |
try { |
|
113 |
MBeanServerNotification mbs = (MBeanServerNotification) notification; |
|
114 |
if (MBeanServerNotification.REGISTRATION_NOTIFICATION.equals(mbs.getType())) { |
|
115 |
ObjectName mBeanName = mbs.getMBeanName(); |
|
116 |
if (isMBeanAllowed(mBeanName)) { |
|
117 |
allowedMbeans.add(mBeanName); |
|
118 |
} |
|
119 |
} else if (MBeanServerNotification.UNREGISTRATION_NOTIFICATION.equals(mbs.getType())) { |
|
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
120 |
if (allowedMbeans.contains(mbs.getMBeanName())) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
121 |
allowedMbeans.remove(mbs.getMBeanName()); |
55994 | 122 |
} |
123 |
} |
|
124 |
} catch (Exception e) { |
|
125 |
} |
|
126 |
} |
|
127 |
||
128 |
public MBeanCollectionResource(MBeanServer mBeanServer) { |
|
129 |
this.mBeanServer = mBeanServer; |
|
130 |
allowedMbeans = new ArrayList<>(); |
|
131 |
introspectMBeanTypes(mBeanServer); |
|
132 |
allowedMbeans = new CopyOnWriteArrayList<>(allowedMbeans); |
|
56000 | 133 |
|
134 |
// Create a REST handler for each MBean |
|
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
135 |
allowedMbeans.forEach(objectName -> mBeanResourceMap.put(objectName, |
55994 | 136 |
new MBeanResource(mBeanServer, objectName))); |
137 |
} |
|
138 |
||
139 |
@Override |
|
140 |
public void handle(HttpExchange exchange) throws IOException { |
|
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
141 |
String path = URLDecoder.decode(exchange.getRequestURI().getPath(), StandardCharsets.UTF_8.name()); |
56006 | 142 |
|
55995 | 143 |
if (path.matches(pathPrefix + "/?$")) { |
55994 | 144 |
RestResource.super.handle(exchange); |
55995 | 145 |
} else if (path.matches(pathPrefix + "/[^/]+/?.*")) { |
55994 | 146 |
// Extract mbean name |
147 |
// Forward the request to its corresponding rest resource |
|
55995 | 148 |
Pattern mbeans = Pattern.compile(pathPrefix + "/"); |
55994 | 149 |
Matcher matcher = mbeans.matcher(path); |
150 |
||
151 |
if (matcher.find()) { |
|
152 |
String ss = path.substring(matcher.end()); |
|
153 |
String mBeanName = ss; |
|
154 |
if (ss.indexOf('/') != -1) { |
|
155 |
mBeanName = ss.substring(0, ss.indexOf('/')); |
|
156 |
} |
|
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
157 |
try { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
158 |
MBeanResource mBeanResource = mBeanResourceMap.get(new ObjectName(mBeanName)); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
159 |
if (mBeanResource == null) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
160 |
HttpUtil.sendResponse(exchange, HttpResponse.REQUEST_NOT_FOUND); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
161 |
return; |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
162 |
} |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
163 |
mBeanResource.handle(exchange); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
164 |
} catch (MalformedObjectNameException e) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
165 |
HttpUtil.sendResponse(exchange, HttpResponse.BAD_REQUEST); |
55994 | 166 |
} |
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
167 |
|
55994 | 168 |
} |
169 |
} |
|
170 |
} |
|
171 |
||
172 |
@Override |
|
173 |
public HttpResponse doGet(HttpExchange exchange) { |
|
174 |
try { |
|
56006 | 175 |
final String path = PlatformRestAdapter.getDomain() |
176 |
+ URLDecoder.decode(exchange.getRequestURI().getPath(), StandardCharsets.UTF_8.displayName()) |
|
177 |
.replaceAll("/$", ""); |
|
56000 | 178 |
List<ObjectName> filteredMBeans = allowedMbeans; |
55994 | 179 |
Map<String, String> queryMap = HttpUtil.getGetRequestQueryMap(exchange); |
56002 | 180 |
String query = exchange.getRequestURI().getQuery(); |
56006 | 181 |
if (query != null && queryMap.isEmpty()) { |
56002 | 182 |
return new HttpResponse(HttpResponse.BAD_REQUEST, |
56006 | 183 |
"Invalid query params : Allowed query keys [objectname,page]"); |
184 |
} else if (query != null && !queryMap.isEmpty()) { |
|
56002 | 185 |
Map<String, String> newMap = new HashMap<>(queryMap); |
56006 | 186 |
newMap.remove("objectname"); |
56002 | 187 |
newMap.remove("page"); |
56006 | 188 |
if (!newMap.isEmpty()) { // Invalid query params |
56002 | 189 |
return new HttpResponse(HttpResponse.BAD_REQUEST, |
56006 | 190 |
"Invalid query params : Allowed query keys [objectname,page]"); |
56002 | 191 |
} |
192 |
} |
|
56006 | 193 |
if (queryMap.containsKey("objectname")) { // Filter based on ObjectName query |
56000 | 194 |
Set<ObjectName> queryMBeans = mBeanServer |
56006 | 195 |
.queryNames(new ObjectName(queryMap.get("objectname")), null); |
56000 | 196 |
queryMBeans.retainAll(allowedMbeans); // Intersection of two lists |
197 |
filteredMBeans = new ArrayList<>(queryMBeans); |
|
55994 | 198 |
} |
199 |
||
56000 | 200 |
JSONObject _links = HttpUtil.getPaginationLinks(exchange, filteredMBeans, pageSize); |
56002 | 201 |
List<ObjectName> mbeanPage = HttpUtil.filterByPage(exchange, filteredMBeans, pageSize); |
56000 | 202 |
|
55994 | 203 |
List<Map<String, String>> items = new ArrayList<>(filteredMBeans.size()); |
56006 | 204 |
for (ObjectName objectName : mbeanPage) { |
205 |
Map<String, String> item = new LinkedHashMap<>(); |
|
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
206 |
item.put("name", objectName.getCanonicalName()); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
207 |
MBeanResource mBeanResource = mBeanResourceMap.get(objectName); |
56006 | 208 |
try { |
209 |
JSONObject mBeanInfo = mBeanResource.getMBeanInfo(mBeanServer, objectName); |
|
210 |
JSONElement element = mBeanInfo.get("descriptor"); |
|
211 |
if (element != null) { |
|
212 |
JSONElement element1 = ((JSONObject) element).get("interfaceClassName"); |
|
213 |
if (element1 != null) { |
|
214 |
String intfName = (String) ((JSONPrimitive) element1).getValue(); |
|
215 |
item.put("interfaceClassName", intfName); |
|
216 |
} |
|
217 |
} |
|
218 |
element = mBeanInfo.get("className"); |
|
219 |
if (element != null) { |
|
220 |
String className = (String) ((JSONPrimitive) element).getValue(); |
|
221 |
item.put("className", className); |
|
222 |
} |
|
223 |
element = mBeanInfo.get("description"); |
|
224 |
if (element != null) { |
|
225 |
String description = (String) ((JSONPrimitive) element).getValue(); |
|
226 |
item.put("description", description); |
|
227 |
} |
|
228 |
element = mBeanInfo.get("attributeInfo"); |
|
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
229 |
if (element != null) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
230 |
item.put("attributeCount", ((JSONArray) element).size() + ""); |
56006 | 231 |
} |
232 |
element = mBeanInfo.get("operationInfo"); |
|
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
233 |
if (element != null) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
234 |
item.put("operationCount", ((JSONArray) element).size() + ""); |
56006 | 235 |
} |
236 |
||
237 |
} catch (InstanceNotFoundException | IntrospectionException | ReflectionException e) { |
|
238 |
} |
|
239 |
||
55994 | 240 |
String href = path + "/" + objectName.toString(); |
241 |
href = HttpUtil.escapeUrl(href); |
|
242 |
item.put("href", href); |
|
243 |
items.add(item); |
|
56006 | 244 |
String info = HttpUtil.escapeUrl(href + "/info"); |
245 |
item.put("info", info); |
|
246 |
} |
|
55994 | 247 |
|
248 |
Map<String, String> properties = new HashMap<>(); |
|
249 |
||
56002 | 250 |
properties.put("mbeanCount", Integer.toString(filteredMBeans.size())); |
55994 | 251 |
|
252 |
JSONMapper typeMapper1 = JSONMappingFactory.INSTANCE.getTypeMapper(items); |
|
253 |
JSONMapper typeMapper2 = JSONMappingFactory.INSTANCE.getTypeMapper(properties); |
|
254 |
||
255 |
JSONElement linkElem = typeMapper1.toJsonValue(items); |
|
256 |
JSONElement propElem = typeMapper2.toJsonValue(properties); |
|
257 |
JSONObject jobj = new JSONObject(); |
|
56000 | 258 |
|
259 |
jobj.putAll((JSONObject) propElem); |
|
260 |
jobj.put("mbeans", linkElem); |
|
261 |
||
55996 | 262 |
if (_links != null && !_links.isEmpty()) { |
263 |
jobj.put("_links", _links); |
|
55994 | 264 |
} |
56000 | 265 |
return new HttpResponse(jobj.toJsonString()); |
55994 | 266 |
} catch (JSONMappingException e) { |
56000 | 267 |
return HttpResponse.SERVER_ERROR; |
55994 | 268 |
} catch (UnsupportedEncodingException e) { |
56000 | 269 |
return HttpResponse.BAD_REQUEST; |
55994 | 270 |
} catch (MalformedObjectNameException e) { |
271 |
return new HttpResponse(HttpResponse.BAD_REQUEST, "Invalid query string"); |
|
272 |
} |
|
273 |
} |
|
274 |
||
275 |
@Override |
|
276 |
public HttpResponse doPost(HttpExchange exchange) { |
|
55996 | 277 |
try { |
56006 | 278 |
String path = URLDecoder.decode(exchange.getRequestURI().getPath(),StandardCharsets.UTF_8.displayName()); |
279 |
String reqBody = null; |
|
55996 | 280 |
if (path.matches(pathPrefix + "/?$")) { // POST to current URL |
281 |
reqBody = HttpUtil.readRequestBody(exchange); |
|
282 |
if (reqBody == null || reqBody.isEmpty()) { // No Parameters |
|
283 |
return HttpResponse.BAD_REQUEST; |
|
284 |
} |
|
285 |
||
286 |
JSONParser parser = new JSONParser(reqBody); |
|
287 |
JSONElement jsonElement = parser.parse(); |
|
288 |
if (!(jsonElement instanceof JSONObject)) { |
|
289 |
return new HttpResponse(HttpResponse.BAD_REQUEST, |
|
290 |
"Invalid parameters : [" + reqBody + "]"); |
|
291 |
} |
|
292 |
||
293 |
JSONObject jsonObject = (JSONObject) jsonElement; |
|
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
294 |
JSONObject normalizedObject = new JSONObject(jsonObject); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
295 |
|
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
296 |
// Normalize the input MBean names |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
297 |
for (String objectNameString : jsonObject.keySet()) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
298 |
if (!objectNameString.startsWith("?")) { // Ignore object name patterns |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
299 |
JSONElement element = jsonObject.get(objectNameString); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
300 |
normalizedObject.remove(objectNameString); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
301 |
normalizedObject.put(new ObjectName(objectNameString).getCanonicalName(), element); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
302 |
} |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
303 |
} |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
304 |
|
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
305 |
jsonObject.clear(); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
306 |
jsonObject = normalizedObject; |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
307 |
|
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
308 |
Set<String> objectNamePatterns = jsonObject.keySet() |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
309 |
.stream() |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
310 |
.filter(a -> a.startsWith("?")) |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
311 |
.collect(Collectors.toSet()); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
312 |
|
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
313 |
if (!objectNamePatterns.isEmpty()) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
314 |
for (String pattern : objectNamePatterns) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
315 |
Set<ObjectName> queryMBeans = mBeanServer |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
316 |
.queryNames(new ObjectName(pattern.substring(1)), null); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
317 |
queryMBeans.retainAll(allowedMbeans); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
318 |
JSONElement patternNode = jsonObject.get(pattern); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
319 |
jsonObject.remove(pattern); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
320 |
for (ObjectName queryMBean : queryMBeans) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
321 |
String name = queryMBean.getCanonicalName(); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
322 |
if (jsonObject.containsKey(name)) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
323 |
JSONObject obj = new JSONObject(); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
324 |
obj.put(name, patternNode); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
325 |
deepMerge(jsonObject, obj); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
326 |
} else { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
327 |
jsonObject.put(name, patternNode); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
328 |
} |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
329 |
} |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
330 |
} |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
331 |
} |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
332 |
|
55996 | 333 |
JSONObject result = new JSONObject(); |
334 |
for (String mBeanName : jsonObject.keySet()) { |
|
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
335 |
MBeanResource mBeanResource = mBeanResourceMap.get(new ObjectName(mBeanName)); |
56000 | 336 |
if (mBeanResource != null) { |
55996 | 337 |
JSONElement element = jsonObject.get(mBeanName); |
338 |
if (element instanceof JSONObject) { |
|
56000 | 339 |
JSONElement res = mBeanResource.handleBulkRequest |
340 |
((JSONObject) element); |
|
55996 | 341 |
result.put(mBeanName, res); |
342 |
} else { |
|
343 |
result.put(mBeanName, "Invalid input"); |
|
344 |
} |
|
56000 | 345 |
} else { |
346 |
result.put(mBeanName, "Invalid MBean"); |
|
55996 | 347 |
} |
348 |
} |
|
56000 | 349 |
return new HttpResponse(result.toJsonString()); |
55996 | 350 |
} else { |
351 |
return HttpResponse.METHOD_NOT_ALLOWED; |
|
352 |
} |
|
353 |
} catch (ParseException e) { |
|
354 |
return new HttpResponse(HttpResponse.BAD_REQUEST, "Invalid JSON String for request body"); |
|
355 |
} catch (IOException e) { |
|
356 |
return HttpResponse.BAD_REQUEST; |
|
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
357 |
} catch (MalformedObjectNameException e) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
358 |
return new HttpResponse(HttpResponse.BAD_REQUEST, "Invalid query string"); |
55996 | 359 |
} |
55994 | 360 |
} |
56007
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
361 |
|
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
362 |
private JSONObject deepMerge(JSONObject jsonObject1, JSONObject jsonObject2) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
363 |
for (String key : jsonObject2.keySet()) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
364 |
if (jsonObject1.containsKey(key)) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
365 |
if (jsonObject2.get(key) instanceof JSONObject && jsonObject1.get(key) instanceof JSONObject) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
366 |
JSONObject jobj1 = (JSONObject) jsonObject1.get(key); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
367 |
JSONObject jobj2 = (JSONObject) jsonObject2.get(key); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
368 |
jsonObject1.put(key, deepMerge(jobj1, jobj2)); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
369 |
} else if (jsonObject2.get(key) instanceof JSONArray && jsonObject1.get(key) instanceof JSONArray) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
370 |
JSONArray array1 = (JSONArray) jsonObject1.get(key); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
371 |
JSONArray array2 = (JSONArray) jsonObject2.get(key); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
372 |
for (JSONElement each : array2) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
373 |
if (!array1.contains(each)) { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
374 |
array1.add(each); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
375 |
} |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
376 |
} |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
377 |
} else { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
378 |
JSONArray array = new JSONArray(); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
379 |
array.add(jsonObject1.get(key)); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
380 |
array.add(jsonObject2.get(key)); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
381 |
jsonObject1.put(key, array); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
382 |
} |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
383 |
} else { |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
384 |
jsonObject1.put(key, jsonObject2.get(key)); |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
385 |
} |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
386 |
} |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
387 |
return jsonObject1; |
d6cbabcaf518
1. MBeanCollection Bulk operation via objectname query
hb
parents:
56006
diff
changeset
|
388 |
} |
55994 | 389 |
} |