org.apache.synapse.mediators.elementary.Source Maven / Gradle / Ivy
/*
* 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.synapse.mediators.elementary;
import org.apache.axiom.om.*;
import org.apache.axiom.soap.SOAPEnvelope;
import org.apache.synapse.MessageContext;
import org.apache.synapse.SynapseLog;
import org.apache.synapse.util.MessageHelper;
import org.apache.synapse.util.xpath.SynapseXPath;
import org.jaxen.JaxenException;
import java.util.ArrayList;
import java.util.List;
/**
* The source of the XML node to be stored. The source can be a
* 1. Property
* 2. XPath Expression
* 3. SOAP Envelope
* 4. SOAP Body
*
* If clone is true a clone will be create and stored from the original content. Otherwise a
* reference will be stored.
*
* In case of property a OMElement is stored in a property and it will be fetched.
*
* In case of a XPath expression, it will be evaluated to get the OMElement
*
* In case of SOAPEnvelope entire SOAP envelope will be stored
*
* In case of Body, the first child of body will be stored
*/
public class Source {
private SynapseXPath xpath = null;
private String property = null;
private int sourceType = EnrichMediator.CUSTOM;
private boolean clone = true;
private OMNode inlineOMNode = null;
private String inlineKey = null;
public ArrayList evaluate(MessageContext synCtx, SynapseLog synLog)
throws JaxenException {
ArrayList sourceNodeList = new ArrayList();
if (sourceType == EnrichMediator.CUSTOM) {
assert xpath != null : "XPath should be non null in case of CUSTOM";
List selectedNodeList = xpath.selectNodes(synCtx);
if (selectedNodeList != null && selectedNodeList.size() != 0) {
for (Object o : selectedNodeList) {
if (o instanceof OMElement) {
if (clone) {
sourceNodeList.add(((OMElement) o).cloneOMElement());
} else {
sourceNodeList.add((OMElement) o);
}
} else if (o instanceof OMText) {
sourceNodeList.add((OMText) o);
} else if (o instanceof String) {
OMFactory fac = OMAbstractFactory.getOMFactory();
sourceNodeList.add(fac.createOMText(o.toString()));
}
}
} else {
synLog.error("Specified node by xpath cannot be found.");
}
} else if (sourceType == EnrichMediator.BODY) {
if (clone) {
sourceNodeList.add(synCtx.getEnvelope().getBody().getFirstElement().cloneOMElement());
} else {
sourceNodeList.add(synCtx.getEnvelope().getBody().getFirstElement());
}
} else if (sourceType == EnrichMediator.ENVELOPE) {
if (clone) {
sourceNodeList.add(MessageHelper.cloneSOAPEnvelope(synCtx.getEnvelope()));
} else {
sourceNodeList.add(synCtx.getEnvelope());
}
} else if (sourceType == EnrichMediator.PROPERTY) {
assert property != null : "property shouldn't be null when type is PROPERTY";
Object o = synCtx.getProperty(property);
if (o instanceof OMElement) {
if (clone) {
sourceNodeList.add(((OMElement) o).cloneOMElement());
} else {
sourceNodeList.add((OMElement) o);
}
} else if (o instanceof String) {
String sourceStr = (String) o;
OMFactory fac = OMAbstractFactory.getOMFactory();
sourceNodeList.add(fac.createOMText(sourceStr));
} else if (o instanceof ArrayList) {
ArrayList nodesList = (ArrayList) o;
for (Object node : nodesList) {
if (node instanceof OMElement) {
if (node instanceof SOAPEnvelope) {
SOAPEnvelope soapEnvelope = (SOAPEnvelope) node;
soapEnvelope.getOrCreateHeader();
sourceNodeList.add(soapEnvelope);
} else {
OMElement ele = (OMElement) node;
sourceNodeList.add(ele);
}
} else if (node instanceof OMText) {
sourceNodeList.add((OMText) node);
}
}
} else {
synLog.error("Invalid source property type.");
}
} else if (sourceType == EnrichMediator.INLINE) {
if (inlineOMNode instanceof OMElement) {
OMElement inlineOMElement = (OMElement) inlineOMNode;
if (inlineOMElement.getLocalName().equals("Envelope")) {
SOAPEnvelope soapEnvelope = getSOAPEnvFromOM(inlineOMElement);
if (soapEnvelope != null) {
sourceNodeList.add(soapEnvelope);
} else {
synLog.error("Inline Source is not a valid SOAPEnvelope.");
}
} else {
sourceNodeList.add(inlineOMElement.cloneOMElement());
}
} else if (inlineOMNode instanceof OMText) {
sourceNodeList.add(inlineOMNode);
} else if (inlineKey != null) {
Object inlineObj = synCtx.getEntry(inlineKey);
if (inlineObj instanceof OMElement) {
if (((OMElement) inlineObj).getLocalName().equals("Envelope")) {
SOAPEnvelope soapEnvelope = getSOAPEnvFromOM((OMElement) inlineObj);
if (soapEnvelope != null) {
sourceNodeList.add(soapEnvelope);
} else {
synLog.error("Specified Resource as Source is not a valid SOAPEnvelope.");
}
} else {
sourceNodeList.add((OMElement) inlineObj);
}
} else if (inlineObj instanceof OMText) {
sourceNodeList.add((OMText) inlineObj);
} else if (inlineObj instanceof String) {
sourceNodeList.add(
OMAbstractFactory.getOMFactory().createOMText(inlineObj.toString()));
} else {
synLog.error("Specified Resource as Source is not valid.");
}
} else {
synLog.error("Inline Source Content is not valid.");
}
}
return sourceNodeList;
}
private SOAPEnvelope getSOAPEnvFromOM(OMElement inlineElement) {
return OMXMLBuilderFactory.createStAXSOAPModelBuilder(
inlineElement.getXMLStreamReader()).getSOAPEnvelope();
}
public SynapseXPath getXpath() {
return xpath;
}
public void setXpath(SynapseXPath xpath) {
this.xpath = xpath;
}
public int getSourceType() {
return sourceType;
}
public void setSourceType(int sourceType) {
this.sourceType = sourceType;
}
public String getProperty() {
return property;
}
public void setProperty(String property) {
this.property = property;
}
public boolean isClone() {
return clone;
}
public void setClone(boolean clone) {
this.clone = clone;
}
public void setInlineOMNode(OMNode inlineOMNode) {
this.inlineOMNode = inlineOMNode;
}
public OMNode getInlineOMNode() {
return inlineOMNode;
}
public String getInlineKey() {
return inlineKey;
}
public void setInlineKey(String inlineKey) {
this.inlineKey = inlineKey;
}
}