bodyNS = body.getAllDeclaredNamespaces();
if (rootElement != null) {
rootElement.declareNamespace(namespace);
while (envNS.hasNext()) {
rootElement.declareNamespace((OMNamespace) envNS.next());
}
while (bodyNS.hasNext()) {
rootElement.declareNamespace((OMNamespace) bodyNS.next());
}
try (final EasyByteArrayOutputStream os = new EasyByteArrayOutputStream();) {
final XMLStreamWriter writer = XMLOutputFactories.createXMLStreamWriter(os);
try {
final XMLStreamWriter encoder = new XOPEncodingStreamWriter(writer, ContentIDGenerator.DEFAULT,
OptimizationPolicy.ALL);
rootElement.serialize(encoder);
} finally {
writer.close();
}
return new BytesSource(os.toRawByteArray());
} catch (final XMLStreamException e) {
throw new MessagingException(e);
}
}
return null;
}
/**
* Create a the JBI payload from a Business SOAP fault.
*
* @param firstDetailFaultElt
* the 1st element of SOAP fault details. Not {@code null}.
* @param exchange
* the JBI exchange that will be filled. Not {@code null}.
* @throws MessagingException
* if there is an error when creating the source content
*/
public static void fillJBIBusinessFault(final OMElement firstDetailFaultElt, final Exchange exchange)
throws MessagingException {
assert firstDetailFaultElt != null;
assert exchange != null;
assert firstDetailFaultElt.getType() == OMNode.ELEMENT_NODE;
final Fault fault = exchange.createFault();
exchange.setFault(fault);
try (final EasyByteArrayOutputStream os = new EasyByteArrayOutputStream()) {
firstDetailFaultElt.serialize(os);
fault.setContent(new BytesSource(os.toRawByteArray()));
} catch (final XMLStreamException e) {
throw new MessagingException(e);
}
}
/**
* Create a the JBI error from a Technical SOAP fault.
*
* @param soapFault
* the SOAP fault. Not {@code null}.
* @param exchange
* the JBI exchange that will be filled. Not {@code null}.
* @throws MessagingException
* if there is an error when creating the source content
*/
public static void fillJBITechnicalError(final SOAPFault soapFault, final Exchange exchange)
throws MessagingException {
assert soapFault != null;
assert exchange != null;
if (soapFault.getDetail() == null) {
exchange.setError(new MessagingException(
"Technical error returned by external service provider: " + soapFault.getReason().getText()));
} else {
assert soapFault.getDetail().getFirstElement() == null;
exchange.setError(new MessagingException(
"Technical error returned by external service provider: " + soapFault.getDetail().getText()));
}
}
/**
*
* Handles attachments. Two cases can occurs:
*
*
* - NMR attachments are declared in the XML of the NMR message using MTOM/XOP,
* - NMR are not declared in the XML.
*
*
* In the first case, it is needed to replace the XML node that declares the attachment by the same using OMElement
* (it's needed by Axis API).
*
*
* In the other case, each attachment is added, using MTOM/XOP, in the special node inside the SOAP Body:
* <soapbc:attachments xmlns:soapbc="http://petals.ow2.org/ns/soapbc">/<soapbc:attachment>
*
*
* @param nm
*
* @param soapFactory
* @param messageContext
* @throws AxisFault
*/
public static final void fillSOAPBodyWithAttachments(final NormalizedMessage nm,
final SOAPFactory soapFactory, final MessageContext messageContext) throws AxisFault {
final SOAPEnvelope env = messageContext.getEnvelope();
if ((nm.getAttachmentNames() != null) && (!nm.getAttachmentNames().isEmpty())) {
SOAPBody body = env.getBody();
if (body == null) {
body = soapFactory.createSOAPBody(env);
}
final OMNamespace omNs = soapFactory.createOMNamespace(SoapConstants.Component.NS_URI,
SoapConstants.Component.NS_PREFIX);
OMElement rootElement = body.getFirstElement();
if (rootElement == null) {
rootElement = soapFactory.createOMElement("response", omNs, body);
}
// set property
messageContext.setDoingMTOM(true);
messageContext.setProperty(Constants.Configuration.ENABLE_MTOM, Constants.VALUE_TRUE);
// Add JBI attachments to the document element
final Set names = nm.getAttachmentNames();
for (final Object key : names) {
final DataHandler attachment = nm.getAttachment((String) key);
OMElement attachRefElt;
try {
attachRefElt = AttachmentHelper.hasAttachmentElement(rootElement,
attachment, (String) key);
if (attachRefElt != null) {
// An element references the attachment, we replace it by
// itself using AXIOM API (It's a requirement of Axis2)
OMElement firstElement = attachRefElt.getFirstChildWithName(new QName(
"http://www.w3.org/2004/08/xop/include", "Include"));
// TODO: should we go through all the children and check
// the type and name?
if (firstElement == null) {
firstElement = attachRefElt.getFirstChildWithName(new QName(
"http://www.w3.org/2004/08/xop/include", "include"));
}
// TODO: if the element is null, should we set a new
// attachment anyway?
// It seemed to work if we did not detached...
if (firstElement != null) {
firstElement.detach();
final OMText attach = soapFactory.createOMText(attachment, true);
attachRefElt.addChild(attach);
}
// TODO: log an error otherwise?
}
} catch (UnsupportedEncodingException uee) {
throw new AxisFault(uee.getMessage(), FAULT_CLIENT, uee);
}
}
}
}
/**
* Put the SOAP attachments in the normalized message
*
* @param attachments
* @param to
* @throws MessagingException
*/
private static void setAttachments(final Attachments attachments, final NormalizedMessage to, final Logger logger)
throws MessagingException {
final String rootPartContentId = attachments.getRootPartContentID();
for (final Object contentIdObj : attachments.getContentIDSet()) {
final String contentId = (String) contentIdObj;
// Avoid to copy the root part as attachment
if (!contentId.equals(rootPartContentId)) {
final DataHandler dh = attachments.getDataHandler(contentId);
to.addAttachment(contentId, dh);
if (logger.isLoggable(Level.FINE)) {
try {
logger.fine(String.format("Attachment '%s' added (size: %d bytes)", contentId,
dh.getInputStream().available()));
} catch (final IOException e) {
logger.log(Level.WARNING,
String.format(
"Attachment '%s' added (size: An error occurs retrieving attachment size)",
contentId),
e);
}
}
}
}
}
/**
* Copy properties from a SOAP message to the normalized message
*
* @param from
* @param to
* Normalized message
* @throws MessagingException
*/
private static void setProperties(final MessageContext from, final NormalizedMessage to) throws MessagingException {
// get the SOAP header from envelope and add it as normalized message
// property if
// it exists
SOAPEnvelope env = from.getEnvelope();
SOAPHeader header = env.getHeader();
if (header != null) {
final Iterator elements = header.getChildElements();
final Map soapHeaderElementsMap = new HashMap();
// We need to use the DocumentBuilder provided by the JVM to have a
// DocumentFragment implementation provided by JVM, otherwise we can
// have ClassNotFoundException on outside container because the
// DocumentFragment implementation is not available on the other
// side.
final Document doc = JVMDocumentBuilders.newDocument();
while ((elements != null) && elements.hasNext()) {
OMElement element = elements.next();
if (!element.getQName().equals(WSSE_QNAME)) {
try {
Element elt = XMLUtils.toDOM(element);
DocumentFragment docfrag = doc.createDocumentFragment();
docfrag.appendChild(doc.importNode(elt, true));
docfrag.normalize();
soapHeaderElementsMap.put(element.getQName().toString(), docfrag);
} catch (Exception e) {
throw new MessagingException(e);
}
}
}
to.setProperty(PROTOCOL_HEADERS, soapHeaderElementsMap);
}
}
}