diff --git a/artemis-commons/src/main/java/org/apache/activemq/artemis/utils/CompositeAddress.java b/artemis-commons/src/main/java/org/apache/activemq/artemis/utils/CompositeAddress.java new file mode 100644 index 0000000000..d06ff8edd7 --- /dev/null +++ b/artemis-commons/src/main/java/org/apache/activemq/artemis/utils/CompositeAddress.java @@ -0,0 +1,97 @@ +/** + * 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.artemis.utils; + +import org.apache.activemq.artemis.api.core.SimpleString; + +public class CompositeAddress { + + public static SimpleString toFullQN(SimpleString address, SimpleString qName) { + return address.concat(SEPARATOR).concat(qName); + } + + public static String toFullQN(String address, String qName) { + return address + SEPARATOR + qName; + } + + public static String SEPARATOR = "::"; + private final String address; + private final String queueName; + private final boolean fqqn; + + public String getAddress() { + return address; + } + + public String getQueueName() { + return queueName; + } + + public CompositeAddress(String address, String queueName) { + + this.address = address; + this.queueName = queueName; + this.fqqn = address != null && !address.isEmpty(); + } + + public CompositeAddress(String singleName) { + int index = singleName.indexOf(SEPARATOR); + if (index == -1) { + this.fqqn = false; + this.address = null; + this.queueName = singleName; + } else { + this.fqqn = true; + this.address = singleName.substring(0, index); + this.queueName = singleName.substring(index + 2); + } + } + + public boolean isFqqn() { + return fqqn; + } + + public static boolean isFullyQualified(String address) { + return address.contains(SEPARATOR); + } + + public static CompositeAddress getQueueName(String address) { + + int index = address.indexOf(SEPARATOR); + if (index == -1) { + throw new IllegalStateException("Not A Fully Qualified Name"); + } + return new CompositeAddress(address.substring(0, index), address.substring(index + 2)); + } + + public static String extractQueueName(String name) { + int index = name.indexOf(SEPARATOR); + if (index != -1) { + return name.substring(index + 2); + } + return name; + } + + public static SimpleString extractQueueName(SimpleString name) { + return new SimpleString(extractQueueName(name.toString())); + } + + public static String extractAddressName(String address) { + String[] split = address.split(SEPARATOR); + return split[0]; + } +} diff --git a/artemis-protocols/artemis-amqp-protocol/src/main/java/org/apache/activemq/artemis/protocol/amqp/proton/ProtonServerSenderContext.java b/artemis-protocols/artemis-amqp-protocol/src/main/java/org/apache/activemq/artemis/protocol/amqp/proton/ProtonServerSenderContext.java index f8fa473720..2743290a2f 100644 --- a/artemis-protocols/artemis-amqp-protocol/src/main/java/org/apache/activemq/artemis/protocol/amqp/proton/ProtonServerSenderContext.java +++ b/artemis-protocols/artemis-amqp-protocol/src/main/java/org/apache/activemq/artemis/protocol/amqp/proton/ProtonServerSenderContext.java @@ -29,7 +29,6 @@ import org.apache.activemq.artemis.api.core.RoutingType; import org.apache.activemq.artemis.api.core.SimpleString; import org.apache.activemq.artemis.core.io.IOCallback; import org.apache.activemq.artemis.core.persistence.OperationContext; -import org.apache.activemq.artemis.core.postoffice.impl.CompositeAddress; import org.apache.activemq.artemis.core.server.AddressQueryResult; import org.apache.activemq.artemis.core.server.Consumer; import org.apache.activemq.artemis.core.server.MessageReference; @@ -47,6 +46,7 @@ import org.apache.activemq.artemis.protocol.amqp.proton.transaction.ProtonTransa import org.apache.activemq.artemis.reader.MessageUtil; import org.apache.activemq.artemis.selector.filter.FilterException; import org.apache.activemq.artemis.selector.impl.SelectorParser; +import org.apache.activemq.artemis.utils.CompositeAddress; import org.apache.qpid.proton.amqp.DescribedType; import org.apache.qpid.proton.amqp.Symbol; import org.apache.qpid.proton.amqp.messaging.Accepted; diff --git a/artemis-protocols/artemis-mqtt-protocol/src/main/java/org/apache/activemq/artemis/core/protocol/mqtt/MQTTSubscriptionManager.java b/artemis-protocols/artemis-mqtt-protocol/src/main/java/org/apache/activemq/artemis/core/protocol/mqtt/MQTTSubscriptionManager.java index c9e7a942fe..3231cb15ce 100644 --- a/artemis-protocols/artemis-mqtt-protocol/src/main/java/org/apache/activemq/artemis/core/protocol/mqtt/MQTTSubscriptionManager.java +++ b/artemis-protocols/artemis-mqtt-protocol/src/main/java/org/apache/activemq/artemis/core/protocol/mqtt/MQTTSubscriptionManager.java @@ -34,6 +34,7 @@ import org.apache.activemq.artemis.core.server.Queue; import org.apache.activemq.artemis.api.core.RoutingType; import org.apache.activemq.artemis.core.server.ServerConsumer; import org.apache.activemq.artemis.core.server.impl.AddressInfo; +import org.apache.activemq.artemis.utils.CompositeAddress; public class MQTTSubscriptionManager { @@ -159,23 +160,23 @@ public class MQTTSubscriptionManager { } private void addSubscription(MqttTopicSubscription subscription) throws Exception { - MqttTopicSubscription s = session.getSessionState().getSubscription(subscription.topicName()); + String topicName = CompositeAddress.extractAddressName(subscription.topicName()); + MqttTopicSubscription s = session.getSessionState().getSubscription(topicName); int qos = subscription.qualityOfService().value(); - String topic = subscription.topicName(); - String coreAddress = MQTTUtil.convertMQTTAddressFilterToCore(topic, session.getWildcardConfiguration()); + String coreAddress = MQTTUtil.convertMQTTAddressFilterToCore(topicName, session.getWildcardConfiguration()); session.getSessionState().addSubscription(subscription, session.getWildcardConfiguration()); Queue q = createQueueForSubscription(coreAddress, qos); if (s == null) { - createConsumerForSubscriptionQueue(q, topic, qos); + createConsumerForSubscriptionQueue(q, topicName, qos); } else { - consumerQoSLevels.put(consumers.get(topic).getID(), qos); + consumerQoSLevels.put(consumers.get(topicName).getID(), qos); } - session.getRetainMessageManager().addRetainedMessagesToQueue(q, topic); + session.getRetainMessageManager().addRetainedMessagesToQueue(q, topicName); } void removeSubscriptions(List topics) throws Exception { diff --git a/artemis-server/src/main/java/org/apache/activemq/artemis/core/postoffice/impl/CompositeAddress.java b/artemis-server/src/main/java/org/apache/activemq/artemis/core/postoffice/impl/CompositeAddress.java deleted file mode 100644 index 32083a5581..0000000000 --- a/artemis-server/src/main/java/org/apache/activemq/artemis/core/postoffice/impl/CompositeAddress.java +++ /dev/null @@ -1,50 +0,0 @@ -/** - * 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.artemis.core.postoffice.impl; - -public class CompositeAddress { - - public static String SEPARATOR = "::"; - private final String address; - private final String queueName; - - public String getAddress() { - return address; - } - - public String getQueueName() { - return queueName; - } - - public CompositeAddress(String address, String queueName) { - - this.address = address; - this.queueName = queueName; - } - - public static boolean isFullyQualified(String address) { - return address.toString().contains(SEPARATOR); - } - - public static CompositeAddress getQueueName(String address) { - String[] split = address.split(SEPARATOR); - if (split.length <= 0) { - throw new IllegalStateException("Nott A Fully Qualified Name"); - } - return new CompositeAddress(split[0], split[1]); - } -} diff --git a/artemis-server/src/main/java/org/apache/activemq/artemis/core/postoffice/impl/SimpleAddressManager.java b/artemis-server/src/main/java/org/apache/activemq/artemis/core/postoffice/impl/SimpleAddressManager.java index fa34594284..e95dd112f1 100644 --- a/artemis-server/src/main/java/org/apache/activemq/artemis/core/postoffice/impl/SimpleAddressManager.java +++ b/artemis-server/src/main/java/org/apache/activemq/artemis/core/postoffice/impl/SimpleAddressManager.java @@ -36,6 +36,7 @@ import org.apache.activemq.artemis.core.server.ActiveMQMessageBundle; import org.apache.activemq.artemis.api.core.RoutingType; import org.apache.activemq.artemis.core.server.impl.AddressInfo; import org.apache.activemq.artemis.core.transaction.Transaction; +import org.apache.activemq.artemis.utils.CompositeAddress; import org.jboss.logging.Logger; /** @@ -103,7 +104,7 @@ public class SimpleAddressManager implements AddressManager { @Override public Binding getBinding(final SimpleString bindableName) { - return nameMap.get(bindableName); + return nameMap.get(CompositeAddress.extractQueueName(bindableName)); } @Override @@ -131,7 +132,7 @@ public class SimpleAddressManager implements AddressManager { @Override public SimpleString getMatchingQueue(final SimpleString address, RoutingType routingType) throws Exception { - Binding binding = nameMap.get(address); + Binding binding = getBinding(address); if (binding == null || !(binding instanceof LocalQueueBinding) || !binding.getAddress().equals(address)) { @@ -151,9 +152,9 @@ public class SimpleAddressManager implements AddressManager { @Override public SimpleString getMatchingQueue(final SimpleString address, final SimpleString queueName, RoutingType routingType) throws Exception { - Binding binding = nameMap.get(queueName); + Binding binding = getBinding(queueName); - if (binding != null && !binding.getAddress().equals(address)) { + if (binding != null && !binding.getAddress().equals(address) && !address.toString().isEmpty()) { throw new IllegalStateException("queue belongs to address" + binding.getAddress()); } return binding != null ? binding.getUniqueName() : null; @@ -188,9 +189,8 @@ public class SimpleAddressManager implements AddressManager { Binding theBinding = null; for (Binding binding : bindings.getBindings()) { - if (binding.getUniqueName().equals(bindableName)) { + if (binding.getUniqueName().equals(CompositeAddress.extractQueueName(bindableName))) { theBinding = binding; - break; } } diff --git a/artemis-server/src/main/java/org/apache/activemq/artemis/core/server/cluster/impl/ClusterConnectionBridge.java b/artemis-server/src/main/java/org/apache/activemq/artemis/core/server/cluster/impl/ClusterConnectionBridge.java index a870ea6767..614e06656b 100644 --- a/artemis-server/src/main/java/org/apache/activemq/artemis/core/server/cluster/impl/ClusterConnectionBridge.java +++ b/artemis-server/src/main/java/org/apache/activemq/artemis/core/server/cluster/impl/ClusterConnectionBridge.java @@ -46,6 +46,7 @@ import org.apache.activemq.artemis.core.server.cluster.ClusterConnection; import org.apache.activemq.artemis.core.server.cluster.ClusterManager; import org.apache.activemq.artemis.core.server.cluster.MessageFlowRecord; import org.apache.activemq.artemis.core.server.cluster.Transformer; +import org.apache.activemq.artemis.utils.CompositeAddress; import org.apache.activemq.artemis.utils.UUID; import org.apache.activemq.artemis.utils.UUIDGenerator; import org.jboss.logging.Logger; @@ -213,10 +214,9 @@ public class ClusterConnectionBridge extends BridgeImpl { } // Get the queue data - String qName = "notif." + UUIDGenerator.getInstance().generateStringUUID() + "." + - clusterConnection.getServer(); + clusterConnection.getServer().toString().replaceAll(CompositeAddress.SEPARATOR, "_"); SimpleString notifQueueName = new SimpleString(qName); diff --git a/artemis-server/src/main/java/org/apache/activemq/artemis/core/server/impl/ActiveMQServerImpl.java b/artemis-server/src/main/java/org/apache/activemq/artemis/core/server/impl/ActiveMQServerImpl.java index 3fc5dd33b4..8482cb3705 100644 --- a/artemis-server/src/main/java/org/apache/activemq/artemis/core/server/impl/ActiveMQServerImpl.java +++ b/artemis-server/src/main/java/org/apache/activemq/artemis/core/server/impl/ActiveMQServerImpl.java @@ -161,6 +161,7 @@ import org.apache.activemq.artemis.spi.core.security.ActiveMQSecurityManager; import org.apache.activemq.artemis.utils.ActiveMQThreadFactory; import org.apache.activemq.artemis.utils.ActiveMQThreadPoolExecutor; import org.apache.activemq.artemis.utils.CertificateUtil; +import org.apache.activemq.artemis.utils.CompositeAddress; import org.apache.activemq.artemis.utils.ConcurrentHashSet; import org.apache.activemq.artemis.utils.ExecutorFactory; import org.apache.activemq.artemis.utils.OrderedExecutorFactory; @@ -699,7 +700,9 @@ public class ActiveMQServerImpl implements ActiveMQServer { throw ActiveMQMessageBundle.BUNDLE.addressIsNull(); } - AddressSettings addressSettings = getAddressSettingsRepository().getMatch(address.toString()); + CompositeAddress addressKey = new CompositeAddress(address.toString()); + String realAddress = addressKey.isFqqn() ? addressKey.getAddress() : addressKey.getQueueName(); + AddressSettings addressSettings = getAddressSettingsRepository().getMatch(realAddress); boolean autoCreateQeueus = addressSettings.isAutoCreateQueues(); boolean autoCreateAddresses = addressSettings.isAutoCreateAddresses(); @@ -710,21 +713,26 @@ public class ActiveMQServerImpl implements ActiveMQServer { // make an exception for the management address (see HORNETQ-29) ManagementService managementService = getManagementService(); + SimpleString bindAddress = new SimpleString(realAddress); if (managementService != null) { - if (address.equals(managementService.getManagementAddress())) { + if (bindAddress.equals(managementService.getManagementAddress())) { return new BindingQueryResult(true, names, autoCreateQeueus, autoCreateAddresses, defaultPurgeOnNoConsumers, defaultMaxConsumers); } } - Bindings bindings = getPostOffice().getMatchingBindings(address); + Bindings bindings = getPostOffice().getMatchingBindings(bindAddress); for (Binding binding : bindings.getBindings()) { if (binding.getType() == BindingType.LOCAL_QUEUE || binding.getType() == BindingType.REMOTE_QUEUE) { - names.add(binding.getUniqueName()); + if (addressKey.isFqqn()) { + names.add(new SimpleString(addressKey.getAddress()).concat(CompositeAddress.SEPARATOR).concat(binding.getUniqueName())); + } else { + names.add(binding.getUniqueName()); + } } } - return new BindingQueryResult(getAddressInfo(address) != null, names, autoCreateQeueus, autoCreateAddresses, defaultPurgeOnNoConsumers, defaultMaxConsumers); + return new BindingQueryResult(getAddressInfo(bindAddress) != null, names, autoCreateQeueus, autoCreateAddresses, defaultPurgeOnNoConsumers, defaultMaxConsumers); } @Override diff --git a/artemis-server/src/test/java/org/apache/activemq/artemis/core/server/impl/CompositeAddressTest.java b/artemis-server/src/test/java/org/apache/activemq/artemis/core/server/impl/CompositeAddressTest.java new file mode 100644 index 0000000000..da6482a1fb --- /dev/null +++ b/artemis-server/src/test/java/org/apache/activemq/artemis/core/server/impl/CompositeAddressTest.java @@ -0,0 +1,39 @@ +/* + * 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.artemis.core.server.impl; + +import org.apache.activemq.artemis.utils.CompositeAddress; +import org.junit.Test; + +import static org.junit.Assert.assertEquals; + +public class CompositeAddressTest { + + @Test + public void testSplit() throws Exception { + String name = "someQueue"; + String name2 = "someAddress::someQueue"; + String qname = CompositeAddress.extractQueueName(name); + assertEquals(name, qname); + qname = CompositeAddress.extractQueueName(name2); + assertEquals(name, qname); + assertEquals("", CompositeAddress.extractQueueName("address::")); + assertEquals("", CompositeAddress.extractQueueName("::")); + assertEquals("queue", CompositeAddress.extractQueueName("::queue")); + assertEquals("address", CompositeAddress.extractAddressName("address::")); + } +} diff --git a/docs/user-manual/en/address-model.md b/docs/user-manual/en/address-model.md index 8de3b67fb5..8d994d2fdf 100644 --- a/docs/user-manual/en/address-model.md +++ b/docs/user-manual/en/address-model.md @@ -289,6 +289,9 @@ The example below configures an address-setting to be automatically deleted by t Internally the broker maps a client’s request for an address to specific queues. The broker decides on behalf of the client which queues to send messages to or from which queue to receive messages. However, more advanced use cases might require that the client specify a queue directly. In these situations the client and use a fully qualified queue name, by specifying both the address name and the queue name, separated by a ::. +Currently Artemis supports fully qualified queue names on Core, AMQP, JMS, OpenWire, MQTT and Stomp protocols for + receiving messages only. + ### Specifying a Fully Qualified Queue Name In this example, the address foo is configured with two queues q1, q2 as shown in the configuration below. diff --git a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/amqp/ProtonFullQualifiedNameTest.java b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/amqp/ProtonFullQualifiedNameTest.java new file mode 100644 index 0000000000..22ba64dfc2 --- /dev/null +++ b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/amqp/ProtonFullQualifiedNameTest.java @@ -0,0 +1,237 @@ +/* + * 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.artemis.tests.integration.amqp; + +import org.apache.activemq.artemis.api.core.RoutingType; +import org.apache.activemq.artemis.api.core.SimpleString; +import org.apache.activemq.artemis.api.core.TransportConfiguration; +import org.apache.activemq.artemis.api.core.client.ClientProducer; +import org.apache.activemq.artemis.api.core.client.ClientSession; +import org.apache.activemq.artemis.api.core.client.ClientSessionFactory; +import org.apache.activemq.artemis.api.core.client.ServerLocator; +import org.apache.activemq.artemis.core.config.Configuration; +import org.apache.activemq.artemis.core.postoffice.Binding; +import org.apache.activemq.artemis.core.postoffice.Bindings; +import org.apache.activemq.artemis.core.postoffice.impl.LocalQueueBinding; +import org.apache.activemq.artemis.core.server.QueueQueryResult; +import org.apache.activemq.artemis.core.settings.impl.AddressSettings; +import org.apache.activemq.artemis.utils.CompositeAddress; +import org.apache.qpid.jms.JmsConnectionFactory; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; + +import javax.jms.Connection; +import javax.jms.InvalidDestinationException; +import javax.jms.Message; +import javax.jms.MessageConsumer; +import javax.jms.MessageProducer; +import javax.jms.Queue; +import javax.jms.Session; +import javax.jms.Topic; +import java.util.HashMap; +import java.util.Map; + +public class ProtonFullQualifiedNameTest extends ProtonTestBase { + + private static final String amqpConnectionUri = "amqp://localhost:5672"; + + private SimpleString anycastAddress = new SimpleString("address.anycast"); + private SimpleString multicastAddress = new SimpleString("address.multicast"); + + private SimpleString anycastQ1 = new SimpleString("q1"); + private SimpleString anycastQ2 = new SimpleString("q2"); + private SimpleString anycastQ3 = new SimpleString("q3"); + + JmsConnectionFactory factory = new JmsConnectionFactory(amqpConnectionUri); + private ServerLocator locator; + + @Override + @Before + public void setUp() throws Exception { + super.setUp(); + + Configuration serverConfig = server.getConfiguration(); + + Map settings = serverConfig.getAddressesSettings(); + assertNotNull(settings); + AddressSettings addressSetting = settings.get("#"); + if (addressSetting == null) { + addressSetting = new AddressSettings(); + settings.put("#", addressSetting); + } + addressSetting.setAutoCreateQueues(true); + locator = createNettyNonHALocator(); + } + + @Override + @After + public void tearDown() throws Exception { + super.tearDown(); + } + + @Override + protected void configureServer(Configuration serverConfig) { + serverConfig.addAcceptorConfiguration(new TransportConfiguration(NETTY_ACCEPTOR_FACTORY, new HashMap(), "netty", new HashMap())); + } + + @Test + //there isn't much use of FQQN for topics + //however we can test query functionality + public void testTopic() throws Exception { + + Connection connection = factory.createConnection(); + try { + connection.setClientID("FQQNconn"); + connection.start(); + Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE); + Topic topic = session.createTopic(multicastAddress.toString()); + + MessageConsumer consumer1 = session.createConsumer(topic); + MessageConsumer consumer2 = session.createConsumer(topic); + MessageConsumer consumer3 = session.createConsumer(topic); + + MessageProducer producer = session.createProducer(topic); + + producer.send(session.createMessage()); + + //each consumer receives one + Message m = consumer1.receive(2000); + assertNotNull(m); + m = consumer2.receive(2000); + assertNotNull(m); + m = consumer3.receive(2000); + assertNotNull(m); + + Bindings bindings = server.getPostOffice().getBindingsForAddress(multicastAddress); + for (Binding b : bindings.getBindings()) { + System.out.println("checking binidng " + b.getUniqueName() + " " + ((LocalQueueBinding)b).getQueue().getDeliveringMessages()); + SimpleString qName = b.getUniqueName(); + //do FQQN query + QueueQueryResult result = server.queueQuery(CompositeAddress.toFullQN(multicastAddress, qName)); + assertTrue(result.isExists()); + assertEquals(result.getName(), CompositeAddress.toFullQN(multicastAddress, qName)); + //do qname query + result = server.queueQuery(qName); + assertTrue(result.isExists()); + assertEquals(result.getName(), qName); + } + } finally { + connection.close(); + } + } + + @Test + public void testQueue() throws Exception { + server.createQueue(anycastAddress, RoutingType.ANYCAST, anycastQ1, null, true, false, -1, false, true); + server.createQueue(anycastAddress, RoutingType.ANYCAST, anycastQ2, null, true, false, -1, false, true); + server.createQueue(anycastAddress, RoutingType.ANYCAST, anycastQ3, null, true, false, -1, false, true); + + Connection connection = factory.createConnection(); + try { + connection.start(); + Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE); + + Queue q1 = session.createQueue(CompositeAddress.toFullQN(anycastAddress, anycastQ1).toString()); + Queue q2 = session.createQueue(CompositeAddress.toFullQN(anycastAddress, anycastQ2).toString()); + Queue q3 = session.createQueue(CompositeAddress.toFullQN(anycastAddress, anycastQ3).toString()); + + //send 3 messages to anycastAddress + ClientSessionFactory cf = createSessionFactory(locator); + ClientSession coreSession = cf.createSession(); + + //send 3 messages + ClientProducer coreProducer = coreSession.createProducer(anycastAddress); + sendMessages(coreSession, coreProducer, 3); + + MessageConsumer consumer1 = session.createConsumer(q1); + MessageConsumer consumer2 = session.createConsumer(q2); + MessageConsumer consumer3 = session.createConsumer(q3); + + //each consumer receives one + assertNotNull(consumer1.receive(2000)); + assertNotNull(consumer2.receive(2000)); + assertNotNull(consumer3.receive(2000)); + + connection.close(); + //queues are empty now + for (SimpleString q : new SimpleString[]{anycastQ1, anycastQ2, anycastQ3}) { + //FQQN query + QueueQueryResult query = server.queueQuery(CompositeAddress.toFullQN(anycastAddress, q)); + assertTrue(query.isExists()); + assertEquals(anycastAddress, query.getAddress()); + assertEquals(CompositeAddress.toFullQN(anycastAddress, q), query.getName()); + assertEquals(0, query.getMessageCount()); + //try query again using qName + query = server.queueQuery(q); + assertEquals(q, query.getName()); + } + } finally { + connection.close(); + } + } + + @Test + public void testQueueSpecial() throws Exception { + server.createQueue(anycastAddress, RoutingType.ANYCAST, anycastQ1, null, true, false, -1, false, true); + + Connection connection = factory.createConnection(); + try { + connection.start(); + Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE); + + //::queue ok! + String specialName = CompositeAddress.toFullQN(new SimpleString(""), anycastQ1).toString(); + Queue q1 = session.createQueue(specialName); + + ClientSessionFactory cf = createSessionFactory(locator); + ClientSession coreSession = cf.createSession(); + + ClientProducer coreProducer = coreSession.createProducer(anycastAddress); + sendMessages(coreSession, coreProducer, 1); + + System.out.println("create consumer: " + q1); + MessageConsumer consumer1 = session.createConsumer(q1); + + assertNotNull(consumer1.receive(2000)); + + //queue:: + specialName = CompositeAddress.toFullQN(anycastQ1, new SimpleString("")).toString(); + q1 = session.createQueue(specialName); + try { + session.createConsumer(q1); + fail("should get exception"); + } catch (InvalidDestinationException e) { + //expected + } + + //:: + specialName = CompositeAddress.toFullQN(new SimpleString(""), new SimpleString("")).toString(); + q1 = session.createQueue(specialName); + try { + session.createConsumer(q1); + fail("should get exception"); + } catch (InvalidDestinationException e) { + //expected + } + + } finally { + connection.close(); + } + } + +} diff --git a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/amqp/ProtonTestBase.java b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/amqp/ProtonTestBase.java index 599022ed1e..ab8d2d3ced 100644 --- a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/amqp/ProtonTestBase.java +++ b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/amqp/ProtonTestBase.java @@ -20,6 +20,7 @@ import java.util.HashMap; import java.util.Map; import org.apache.activemq.artemis.api.core.TransportConfiguration; +import org.apache.activemq.artemis.core.config.Configuration; import org.apache.activemq.artemis.core.remoting.impl.netty.TransportConstants; import org.apache.activemq.artemis.core.server.ActiveMQServer; import org.apache.activemq.artemis.core.settings.impl.AddressFullMessagePolicy; @@ -68,9 +69,13 @@ public class ProtonTestBase extends ActiveMQTestBase { AddressSettings addressSettings = new AddressSettings(); addressSettings.setAddressFullMessagePolicy(AddressFullMessagePolicy.PAGE); amqpServer.getConfiguration().getAddressesSettings().put("#", addressSettings); + configureServer(amqpServer.getConfiguration()); return amqpServer; } + protected void configureServer(Configuration serverConfig) { + } + protected void configureAmqp(Map params) { } diff --git a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/client/FullQualifiedQueueTest.java b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/client/FullQualifiedQueueTest.java new file mode 100644 index 0000000000..dfc7f59744 --- /dev/null +++ b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/client/FullQualifiedQueueTest.java @@ -0,0 +1,257 @@ +/* + * 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.artemis.tests.integration.client; + +import org.apache.activemq.artemis.api.core.ActiveMQNonExistentQueueException; +import org.apache.activemq.artemis.api.core.RoutingType; +import org.apache.activemq.artemis.api.core.SimpleString; +import org.apache.activemq.artemis.api.core.client.ClientConsumer; +import org.apache.activemq.artemis.api.core.client.ClientMessage; +import org.apache.activemq.artemis.api.core.client.ClientProducer; +import org.apache.activemq.artemis.api.core.client.ClientSession; +import org.apache.activemq.artemis.api.core.client.ClientSessionFactory; +import org.apache.activemq.artemis.api.core.client.ServerLocator; +import org.apache.activemq.artemis.core.server.ActiveMQServer; +import org.apache.activemq.artemis.core.server.QueueQueryResult; +import org.apache.activemq.artemis.tests.util.ActiveMQTestBase; +import org.apache.activemq.artemis.utils.CompositeAddress; +import org.junit.Before; +import org.junit.Test; + +public class FullQualifiedQueueTest extends ActiveMQTestBase { + + private SimpleString anycastAddress = new SimpleString("address.anycast"); + private SimpleString multicastAddress = new SimpleString("address.multicast"); + private SimpleString mixedAddress = new SimpleString("address.mixed"); + + private SimpleString anycastQ1 = new SimpleString("q1"); + private SimpleString anycastQ2 = new SimpleString("q2"); + private SimpleString anycastQ3 = new SimpleString("q3"); + + private SimpleString multicastQ1 = new SimpleString("q4"); + private SimpleString multicastQ2 = new SimpleString("q5"); + private SimpleString multicastQ3 = new SimpleString("q6"); + + private ActiveMQServer server; + private ServerLocator locator; + + @Override + @Before + public void setUp() throws Exception { + super.setUp(); + + server = createServer(false, true); + + server.start(); + locator = createNettyNonHALocator(); + } + + @Test + public void testMixedQueues() throws Exception { + server.createQueue(mixedAddress, RoutingType.MULTICAST, multicastQ1, null, true, false, -1, false, true); + server.createQueue(mixedAddress, RoutingType.MULTICAST, multicastQ2, null, true, false, -1, false, true); + server.createQueue(mixedAddress, RoutingType.MULTICAST, multicastQ3, null, true, false, -1, false, true); + server.createQueue(mixedAddress, RoutingType.ANYCAST, anycastQ1, null, true, false, -1, false, true); + server.createQueue(mixedAddress, RoutingType.ANYCAST, anycastQ2, null, true, false, -1, false, true); + server.createQueue(mixedAddress, RoutingType.ANYCAST, anycastQ3, null, true, false, -1, false, true); + + ClientSessionFactory cf = createSessionFactory(locator); + ClientSession session = cf.createSession(); + session.start(); + + //send 3 messages + ClientProducer producer = session.createProducer(mixedAddress); + final int num = 3; + sendMessages(session, producer, num); + + ClientConsumer consumer1 = session.createConsumer(toFullQN(mixedAddress, anycastQ1)); + ClientConsumer consumer2 = session.createConsumer(toFullQN(mixedAddress, anycastQ2)); + ClientConsumer consumer3 = session.createConsumer(toFullQN(mixedAddress, anycastQ3)); + ClientConsumer consumer4 = session.createConsumer(toFullQN(mixedAddress, multicastQ1)); + ClientConsumer consumer5 = session.createConsumer(toFullQN(mixedAddress, multicastQ2)); + ClientConsumer consumer6 = session.createConsumer(toFullQN(mixedAddress, multicastQ3)); + + session.start(); + + //each anycast consumer receives one, each multicast receives three. + ClientMessage m = consumer1.receive(2000); + assertNotNull(m); + System.out.println("consumer1 : " + m); + m.acknowledge(); + + m = consumer2.receive(2000); + assertNotNull(m); + System.out.println("consumer2 : " + m); + m.acknowledge(); + + m = consumer3.receive(2000); + assertNotNull(m); + System.out.println("consumer3 : " + m); + m.acknowledge(); + + for (int i = 0; i < num; i++) { + m = consumer4.receive(2000); + assertNotNull(m); + System.out.println("consumer4 : " + m); + m.acknowledge(); + m = consumer5.receive(2000); + assertNotNull(m); + System.out.println("consumer5 : " + m); + m.acknowledge(); + m = consumer6.receive(2000); + assertNotNull(m); + System.out.println("consumer6 : " + m); + m.acknowledge(); + } + + session.commit(); + + //queues are empty now + for (SimpleString q : new SimpleString[]{anycastQ1, anycastQ2, anycastQ3, multicastQ1, multicastQ2, multicastQ3}) { + QueueQueryResult query = server.queueQuery(toFullQN(mixedAddress, q)); + assertTrue(query.isExists()); + assertEquals(mixedAddress, query.getAddress()); + assertEquals(toFullQN(mixedAddress, q), query.getName()); + assertEquals(0, query.getMessageCount()); + } + } + + @Test + public void testMulticastQueues() throws Exception { + server.createQueue(multicastAddress, RoutingType.MULTICAST, multicastQ1, null, true, false, -1, false, true); + server.createQueue(multicastAddress, RoutingType.MULTICAST, multicastQ2, null, true, false, -1, false, true); + server.createQueue(multicastAddress, RoutingType.MULTICAST, multicastQ3, null, true, false, -1, false, true); + + ClientSessionFactory cf = createSessionFactory(locator); + ClientSession session = cf.createSession(); + session.start(); + + //send 3 messages + ClientProducer producer = session.createProducer(multicastAddress); + sendMessages(session, producer, 1); + + ClientConsumer consumer1 = session.createConsumer(toFullQN(multicastAddress, multicastQ1)); + ClientConsumer consumer2 = session.createConsumer(toFullQN(multicastAddress, multicastQ2)); + ClientConsumer consumer3 = session.createConsumer(toFullQN(multicastAddress, multicastQ3)); + session.start(); + + //each consumer receives one + ClientMessage m = consumer1.receive(2000); + assertNotNull(m); + m.acknowledge(); + m = consumer2.receive(2000); + assertNotNull(m); + m.acknowledge(); + m = consumer3.receive(2000); + assertNotNull(m); + m.acknowledge(); + + session.commit(); + //queues are empty now + for (SimpleString q : new SimpleString[]{multicastQ1, multicastQ2, multicastQ3}) { + QueueQueryResult query = server.queueQuery(toFullQN(multicastAddress, q)); + assertTrue(query.isExists()); + assertEquals(multicastAddress, query.getAddress()); + assertEquals(toFullQN(multicastAddress, q), query.getName()); + assertEquals(0, query.getMessageCount()); + } + } + + @Test + public void testAnycastQueues() throws Exception { + server.createQueue(anycastAddress, RoutingType.ANYCAST, anycastQ1, null, true, false, -1, false, true); + server.createQueue(anycastAddress, RoutingType.ANYCAST, anycastQ2, null, true, false, -1, false, true); + server.createQueue(anycastAddress, RoutingType.ANYCAST, anycastQ3, null, true, false, -1, false, true); + + ClientSessionFactory cf = createSessionFactory(locator); + ClientSession session = cf.createSession(); + session.start(); + + //send 3 messages + ClientProducer producer = session.createProducer(anycastAddress); + sendMessages(session, producer, 3); + + ClientConsumer consumer1 = session.createConsumer(toFullQN(anycastAddress, anycastQ1)); + ClientConsumer consumer2 = session.createConsumer(toFullQN(anycastAddress, anycastQ2)); + ClientConsumer consumer3 = session.createConsumer(toFullQN(anycastAddress, anycastQ3)); + session.start(); + + //each consumer receives one + ClientMessage m = consumer1.receive(2000); + assertNotNull(m); + m.acknowledge(); + m = consumer2.receive(2000); + assertNotNull(m); + m.acknowledge(); + m = consumer3.receive(2000); + assertNotNull(m); + m.acknowledge(); + + session.commit(); + //queues are empty now + for (SimpleString q : new SimpleString[]{anycastQ1, anycastQ2, anycastQ3}) { + QueueQueryResult query = server.queueQuery(toFullQN(anycastAddress, q)); + assertTrue(query.isExists()); + assertEquals(anycastAddress, query.getAddress()); + assertEquals(toFullQN(anycastAddress, q), query.getName()); + assertEquals(0, query.getMessageCount()); + } + } + + @Test + public void testSpecialCase() throws Exception { + server.createQueue(anycastAddress, RoutingType.ANYCAST, anycastQ1, null, true, false, -1, false, true); + + ClientSessionFactory cf = createSessionFactory(locator); + ClientSession session = cf.createSession(); + session.start(); + + ClientProducer producer = session.createProducer(anycastAddress); + sendMessages(session, producer, 1); + + //::queue + ClientConsumer consumer1 = session.createConsumer(toFullQN(new SimpleString(""), anycastQ1)); + session.start(); + + ClientMessage m = consumer1.receive(2000); + assertNotNull(m); + m.acknowledge(); + + session.commit(); + consumer1.close(); + + try { + //queue:: + session.createConsumer(toFullQN(anycastQ1, new SimpleString(""))); + fail("should get exception"); + } catch (ActiveMQNonExistentQueueException e) { + //expected. + } + + try { + //:: + session.createConsumer(toFullQN(new SimpleString(""), new SimpleString(""))); + fail("should get exception"); + } catch (ActiveMQNonExistentQueueException e) { + //expected. + } + } + + private SimpleString toFullQN(SimpleString address, SimpleString qName) { + return address.concat(CompositeAddress.SEPARATOR).concat(qName); + } +} diff --git a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/management/AddressControlTest.java b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/management/AddressControlTest.java index bf505ede28..7bc648f635 100644 --- a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/management/AddressControlTest.java +++ b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/management/AddressControlTest.java @@ -61,6 +61,12 @@ public class AddressControlTest extends ManagementTestBase { return false; } + @Test + public void testManagementAddressAlwaysExists() throws Exception { + ClientSession.AddressQuery query = session.addressQuery(new SimpleString("activemq.management")); + assertTrue(query.isExists()); + } + @Test public void testGetAddress() throws Exception { SimpleString address = RandomUtil.randomSimpleString(); diff --git a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/mqtt/imported/MQTTFQQNTest.java b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/mqtt/imported/MQTTFQQNTest.java new file mode 100644 index 0000000000..a1fa0071b3 --- /dev/null +++ b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/mqtt/imported/MQTTFQQNTest.java @@ -0,0 +1,184 @@ +/** + * 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.artemis.tests.integration.mqtt.imported; + +import java.lang.reflect.Field; +import java.util.Map; +import java.util.concurrent.ConcurrentHashMap; +import java.util.concurrent.CountDownLatch; +import java.util.concurrent.TimeUnit; + +import org.apache.activemq.artemis.api.core.SimpleString; +import org.apache.activemq.artemis.core.postoffice.Binding; +import org.apache.activemq.artemis.core.protocol.mqtt.MQTTConnectionManager; +import org.apache.activemq.artemis.core.protocol.mqtt.MQTTSession; +import org.apache.activemq.artemis.core.server.QueueQueryResult; +import org.apache.activemq.artemis.utils.ConcurrentHashSet; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +public class MQTTFQQNTest extends MQTTTestSupport { + + private static final Logger LOG = LoggerFactory.getLogger(MQTTFQQNTest.class); + + @Override + @Before + public void setUp() throws Exception { + Field sessions = MQTTSession.class.getDeclaredField("SESSIONS"); + sessions.setAccessible(true); + sessions.set(null, new ConcurrentHashMap<>()); + + Field connectedClients = MQTTConnectionManager.class.getDeclaredField("CONNECTED_CLIENTS"); + connectedClients.setAccessible(true); + connectedClients.set(null, new ConcurrentHashSet<>()); + super.setUp(); + + } + + @Override + @After + public void tearDown() throws Exception { + super.tearDown(); + + } + + @Test + public void testMQTTSubNames() throws Exception { + final MQTTClientProvider subscriptionProvider = getMQTTClientProvider(); + initializeConnection(subscriptionProvider); + + try { + subscriptionProvider.subscribe("foo/bah", AT_MOST_ONCE); + + Map allBindings = server.getPostOffice().getAllBindings(); + assertEquals(1, allBindings.size()); + Binding b = allBindings.values().iterator().next(); + //check that query using bare queue name works as before + QueueQueryResult result = server.queueQuery(b.getUniqueName()); + assertTrue(result.isExists()); + assertEquals(result.getAddress(), new SimpleString("foo.bah")); + assertEquals(b.getUniqueName(), result.getName()); + //check that queue query using FQQN returns FQQN + result = server.queueQuery(new SimpleString("foo.bah::" + b.getUniqueName())); + assertTrue(result.isExists()); + assertEquals(new SimpleString("foo.bah"), result.getAddress()); + assertEquals(new SimpleString("foo.bah::" + b.getUniqueName()), result.getName()); + } finally { + subscriptionProvider.disconnect(); + } + } + + @Test(timeout = 60 * 1000) + public void testSendAndReceiveMQTTSpecial1() throws Exception { + final MQTTClientProvider subscriptionProvider = getMQTTClientProvider(); + initializeConnection(subscriptionProvider); + + subscriptionProvider.subscribe("foo/bah::", AT_MOST_ONCE); + + final CountDownLatch latch = new CountDownLatch(NUM_MESSAGES); + + Thread thread = new Thread(new Runnable() { + @Override + public void run() { + for (int i = 0; i < NUM_MESSAGES; i++) { + try { + byte[] payload = subscriptionProvider.receive(10000); + assertNotNull("Should get a message", payload); + latch.countDown(); + } catch (Exception e) { + e.printStackTrace(); + break; + } + + } + } + }); + thread.start(); + + final MQTTClientProvider publishProvider = getMQTTClientProvider(); + initializeConnection(publishProvider); + + for (int i = 0; i < NUM_MESSAGES; i++) { + String payload = "Message " + i; + publishProvider.publish("foo/bah", payload.getBytes(), AT_LEAST_ONCE); + } + + latch.await(10, TimeUnit.SECONDS); + assertEquals(0, latch.getCount()); + subscriptionProvider.disconnect(); + publishProvider.disconnect(); + } + + @Test(timeout = 60 * 1000) + public void testSendAndReceiveMQTTSpecial2() throws Exception { + final MQTTClientProvider subscriptionProvider = getMQTTClientProvider(); + initializeConnection(subscriptionProvider); + + try { + subscriptionProvider.subscribe("::foo/bah", AT_MOST_ONCE); + fail("should get exception!"); + } catch (Exception e) { + //expected + } finally { + subscriptionProvider.disconnect(); + } + + //:: + initializeConnection(subscriptionProvider); + try { + subscriptionProvider.subscribe("::", AT_MOST_ONCE); + fail("should get exception!"); + } catch (Exception e) { + //expected + } finally { + subscriptionProvider.disconnect(); + } + } + + @Test + public void testMQTTSubNamesSpecial() throws Exception { + final MQTTClientProvider subscriptionProvider = getMQTTClientProvider(); + initializeConnection(subscriptionProvider); + + try { + subscriptionProvider.subscribe("foo/bah", AT_MOST_ONCE); + + Map allBindings = server.getPostOffice().getAllBindings(); + assertEquals(1, allBindings.size()); + Binding b = allBindings.values().iterator().next(); + + //check ::queue + QueueQueryResult result = server.queueQuery(new SimpleString("::" + b.getUniqueName())); + assertTrue(result.isExists()); + assertEquals(new SimpleString("foo.bah"), result.getAddress()); + assertEquals(new SimpleString("::" + b.getUniqueName()), result.getName()); + + //check queue:: + result = server.queueQuery(new SimpleString(b.getUniqueName() + "::")); + assertFalse(result.isExists()); + + //check :: + result = server.queueQuery(new SimpleString("::")); + assertFalse(result.isExists()); + } finally { + subscriptionProvider.disconnect(); + } + } +} diff --git a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/openwire/FQQNOpenWireTest.java b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/openwire/FQQNOpenWireTest.java new file mode 100644 index 0000000000..44be8054d8 --- /dev/null +++ b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/openwire/FQQNOpenWireTest.java @@ -0,0 +1,272 @@ +/** + * 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.artemis.tests.integration.openwire; + +import javax.jms.Connection; +import javax.jms.ConnectionFactory; +import javax.jms.Destination; +import javax.jms.InvalidDestinationException; +import javax.jms.Message; +import javax.jms.MessageConsumer; +import javax.jms.MessageProducer; +import javax.jms.Queue; +import javax.jms.Session; +import javax.jms.TextMessage; +import javax.jms.Topic; + +import org.apache.activemq.ActiveMQConnectionFactory; +import org.apache.activemq.artemis.api.core.SimpleString; +import org.apache.activemq.artemis.api.core.RoutingType; +import org.apache.activemq.artemis.api.core.client.ClientProducer; +import org.apache.activemq.artemis.api.core.client.ClientSession; +import org.apache.activemq.artemis.api.core.client.ClientSessionFactory; +import org.apache.activemq.artemis.api.core.client.ServerLocator; +import org.apache.activemq.artemis.core.postoffice.Binding; +import org.apache.activemq.artemis.core.postoffice.Bindings; +import org.apache.activemq.artemis.core.postoffice.impl.LocalQueueBinding; +import org.apache.activemq.artemis.core.server.QueueQueryResult; +import org.apache.activemq.artemis.utils.CompositeAddress; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.junit.runners.Parameterized; + +import java.util.Arrays; +import java.util.Collection; + +/** + * Verify FQQN queues work with openwire/artemis JMS API + */ +@RunWith(Parameterized.class) +public class FQQNOpenWireTest extends OpenWireTestBase { + + protected static final String urlString = "tcp://" + OWHOST + ":" + OWPORT + "?wireFormat.cacheEnabled=true"; + + + @Parameterized.Parameters(name = "{0}") + public static Collection params() { + return Arrays.asList(new Object[][]{{"OpenWire"}, {"Artemis"}}); + } + + private SimpleString anycastAddress = new SimpleString("address.anycast"); + private SimpleString multicastAddress = new SimpleString("address.multicast"); + + private SimpleString anycastQ1 = new SimpleString("q1"); + private SimpleString anycastQ2 = new SimpleString("q2"); + private SimpleString anycastQ3 = new SimpleString("q3"); + + private ConnectionFactory factory; + + private ServerLocator locator; + + public FQQNOpenWireTest(String factoryType) { + if ("OpenWire".equals(factoryType)) { + factory = new ActiveMQConnectionFactory(urlString); + } else if ("Artemis".equals(factoryType)) { + factory = new org.apache.activemq.artemis.jms.client.ActiveMQConnectionFactory(urlString); + } + } + + @Test + //there isn't much use of FQQN for topics + //however we can test query functionality + public void testTopic() throws Exception { + + Connection connection = factory.createConnection(); + try { + connection.setClientID("FQQNconn"); + connection.start(); + Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE); + Topic topic = session.createTopic(multicastAddress.toString()); + + MessageConsumer consumer1 = session.createConsumer(topic); + MessageConsumer consumer2 = session.createConsumer(topic); + MessageConsumer consumer3 = session.createConsumer(topic); + + MessageProducer producer = session.createProducer(topic); + + producer.send(session.createMessage()); + + //each consumer receives one + Message m = consumer1.receive(2000); + assertNotNull(m); + m = consumer2.receive(2000); + assertNotNull(m); + m = consumer3.receive(2000); + assertNotNull(m); + + Bindings bindings = server.getPostOffice().getBindingsForAddress(multicastAddress); + for (Binding b : bindings.getBindings()) { + System.out.println("checking binidng " + b.getUniqueName() + " " + ((LocalQueueBinding)b).getQueue().getDeliveringMessages()); + SimpleString qName = b.getUniqueName(); + //do FQQN query + QueueQueryResult result = server.queueQuery(CompositeAddress.toFullQN(multicastAddress, qName)); + assertTrue(result.isExists()); + assertEquals(result.getName(), CompositeAddress.toFullQN(multicastAddress, qName)); + //do qname query + result = server.queueQuery(qName); + assertTrue(result.isExists()); + assertEquals(result.getName(), qName); + } + } finally { + connection.close(); + } + } + + @Test + //jms queues know no addresses, this test only shows + //that it is possible for jms clients to receive from + //core queues by its FQQN. + public void testQueue() throws Exception { + server.createQueue(anycastAddress, RoutingType.ANYCAST, anycastQ1, null, true, false, -1, false, true); + server.createQueue(anycastAddress, RoutingType.ANYCAST, anycastQ2, null, true, false, -1, false, true); + server.createQueue(anycastAddress, RoutingType.ANYCAST, anycastQ3, null, true, false, -1, false, true); + + Connection connection = factory.createConnection(); + try { + connection.start(); + Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE); + + Queue q1 = session.createQueue(CompositeAddress.toFullQN(anycastAddress, anycastQ1).toString()); + Queue q2 = session.createQueue(CompositeAddress.toFullQN(anycastAddress, anycastQ2).toString()); + Queue q3 = session.createQueue(CompositeAddress.toFullQN(anycastAddress, anycastQ3).toString()); + + //send 3 messages to anycastAddress + locator = createNonHALocator(true); + ClientSessionFactory cf = createSessionFactory(locator); + ClientSession coreSession = cf.createSession(); + + //send 3 messages + ClientProducer coreProducer = coreSession.createProducer(anycastAddress); + sendMessages(coreSession, coreProducer, 3); + + System.out.println("Queue is: " + q1); + MessageConsumer consumer1 = session.createConsumer(q1); + MessageConsumer consumer2 = session.createConsumer(q2); + MessageConsumer consumer3 = session.createConsumer(q3); + + //each consumer receives one + assertNotNull(consumer1.receive(2000)); + assertNotNull(consumer2.receive(2000)); + assertNotNull(consumer3.receive(2000)); + + connection.close(); + //queues are empty now + for (SimpleString q : new SimpleString[]{anycastQ1, anycastQ2, anycastQ3}) { + //FQQN query + QueueQueryResult query = server.queueQuery(CompositeAddress.toFullQN(anycastAddress, q)); + assertTrue(query.isExists()); + assertEquals(anycastAddress, query.getAddress()); + assertEquals(CompositeAddress.toFullQN(anycastAddress, q), query.getName()); + assertEquals(0, query.getMessageCount()); + //try query again using qName + query = server.queueQuery(q); + assertEquals(q, query.getName()); + } + } finally { + connection.close(); + if (locator != null) { + locator.close(); + } + } + } + + @Test + public void testFQNConsumer() throws Exception { + Connection exConn = null; + + SimpleString durableQueue = new SimpleString("myqueue"); + this.server.createQueue(durableQueue, RoutingType.ANYCAST, durableQueue, null, true, false, -1, false, true); + + try { + ActiveMQConnectionFactory exFact = new ActiveMQConnectionFactory(); + + exConn = exFact.createConnection(); + + exConn.start(); + + Session session = exConn.createSession(false, Session.AUTO_ACKNOWLEDGE); + + Destination destination = session.createQueue(durableQueue.toString()); + + MessageProducer producer = session.createProducer(destination); + + TextMessage message = session.createTextMessage("This is a text message"); + + producer.send(message); + + Destination destinationFQN = session.createQueue(CompositeAddress.toFullQN(durableQueue, durableQueue).toString()); + + MessageConsumer messageConsumer = session.createConsumer(destinationFQN); + + TextMessage messageReceived = (TextMessage) messageConsumer.receive(5000); + + assertEquals("This is a text message", messageReceived.getText()); + } finally { + if (exConn != null) { + exConn.close(); + } + } + } + + @Test + public void testSpecialFQQNCase() throws Exception { + Connection exConn = null; + + SimpleString durableQueue = new SimpleString("myqueue"); + this.server.createQueue(durableQueue, RoutingType.ANYCAST, durableQueue, null, true, false, -1, false, true); + + try { + ActiveMQConnectionFactory exFact = new ActiveMQConnectionFactory(); + exConn = exFact.createConnection(); + exConn.start(); + + Session session = exConn.createSession(false, Session.AUTO_ACKNOWLEDGE); + Destination destination = session.createQueue(durableQueue.toString()); + + MessageProducer producer = session.createProducer(destination); + TextMessage message = session.createTextMessage("This is a text message"); + producer.send(message); + + //this should work as if only queue names is given + Destination destinationFQN = session.createQueue(CompositeAddress.SEPARATOR + durableQueue); + MessageConsumer messageConsumer = session.createConsumer(destinationFQN); + TextMessage messageReceived = (TextMessage) messageConsumer.receive(5000); + assertEquals("This is a text message", messageReceived.getText()); + messageConsumer.close(); + + destinationFQN = session.createQueue(durableQueue + CompositeAddress.SEPARATOR); + try { + session.createConsumer(destinationFQN); + fail("should get exception"); + } catch (InvalidDestinationException e) { + //expected. + } + destinationFQN = session.createQueue(CompositeAddress.SEPARATOR); + try { + session.createConsumer(destinationFQN); + fail("should get exception"); + } catch (InvalidDestinationException e) { + //expected. + } + + } finally { + if (exConn != null) { + exConn.close(); + } + } + } +} diff --git a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/stomp/FQQNStompTest.java b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/stomp/FQQNStompTest.java new file mode 100644 index 0000000000..c29db665a2 --- /dev/null +++ b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/stomp/FQQNStompTest.java @@ -0,0 +1,103 @@ +/* + * 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.artemis.tests.integration.stomp; + +import org.apache.activemq.artemis.api.core.SimpleString; +import org.apache.activemq.artemis.core.server.QueueQueryResult; +import org.apache.activemq.artemis.tests.integration.stomp.util.ClientStompFrame; +import org.apache.activemq.artemis.tests.integration.stomp.util.StompClientConnection; +import org.apache.activemq.artemis.tests.integration.stomp.util.StompClientConnectionFactory; +import org.junit.After; +import org.junit.Before; +import org.junit.Test; + +public class FQQNStompTest extends StompTestBase { + + private StompClientConnection conn; + + @Override + @Before + public void setUp() throws Exception { + super.setUp(); + conn = StompClientConnectionFactory.createClientConnection("1.2", hostname, port); + QueueQueryResult result = server.getActiveMQServer().queueQuery(new SimpleString(getQueueName())); + assertTrue(result.isExists()); + System.out.println("address: " + result.getAddress() + " queue " + result.getName()); + } + + @Override + @After + public void tearDown() throws Exception { + try { + boolean connected = conn != null && conn.isConnected(); + if (connected) { + try { + conn.disconnect(); + } catch (Exception e) { + } + } + } finally { + super.tearDown(); + } + } + + @Test + //to receive from a FQQN queue like testQueue::testQueue + //special care is needed as ":" is a reserved character + //in STOMP. Clients need to escape it. + public void testReceiveFQQN() throws Exception { + conn.connect(defUser, defPass); + subscribeQueue(conn, "sub-01", getQueueName() + "\\c\\c" + getQueueName()); + sendJmsMessage("Hello World!"); + ClientStompFrame frame = conn.receiveFrame(2000); + assertNotNull(frame); + assertEquals("Hello World!", frame.getBody()); + System.out.println("frame: " + frame); + unsubscribe(conn, "sub-01"); + } + + @Test + public void testReceiveFQQNSpecial() throws Exception { + conn.connect(defUser, defPass); + //::queue + subscribeQueue(conn, "sub-01", "\\c\\c" + getQueueName()); + sendJmsMessage("Hello World!"); + ClientStompFrame frame = conn.receiveFrame(2000); + assertNotNull(frame); + assertEquals("Hello World!", frame.getBody()); + System.out.println("frame: " + frame); + unsubscribe(conn, "sub-01"); + + //queue:: + subscribeQueue(conn, "sub-01", getQueueName() + "\\c\\c"); + sendJmsMessage("Hello World!"); + frame = conn.receiveFrame(2000); + assertNotNull(frame); + assertEquals("ERROR", frame.getCommand()); + assertTrue(frame.getBody().contains(getQueueName())); + assertTrue(frame.getBody().contains("not exist")); + + //need reconnect because stomp disconnect on error + conn = StompClientConnectionFactory.createClientConnection("1.2", hostname, port); + //:: will subscribe to no queue so no message received. + subscribeQueue(conn, "sub-01", "\\c\\c"); + sendJmsMessage("Hello World!"); + frame = conn.receiveFrame(2000); + assertNull(frame); + } + +} diff --git a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/stomp/StompTestBase.java b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/stomp/StompTestBase.java index 0641aedbed..f885659194 100644 --- a/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/stomp/StompTestBase.java +++ b/tests/integration-tests/src/test/java/org/apache/activemq/artemis/tests/integration/stomp/StompTestBase.java @@ -355,6 +355,10 @@ public abstract class StompTestBase extends ActiveMQTestBase { return subscribe(conn, subscriptionId, ack, durableId, selector, getQueuePrefix() + getQueueName(), receipt); } + public void subscribeQueue(StompClientConnection conn, String subId, String destination) throws IOException, InterruptedException { + subscribe(conn, subId, Stomp.Headers.Subscribe.AckModeValues.AUTO, null, null, destination, false); + } + public ClientStompFrame subscribe(StompClientConnection conn, String subscriptionId, String ack, diff --git a/tests/unit-tests/src/test/java/org/apache/activemq/artemis/tests/unit/core/postoffice/impl/WildcardAddressManagerUnitTest.java b/tests/unit-tests/src/test/java/org/apache/activemq/artemis/tests/unit/core/postoffice/impl/WildcardAddressManagerUnitTest.java index b74a4c0323..7fc9b232ba 100644 --- a/tests/unit-tests/src/test/java/org/apache/activemq/artemis/tests/unit/core/postoffice/impl/WildcardAddressManagerUnitTest.java +++ b/tests/unit-tests/src/test/java/org/apache/activemq/artemis/tests/unit/core/postoffice/impl/WildcardAddressManagerUnitTest.java @@ -69,6 +69,36 @@ public class WildcardAddressManagerUnitTest extends ActiveMQTestBase { assertEquals("Exception happened during the process", 0, errors); } + @Test + public void testUnitOnWildCardFailingScenarioFQQN() throws Exception { + int errors = 0; + WildcardAddressManager ad = new WildcardAddressManager(new BindingFactoryFake()); + ad.addBinding(new BindingFake("Topic1", "Topic1")); + ad.addBinding(new BindingFake("Topic1", "one")); + ad.addBinding(new BindingFake("*", "two")); + ad.removeBinding(SimpleString.toSimpleString("Topic1::one"), null); + try { + ad.removeBinding(SimpleString.toSimpleString("*::two"), null); + } catch (Throwable e) { + // We are not failing the test here as this test is replicating the exact scenario + // that was happening under https://issues.jboss.org/browse/HORNETQ-988 + // In which this would be ignored + errors++; + e.printStackTrace(); + } + try { + ad.addBinding(new BindingFake("Topic1", "three")); + } catch (Throwable e) { + // We are not failing the test here as this test is replicating the exact scenario + // that was happening under https://issues.jboss.org/browse/HORNETQ-988 + // In which this would be ignored + errors++; + e.printStackTrace(); + } + + assertEquals("Exception happened during the process", 0, errors); + } + class BindingFactoryFake implements BindingsFactory { @Override