| /** |
| * 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.activemq.web; |
| |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.assertNull; |
| |
| import java.lang.reflect.Field; |
| import java.util.Collection; |
| import java.util.Set; |
| import java.util.concurrent.CountDownLatch; |
| import java.util.concurrent.ExecutorService; |
| import java.util.concurrent.Executors; |
| |
| import javax.management.ObjectName; |
| import javax.management.remote.JMXConnectorServer; |
| |
| import org.apache.activemq.broker.BrokerFactory; |
| import org.apache.activemq.broker.BrokerService; |
| import org.apache.activemq.broker.jmx.DestinationViewMBean; |
| import org.apache.activemq.broker.jmx.ManagementContext; |
| import org.apache.activemq.command.ActiveMQQueue; |
| import org.apache.activemq.util.TestUtils; |
| import org.apache.activemq.web.config.SystemPropertiesConfiguration; |
| import org.junit.After; |
| import org.junit.Before; |
| import org.junit.Test; |
| |
| /** |
| * @author <a href="http://www.christianposta.com/blog">Christian Posta</a> |
| * |
| * You can use this class to connect up to a running web console and run some queries. |
| * Used to work through https://issues.apache.org/jira/browse/AMQ-4272 but would be useful |
| * in any scenario where you need access to the underlying broker in the web-console to hack |
| * at it |
| * |
| */ |
| public class RemoteJMXBrokerTest { |
| |
| |
| private BrokerService brokerService; |
| |
| @Before |
| public void startUp() throws Exception { |
| brokerService = BrokerFactory.createBroker("broker:()/remoteBroker?useJmx=true"); |
| brokerService.getManagementContext().setConnectorPort(TestUtils.findOpenPort()); |
| brokerService.getManagementContext().setCreateConnector(true); |
| brokerService.start(); |
| brokerService.waitUntilStarted(); |
| |
| String jmxUri = getJmxUri(); |
| System.setProperty("webconsole.jmx.url", jmxUri); |
| |
| } |
| |
| @After |
| public void after() throws Exception { |
| brokerService.stop(); |
| brokerService.waitUntilStopped(); |
| } |
| |
| /** |
| * Test that we can query the remote broker... |
| * Specifically this tests that the domain and objectnames are correct (type and brokerName |
| * instead of Type and BrokerName, which they were) |
| * @throws Exception |
| */ |
| @Test |
| public void testConnectRemoteBrokerFacade() throws Exception { |
| RemoteJMXBrokerFacade brokerFacade = new RemoteJMXBrokerFacade(); |
| |
| SystemPropertiesConfiguration configuration = new SystemPropertiesConfiguration(); |
| brokerFacade.setConfiguration(configuration); |
| |
| ObjectName query = new ObjectName("org.apache.activemq:type=Broker,brokerName=remoteBroker"); |
| Set<ObjectName> queryResult = brokerFacade.queryNames(query, null); |
| |
| System.out.println("Number: " + queryResult.size()); |
| assertEquals(1, queryResult.size()); |
| |
| } |
| |
| /** |
| * Before AMQ-5896 there was the possibility of an InstanceNotFoundException when |
| * brokerFacade.getQueue if a destination was deleted after the initial list was looked |
| * up but before iterating over the list to find the right destination by name. |
| * |
| */ |
| @Test(timeout=10000) |
| public void testGetDestinationRaceCondition() throws Exception { |
| final CountDownLatch getQueuesLatch = new CountDownLatch(1); |
| final CountDownLatch destDeletionLatch = new CountDownLatch(1); |
| // Adding a pause so we can test the case where the destination is |
| // deleted in between calling getQueues() and iterating over the list |
| //and calling getName() on the DestinationViewMBean |
| // See AMQ-5896 |
| RemoteJMXBrokerFacade brokerFacade = new RemoteJMXBrokerFacade() { |
| @Override |
| protected DestinationViewMBean getDestinationByName( |
| Collection<? extends DestinationViewMBean> collection, |
| String name) { |
| try { |
| //we are done getting the queue collection so let thread know |
| //to remove destination |
| getQueuesLatch.countDown(); |
| //wait until other thread is done removing destination |
| destDeletionLatch.await(); |
| } catch (InterruptedException e) { |
| } |
| return super.getDestinationByName(collection, name); |
| } |
| |
| }; |
| |
| SystemPropertiesConfiguration configuration = new SystemPropertiesConfiguration(); |
| brokerFacade.setConfiguration(configuration); |
| //Create the destination |
| final ActiveMQQueue queue = new ActiveMQQueue("queue.test"); |
| brokerService.getDestination(queue); |
| |
| //after 1 second delete |
| ExecutorService service = Executors.newCachedThreadPool(); |
| service.submit(new Runnable() { |
| @Override |
| public void run() { |
| try { |
| //wait for confirmation that the queue list was obtained |
| getQueuesLatch.await(); |
| brokerService.removeDestination(queue); |
| //let original thread know destination was deleted |
| destDeletionLatch.countDown(); |
| } catch (Exception e) { |
| } |
| } |
| }); |
| |
| //Assert that the destination is now null because it was deleted in another thread |
| //during iteration |
| assertNull(brokerFacade.getQueue(queue.getPhysicalName())); |
| service.shutdown(); |
| } |
| |
| public String getJmxUri() throws NoSuchFieldException, IllegalAccessException { |
| Field field = ManagementContext.class.getDeclaredField("connectorServer"); |
| field.setAccessible(true); |
| JMXConnectorServer server = (JMXConnectorServer) field.get(brokerService.getManagementContext()); |
| return server.getAddress().toString(); |
| } |
| } |