/*
*
* 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.qpid.client.message;
import java.io.*;
import java.nio.ByteBuffer;
import javax.jms.JMSException;
import javax.jms.MessageFormatException;
import javax.jms.ObjectMessage;
import org.apache.qpid.AMQException;
import org.apache.qpid.client.util.ClassLoadingAwareObjectInputStream;
public class JMSObjectMessage extends AbstractJMSMessage implements ObjectMessage
{
public static final String MIME_TYPE = "application/java-object-stream";
private static final int DEFAULT_OUTPUT_BUFFER_SIZE = 256;
private Serializable _readData;
private ByteBuffer _data;
private Exception _exception;
private static final ByteBuffer EMPTY_BYTE_BUFFER = ByteBuffer.allocate(0);
/**
* Creates empty, writable message for use by producers
* @param delegateFactory
*/
public JMSObjectMessage(AMQMessageDelegateFactory delegateFactory)
{
super(delegateFactory, false);
}
/**
* Creates read only message for delivery to consumers
*/
JMSObjectMessage(AMQMessageDelegate delegate, final ByteBuffer data) throws AMQException
{
super(delegate, data!=null);
try
{
ClassLoadingAwareObjectInputStream in = new ClassLoadingAwareObjectInputStream(new InputStream()
{
@Override
public int read() throws IOException
{
return data.get();
}
@Override
public int read(byte[] b, int off, int len) throws IOException
{
len = data.remaining() < len ? data.remaining() : len;
data.get(b, off, len);
return len;
}
});
_readData = (Serializable) in.readObject();
}
catch (IOException e)
{
_exception = e;
}
catch (ClassNotFoundException e)
{
_exception = e;
}
}
public void clearBody() throws JMSException
{
super.clearBody();
_exception = null;
_readData = null;
_data = null;
}
public String toBodyString() throws JMSException
{
return String.valueOf(getObject());
}
public String getMimeType()
{
return MIME_TYPE;
}
@Override
public ByteBuffer getData() throws JMSException
{
if(_exception != null)
{
final MessageFormatException messageFormatException =
new MessageFormatException("Unable to deserialize message");
messageFormatException.setLinkedException(_exception);
throw messageFormatException;
}
if(_readData == null)
{
return _data == null ? EMPTY_BYTE_BUFFER : _data.duplicate();
}
else
{
try
{
ByteArrayOutputStream baos = new ByteArrayOutputStream(DEFAULT_OUTPUT_BUFFER_SIZE);
ObjectOutputStream oos = new ObjectOutputStream(baos);
oos.writeObject(_readData);
oos.flush();
return ByteBuffer.wrap(baos.toByteArray());
}
catch (IOException e)
{
final JMSException jmsException = new JMSException("Unable to encode object of type: " +
_readData.getClass().getName() + ", value " + _readData);
jmsException.setLinkedException(e);
throw jmsException;
}
}
}
public void setObject(Serializable serializable) throws JMSException
{
checkWritable();
clearBody();
try
{
ByteArrayOutputStream baos = new ByteArrayOutputStream(DEFAULT_OUTPUT_BUFFER_SIZE);
ObjectOutputStream oos = new ObjectOutputStream(baos);
oos.writeObject(serializable);
oos.flush();
_data = ByteBuffer.wrap(baos.toByteArray());
}
catch (IOException e)
{
final JMSException jmsException = new JMSException("Unable to encode object of type: " +
serializable.getClass().getName() + ", value " + serializable);
jmsException.setLinkedException(e);
throw jmsException;
}
}
public Serializable getObject() throws JMSException
{
if(_exception != null)
{
final MessageFormatException messageFormatException = new MessageFormatException("Unable to deserialize message");
messageFormatException.setLinkedException(_exception);
throw messageFormatException;
}
else if(_readData != null || _data == null)
{
return _readData;
}
else
{
Exception exception = null;
final ByteBuffer data = _data.duplicate();
try
{
ClassLoadingAwareObjectInputStream in = new ClassLoadingAwareObjectInputStream(new InputStream()
{
@Override
public int read() throws IOException
{
return data.get();
}
@Override
public int read(byte[] b, int off, int len) throws IOException
{
len = data.remaining() < len ? data.remaining() : len;
data.get(b, off, len);
return len;
}
});
return (Serializable) in.readObject();
}
catch (ClassNotFoundException e)
{
exception = e;
}
catch (IOException e)
{
exception = e;
}
JMSException jmsException = new JMSException("Could not deserialize object");
jmsException.setLinkedException(exception);
throw jmsException;
}
}
}