com.atlantbh.jmeter.plugins.jmstools.JmsUtil.java Source code

Introduction

Here is the source code for com.atlantbh.jmeter.plugins.jmstools.JmsUtil.java

Source

/*!
 * AtlantBH Custom Jmeter Components v1.0.0
 * http://www.atlantbh.com/jmeter-components/
 *
 * Copyright 2011, AtlantBH
 *
 * Licensed under the under the Apache License, Version 2.0.
 */

package com.atlantbh.jmeter.plugins.jmstools;

import java.util.HashMap;
import java.util.Map;

import javax.jms.ConnectionFactory;
import javax.jms.Destination;

import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.protocol.java.sampler.JavaSamplerClient;
import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.qpid.client.AMQAnyDestination;
import org.apache.qpid.client.AMQConnectionFactory;
import org.apache.qpid.url.AMQBindingURL;
import org.springframework.jms.core.JmsTemplate;

public class JmsUtil implements JavaSamplerClient {

    @Override
    public Arguments getDefaultParameters() {

        Arguments args = new Arguments();
        args.addArgument("connection.url", "");
        args.addArgument("binding.url", "");
        args.addArgument("message", "");
        args.addArgument("direction", "");
        args.addArgument("header.properties", "");
        args.addArgument("header.property.reference", "");
        args.addArgument("message.selector", "");
        args.addArgument("receive.timeout", "");
        return args;
    }

    @Override
    public SampleResult runTest(JavaSamplerContext ctx) {

        SampleResult result = new SampleResult();
        result.setContentType("plain/text");
        result.setDataType(SampleResult.TEXT);
        result.setDataEncoding(SampleResult.DEFAULT_HTTP_ENCODING);

        String connectionUrl = ctx.getParameter("connection.url");
        String bindingUrl = ctx.getParameter("binding.url");
        String message = ctx.getParameter("message");

        if (connectionUrl == null || "".equals(connectionUrl)) {
            result.setSuccessful(false);
            result.setResponseMessage("Connection URL cannot be empty.");
            result.setResponseCode("0xDEAD");
        } else {
            if (bindingUrl == null || "".equals(bindingUrl)) {
                result.setSuccessful(false);
                result.setResponseMessage("Binding URL cannot be empty.");
                result.setResponseCode("0xDEAD");
            } else {
                try {
                    ConnectionFactory connectionFactory = new AMQConnectionFactory(connectionUrl);
                    AMQBindingURL burl = new AMQBindingURL(bindingUrl);

                    Destination destinationProducer = new AMQAnyDestination(burl);
                    JmsTemplate sender = new JmsTemplate();
                    sender.setConnectionFactory(connectionFactory);
                    sender.setDefaultDestination(destinationProducer);
                    BinaryMessageConverter bmc = new BinaryMessageConverter();
                    sender.setMessageConverter(bmc);

                    BinaryMessagepostProcessor postProcessor = new BinaryMessagepostProcessor();

                    sender.setDeliveryMode(2);
                    int rt = 30000;
                    try {
                        rt = Integer.valueOf(ctx.getParameter("receive.timeout"));
                    } catch (Exception e) {
                    }

                    sender.setReceiveTimeout(rt);

                    String direction = ctx.getParameter("direction");
                    if (direction == null || "".equals(direction)) {
                        direction = "send";
                    }
                    if (direction.toLowerCase().equals("send")) {
                        Map<String, String> mp = getMessageProperties(ctx.getParameter("header.properties"));
                        postProcessor.setMessageProperties(mp);
                        sender.convertAndSend((Object) message, postProcessor);
                        result.setSuccessful(true);
                        result.setResponseMessage("Message sent.");
                    } else {
                        if (direction.toLowerCase().equals("receive")) {

                            System.out.println("Receive");
                            String messageSelector = ctx.getParameter("message.selector");
                            System.out.println("Selector: " + messageSelector);
                            Object obj = null;
                            if (messageSelector != null && !"".equals(messageSelector)) {
                                obj = sender.receiveSelectedAndConvert(messageSelector);
                            } else {
                                obj = sender.receiveAndConvert();
                            }

                            if (obj != null) {
                                result.setSuccessful(true);
                                result.setResponseData(obj.toString().getBytes());
                                String paramName = ctx.getParameter("header.property.reference");
                                if (paramName != null && !"".equals(paramName))
                                    JMeterUtils.setProperty(paramName,
                                            concatProperties(bmc.getMessageProperties()));
                            } else {
                                result.setSuccessful(false);
                                result.setResponseData("Conection timeout".getBytes());

                            }

                        } else {
                            result.setSuccessful(false);
                            result.setResponseMessage("Unknown direction.");

                        }
                    }
                } catch (Exception e) {
                    e.printStackTrace();
                    result.setSuccessful(!true);
                    result.setResponseMessage("Exception");
                    result.setResponseData(e.getMessage().getBytes());
                }

            }

        }

        return result;
    }

    private String concatProperties(Map<String, String> messageProperties) {
        String ret = "";
        if (messageProperties != null) {
            for (String key : messageProperties.keySet()) {
                ret += key + "=" + messageProperties.get(key) + "&&";
            }
        }
        return ret;
    }

    private Map<String, String> getMessageProperties(String parameter) {
        if (parameter != null && !"".equals(parameter)) {
            Map<String, String> ret = new HashMap<String, String>();
            String[] pairs = parameter.split("&&");
            for (String pair : pairs) {
                String[] parts = pair.split("=");
                if (parts.length >= 2) {
                    ret.put(parts[0], parts[1]);
                }
            }
            return ret;
        } else {
            return new HashMap<String, String>();
        }

    }

    @Override
    public void setupTest(JavaSamplerContext arg0) {
    }

    @Override
    public void teardownTest(JavaSamplerContext arg0) {
    }
}