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 import java.io.IOException;
54
55 import javax.management.Attribute;
56 import javax.management.JMException;
57 import javax.management.MBeanException;
58 import javax.management.MBeanInfo;
59 import javax.management.MBeanOperationInfo;
60 import javax.management.MBeanParameterInfo;
61 import javax.management.ObjectName;
62 import javax.management.MBeanServerConnection;
63
64
65 /***
66 * A GroovyObject facade for an underlying MBean which acts like a normal
67 * groovy object but which is actually implemented via
68 * an underlying JMX MBean.
69 * Properties and normal method invocations
70 * delegate to the MBeanServer to the actual MBean.
71 *
72 * @author <a href="mailto:james@coredevelopers.net">James Strachan</a>
73 * @version $Revision: 1.4 $
74 */
75 public class GroovyMBean extends GroovyObjectSupport {
76
77 private MBeanServerConnection server;
78 private ObjectName name;
79 private MBeanInfo beanInfo;
80 private Map operations = new HashMap();
81
82 public GroovyMBean(MBeanServerConnection server, ObjectName name) throws JMException, IOException {
83 this.server = server;
84 this.name = name;
85 this.beanInfo = server.getMBeanInfo(name);
86
87 MBeanOperationInfo[] operationInfos = beanInfo.getOperations();
88 for (int i = 0; i < operationInfos.length; i++ ) {
89 MBeanOperationInfo info = operationInfos[i];
90 operations.put(info.getName(), createSignature(info));
91 }
92 }
93
94 public MBeanServerConnection server() {
95 return server;
96 }
97
98 public ObjectName name() {
99 return name;
100 }
101
102 public MBeanInfo info() {
103 return beanInfo;
104 }
105
106 public Object getProperty(String property) {
107 try {
108 return server.getAttribute(name, property);
109 }
110 catch (MBeanException e) {
111 throw new GroovyRuntimeException("Could not access property: " + property + ". Reason: " + e, e.getTargetException());
112 }
113 catch (Exception e) {
114 throw new GroovyRuntimeException("Could not access property: " + property + ". Reason: " + e, e);
115 }
116 }
117
118 public void setProperty(String property, Object value) {
119 try {
120 server.setAttribute(name, new Attribute(property, value));
121 }
122 catch (MBeanException e) {
123 throw new GroovyRuntimeException("Could not set property: " + property + ". Reason: " + e, e.getTargetException());
124 }
125 catch (Exception e) {
126 throw new GroovyRuntimeException("Could not set property: " + property + ". Reason: " + e, e);
127 }
128 }
129
130 public Object invokeMethod(String method, Object arguments) {
131 String[] signature = (String[]) operations.get(method);
132 if (signature != null) {
133 Object[] argArray = null;
134 if (arguments instanceof Object[]) {
135 argArray = (Object[]) arguments;
136 }
137 else {
138 argArray = new Object[] { arguments };
139 }
140 try {
141 return server.invoke(name, method, argArray, signature);
142 }
143 catch (MBeanException e) {
144 throw new GroovyRuntimeException("Could not invoke method: " + method + ". Reason: " + e, e.getTargetException());
145 }
146 catch (Exception e) {
147 throw new GroovyRuntimeException("Could not invoke method: " + method + ". Reason: " + e, e);
148 }
149 }
150 else {
151 return super.invokeMethod(method, arguments);
152 }
153 }
154
155 protected String[] createSignature(MBeanOperationInfo info) {
156 MBeanParameterInfo[] params = info.getSignature();
157 String[] answer = new String[params.length];
158 for (int i = 0; i < params.length; i++) {
159 answer[i] = params[i].getType();
160 }
161 return answer;
162 }
163 }