This closes #542
This commit is contained in:
commit
385bd047fe
|
@ -1233,9 +1233,7 @@ public class ServerSessionImpl implements ServerSession, FailureListener {
|
|||
|
||||
SimpleString address = message.getAddress();
|
||||
|
||||
if (defaultAddress == null && address != null) {
|
||||
defaultAddress = address;
|
||||
}
|
||||
checkDefaultAddress(address);
|
||||
|
||||
if (address == null) {
|
||||
if (message.isDurable()) {
|
||||
|
@ -1269,6 +1267,12 @@ public class ServerSessionImpl implements ServerSession, FailureListener {
|
|||
return result;
|
||||
}
|
||||
|
||||
protected void checkDefaultAddress(SimpleString address) {
|
||||
if (defaultAddress == null && address != null) {
|
||||
defaultAddress = address;
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void sendContinuations(final int packetSize,
|
||||
final long messageBodySize,
|
||||
|
@ -1298,6 +1302,10 @@ public class ServerSessionImpl implements ServerSession, FailureListener {
|
|||
|
||||
@Override
|
||||
public void requestProducerCredits(final SimpleString address, final int credits) throws Exception {
|
||||
// When the client gets the producer credits it already has the intention of sending messages
|
||||
// and it will already register the default address on the core protocol
|
||||
// hence we need to set it here when we request credits as well
|
||||
checkDefaultAddress(address);
|
||||
PagingStore store = server.getPagingManager().getPageStore(address);
|
||||
|
||||
if (!store.checkMemory(new Runnable() {
|
||||
|
|
|
@ -0,0 +1,171 @@
|
|||
/**
|
||||
* 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 javax.jms.Connection;
|
||||
import javax.jms.DeliveryMode;
|
||||
import javax.jms.Message;
|
||||
import javax.jms.MessageProducer;
|
||||
import javax.jms.Queue;
|
||||
import javax.jms.Session;
|
||||
|
||||
import org.apache.activemq.artemis.api.core.SimpleString;
|
||||
import org.apache.activemq.artemis.api.core.management.ResourceNames;
|
||||
import org.apache.activemq.artemis.api.jms.management.JMSQueueControl;
|
||||
import org.apache.activemq.artemis.core.settings.HierarchicalRepository;
|
||||
import org.apache.activemq.artemis.core.settings.impl.AddressFullMessagePolicy;
|
||||
import org.apache.activemq.artemis.core.settings.impl.AddressSettings;
|
||||
import org.apache.activemq.artemis.core.settings.impl.SlowConsumerPolicy;
|
||||
import org.apache.activemq.artemis.tests.util.JMSTestBase;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
|
||||
public class MultipleProducersTest extends JMSTestBase {
|
||||
|
||||
public Connection conn;
|
||||
public Queue queueOne = null;
|
||||
public Queue queueTwo = null;
|
||||
public Session session = null;
|
||||
|
||||
public SimpleString dlq = new SimpleString("jms.queue.DLQ");
|
||||
public SimpleString expiryQueue = new SimpleString("jms.queue.ExpiryQueue");
|
||||
|
||||
public SimpleString queueOneName = new SimpleString("jms.queue.queueOne");
|
||||
public SimpleString queueTwoName = new SimpleString("jms.queue.queueTwo");
|
||||
public JMSQueueControl control = null;
|
||||
public long queueOneMsgCount = 0;
|
||||
public long queueTwoMsgCount = 0;
|
||||
|
||||
@Before
|
||||
public void iniTest() throws Exception {
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void wrongQueue() throws Exception {
|
||||
|
||||
conn = cf.createConnection();
|
||||
|
||||
session = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
|
||||
|
||||
HierarchicalRepository<AddressSettings> repos = server.getAddressSettingsRepository();
|
||||
|
||||
AddressSettings addressSettings = new AddressSettings();
|
||||
|
||||
addressSettings.setAddressFullMessagePolicy(AddressFullMessagePolicy.FAIL);
|
||||
addressSettings.setExpiryAddress(dlq);
|
||||
addressSettings.setDeadLetterAddress(expiryQueue);
|
||||
addressSettings.setRedeliveryDelay(0);
|
||||
addressSettings.setMessageCounterHistoryDayLimit(2);
|
||||
addressSettings.setLastValueQueue(false);
|
||||
addressSettings.setMaxDeliveryAttempts(10);
|
||||
addressSettings.setMaxSizeBytes(1048576);
|
||||
addressSettings.setPageCacheMaxSize(5);
|
||||
addressSettings.setPageSizeBytes(2097152);
|
||||
addressSettings.setRedistributionDelay(-1);
|
||||
addressSettings.setSendToDLAOnNoRoute(false);
|
||||
addressSettings.setSlowConsumerCheckPeriod(5);
|
||||
addressSettings.setSlowConsumerPolicy(SlowConsumerPolicy.NOTIFY);
|
||||
addressSettings.setSlowConsumerThreshold(-1);
|
||||
|
||||
repos.setDefault(addressSettings);
|
||||
|
||||
queueOne = createQueue("queueOne");
|
||||
|
||||
queueTwo = createQueue("queueTwo");
|
||||
|
||||
try {
|
||||
while (true) {
|
||||
sendMessage(queueOne, session);
|
||||
}
|
||||
}
|
||||
catch (Throwable t) {
|
||||
// expected
|
||||
}
|
||||
|
||||
session.close();
|
||||
|
||||
conn.close();
|
||||
|
||||
session = null;
|
||||
conn = null;
|
||||
|
||||
conn = cf.createConnection();
|
||||
session = conn.createSession(false, Session.AUTO_ACKNOWLEDGE);
|
||||
|
||||
// send a message to a queue which is already full
|
||||
// result an exception
|
||||
try {
|
||||
sendMessage(queueOne, session);
|
||||
Assert.fail("Exception expected");
|
||||
}
|
||||
catch (Exception t) {
|
||||
}
|
||||
|
||||
// send 5 message to queueTwo
|
||||
// there shoudl be 5 messages on queueTwo
|
||||
for (int i = 0; i < 5; i++) {
|
||||
sendMessage(queueTwo, session);
|
||||
}
|
||||
|
||||
// before sending any messages to queueOne it has to be drained.
|
||||
// after draining queueOne send 5 message to queueOne
|
||||
queueTwoMsgCount = server.locateQueue(queueTwoName).getMessageCount();
|
||||
|
||||
control = (JMSQueueControl) server.getManagementService().getResource(ResourceNames.JMS_QUEUE + queueOne.getQueueName());
|
||||
|
||||
control.removeMessages(null);
|
||||
|
||||
for (int i = 0; i < 5; i++) {
|
||||
sendMessage(queueOne, session);
|
||||
}
|
||||
|
||||
// at the end of the test there shoudl be 5 message on queueOne and 5 messages on queueTwo
|
||||
|
||||
session.close();
|
||||
|
||||
conn.close();
|
||||
|
||||
queueOneMsgCount = server.locateQueue(queueOneName).getMessageCount();
|
||||
|
||||
queueTwoMsgCount = server.locateQueue(queueTwoName).getMessageCount();
|
||||
|
||||
Assert.assertEquals("queueTwo message count", 5, queueTwoMsgCount);
|
||||
Assert.assertEquals("queueOne message count", 5, queueOneMsgCount);
|
||||
|
||||
}
|
||||
|
||||
private void sendMessage(Queue queue, Session session) throws Exception {
|
||||
|
||||
MessageProducer mp = session.createProducer(queue);
|
||||
|
||||
try {
|
||||
mp.setDisableMessageID(true);
|
||||
mp.setDeliveryMode(DeliveryMode.NON_PERSISTENT);
|
||||
mp.setPriority(Message.DEFAULT_PRIORITY);
|
||||
mp.setTimeToLive(Message.DEFAULT_TIME_TO_LIVE);
|
||||
|
||||
mp.send(session.createTextMessage("This is message for " + queue.getQueueName()));
|
||||
}
|
||||
finally {
|
||||
|
||||
mp.close();
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue