/**
* Copyright (c) 2015, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
*
* WSO2 Inc. 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.wso2.carbon.inbound.endpoint.protocol.rabbitmq;
import org.apache.axiom.om.OMElement;
import org.apache.axiom.om.util.UUIDGenerator;
import org.apache.axis2.AxisFault;
import org.apache.axis2.builder.Builder;
import org.apache.axis2.builder.BuilderUtil;
import org.apache.axis2.builder.SOAPBuilder;
import org.apache.axis2.context.MessageContext;
import org.apache.axis2.transport.TransportUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.synapse.SynapseConstants;
import org.apache.synapse.core.SynapseEnvironment;
import org.apache.synapse.mediators.base.SequenceMediator;
import org.wso2.carbon.context.PrivilegedCarbonContext;
import org.wso2.carbon.utils.multitenancy.MultitenantConstants;
import java.io.ByteArrayInputStream;
import java.io.InputStream;
/**
* RabbitMQInjectHandler uses to mediate the received RabbitMQ message
*/
public class RabbitMQInjectHandler {
private static final Log log = LogFactory.getLog(RabbitMQInjectHandler.class);
private String injectingSeq;
private String onErrorSeq;
private boolean sequential;
private SynapseEnvironment synapseEnvironment;
public RabbitMQInjectHandler(String injectingSeq, String onErrorSeq, boolean sequential,
SynapseEnvironment synapseEnvironment) {
this.injectingSeq = injectingSeq;
this.onErrorSeq = onErrorSeq;
this.sequential = sequential;
this.synapseEnvironment = synapseEnvironment;
}
/**
* Determine the message builder to use, set the message payload to the message context and
* inject the message.
* @param inboundName
*/
public boolean invoke(RabbitMQMessage message, String inboundName) {
boolean success = true;
org.apache.synapse.MessageContext msgCtx = createMessageContext();
log.debug("Processed RabbitMQ Message ");
MessageContext axis2MsgCtx = ((org.apache.synapse.core.axis2.Axis2MessageContext) msgCtx)
.getAxis2MessageContext();
// Determine the message builder to use
String amqpCorrelationID = message.getCorrelationId();
if (amqpCorrelationID != null && amqpCorrelationID.length() > 0) {
msgCtx.setProperty(RabbitMQConstants.CORRELATION_ID, amqpCorrelationID);
} else {
msgCtx.setProperty(RabbitMQConstants.CORRELATION_ID, message.getMessageId());
}
String contentType = message.getContentType();
Builder builder = null;
if (contentType == null) {
log.warn("Unable to determine content type for message " +
msgCtx.getMessageID() + " setting to text/plain");
contentType = RabbitMQConstants.DEFAULT_CONTENT_TYPE;
message.setContentType(contentType);
}
int index = contentType.indexOf(';');
String type = index > 0 ? contentType.substring(0, index)
: contentType;
try {
builder = BuilderUtil.getBuilderFromSelector(type, axis2MsgCtx);
} catch (AxisFault axisFault) {
log.error("Error while creating message builder :: "
+ axisFault.getMessage());
}
if (builder == null) {
if (log.isDebugEnabled()) {
log.debug("No message builder found for type '" + type
+ "'. Falling back to SOAP.");
}
builder = new SOAPBuilder();
}
OMElement documentElement = null;
// set the message payload to the message context
InputStream in = new ByteArrayInputStream(message.getBody());
try {
documentElement = builder.processDocument(in, contentType,
axis2MsgCtx);
} catch (AxisFault axisFault) {
log.error("Error while processing message :: "
+ axisFault.getMessage());
}
try {
msgCtx.setEnvelope(TransportUtils
.createSOAPEnvelope(documentElement));
} catch (AxisFault axisFault) {
log.error("Error while setting message payload to the message context :: "
+ axisFault.getMessage());
}
// Inject the message to the sequence.
if (injectingSeq == null || injectingSeq.equals("")) {
log.error("Sequence name not specified. Sequence : " + injectingSeq);
success = false;
}
SequenceMediator seq = (SequenceMediator) synapseEnvironment
.getSynapseConfiguration().getSequence(injectingSeq);
if (seq != null) {
if (log.isDebugEnabled()) {
log.debug("injecting message to sequence : " + injectingSeq);
}
if (!seq.isInitialized()) {
seq.init(synapseEnvironment);
}
seq.setErrorHandler(onErrorSeq);
msgCtx.setProperty(SynapseConstants.INBOUND_ENDPOINT_NAME, inboundName);
synapseEnvironment.injectInbound(msgCtx, seq, sequential);
} else {
log.error("Sequence: " + injectingSeq + " not found");
}
Object rollbackProperty = msgCtx.getProperty(RabbitMQConstants.SET_ROLLBACK_ONLY);
if (rollbackProperty != null && (rollbackProperty instanceof Boolean && ((Boolean) rollbackProperty))
|| (rollbackProperty instanceof String && Boolean.valueOf((String) rollbackProperty))) {
success = false;
}
return success;
}
/**
* Create the initial message context for rabbitmq
*/
private org.apache.synapse.MessageContext createMessageContext() {
org.apache.synapse.MessageContext msgCtx = synapseEnvironment
.createMessageContext();
MessageContext axis2MsgCtx = ((org.apache.synapse.core.axis2.Axis2MessageContext) msgCtx)
.getAxis2MessageContext();
axis2MsgCtx.setServerSide(true);
axis2MsgCtx.setMessageID(UUIDGenerator.getUUID());
msgCtx.setProperty(MessageContext.CLIENT_API_NON_BLOCKING, true);
PrivilegedCarbonContext carbonContext = PrivilegedCarbonContext.getThreadLocalCarbonContext();
axis2MsgCtx.setProperty(MultitenantConstants.TENANT_DOMAIN, carbonContext.getTenantDomain());
return msgCtx;
}
}