MessageFactory.java

package org.openamq.headers;

import org.openamq.framing.FieldTable;
import org.openamq.client.AMQSession;

import javax.jms.*;

/**
 */
class MessageFactory
{
    private static final char[] DATA = "abcdefghijklmnopqrstuvwxyz".toCharArray();

    private final AMQSession _session;
    private final byte[] _payload;

    private String[] _headerNames;

    MessageFactory(AMQSession session)
    {
        this(session, Integer.getInteger("amqj.test.message_size", 256).intValue(), 5);
    }

    MessageFactory(AMQSession session, int payloadSize, int headerCount)
    {
        if (headerCount < 1)
        {
            throw new IllegalArgumentException("Header count must be positive");
        }
        _session = session;
        _payload = new byte[payloadSize];
        for(int i = 0; i < _payload.length; i++)
        {
            _payload[i] = (byte) DATA[i % DATA.length];
        }
        _headerNames = new String[headerCount];
        // note that with the standard encoding the headers get prefixed with an S to indicate their type
        for (int i = 0; i < _headerNames.length; i++)
        {
            if (i < 10)
            {
                _headerNames[i] = "F000" + i;
            }
            else if (i >= 10 && i < 100)
            {
                _headerNames[i] = "F00" + i;
            }
            else
            {
                _headerNames[i] = "F0" + i;
            }
        }
    }

    Message createEventMessage() throws JMSException
    {
        BytesMessage msg = _session.createBytesMessage();
        if (_payload.length != 0)
        {
            msg.writeBytes(_payload);
        }
        return setHeaders(msg, _headerNames);
    }

    Message createShutdownMessage() throws JMSException
    {
        return setHeaders(_session.createMessage(), new String[]{"F0000", "SHUTDOWN"});
    }

    Message createReportRequestMessage() throws JMSException
    {
        return setHeaders(_session.createMessage(), new String[]{"F0000", "REPORT"});
    }

    Message createReportResponseMessage(String msg) throws JMSException
    {
        return setHeaders(_session.createTextMessage(msg), new String[]{"CONTROL", "REPORT"});
    }

    boolean isShutdown(Message m)
    {
        return checkPresent(m, "SHUTDOWN");
    }

    boolean isReport(Message m)
    {
        return checkPresent(m, "REPORT");
    }

    Object getReport(Message m)
    {
        try
        {
            return ((TextMessage) m).getText();
        }
        catch (JMSException e)
        {
            e.printStackTrace(System.out);
            return e.toString();
        }
    }

    FieldTable getConsumerBinding()
    {
        FieldTable binding = new FieldTable();
        binding.put("SF0000", "value");
        return binding;
    }

    FieldTable getControllerBinding()
    {
        FieldTable binding = new FieldTable();
        binding.put("SCONTROL", "value");
        return binding;
    }

    MessageConsumer createConsumer(Destination source) throws Exception
    {
        return _session.createConsumer(source, 0, false, true, null, getConsumerBinding());
    }

    MessageConsumer createController(Destination source) throws Exception
    {
        return _session.createConsumer(source, 0, false, true, null, getControllerBinding());
    }

    private static boolean checkPresent(Message m, String s)
    {
        try
        {
            return m.getStringProperty(s) != null;
        }
        catch (JMSException e)
        {
            e.printStackTrace(System.out);
            return false;
        }
    }

    private static Message setHeaders(Message m, String[] headers) throws JMSException
    {
        for(int i = 0; i < headers.length; i++)
        {
            // the value in GRM is 5 bytes
            m.setStringProperty(headers[i], "value");
        }
        return m;
    }
}