/*
* Copyright 2013 Red Hat Inc. and/or its affiliates and other contributors.
*
* Licensed 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.switchyard.component.jca.composer;
import java.io.InputStream;
import java.io.Serializable;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;
import javax.jms.BytesMessage;
import javax.jms.MapMessage;
import javax.jms.Message;
import javax.jms.ObjectMessage;
import javax.jms.StreamMessage;
import javax.jms.TextMessage;
import org.switchyard.Exchange;
import org.switchyard.component.jca.JCAMessages;
import org.switchyard.component.common.composer.BaseMessageComposer;
/**
* MessageComposer implementation for JMS Message that is used by JCA component.
*
* @author David Ward <<a href="mailto:dward@jboss.org">dward@jboss.org</a>> (C) 2011 Red Hat Inc.
* @author <a href="mailto:tm.igarashi@gmail.com">Tomohisa Igarashi</a>
*/
public class JMSMessageComposer extends BaseMessageComposer<JMSBindingData> {
/**
* {@inheritDoc}
*/
@Override
public org.switchyard.Message compose(JMSBindingData source, Exchange exchange) throws Exception {
final org.switchyard.Message syMessage = exchange.createMessage();
getContextMapper().mapFrom(source, exchange.getContext(syMessage));
Message jmsMessage = source.getMessage();
if (jmsMessage instanceof BytesMessage) {
BytesMessage sourceBytes = BytesMessage.class.cast(jmsMessage);
if (sourceBytes.getBodyLength() > Integer.MAX_VALUE) {
throw JCAMessages.MESSAGES.theSizeOfMessageContentExceedsBytesThatIsNotSupportedByThisMessageComposer("" + Integer.MAX_VALUE);
}
byte[] bytearr = new byte[(int)sourceBytes.getBodyLength()];
sourceBytes.readBytes(bytearr);
syMessage.setContent(bytearr);
} else if (jmsMessage instanceof MapMessage) {
MapMessage sourceMap = MapMessage.class.cast(jmsMessage);
Map<String,Object> body = new HashMap<String,Object>();
Enumeration<?> e = sourceMap.getMapNames();
while (e.hasMoreElements()) {
String key = String.class.cast(e.nextElement());
body.put(key, sourceMap.getObject(key));
}
syMessage.setContent(body);
} else if (jmsMessage instanceof ObjectMessage) {
ObjectMessage sourceObj = ObjectMessage.class.cast(jmsMessage);
syMessage.setContent(sourceObj.getObject());
} else if (jmsMessage instanceof StreamMessage) {
StreamMessage sourceStream = StreamMessage.class.cast(jmsMessage);
syMessage.setContent(sourceStream);
} else if (jmsMessage instanceof TextMessage) {
TextMessage sourceText = TextMessage.class.cast(jmsMessage);
syMessage.setContent(sourceText.getText());
} else {
// plain javax.jms.Message doesn't have body content
syMessage.setContent(null);
}
return syMessage;
}
/**
* {@inheritDoc}
*/
@Override
public JMSBindingData decompose(Exchange exchange, JMSBindingData target) throws Exception {
org.switchyard.Message syMessage = exchange.getMessage();
getContextMapper().mapTo(exchange.getContext(), target);
Message jmsMessage = target.getMessage();
if (jmsMessage instanceof ObjectMessage) {
ObjectMessage msg = ObjectMessage.class.cast(jmsMessage);
// expect transformer to transform the content into Serializable ...
msg.setObject(syMessage.getContent(Serializable.class));
} else if (jmsMessage instanceof TextMessage) {
TextMessage msg = TextMessage.class.cast(jmsMessage);
msg.setText(syMessage.getContent(String.class));
} else if (jmsMessage instanceof BytesMessage) {
BytesMessage msg = BytesMessage.class.cast(jmsMessage);
msg.writeBytes(syMessage.getContent(byte[].class));
} else if (jmsMessage instanceof StreamMessage) {
StreamMessage msg = StreamMessage.class.cast(jmsMessage);
byte[] buffer = new byte[8192];
int size = 0;
if (syMessage.getContent() instanceof StreamMessage) {
// in case the StreamMessage is passed through from JMS inbound
StreamMessage sm = syMessage.getContent(StreamMessage.class);
while ((size = sm.readBytes(buffer)) > 0) {
msg.writeBytes(buffer, 0, size);
}
} else {
InputStream is = syMessage.getContent(InputStream.class);
while ((size = is.read(buffer)) > 0) {
msg.writeBytes(buffer, 0, size);
}
}
} else if (jmsMessage instanceof MapMessage) {
MapMessage msg = MapMessage.class.cast(jmsMessage);
Map<?,?> map = syMessage.getContent(Map.class);
Iterator entries = (Iterator)map.entrySet().iterator();
while (entries.hasNext()) {
Entry<?,?> entry = (Entry<?, ?>) entries.next();
msg.setObject((String) entry.getKey(), entry.getValue());
}
}
return target;
}
}