1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46 package groovy.util;
47
48 import groovy.lang.GroovyObjectSupport;
49 import groovy.lang.GroovyRuntimeException;
50
51 import java.util.HashMap;
52 import java.util.Map;
53
54 import javax.management.Attribute;
55 import javax.management.JMException;
56 import javax.management.MBeanException;
57 import javax.management.MBeanInfo;
58 import javax.management.MBeanOperationInfo;
59 import javax.management.MBeanParameterInfo;
60 import javax.management.MBeanServer;
61 import javax.management.ObjectName;
62
63
64 /***
65 * A GroovyObject facade for an underlying MBean which acts like a normal
66 * groovy object but which is actually implemented via
67 * an underlying JMX MBean.
68 * Properties and normal method invocations
69 * delegate to the MBeanServer to the actual MBean.
70 *
71 * @author <a href="mailto:james@coredevelopers.net">James Strachan</a>
72 * @version $Revision: 1.2 $
73 */
74 public class GroovyMBean extends GroovyObjectSupport {
75
76 private MBeanServer server;
77 private ObjectName name;
78 private MBeanInfo beanInfo;
79 private Map operations = new HashMap();
80
81 public GroovyMBean(MBeanServer server, ObjectName name) throws JMException {
82 this.server = server;
83 this.name = name;
84 this.beanInfo = server.getMBeanInfo(name);
85
86 MBeanOperationInfo[] operationInfos = beanInfo.getOperations();
87 for (int i = 0; i < operationInfos.length; i++ ) {
88 MBeanOperationInfo info = operationInfos[i];
89 operations.put(info.getName(), createSignature(info));
90 }
91 }
92
93 public MBeanServer server() {
94 return server;
95 }
96
97 public ObjectName name() {
98 return name;
99 }
100
101 public MBeanInfo info() {
102 return beanInfo;
103 }
104
105 public Object getProperty(String property) {
106 try {
107 return server.getAttribute(name, property);
108 }
109 catch (MBeanException e) {
110 throw new GroovyRuntimeException("Could not access property: " + property + ". Reason: " + e, e.getTargetException());
111 }
112 catch (Exception e) {
113 throw new GroovyRuntimeException("Could not access property: " + property + ". Reason: " + e, e);
114 }
115 }
116
117 public void setProperty(String property, Object value) {
118 try {
119 server.setAttribute(name, new Attribute(property, value));
120 }
121 catch (MBeanException e) {
122 throw new GroovyRuntimeException("Could not set property: " + property + ". Reason: " + e, e.getTargetException());
123 }
124 catch (Exception e) {
125 throw new GroovyRuntimeException("Could not set property: " + property + ". Reason: " + e, e);
126 }
127 }
128
129 public Object invokeMethod(String method, Object arguments) {
130 String[] signature = (String[]) operations.get(method);
131 if (signature != null) {
132 Object[] argArray = null;
133 if (arguments instanceof Object[]) {
134 argArray = (Object[]) arguments;
135 }
136 else {
137 argArray = new Object[] { arguments };
138 }
139 try {
140 return server.invoke(name, method, argArray, signature);
141 }
142 catch (MBeanException e) {
143 throw new GroovyRuntimeException("Could not invoke method: " + method + ". Reason: " + e, e.getTargetException());
144 }
145 catch (Exception e) {
146 throw new GroovyRuntimeException("Could not invoke method: " + method + ". Reason: " + e, e);
147 }
148 }
149 else {
150 return super.invokeMethod(method, arguments);
151 }
152 }
153
154 protected String[] createSignature(MBeanOperationInfo info) {
155 MBeanParameterInfo[] params = info.getSignature();
156 String[] answer = new String[params.length];
157 for (int i = 0; i < params.length; i++) {
158 answer[i] = params[i].getType();
159 }
160 return answer;
161 }
162 }