| /* |
| * 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.camel.itest.jms; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| import javax.jms.ConnectionFactory; |
| |
| import org.apache.camel.Header; |
| import org.apache.camel.builder.RouteBuilder; |
| import org.apache.camel.component.jms.JmsComponent; |
| import org.apache.camel.itest.CamelJmsTestHelper; |
| import org.apache.camel.spi.Registry; |
| import org.apache.camel.test.junit4.CamelTestSupport; |
| import org.junit.Test; |
| |
| import static org.apache.camel.component.jms.JmsComponent.jmsComponentAutoAcknowledge; |
| |
| public class JmsPerformanceTest extends CamelTestSupport { |
| private List<Integer> receivedHeaders = new ArrayList<>(getMessageCount()); |
| private List<Object> receivedMessages = new ArrayList<>(getMessageCount()); |
| |
| @Test |
| public void testSendingAndReceivingMessages() throws Exception { |
| log.info("Sending {} messages", getMessageCount()); |
| |
| sendLoop(getMessageCount()); |
| |
| log.info("Sending {} messages completed, now will assert on their content as well as the order of their receipt", getMessageCount()); |
| |
| // should wait a bit to make sure all messages have been received by the MyBean#onMessage() method |
| // as this happens asynchronously, that's not inside the 'main' thread |
| Thread.sleep(3000); |
| |
| assertExpectedMessagesReceived(); |
| } |
| |
| protected int getMessageCount() { |
| return 100; |
| } |
| |
| protected void sendLoop(int messageCount) { |
| for (int i = 1; i <= messageCount; i++) { |
| sendMessage(i); |
| } |
| } |
| |
| protected void sendMessage(int messageCount) { |
| template.sendBodyAndHeader("activemq:" + getQueueName(), "Hello:" + messageCount, "counter", messageCount); |
| } |
| |
| protected String getQueueName() { |
| return "testSendingAndReceivingMessages"; |
| } |
| |
| protected void assertExpectedMessagesReceived() throws InterruptedException { |
| // assert on the expected message count |
| assertEquals("The expected message count does not match!", getMessageCount(), receivedMessages.size()); |
| |
| // assert on the expected message order |
| List<Integer> expectedHeaders = new ArrayList<>(getMessageCount()); |
| for (int i = 1; i <= getMessageCount(); i++) { |
| expectedHeaders.add(i); |
| } |
| |
| List<Object> expectedMessages = new ArrayList<>(getMessageCount()); |
| for (int i = 1; i <= getMessageCount(); i++) { |
| expectedMessages.add("Hello:" + i); |
| } |
| |
| assertEquals("The expected header order does not match!", expectedHeaders, receivedHeaders); |
| assertEquals("The expected message order does not match!", expectedMessages, receivedMessages); |
| } |
| |
| @Override |
| protected void bindToRegistry(Registry registry) throws Exception { |
| // add AMQ client and make use of connection pooling we depend on because of the (large) number |
| // of the JMS messages we do produce |
| // add ActiveMQ with embedded broker |
| ConnectionFactory connectionFactory = CamelJmsTestHelper.createConnectionFactory(); |
| JmsComponent amq = jmsComponentAutoAcknowledge(connectionFactory); |
| amq.setCamelContext(context); |
| |
| registry.bind("myBean", new MyBean()); |
| registry.bind("activemq", amq); |
| } |
| |
| @Override |
| protected RouteBuilder createRouteBuilder() throws Exception { |
| return new RouteBuilder() { |
| public void configure() { |
| from("activemq:" + getQueueName()).to("bean:myBean"); |
| } |
| }; |
| } |
| |
| protected class MyBean { |
| public void onMessage(@Header("counter") int counter, Object body) { |
| // the invocation of this method happens inside the same thread so no need for a thread-safe list here |
| receivedHeaders.add(counter); |
| receivedMessages.add(body); |
| } |
| } |
| } |