circuitbreaker/jmx/JMXWrapper.java

package dk.thoerup.circuitbreaker.jmx;

import java.lang.management.ManagementFactory;
import java.util.Set;
import java.util.TreeSet;
import java.util.logging.Level;
import java.util.logging.Logger;

import javax.management.MBeanServer;
import javax.management.ObjectName;

import dk.thoerup.circuitbreaker.CircuitBreaker;
import dk.thoerup.circuitbreaker.CircuitBreakerManager;

public class JMXWrapper {
        static Logger logger = Logger.getLogger(JMXWrapper.class.getName());
        
        private static Set<ObjectName> objectNames = new TreeSet<ObjectName>();
        
        
        public static ObjectName registerCircuitBreaker(CircuitBreaker cb) {
                Object jmxBean;
                String className;
                

                jmxBean = new JmxCircuitBreaker(cb);
                className = "CircuitBreaker";                   
                
                
                String name = cb.getName();
                
                try {
                        
                        ObjectName beanName = new ObjectName("dk.thoerup.circuitbreaker:type=" + className + ",name=" + name);
                        
                        MBeanServer server = ManagementFactory.getPlatformMBeanServer();
                        if (server.isRegistered(beanName))
                                server.unregisterMBean(beanName);
                        server.registerMBean(jmxBean, beanName);
                        objectNames.add(beanName);
                        return beanName;
                        
                } catch (Exception e) {
                        logger.log(Level.SEVERE, "Mbean failure",e );
                        return null;
                }
        }
        
        public static void registerAllCircuitBreakers() {
                CircuitBreakerManager mgr = CircuitBreakerManager.getManager();
                
                for (String cbName : mgr.getCircuitBreakerNames()) {
                        CircuitBreaker cb = mgr.getCircuitBreaker(cbName);
                        registerCircuitBreaker(cb);
                }
        }
        
        public static void unregisterAllCircuitBreakers() {
                MBeanServer server = ManagementFactory.getPlatformMBeanServer();
                
                for (ObjectName objName : objectNames) {
                        if (server.isRegistered(objName)) {
                                try {
                                        server.unregisterMBean(objName);
                                } catch (Exception e) {
                                        //At this point we don't care why it wasn't in the system - all that matter is that it's gone
                                }
                        }
                }
        }
}
1	torben	467	package dk.thoerup.circuitbreaker.jmx;
2	torben	402
3			import java.lang.management.ManagementFactory;
4	torben	408	import java.util.Set;
5			import java.util.TreeSet;
6	torben	402	import java.util.logging.Level;
7			import java.util.logging.Logger;
8
9			import javax.management.MBeanServer;
10			import javax.management.ObjectName;
11
12	torben	467	import dk.thoerup.circuitbreaker.CircuitBreaker;
13			import dk.thoerup.circuitbreaker.CircuitBreakerManager;
14	torben	402
15			public class JMXWrapper {
16			static Logger logger = Logger.getLogger(JMXWrapper.class.getName());
17
18	torben	408	private static Set<ObjectName> objectNames = new TreeSet<ObjectName>();
19
20
21	torben	1154	public static ObjectName registerCircuitBreaker(CircuitBreaker cb) {
22	torben	402	Object jmxBean;
23			String className;
24
25	torben	2570
26			jmxBean = new JmxCircuitBreaker(cb);
27			className = "CircuitBreaker";
28	torben	402
29	torben	2570
30	torben	402	String name = cb.getName();
31	torben	408
32	torben	402	try {
33
34	torben	405	ObjectName beanName = new ObjectName("dk.thoerup.circuitbreaker:type=" + className + ",name=" + name);
35	torben	402
36			MBeanServer server = ManagementFactory.getPlatformMBeanServer();
37			if (server.isRegistered(beanName))
38			server.unregisterMBean(beanName);
39			server.registerMBean(jmxBean, beanName);
40	torben	408	objectNames.add(beanName);
41	torben	1154	return beanName;
42	torben	402
43			} catch (Exception e) {
44			logger.log(Level.SEVERE, "Mbean failure",e );
45	torben	1154	return null;
46	torben	402	}
47	torben	406	}
48
49			public static void registerAllCircuitBreakers() {
50			CircuitBreakerManager mgr = CircuitBreakerManager.getManager();
51	torben	402
52	torben	406	for (String cbName : mgr.getCircuitBreakerNames()) {
53			CircuitBreaker cb = mgr.getCircuitBreaker(cbName);
54			registerCircuitBreaker(cb);
55			}
56	torben	402	}
57	torben	408
58			public static void unregisterAllCircuitBreakers() {
59			MBeanServer server = ManagementFactory.getPlatformMBeanServer();
60
61			for (ObjectName objName : objectNames) {
62			if (server.isRegistered(objName)) {
63			try {
64			server.unregisterMBean(objName);
65			} catch (Exception e) {
66			//At this point we don't care why it wasn't in the system - all that matter is that it's gone
67			}
68			}
69			}
70			}
71	torben	402	}