content: /**
   * 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();
      }
  }
  