| /* |
| * |
| * 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 java.io.ByteArrayOutputStream; |
| import java.io.EOFException; |
| import java.io.IOException; |
| import java.net.URISyntaxException; |
| import java.nio.ByteBuffer; |
| import java.util.ArrayList; |
| import java.util.Collection; |
| import java.util.LinkedHashMap; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.Set; |
| |
| import org.apache.qpid.bytebuffer.QpidByteBuffer; |
| import org.apache.qpid.server.message.AMQMessageHeader; |
| import org.apache.qpid.server.message.internal.InternalMessage; |
| import org.apache.qpid.server.model.NamedAddressSpace; |
| import org.apache.qpid.server.plugin.MessageConverter; |
| import org.apache.qpid.server.plugin.PluggableService; |
| import org.apache.qpid.server.util.ConnectionScopedRuntimeException; |
| import org.apache.qpid.transport.codec.BBDecoder; |
| import org.apache.qpid.typedmessage.TypedBytesContentReader; |
| import org.apache.qpid.typedmessage.TypedBytesFormatException; |
| import org.apache.qpid.url.AMQBindingURL; |
| import org.apache.qpid.util.GZIPUtils; |
| |
| @PluggableService |
| public class MessageConverter_v0_8_to_Internal implements MessageConverter<AMQMessage, InternalMessage> |
| { |
| @Override |
| public Class<AMQMessage> getInputClass() |
| { |
| return AMQMessage.class; |
| } |
| |
| @Override |
| public Class<InternalMessage> getOutputClass() |
| { |
| return InternalMessage.class; |
| } |
| |
| @Override |
| public InternalMessage convert(AMQMessage serverMessage, NamedAddressSpace addressSpace) |
| { |
| final String mimeType = serverMessage.getMessageHeader().getMimeType(); |
| byte[] data = new byte[(int) serverMessage.getSize()]; |
| int total = 0; |
| for(QpidByteBuffer b : serverMessage.getContent(0, (int) serverMessage.getSize())) |
| { |
| int len = b.remaining(); |
| b.get(data, total, len); |
| b.dispose(); |
| total += len; |
| } |
| |
| String encoding = serverMessage.getMessageHeader().getEncoding(); |
| byte[] uncompressed; |
| if (GZIPUtils.GZIP_CONTENT_ENCODING.equals(encoding) |
| && (uncompressed = GZIPUtils.uncompressBufferToArray(ByteBuffer.wrap(data))) != null) |
| { |
| data = uncompressed; |
| encoding = null; |
| } |
| |
| Object body = convertMessageBody(mimeType, data); |
| |
| return InternalMessage.convert(serverMessage.getMessageNumber(), serverMessage.isPersistent(), |
| new DelegatingMessageHeader(serverMessage.getMessageHeader(), encoding), body); |
| } |
| |
| private static class ReplyToComponents |
| { |
| private String _exchange; |
| private String _queue; |
| private String _routingKey; |
| |
| public void setExchange(final String exchange) |
| { |
| _exchange = exchange; |
| } |
| |
| public void setQueue(final String queue) |
| { |
| _queue = queue; |
| } |
| |
| public void setRoutingKey(final String routingKey) |
| { |
| _routingKey = routingKey; |
| } |
| |
| public String getExchange() |
| { |
| return _exchange; |
| } |
| |
| public String getQueue() |
| { |
| return _queue; |
| } |
| |
| public String getRoutingKey() |
| { |
| return _routingKey; |
| } |
| |
| public boolean hasExchange() |
| { |
| return _exchange != null; |
| } |
| |
| public boolean hasQueue() |
| { |
| return _queue != null; |
| } |
| |
| public boolean hasRoutingKey() |
| { |
| return _routingKey != null; |
| } |
| } |
| |
| private static class DelegatingMessageHeader implements AMQMessageHeader |
| { |
| private final AMQMessageHeader _delegate; |
| private final String _encoding; |
| |
| private DelegatingMessageHeader(final AMQMessageHeader delegate, String encoding) |
| { |
| _delegate = delegate; |
| _encoding = encoding; |
| } |
| |
| @Override |
| public String getCorrelationId() |
| { |
| return _delegate.getCorrelationId(); |
| } |
| |
| @Override |
| public long getExpiration() |
| { |
| return _delegate.getExpiration(); |
| } |
| |
| @Override |
| public String getUserId() |
| { |
| return _delegate.getUserId(); |
| } |
| |
| @Override |
| public String getAppId() |
| { |
| return _delegate.getAppId(); |
| } |
| |
| @Override |
| public String getMessageId() |
| { |
| return _delegate.getMessageId(); |
| } |
| |
| @Override |
| public String getMimeType() |
| { |
| return _delegate.getMimeType(); |
| } |
| |
| @Override |
| public String getEncoding() |
| { |
| return _encoding; |
| } |
| |
| @Override |
| public byte getPriority() |
| { |
| return _delegate.getPriority(); |
| } |
| |
| @Override |
| public long getTimestamp() |
| { |
| return _delegate.getTimestamp(); |
| } |
| |
| @Override |
| public long getNotValidBefore() |
| { |
| return _delegate.getNotValidBefore(); |
| } |
| |
| @Override |
| public String getType() |
| { |
| return _delegate.getType(); |
| } |
| |
| @Override |
| public String getReplyTo() |
| { |
| String originalReplyTo = _delegate.getReplyTo(); |
| ReplyToComponents replyTo = convertReplyTo(originalReplyTo); |
| if(replyTo != null) |
| { |
| if(replyTo.hasExchange()) |
| { |
| return replyTo.getExchange() + (replyTo.hasRoutingKey() ? "/" + replyTo.getRoutingKey() : ""); |
| } |
| else |
| { |
| return replyTo.hasQueue() ? replyTo.getQueue() : replyTo.getRoutingKey(); |
| } |
| } |
| else |
| { |
| return originalReplyTo; |
| } |
| } |
| |
| private ReplyToComponents convertReplyTo(final String origReplyToString) |
| { |
| try |
| { |
| AMQBindingURL burl = new AMQBindingURL(origReplyToString); |
| ReplyToComponents replyTo = new ReplyToComponents(); |
| String routingKey = burl.getRoutingKey(); |
| if(routingKey != null) |
| { |
| replyTo.setRoutingKey(routingKey); |
| } |
| |
| String exchangeName = burl.getExchangeName(); |
| if(exchangeName != null) |
| { |
| replyTo.setExchange(exchangeName); |
| } |
| |
| String queueName = burl.getQueueName(); |
| if(queueName != null) |
| { |
| replyTo.setQueue(queueName); |
| } |
| return replyTo; |
| } |
| catch (URISyntaxException e) |
| { |
| return null; |
| } |
| } |
| |
| @Override |
| public Object getHeader(final String name) |
| { |
| return _delegate.getHeader(name); |
| } |
| |
| @Override |
| public boolean containsHeaders(final Set<String> names) |
| { |
| return _delegate.containsHeaders(names); |
| } |
| |
| @Override |
| public boolean containsHeader(final String name) |
| { |
| return _delegate.containsHeader(name); |
| } |
| |
| @Override |
| public Collection<String> getHeaderNames() |
| { |
| return _delegate.getHeaderNames(); |
| } |
| } |
| |
| |
| private static Object convertMessageBody(String mimeType, byte[] data) |
| { |
| if("text/plain".equals(mimeType) || "text/xml".equals(mimeType)) |
| { |
| String text = new String(data); |
| return text; |
| } |
| else if("jms/map-message".equals(mimeType)) |
| { |
| TypedBytesContentReader reader = new TypedBytesContentReader(ByteBuffer.wrap(data)); |
| |
| LinkedHashMap map = new LinkedHashMap(); |
| final int entries = reader.readIntImpl(); |
| for (int i = 0; i < entries; i++) |
| { |
| try |
| { |
| String propName = reader.readStringImpl(); |
| Object value = reader.readObject(); |
| |
| map.put(propName, value); |
| } |
| catch (EOFException e) |
| { |
| throw new IllegalArgumentException(e); |
| } |
| catch (TypedBytesFormatException e) |
| { |
| throw new IllegalArgumentException(e); |
| } |
| |
| } |
| |
| return map; |
| |
| } |
| else if("amqp/map".equals(mimeType)) |
| { |
| BBDecoder decoder = new BBDecoder(); |
| decoder.init(ByteBuffer.wrap(data)); |
| final Map<String,Object> map = decoder.readMap(); |
| |
| return map; |
| |
| } |
| else if("amqp/list".equals(mimeType)) |
| { |
| BBDecoder decoder = new BBDecoder(); |
| decoder.init(ByteBuffer.wrap(data)); |
| return decoder.readList(); |
| } |
| else if("jms/stream-message".equals(mimeType)) |
| { |
| TypedBytesContentReader reader = new TypedBytesContentReader(ByteBuffer.wrap(data)); |
| |
| List list = new ArrayList(); |
| while (reader.remaining() != 0) |
| { |
| try |
| { |
| list.add(reader.readObject()); |
| } |
| catch (TypedBytesFormatException e) |
| { |
| throw new ConnectionScopedRuntimeException(e); |
| } |
| catch (EOFException e) |
| { |
| throw new ConnectionScopedRuntimeException(e); |
| } |
| } |
| return list; |
| } |
| else |
| { |
| return data; |
| |
| } |
| } |
| |
| @Override |
| public String getType() |
| { |
| return "v0-8 to Internal"; |
| } |
| } |