com.sitewhere.sources.ActiveMQTests.java Source code

Introduction

Here is the source code for com.sitewhere.sources.ActiveMQTests.java

Source

/*
 * Copyright (c) SiteWhere, LLC. All rights reserved. http://www.sitewhere.com
 *
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */
package com.sitewhere.sources;

import java.util.concurrent.Callable;
import java.util.concurrent.CompletionService;
import java.util.concurrent.ExecutorCompletionService;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

import javax.jms.BytesMessage;
import javax.jms.Destination;
import javax.jms.MessageProducer;
import javax.jms.Session;

import org.apache.activemq.ActiveMQConnectionFactory;
import org.apache.qpid.proton.Proton;
import org.apache.qpid.proton.amqp.Binary;
import org.apache.qpid.proton.amqp.messaging.Data;
import org.apache.qpid.proton.message.Message;
import org.apache.qpid.proton.messenger.Messenger;
import org.junit.Test;

import com.rabbitmq.client.Channel;
import com.rabbitmq.client.Connection;
import com.rabbitmq.client.ConnectionFactory;

public class ActiveMQTests {

    /** Hardware id for test message */
    private static final String HARDWARE_ID = "4eceeda1-9ab2-4069-9372-7aa007a3dccc";

    /** Nunber of threads for multithreaded tests */
    private static final int NUM_THREADS = 150;

    /** Nunber of calls performed per thread */
    private static final int NUM_CALLS_PER_THREAD = 400;

    @Test
    public void doJMSTest() throws Exception {
        (new JmsTester(1)).call();
    }

    @Test
    public void doMultithreadedJmsTest() throws Exception {
        ExecutorService executor = Executors.newFixedThreadPool(NUM_THREADS);
        CompletionService<Void> completionService = new ExecutorCompletionService<Void>(executor);

        for (int i = 0; i < NUM_THREADS; i++) {
            completionService.submit(new JmsTester(NUM_CALLS_PER_THREAD));
        }
        for (int i = 0; i < NUM_THREADS; ++i) {
            completionService.take().get();
        }
    }

    @Test
    public void doRabbitMQTest() throws Exception {
        String exchangeName = "sitewhere";
        String queueName = "sitewhere.input";
        String routingKey = "sitewhere";

        ConnectionFactory factory = new ConnectionFactory();
        factory.setUri("amqp://localhost:5672");
        Connection connection = factory.newConnection();
        Channel channel = connection.createChannel();

        channel.exchangeDeclare(exchangeName, "direct", true);
        channel.queueDeclare(queueName, true, false, false, null);
        channel.queueBind(queueName, exchangeName, routingKey);

        byte[] messageBodyBytes = EventsHelper.generateEncodedMeasurementsMessage(HARDWARE_ID);
        channel.basicPublish(exchangeName, routingKey, null, messageBodyBytes);

        channel.close();
        connection.close();
    }

    @Test
    public void doQpidTest() throws Exception {
        Messenger messenger = Proton.messenger();
        messenger.start();

        Data data = new Data(new Binary(EventsHelper.generateEncodedMeasurementsMessage(HARDWARE_ID)));

        Message message = Proton.message();
        message.setAddress("amqp://127.0.0.1:5672/SITEWHERE.IN");
        message.setBody(data);
        messenger.put(message);

        messenger.send();
        messenger.stop();
    }

    public class JmsTester implements Callable<Void> {

        private int messageCount;

        public JmsTester(int messageCount) {
            this.messageCount = messageCount;
        }

        @Override
        public Void call() throws Exception {
            ActiveMQConnectionFactory connectionFactory = new ActiveMQConnectionFactory("tcp://localhost:1234");
            javax.jms.Connection connection = connectionFactory.createConnection();
            connection.start();

            Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
            Destination destination = session.createQueue("SITEWHERE.IN");

            MessageProducer producer = session.createProducer(destination);

            for (int i = 0; i < messageCount; i++) {
                BytesMessage message = session.createBytesMessage();
                // message.writeBytes(EventsHelper.generateEncodedMeasurementsMessage(HARDWARE_ID));
                message.writeBytes(EventsHelper.generateEncodedRegistrationMessage());
                producer.send(message);
            }

            session.close();
            connection.close();
            return null;
        }
    }
}