/*
*
* 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.server.protocol.v0_8;
import org.apache.qpid.framing.AMQShortString;
import org.apache.qpid.framing.BasicContentHeaderProperties;
import org.apache.qpid.framing.ContentHeaderBody;
import org.apache.qpid.framing.FieldTable;
import org.apache.qpid.framing.abstraction.MessagePublishInfo;
import org.apache.qpid.server.message.internal.InternalMessage;
import org.apache.qpid.server.plugin.MessageConverter;
import org.apache.qpid.server.store.StoreFuture;
import org.apache.qpid.server.store.StoredMessage;
import org.apache.qpid.server.util.ConnectionScopedRuntimeException;
import org.apache.qpid.server.virtualhost.VirtualHost;
import org.apache.qpid.transport.codec.BBEncoder;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.ObjectOutputStream;
import java.nio.ByteBuffer;
import java.nio.charset.Charset;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
public class MessageConverter_Internal_to_v0_8 implements MessageConverter<InternalMessage, AMQMessage>
{
private static final int BASIC_CLASS_ID = 60;
private static final Charset UTF_8 = Charset.forName("UTF-8");
public Class<InternalMessage> getInputClass()
{
return InternalMessage.class;
}
@Override
public Class<AMQMessage> getOutputClass()
{
return AMQMessage.class;
}
@Override
public AMQMessage convert(InternalMessage serverMsg, VirtualHost vhost)
{
return new AMQMessage(convertToStoredMessage(serverMsg), null);
}
private StoredMessage<MessageMetaData> convertToStoredMessage(final InternalMessage serverMsg)
{
final byte[] messageContent = convertToBody(serverMsg.getMessageBody());
final MessageMetaData messageMetaData_0_8 = convertMetaData(serverMsg,
getBodyMimeType(serverMsg.getMessageBody()),
messageContent.length);
return new StoredMessage<MessageMetaData>()
{
@Override
public MessageMetaData getMetaData()
{
return messageMetaData_0_8;
}
@Override
public long getMessageNumber()
{
return serverMsg.getMessageNumber();
}
@Override
public void addContent(int offsetInMessage, ByteBuffer src)
{
throw new UnsupportedOperationException();
}
@Override
public int getContent(int offsetInMessage, ByteBuffer dst)
{
int size = messageContent.length - offsetInMessage;
if(dst.remaining() < size)
{
size = dst.remaining();
}
ByteBuffer buf = ByteBuffer.wrap(messageContent, offsetInMessage, size);
dst.put(buf);
return size;
}
@Override
public ByteBuffer getContent(int offsetInMessage, int size)
{
return ByteBuffer.wrap(messageContent, offsetInMessage, size);
}
@Override
public StoreFuture flushToStore()
{
return StoreFuture.IMMEDIATE_FUTURE;
}
@Override
public void remove()
{
throw new UnsupportedOperationException();
}
};
}
private MessageMetaData convertMetaData(final InternalMessage serverMsg, final String bodyMimeType, final int size)
{
MessagePublishInfo publishInfo = new MessagePublishInfo()
{
@Override
public AMQShortString getExchange()
{
return AMQShortString.EMPTY_STRING;
}
@Override
public void setExchange(final AMQShortString amqShortString)
{
throw new UnsupportedOperationException();
}
@Override
public boolean isImmediate()
{
return false;
}
@Override
public boolean isMandatory()
{
return false;
}
@Override
public AMQShortString getRoutingKey()
{
return AMQShortString.valueOf(serverMsg.getInitialRoutingAddress());
}
};
final BasicContentHeaderProperties props = new BasicContentHeaderProperties();
props.setAppId(serverMsg.getMessageHeader().getAppId());
props.setContentType(bodyMimeType);
props.setCorrelationId(serverMsg.getMessageHeader().getCorrelationId());
props.setDeliveryMode(serverMsg.isPersistent() ? BasicContentHeaderProperties.PERSISTENT : BasicContentHeaderProperties.NON_PERSISTENT);
props.setExpiration(serverMsg.getExpiration());
props.setMessageId(serverMsg.getMessageHeader().getMessageId());
props.setPriority(serverMsg.getMessageHeader().getPriority());
props.setReplyTo(serverMsg.getMessageHeader().getReplyTo());
props.setTimestamp(serverMsg.getMessageHeader().getTimestamp());
props.setUserId(serverMsg.getMessageHeader().getUserId());
Map<String,Object> headerProps = new LinkedHashMap<String, Object>();
for(String headerName : serverMsg.getMessageHeader().getHeaderNames())
{
headerProps.put(headerName, serverMsg.getMessageHeader().getHeader(headerName));
}
props.setHeaders(FieldTable.convertToFieldTable(headerProps));
final ContentHeaderBody chb = new ContentHeaderBody(props, BASIC_CLASS_ID);
chb.setBodySize(size);
return new MessageMetaData(publishInfo, chb, serverMsg.getArrivalTime());
}
@Override
public String getType()
{
return "Internal to v0-8";
}
public static byte[] convertToBody(Object object)
{
if(object instanceof String)
{
return ((String)object).getBytes(UTF_8);
}
else if(object instanceof byte[])
{
return (byte[]) object;
}
else if(object instanceof Map)
{
BBEncoder encoder = new BBEncoder(1024);
encoder.writeMap((Map)object);
ByteBuffer buf = encoder.segment();
int remaining = buf.remaining();
byte[] data = new byte[remaining];
buf.get(data);
return data;
}
else if(object instanceof List)
{
BBEncoder encoder = new BBEncoder(1024);
encoder.writeList((List) object);
ByteBuffer buf = encoder.segment();
int remaining = buf.remaining();
byte[] data = new byte[remaining];
buf.get(data);
return data;
}
else
{
ByteArrayOutputStream bytesOut = new ByteArrayOutputStream();
try
{
ObjectOutputStream os = new ObjectOutputStream(bytesOut);
os.writeObject(object);
return bytesOut.toByteArray();
}
catch (IOException e)
{
throw new ConnectionScopedRuntimeException(e);
}
}
}
public static String getBodyMimeType(Object object)
{
if(object instanceof String)
{
return "text/plain";
}
else if(object instanceof byte[])
{
return "application/octet-stream";
}
else if(object instanceof Map)
{
return "amqp/map";
}
else if(object instanceof List)
{
return "amqp/list";
}
else
{
return "application/java-object-stream";
}
}
}