/*
* 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.scheduling;
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.settings.impl.AddressSettings;
import org.apache.activemq.artemis.jms.client.ActiveMQTextMessage;
import org.apache.activemq.artemis.tests.integration.IntegrationTestLogger;
import org.apache.activemq.artemis.tests.util.ActiveMQTestBase;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
public class MultipliedDelayedMessageTest extends ActiveMQTestBase {
private static final IntegrationTestLogger log = IntegrationTestLogger.LOGGER;
private ActiveMQServer server;
private static final long DELAY = 1000;
private static final double MULTIPLIER = 2.0;
private static final long MAX_DELAY = 17000;
private final String queueName = "MultipliedDelayedMessageTestQueue";
private ServerLocator locator;
@Override
@Before
public void setUp() throws Exception {
super.setUp();
initServer();
}
/**
* @throws Exception
*/
protected void initServer() throws Exception {
server = createServer(true, createDefaultInVMConfig());
server.start();
// Create settings to enable multiplied redelivery delay
AddressSettings addressSettings = server.getAddressSettingsRepository().getMatch("*");
AddressSettings newAddressSettings = new AddressSettings().setRedeliveryDelay(DELAY).setRedeliveryMultiplier(MULTIPLIER).setMaxRedeliveryDelay(MAX_DELAY);
newAddressSettings.merge(addressSettings);
server.getAddressSettingsRepository().addMatch(queueName, newAddressSettings);
locator = createInVMNonHALocator();
}
@Test
public void testMultipliedDelayedRedeliveryOnClose() throws Exception {
ClientSessionFactory sessionFactory = createSessionFactory(locator);
// Session for creating the queue
ClientSession session = sessionFactory.createSession(false, false, false);
session.createQueue(queueName, queueName, null, true);
session.close();
// Session for sending the message
session = sessionFactory.createSession(false, true, true);
ClientProducer producer = session.createProducer(queueName);
ActiveMQTestBase.forceGC();
ClientMessage tm = createDurableMessage(session, "message");
producer.send(tm);
session.close();
// Session for consuming the message
session = sessionFactory.createSession(false, false, false);
ClientConsumer consumer = session.createConsumer(queueName);
session.start();
tm = consumer.receive(500);
Assert.assertNotNull(tm);
for (int i = 1; i <= 6; i++) {
// Ack the message, but rollback the session to trigger redelivery with increasing delivery count
long start = System.currentTimeMillis();
tm.acknowledge();
session.rollback();
long expectedDelay = calculateExpectedDelay(DELAY, MAX_DELAY, MULTIPLIER, i);
log.info("\nExpected delay: " + expectedDelay);
tm = consumer.receive(expectedDelay + 500);
long stop = System.currentTimeMillis();
Assert.assertNotNull(tm);
log.info("Actual delay: " + (stop - start));
Assert.assertTrue(stop - start >= expectedDelay);
}
tm.acknowledge();
session.commit();
session.close();
}
// Private -------------------------------------------------------
private ClientMessage createDurableMessage(final ClientSession session, final String body) {
ClientMessage message = session.createMessage(ActiveMQTextMessage.TYPE, true, 0, System.currentTimeMillis(), (byte) 1);
message.getBodyBuffer().writeString(body);
return message;
}
// This is based on org.apache.activemq.artemis.core.server.impl.QueueImpl.calculateRedeliveryDelay()
private long calculateExpectedDelay(final long redeliveryDelay,
final long maxRedeliveryDelay,
final double redeliveryMultiplier,
final int deliveryCount) {
int tmpDeliveryCount = deliveryCount > 0 ? deliveryCount - 1 : 0;
long delay = (long) (redeliveryDelay * (Math.pow(redeliveryMultiplier, tmpDeliveryCount)));
if (delay > maxRedeliveryDelay) {
delay = maxRedeliveryDelay;
}
return delay;
}
}