All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.skyscreamer.nevado.jms.message.NevadoBytesMessage Maven / Gradle / Ivy

There is a newer version: 1.3.2
Show newest version
package org.skyscreamer.nevado.jms.message;

import org.apache.commons.lang.builder.HashCodeBuilder;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import javax.jms.*;
import java.io.*;

/**
 * This class borrrows heavily from ActiveMQStreamMessage, Copyright 2004 Protique Ltd,
 * with great appreciation for its creators.  It is also licensed under Apache 2.0.
 *
 * 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.
 */
public class NevadoBytesMessage extends NevadoMessage implements BytesMessage {
    transient private Log _log = LogFactory.getLog(NevadoBytesMessage.class);

    private ByteArray _body;

    public NevadoBytesMessage() {}

    protected NevadoBytesMessage(BytesMessage message) throws JMSException {
        super(message);
        message.reset();
        for(int count = 0 ; count < message.getBodyLength() ; ) {
            byte[] buffer = new byte[10240];
            int numRead = message.readBytes(buffer);
            writeBytes(buffer, 0, numRead);
            count += numRead;
        }
    }

    @Override
    public void internalClearBody() throws JMSException {
        _body = null;
        this.dataOut = null;
        this.dataIn = null;
        this.bytesOut = null;
    }

    private transient DataOutputStream dataOut;
    private transient ByteArrayOutputStream bytesOut;
    private transient DataInputStream dataIn;

    public void onSend() {
        super.onSend();
        storeContent();
    }

    private void storeContent() {
        if (dataOut != null) {
            try {
                dataOut.close();
                _body = new ByteArray(bytesOut.toByteArray());
                bytesOut = null;
                dataOut = null;
            } catch (IOException ioe) {
                // Since this is backed by memory, it's rare enough to turn to a runtime
                throw new RuntimeException(ioe);
            }
        }
    }

    /**
     * @param bodyAsBytes The bodyAsBytes to set.
     * @param offset
     * @param length
     */
    public void setBodyAsBytes(byte[] bodyAsBytes,int offset, int length) {
        _body = new ByteArray(bodyAsBytes, offset, length);
        dataOut = null;
        dataIn = null;
        bytesOut = null;
    }

    /**
     * Gets the number of bytes of the message body when the message is in read-only mode. The value returned can be
     * used to allocate a byte array. The value returned is the entire length of the message body, regardless of where
     * the pointer for reading the message is currently located.
     *
     * @return number of bytes in the message
     * @throws JMSException                if the JMS provider fails to read the message due to some internal error.
     * @throws MessageNotReadableException if the message is in write-only mode.
     * @since 1.1
     */
    public long getBodyLength() throws JMSException {
        checkWriteOnlyBody();
        return _body.size();
    }

    /**
     * Reads a boolean from the bytes message stream.
     *
     * @return the boolean value read
     * @throws JMSException                if the JMS provider fails to read the message due to some internal error.
     * @throws MessageEOFException         if unexpected end of bytes stream has been reached.
     * @throws MessageNotReadableException if the message is in write-only mode.
     */
    public boolean readBoolean() throws JMSException {
        initializeReading();
        try {
            return this.dataIn.readBoolean();
        }
        catch (EOFException eof) {
            JMSException jmsEx = new MessageEOFException(eof.getMessage());
            jmsEx.setLinkedException(eof);
            throw jmsEx;
        }
        catch (IOException ioe) {
            JMSException jmsEx = new JMSException("Format error occured" + ioe.getMessage());
            jmsEx.setLinkedException(ioe);
            throw jmsEx;
        }
    }

    /**
     * Reads a signed 8-bit value from the bytes message stream.
     *
     * @return the next byte from the bytes message stream as a signed 8-bit byte
     * @throws JMSException                if the JMS provider fails to read the message due to some internal error.
     * @throws MessageEOFException         if unexpected end of bytes stream has been reached.
     * @throws MessageNotReadableException if the message is in write-only mode.
     */
    public byte readByte() throws JMSException {
        initializeReading();
        try {
            return this.dataIn.readByte();
        }
        catch (EOFException eof) {
            JMSException jmsEx = new MessageEOFException(eof.getMessage());
            jmsEx.setLinkedException(eof);
            throw jmsEx;
        }
        catch (IOException ioe) {
            JMSException jmsEx = new JMSException("Format error occured" + ioe.getMessage());
            jmsEx.setLinkedException(ioe);
            throw jmsEx;
        }
    }

    /**
     * Reads an unsigned 8-bit number from the bytes message stream.
     *
     * @return the next byte from the bytes message stream, interpreted as an unsigned 8-bit number
     * @throws JMSException                if the JMS provider fails to read the message due to some internal error.
     * @throws MessageEOFException         if unexpected end of bytes stream has been reached.
     * @throws MessageNotReadableException if the message is in write-only mode.
     */
    public int readUnsignedByte() throws JMSException {
        initializeReading();
        try {
            return this.dataIn.readUnsignedByte();
        }
        catch (EOFException eof) {
            JMSException jmsEx = new MessageEOFException(eof.getMessage());
            jmsEx.setLinkedException(eof);
            throw jmsEx;
        }
        catch (IOException ioe) {
            JMSException jmsEx = new JMSException("Format error occured" + ioe.getMessage());
            jmsEx.setLinkedException(ioe);
            throw jmsEx;
        }
    }

    /**
     * Reads a signed 16-bit number from the bytes message stream.
     *
     * @return the next two bytes from the bytes message stream, interpreted as a signed 16-bit number
     * @throws JMSException                if the JMS provider fails to read the message due to some internal error.
     * @throws MessageEOFException         if unexpected end of bytes stream has been reached.
     * @throws MessageNotReadableException if the message is in write-only mode.
     */
    public short readShort() throws JMSException {
        initializeReading();
        try {
            return this.dataIn.readShort();
        }
        catch (EOFException eof) {
            JMSException jmsEx = new MessageEOFException(eof.getMessage());
            jmsEx.setLinkedException(eof);
            throw jmsEx;
        }
        catch (IOException ioe) {
            JMSException jmsEx = new JMSException("Format error occured" + ioe.getMessage());
            jmsEx.setLinkedException(ioe);
            throw jmsEx;
        }
    }

    /**
     * Reads an unsigned 16-bit number from the bytes message stream.
     *
     * @return the next two bytes from the bytes message stream, interpreted as an unsigned 16-bit integer
     * @throws JMSException                if the JMS provider fails to read the message due to some internal error.
     * @throws MessageEOFException         if unexpected end of bytes stream has been reached.
     * @throws MessageNotReadableException if the message is in write-only mode.
     */
    public int readUnsignedShort() throws JMSException {
        initializeReading();
        try {
            return this.dataIn.readUnsignedShort();
        }
        catch (EOFException eof) {
            JMSException jmsEx = new MessageEOFException(eof.getMessage());
            jmsEx.setLinkedException(eof);
            throw jmsEx;
        }
        catch (IOException ioe) {
            JMSException jmsEx = new JMSException("Format error occured" + ioe.getMessage());
            jmsEx.setLinkedException(ioe);
            throw jmsEx;
        }
    }

    /**
     * Reads a Unicode character value from the bytes message stream.
     *
     * @return the next two bytes from the bytes message stream as a Unicode character
     * @throws JMSException                if the JMS provider fails to read the message due to some internal error.
     * @throws MessageEOFException         if unexpected end of bytes stream has been reached.
     * @throws MessageNotReadableException if the message is in write-only mode.
     */
    public char readChar() throws JMSException {
        initializeReading();
        try {
            return this.dataIn.readChar();
        }
        catch (EOFException eof) {
            JMSException jmsEx = new MessageEOFException(eof.getMessage());
            jmsEx.setLinkedException(eof);
            throw jmsEx;
        }
        catch (IOException ioe) {
            JMSException jmsEx = new JMSException("Format error occured" + ioe.getMessage());
            jmsEx.setLinkedException(ioe);
            throw jmsEx;
        }
    }

    /**
     * Reads a signed 32-bit integer from the bytes message stream.
     *
     * @return the next four bytes from the bytes message stream, interpreted as an int
     * @throws JMSException                if the JMS provider fails to read the message due to some internal error.
     * @throws MessageEOFException         if unexpected end of bytes stream has been reached.
     * @throws MessageNotReadableException if the message is in write-only mode.
     */
    public int readInt() throws JMSException {
        initializeReading();
        try {
            return this.dataIn.readInt();
        }
        catch (EOFException eof) {
            JMSException jmsEx = new MessageEOFException(eof.getMessage());
            jmsEx.setLinkedException(eof);
            throw jmsEx;
        }
        catch (IOException ioe) {
            JMSException jmsEx = new JMSException("Format error occured" + ioe.getMessage());
            jmsEx.setLinkedException(ioe);
            throw jmsEx;
        }
    }

    /**
     * Reads a signed 64-bit integer from the bytes message stream.
     *
     * @return the next eight bytes from the bytes message stream, interpreted as a long
     * @throws JMSException                if the JMS provider fails to read the message due to some internal error.
     * @throws MessageEOFException         if unexpected end of bytes stream has been reached.
     * @throws MessageNotReadableException if the message is in write-only mode.
     */
    public long readLong() throws JMSException {
        initializeReading();
        try {
            return this.dataIn.readLong();
        }
        catch (EOFException eof) {
            JMSException jmsEx = new MessageEOFException(eof.getMessage());
            jmsEx.setLinkedException(eof);
            throw jmsEx;
        }
        catch (IOException ioe) {
            JMSException jmsEx = new JMSException("Format error occured" + ioe.getMessage());
            jmsEx.setLinkedException(ioe);
            throw jmsEx;
        }
    }

    /**
     * Reads a float from the bytes message stream.
     *
     * @return the next four bytes from the bytes message stream, interpreted as a float
     * @throws JMSException                if the JMS provider fails to read the message due to some internal error.
     * @throws MessageEOFException         if unexpected end of bytes stream has been reached.
     * @throws MessageNotReadableException if the message is in write-only mode.
     */
    public float readFloat() throws JMSException {
        initializeReading();
        try {
            return this.dataIn.readFloat();
        }
        catch (EOFException eof) {
            JMSException jmsEx = new MessageEOFException(eof.getMessage());
            jmsEx.setLinkedException(eof);
            throw jmsEx;
        }
        catch (IOException ioe) {
            JMSException jmsEx = new JMSException("Format error occured" + ioe.getMessage());
            jmsEx.setLinkedException(ioe);
            throw jmsEx;
        }
    }

    /**
     * Reads a double from the bytes message stream.
     *
     * @return the next eight bytes from the bytes message stream, interpreted as a double
     * @throws JMSException                if the JMS provider fails to read the message due to some internal error.
     * @throws MessageEOFException         if unexpected end of bytes stream has been reached.
     * @throws MessageNotReadableException if the message is in write-only mode.
     */
    public double readDouble() throws JMSException {
        initializeReading();
        try {
            return this.dataIn.readDouble();
        }
        catch (EOFException eof) {
            JMSException jmsEx = new MessageEOFException(eof.getMessage());
            jmsEx.setLinkedException(eof);
            throw jmsEx;
        }
        catch (IOException ioe) {
            JMSException jmsEx = new JMSException("Format error occured" + ioe.getMessage());
            jmsEx.setLinkedException(ioe);
            throw jmsEx;
        }
    }

    /**
     * Reads a string that has been encoded using a modified UTF-8 format from the bytes message stream.
     * 

* For more information on the UTF-8 format, see "File System Safe UCS Transformation Format (FSS_UTF)", X/Open * Preliminary Specification, X/Open Company Ltd., Document Number: P316. This information also appears in ISO/IEC * 10646, Annex P. * * @return a Unicode string from the bytes message stream * @throws JMSException if the JMS provider fails to read the message due to some internal error. * @throws MessageEOFException if unexpected end of bytes stream has been reached. * @throws MessageNotReadableException if the message is in write-only mode. */ public String readUTF() throws JMSException { initializeReading(); try { return this.dataIn.readUTF(); } catch (EOFException eof) { JMSException jmsEx = new MessageEOFException(eof.getMessage()); jmsEx.setLinkedException(eof); throw jmsEx; } catch (IOException ioe) { JMSException jmsEx = new JMSException("Format error occured" + ioe.getMessage()); jmsEx.setLinkedException(ioe); throw jmsEx; } } /** * Reads a byte array from the bytes message stream. *

* If the length of array value is less than the number of bytes remaining to be read from the * stream, the array should be filled. A subsequent call reads the next increment, and so on. *

* If the number of bytes remaining in the stream is less than the length of array value, the bytes * should be read into the array. The return value of the total number of bytes read will be less than the length * of the array, indicating that there are no more bytes left to be read from the stream. The next read of the * stream returns -1. * * @param value the buffer into which the data is read * @return the total number of bytes read into the buffer, or -1 if there is no more data because the end of the * stream has been reached * @throws JMSException if the JMS provider fails to read the message due to some internal error. * @throws MessageNotReadableException if the message is in write-only mode. */ public int readBytes(byte[] value) throws JMSException { return readBytes(value, value.length); } /** * Reads a portion of the bytes message stream. *

* If the length of array value is less than the number of bytes remaining to be read from the * stream, the array should be filled. A subsequent call reads the next increment, and so on. *

* If the number of bytes remaining in the stream is less than the length of array value, the bytes * should be read into the array. The return value of the total number of bytes read will be less than the length * of the array, indicating that there are no more bytes left to be read from the stream. The next read of the * stream returns -1. *

* If length is negative, or length is greater than the length of the array value, * then an IndexOutOfBoundsException is thrown. No bytes will be read from the stream for this * exception case. * * @param value the buffer into which the data is read * @param length the number of bytes to read; must be less than or equal to value.length * @return the total number of bytes read into the buffer, or -1 if there is no more data because the end of the * stream has been reached * @throws JMSException if the JMS provider fails to read the message due to some internal error. * @throws MessageNotReadableException if the message is in write-only mode. */ public int readBytes(byte[] value, int length) throws JMSException { initializeReading(); try { int n = 0; while (n < length) { int count = this.dataIn.read(value, n, length - n); if (count < 0) { break; } n += count; } if (n == 0 && length > 0) { n = -1; } return n; } catch (EOFException eof) { JMSException jmsEx = new MessageEOFException(eof.getMessage()); jmsEx.setLinkedException(eof); throw jmsEx; } catch (IOException ioe) { JMSException jmsEx = new JMSException("Format error occured" + ioe.getMessage()); jmsEx.setLinkedException(ioe); throw jmsEx; } } /** * Writes a boolean to the bytes message stream as a 1-byte value. The value true is * written as the value (byte)1; the value false is written as the value (byte)0. * * @param value the boolean value to be written * @throws JMSException if the JMS provider fails to write the message due to some internal error. * @throws MessageNotWriteableException if the message is in read-only mode. */ public void writeBoolean(boolean value) throws JMSException { initializeWriting(); try { this.dataOut.writeBoolean(value); } catch (IOException ioe) { JMSException jmsEx = new JMSException("Could not write data:" + ioe.getMessage()); jmsEx.setLinkedException(ioe); throw jmsEx; } } /** * Writes a byte to the bytes message stream as a 1-byte value. * * @param value the byte value to be written * @throws JMSException if the JMS provider fails to write the message due to some internal error. * @throws MessageNotWriteableException if the message is in read-only mode. */ public void writeByte(byte value) throws JMSException { initializeWriting(); try { this.dataOut.writeByte(value); } catch (IOException ioe) { JMSException jmsEx = new JMSException("Could not write data:" + ioe.getMessage()); jmsEx.setLinkedException(ioe); throw jmsEx; } } /** * Writes a short to the bytes message stream as two bytes, high byte first. * * @param value the short to be written * @throws JMSException if the JMS provider fails to write the message due to some internal error. * @throws MessageNotWriteableException if the message is in read-only mode. */ public void writeShort(short value) throws JMSException { initializeWriting(); try { this.dataOut.writeShort(value); } catch (IOException ioe) { JMSException jmsEx = new JMSException("Could not write data:" + ioe.getMessage()); jmsEx.setLinkedException(ioe); throw jmsEx; } } /** * Writes a char to the bytes message stream as a 2-byte value, high byte first. * * @param value the char value to be written * @throws JMSException if the JMS provider fails to write the message due to some internal error. * @throws MessageNotWriteableException if the message is in read-only mode. */ public void writeChar(char value) throws JMSException { initializeWriting(); try { this.dataOut.writeChar(value); } catch (IOException ioe) { JMSException jmsEx = new JMSException("Could not write data:" + ioe.getMessage()); jmsEx.setLinkedException(ioe); throw jmsEx; } } /** * Writes an int to the bytes message stream as four bytes, high byte first. * * @param value the int to be written * @throws JMSException if the JMS provider fails to write the message due to some internal error. * @throws MessageNotWriteableException if the message is in read-only mode. */ public void writeInt(int value) throws JMSException { initializeWriting(); try { this.dataOut.writeInt(value); } catch (IOException ioe) { JMSException jmsEx = new JMSException("Could not write data:" + ioe.getMessage()); jmsEx.setLinkedException(ioe); throw jmsEx; } } /** * Writes a long to the bytes message stream as eight bytes, high byte first. * * @param value the long to be written * @throws JMSException if the JMS provider fails to write the message due to some internal error. * @throws MessageNotWriteableException if the message is in read-only mode. */ public void writeLong(long value) throws JMSException { initializeWriting(); try { this.dataOut.writeLong(value); } catch (IOException ioe) { JMSException jmsEx = new JMSException("Could not write data:" + ioe.getMessage()); jmsEx.setLinkedException(ioe); throw jmsEx; } } /** * Converts the float argument to an int using the floatToIntBits * method in class Float, and then writes that int value to the bytes message stream * as a 4-byte quantity, high byte first. * * @param value the float value to be written * @throws JMSException if the JMS provider fails to write the message due to some internal error. * @throws MessageNotWriteableException if the message is in read-only mode. */ public void writeFloat(float value) throws JMSException { initializeWriting(); try { this.dataOut.writeFloat(value); } catch (IOException ioe) { JMSException jmsEx = new JMSException("Could not write data:" + ioe.getMessage()); jmsEx.setLinkedException(ioe); throw jmsEx; } } /** * Converts the double argument to a long using the doubleToLongBits * method in class Double, and then writes that long value to the bytes message * stream as an 8-byte quantity, high byte first. * * @param value the double value to be written * @throws JMSException if the JMS provider fails to write the message due to some internal error. * @throws MessageNotWriteableException if the message is in read-only mode. */ public void writeDouble(double value) throws JMSException { initializeWriting(); try { this.dataOut.writeDouble(value); } catch (IOException ioe) { JMSException jmsEx = new JMSException("Could not write data:" + ioe.getMessage()); jmsEx.setLinkedException(ioe); throw jmsEx; } } /** * Writes a string to the bytes message stream using UTF-8 encoding in a machine-independent manner. *

* For more information on the UTF-8 format, see "File System Safe UCS Transformation Format (FSS_UTF)", X/Open * Preliminary Specification, X/Open Company Ltd., Document Number: P316. This information also appears in ISO/IEC * 10646, Annex P. * * @param value the String value to be written * @throws JMSException if the JMS provider fails to write the message due to some internal error. * @throws MessageNotWriteableException if the message is in read-only mode. */ public void writeUTF(String value) throws JMSException { initializeWriting(); try { this.dataOut.writeUTF(value); } catch (IOException ioe) { JMSException jmsEx = new JMSException("Could not write data:" + ioe.getMessage()); jmsEx.setLinkedException(ioe); throw jmsEx; } } /** * Writes a byte array to the bytes message stream. * * @param value the byte array to be written * @throws JMSException if the JMS provider fails to write the message due to some internal error. * @throws MessageNotWriteableException if the message is in read-only mode. */ public void writeBytes(byte[] value) throws JMSException { initializeWriting(); try { this.dataOut.write(value); } catch (IOException ioe) { JMSException jmsEx = new JMSException("Could not write data:" + ioe.getMessage()); jmsEx.setLinkedException(ioe); throw jmsEx; } } /** * Writes a portion of a byte array to the bytes message stream. * * @param value the byte array value to be written * @param offset the initial offset within the byte array * @param length the number of bytes to use * @throws JMSException if the JMS provider fails to write the message due to some internal error. * @throws MessageNotWriteableException if the message is in read-only mode. */ public void writeBytes(byte[] value, int offset, int length) throws JMSException { initializeWriting(); try { this.dataOut.write(value, offset, length); } catch (IOException ioe) { JMSException jmsEx = new JMSException("Could not write data:" + ioe.getMessage()); jmsEx.setLinkedException(ioe); throw jmsEx; } } /** * Writes an object to the bytes message stream. *

* This method works only for the objectified primitive object types (Integer,Double, * Long  ...), String objects, and byte arrays. * * @param value the object in the Java programming language ("Java object") to be written; it must not be null * @throws JMSException if the JMS provider fails to write the message due to some internal error. * @throws MessageFormatException if the object is of an invalid type. * @throws MessageNotWriteableException if the message is in read-only mode. * @throws java.lang.NullPointerException if the parameter value is null. */ public void writeObject(Object value) throws JMSException { if (value == null) { throw new NullPointerException(); } initializeWriting(); if (value instanceof Boolean) { writeBoolean(((Boolean) value).booleanValue()); } else if (value instanceof Character) { writeChar(((Character) value).charValue()); } else if (value instanceof Byte) { writeByte(((Byte) value).byteValue()); } else if (value instanceof Short) { writeShort(((Short) value).shortValue()); } else if (value instanceof Integer) { writeInt(((Integer) value).intValue()); } else if (value instanceof Double) { writeDouble(((Double) value).doubleValue()); } else if (value instanceof Long) { writeLong(((Long) value).longValue()); } else if (value instanceof Float) { writeFloat(((Float) value).floatValue()); } else if (value instanceof Double) { writeDouble(((Double) value).doubleValue()); } else if (value instanceof String) { writeUTF(value.toString()); } else if (value instanceof byte[]) { writeBytes((byte[]) value); } else { throw new MessageFormatException("Cannot write non-primitive type:" + value.getClass()); } } /** * Puts the message body in read-only mode and repositions the stream of bytes to the beginning. * * @throws JMSException if an internal error occurs */ public void reset() throws JMSException { storeContent(); this.bytesOut = null; this.dataIn = null; this.dataOut = null; } private void initializeWriting() throws MessageNotWriteableException { checkReadOnlyBody(); if (this.dataOut == null) { this.bytesOut = new ByteArrayOutputStream(); this.dataOut = new DataOutputStream(this.bytesOut); } } private void initializeReading() throws MessageNotReadableException { checkWriteOnlyBody(); if (this.dataIn == null) { ByteArrayInputStream bytesIn = new ByteArrayInputStream(_body.toByteArray()); this.dataIn = new DataInputStream(bytesIn); } } @Override public boolean equals(Object o) { if (this == o) return true; if (o == null || getClass() != o.getClass()) return false; NevadoBytesMessage that = (NevadoBytesMessage) o; if (_messageID != null ? !_messageID.equals(that._messageID) : that._messageID != null) return false; if (_body != null ? !_body.equals(that._body) : that._body != null) return false; return true; } @Override public int hashCode() { return new HashCodeBuilder().append(_messageID).append(_body).toHashCode(); } }





© 2015 - 2024 Weber Informatics LLC | Privacy Policy