FELIX-2349: JMX notifications
git-svn-id: https://svn.apache.org/repos/asf/felix/trunk@952712 13f79535-47bb-0310-9956-ffa450edef68
diff --git a/karaf/management/pom.xml b/karaf/management/pom.xml
index 0a6ffef..5c62e99 100644
--- a/karaf/management/pom.xml
+++ b/karaf/management/pom.xml
@@ -58,6 +58,17 @@
<artifactId>commons-logging</artifactId>
<scope>test</scope>
</dependency>
+ <dependency>
+ <groupId>org.easymock</groupId>
+ <artifactId>easymock</artifactId>
+ <scope>test</scope>
+ </dependency>
+ <dependency>
+ <groupId>org.easymock</groupId>
+ <artifactId>easymockclassextension</artifactId>
+ <version>${easymock.version}</version>
+ <scope>test</scope>
+ </dependency>
</dependencies>
<build>
diff --git a/karaf/management/src/main/java/org/apache/felix/karaf/management/MBeanRegistrer.java b/karaf/management/src/main/java/org/apache/felix/karaf/management/MBeanRegistrer.java
new file mode 100644
index 0000000..10645c6
--- /dev/null
+++ b/karaf/management/src/main/java/org/apache/felix/karaf/management/MBeanRegistrer.java
@@ -0,0 +1,80 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.felix.karaf.management;
+
+import java.util.Map;
+
+import javax.management.JMException;
+import javax.management.MBeanServer;
+import javax.management.ObjectName;
+
+public class MBeanRegistrer {
+
+ private MBeanServer mbeanServer;
+
+ private Map<Object, String> mbeans;
+
+ public void setMbeans(Map<Object, String> mbeans) {
+ this.mbeans = mbeans;
+ }
+
+ public void registerMBeanServer(MBeanServer mbeanServer) throws JMException {
+ if (this.mbeanServer != mbeanServer) {
+ unregisterMBeans();
+ }
+ this.mbeanServer = mbeanServer;
+ registerMBeans();
+ }
+
+ public void unregisterMBeanServer(MBeanServer mbeanServer) throws JMException {
+ unregisterMBeans();
+ this.mbeanServer = null;
+ }
+
+ public void init() throws Exception {
+ registerMBeans();
+ }
+
+ protected void registerMBeans() throws JMException {
+ if (mbeanServer != null && mbeans != null) {
+ for (Map.Entry<Object, String> entry : mbeans.entrySet()) {
+ String value = parseProperty(entry.getValue());
+ mbeanServer.registerMBean(entry.getKey(), new ObjectName(value));
+ }
+ }
+ }
+
+ protected void unregisterMBeans() throws JMException {
+ if (mbeanServer != null && mbeans != null) {
+ for (Map.Entry<Object, String> entry : mbeans.entrySet()) {
+ String value = parseProperty(entry.getValue());
+ mbeanServer.unregisterMBean(new ObjectName(value));
+ }
+ }
+ }
+
+ protected String parseProperty(String raw) {
+ if (raw.indexOf("${") > -1 && raw.indexOf("}", raw.indexOf("${")) > -1) {
+ String var = raw.substring(raw.indexOf("${") + 2, raw.indexOf("}"));
+ String val = System.getProperty(var);
+ if (val != null) {
+ raw = raw.replace("${" + var + "}", val);
+ }
+ }
+ return raw;
+ }
+}
diff --git a/karaf/management/src/test/java/org/apache/felix/karaf/management/MBeanRegistererTest.java b/karaf/management/src/test/java/org/apache/felix/karaf/management/MBeanRegistererTest.java
new file mode 100644
index 0000000..dc95bdc
--- /dev/null
+++ b/karaf/management/src/test/java/org/apache/felix/karaf/management/MBeanRegistererTest.java
@@ -0,0 +1,67 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.felix.karaf.management;
+
+import java.util.Map;
+import java.util.HashMap;
+
+import javax.management.MBeanServer;
+import javax.management.ObjectName;
+
+import static org.easymock.classextension.EasyMock.*;
+import org.easymock.classextension.IMocksControl;
+
+import junit.framework.TestCase;
+import org.junit.Assert;
+
+public class MBeanRegistererTest extends TestCase {
+ public void testRegistration() throws Exception {
+ IMocksControl control = createNiceControl();
+ MBeanServer mbeanServer = control.createMock(MBeanServer.class);
+ String rawName = "org.apache.felix.karaf:type=admin,name=${karaf.name}";
+ AdminServiceMBean mbean = control.createMock(AdminServiceMBean.class);
+ ObjectName name = new ObjectName("org.apache.felix.karaf:type=admin,name=foobar");
+ expect(mbeanServer.registerMBean(isA(AdminServiceMBean.class), eq(name))).andReturn(null);
+ mbeanServer.unregisterMBean(eq(name));
+ expectLastCall();
+ control.replay();
+
+ String old = System.getProperty("karaf.name");
+ System.setProperty("karaf.name", "foobar");
+ MBeanRegistrer registerer = new MBeanRegistrer();
+ Map<Object, String> mbeans = new HashMap<Object, String>();
+ mbeans.put(mbean, rawName);
+ registerer.setMbeans(mbeans);
+ registerer.registerMBeanServer(mbeanServer);
+ registerer.unregisterMBeanServer(mbeanServer);
+
+ restoreProperties(old);
+ control.verify();
+ }
+
+ private void restoreProperties(String old) {
+ if (old != null) {
+ System.setProperty("karaf.name", old);
+ } else {
+ System.getProperties().remove("karaf.name");
+ }
+ }
+
+ public static interface AdminServiceMBean {
+ int createInstance(String name) throws Exception;
+ }
+}