001    /*
002     $Id: GroovyMBean.java,v 1.4 2005/06/10 09:55:30 cstein Exp $
003    
004     Copyright 2003 (C) James Strachan and Bob Mcwhirter. All Rights Reserved.
005    
006     Redistribution and use of this software and associated documentation
007     ("Software"), with or without modification, are permitted provided
008     that the following conditions are met:
009    
010     1. Redistributions of source code must retain copyright
011        statements and notices.  Redistributions must also contain a
012        copy of this document.
013    
014     2. Redistributions in binary form must reproduce the
015        above copyright notice, this list of conditions and the
016        following disclaimer in the documentation and/or other
017        materials provided with the distribution.
018    
019     3. The name "groovy" must not be used to endorse or promote
020        products derived from this Software without prior written
021        permission of The Codehaus.  For written permission,
022        please contact info@codehaus.org.
023    
024     4. Products derived from this Software may not be called "groovy"
025        nor may "groovy" appear in their names without prior written
026        permission of The Codehaus. "groovy" is a registered
027        trademark of The Codehaus.
028    
029     5. Due credit should be given to The Codehaus -
030        http://groovy.codehaus.org/
031    
032     THIS SOFTWARE IS PROVIDED BY THE CODEHAUS AND CONTRIBUTORS
033     ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT
034     NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
035     FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
036     THE CODEHAUS OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
037     INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
038     (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
039     SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
040     HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
041     STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
042     ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
043     OF THE POSSIBILITY OF SUCH DAMAGE.
044    
045     */
046    package groovy.util;
047    
048    import groovy.lang.GroovyObjectSupport;
049    import groovy.lang.GroovyRuntimeException;
050    
051    import java.util.HashMap;
052    import java.util.Map;
053    import java.io.IOException;
054    
055    import javax.management.Attribute;
056    import javax.management.JMException;
057    import javax.management.MBeanException;
058    import javax.management.MBeanInfo;
059    import javax.management.MBeanOperationInfo;
060    import javax.management.MBeanParameterInfo;
061    import javax.management.ObjectName;
062    import javax.management.MBeanServerConnection;
063    
064    
065    /**
066     * A GroovyObject facade for an underlying MBean which acts like a normal 
067     * groovy object but which is actually implemented via 
068     * an underlying JMX MBean. 
069     * Properties and normal method invocations
070     * delegate to the MBeanServer to the actual MBean.
071     * 
072     * @author <a href="mailto:james@coredevelopers.net">James Strachan</a>
073     * @version $Revision: 1.4 $
074     */
075    public class GroovyMBean extends GroovyObjectSupport {
076    
077        private MBeanServerConnection server;
078        private ObjectName name;
079        private MBeanInfo beanInfo;
080        private Map operations = new HashMap();
081    
082        public GroovyMBean(MBeanServerConnection server, ObjectName name) throws JMException, IOException {
083            this.server = server;
084            this.name = name;
085            this.beanInfo = server.getMBeanInfo(name);
086            
087            MBeanOperationInfo[] operationInfos = beanInfo.getOperations();
088            for (int i = 0; i < operationInfos.length; i++ ) {
089                MBeanOperationInfo info = operationInfos[i];
090                operations.put(info.getName(), createSignature(info));
091            }
092        }
093    
094        public MBeanServerConnection server() {
095            return server;
096        }
097        
098        public ObjectName name() {
099            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    }