ARTEMIS-4085 exclusive LVQ sending all messages to consumer

This commit is contained in:
Justin Bertram 2022-11-07 15:07:35 -06:00 committed by clebertsuconic
parent 3a5e4ce363
commit ca580814de
3 changed files with 159 additions and 0 deletions

View File

@ -3155,6 +3155,7 @@ public class QueueImpl extends CriticalComponentImpl implements Queue {
if (groupConsumer != null) {
if (noDelivery > 0) {
pruneLastValues();
break;
}
noDelivery = 0;

View File

@ -86,6 +86,30 @@ public class LVQTest extends ActiveMQTestBase {
Assert.assertEquals(m.getBodyBuffer().readString(), "m2");
}
@Test
public void testSimpleExclusive() throws Exception {
ServerLocator locator = createNettyNonHALocator().setConsumerWindowSize(0);
ClientSessionFactory sf = createSessionFactory(locator);
ClientSession clientSession = addClientSession(sf.createSession(false, true, true));
final String EXCLUSIVE_QUEUE = "exclusiveQueue";
clientSession.createQueue(new QueueConfiguration(EXCLUSIVE_QUEUE).setExclusive(true).setLastValue(true));
ClientProducer producer = clientSession.createProducer(EXCLUSIVE_QUEUE);
ClientConsumer consumer = clientSession.createConsumer(EXCLUSIVE_QUEUE);
clientSession.start();
ClientMessage m1 = createTextMessage(clientSession, "m1");
SimpleString rh = new SimpleString("SMID1");
m1.putStringProperty(Message.HDR_LAST_VALUE_NAME, rh);
ClientMessage m2 = createTextMessage(clientSession, "m2");
m2.putStringProperty(Message.HDR_LAST_VALUE_NAME, rh);
producer.send(m1);
producer.send(m2);
ClientMessage m = consumer.receive(1000);
Assert.assertNotNull(m);
m.acknowledge();
Assert.assertEquals(m.getBodyBuffer().readString(), "m2");
}
@Test
public void testSimpleRestart() throws Exception {
ClientProducer producer = clientSession.createProducer(address);

View File

@ -0,0 +1,134 @@
/*
* 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 java.lang.invoke.MethodHandles;
import java.util.ArrayList;
import java.util.List;
import java.util.UUID;
import org.apache.activemq.artemis.api.core.Message;
import org.apache.activemq.artemis.api.core.QueueConfiguration;
import org.apache.activemq.artemis.core.protocol.stomp.Stomp;
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.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@RunWith(Parameterized.class)
public class StompLVQTest extends StompTestBase {
private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
protected StompClientConnection producerConn;
protected StompClientConnection consumerConn;
private final String queue = "lvq";
@Override
@Before
public void setUp() throws Exception {
super.setUp();
server.createQueue(new QueueConfiguration(queue).setLastValue(true).setExclusive(true));
producerConn = StompClientConnectionFactory.createClientConnection(uri);
consumerConn = StompClientConnectionFactory.createClientConnection(uri);
}
@Override
@After
public void tearDown() throws Exception {
try {
if (producerConn != null && producerConn.isConnected()) {
try {
producerConn.disconnect();
} catch (Exception e) {
// ignore
}
}
} finally {
producerConn.closeTransport();
}
try {
if (consumerConn != null && consumerConn.isConnected()) {
try {
consumerConn.disconnect();
} catch (Exception e) {
// ignore
}
}
} finally {
consumerConn.closeTransport();
}
super.tearDown();
}
@Test
public void testLVQ() throws Exception {
producerConn.connect(defUser, defPass);
consumerConn.connect(defUser, defPass);
subscribe(consumerConn, "lvqtest", Stomp.Headers.Subscribe.AckModeValues.CLIENT, null, null, queue, true, 0);
try {
for (int i = 1; i <= 100; i++) {
String uuid = UUID.randomUUID().toString();
ClientStompFrame frame = producerConn.sendFrame(producerConn.createFrame(Stomp.Commands.SEND)
.addHeader(Stomp.Headers.Send.DESTINATION, queue)
.addHeader(Message.HDR_LAST_VALUE_NAME.toString(), "test")
.addHeader(Stomp.Headers.RECEIPT_REQUESTED, uuid)
.setBody(String.valueOf(i)));
assertEquals(Stomp.Responses.RECEIPT, frame.getCommand());
assertEquals(uuid, frame.getHeader(Stomp.Headers.Response.RECEIPT_ID));
}
} catch (Exception e) {
logger.error(null, e);
}
List<ClientStompFrame> messages = new ArrayList<>();
try {
ClientStompFrame frame;
while ((frame = consumerConn.receiveFrame(10000)) != null) {
assertEquals(Stomp.Responses.MESSAGE, frame.getCommand());
ack(consumerConn, null, frame);
messages.add(frame);
}
} catch (Exception e) {
logger.error(null, e);
}
Assert.assertEquals(2, messages.size());
Assert.assertEquals("1", messages.get(0).getBody());
Assert.assertEquals("100", messages.get(1).getBody());
}
}