com.sun.xml.ws.handler.LogicalMessageImpl Maven / Gradle / Ivy
Show all versions of webservices-rt Show documentation
/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 1997-2017 Oracle and/or its affiliates. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in compliance with the License. You can
* obtain a copy of the License at
* https://oss.oracle.com/licenses/CDDL+GPL-1.1
* or LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at LICENSE.txt.
*
* GPL Classpath Exception:
* Oracle designates this particular file as subject to the "Classpath"
* exception as provided by Oracle in the GPL Version 2 section of the License
* file that accompanied this code.
*
* Modifications:
* If applicable, add the following below the License Header, with the fields
* enclosed by brackets [] replaced by your own identifying information:
* "Portions Copyright [year] [name of copyright owner]"
*
* Contributor(s):
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your decision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*/
package com.sun.xml.ws.handler;
import com.sun.xml.ws.api.message.MessageHeaders;
import com.sun.xml.ws.api.message.Packet;
import com.sun.xml.ws.api.message.Message;
import com.sun.xml.ws.api.message.AttachmentSet;
import com.sun.xml.ws.api.WSBinding;
import com.sun.xml.ws.spi.db.BindingContext;
import com.sun.xml.ws.spi.db.BindingContextFactory;
import com.sun.xml.ws.util.xml.XmlUtil;
import com.sun.xml.ws.message.EmptyMessageImpl;
import com.sun.xml.ws.message.DOMMessage;
import com.sun.xml.ws.message.jaxb.JAXBMessage;
import com.sun.xml.ws.message.source.PayloadSourceMessage;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;
import javax.xml.bind.util.JAXBSource;
import javax.xml.transform.Source;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerException;
import javax.xml.transform.dom.DOMResult;
import javax.xml.transform.dom.DOMSource;
import javax.xml.ws.LogicalMessage;
import javax.xml.ws.WebServiceException;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.Document;
/**
* Implementation of {@link LogicalMessage}. This class implements the methods
* used by LogicalHandlers to get/set the request or response either
* as a JAXB object or as javax.xml.transform.Source.
*
* The {@link Message} that is passed into the constructor
* is used to retrieve the payload of the request or response.
*
* @author WS Development Team
* @see Message
* @see LogicalMessageContextImpl
*/
class LogicalMessageImpl implements LogicalMessage {
private Packet packet;
// protected JAXBContext defaultJaxbContext;
protected BindingContext defaultJaxbContext;
private ImmutableLM lm = null;
public LogicalMessageImpl(BindingContext defaultJaxbContext, Packet
packet) {
// don't create extract payload until Users wants it.
this.packet = packet;
this.defaultJaxbContext = defaultJaxbContext;
}
public Source getPayload() {
if (lm == null) {
Source payload = packet.getMessage().copy().readPayloadAsSource();
if (payload instanceof DOMSource) {
lm = createLogicalMessageImpl(payload);
}
return payload;
} else {
return lm.getPayload();
}
}
public void setPayload(Source payload) {
lm = createLogicalMessageImpl(payload);
}
private ImmutableLM createLogicalMessageImpl(Source payload) {
if (payload == null) {
lm = new EmptyLogicalMessageImpl();
} else if (payload instanceof DOMSource) {
lm = new DOMLogicalMessageImpl((DOMSource) payload);
} else {
lm = new SourceLogicalMessageImpl(payload);
}
return lm;
}
public Object getPayload(BindingContext context) {
if (context == null) {
context = defaultJaxbContext;
}
if (context == null)
throw new WebServiceException("JAXBContext parameter cannot be null");
Object o;
if (lm == null) {
try {
o = packet.getMessage().copy().readPayloadAsJAXB(context.createUnmarshaller());
} catch (JAXBException e) {
throw new WebServiceException(e);
}
} else {
o = lm.getPayload(context);
lm = new JAXBLogicalMessageImpl(context.getJAXBContext(), o);
}
return o;
}
public Object getPayload(JAXBContext context) {
if (context == null) {
return getPayload(defaultJaxbContext);
}
if (context == null)
throw new WebServiceException("JAXBContext parameter cannot be null");
Object o;
if (lm == null) {
try {
o = packet.getMessage().copy().readPayloadAsJAXB(context.createUnmarshaller());
} catch (JAXBException e) {
throw new WebServiceException(e);
}
} else {
o = lm.getPayload(context);
lm = new JAXBLogicalMessageImpl(context, o);
}
return o;
}
public void setPayload(Object payload, BindingContext context) {
if (context == null) {
context = defaultJaxbContext;
}
if (payload == null) {
lm = new EmptyLogicalMessageImpl();
} else {
lm = new JAXBLogicalMessageImpl(context.getJAXBContext(), payload);
}
}
public void setPayload(Object payload, JAXBContext context) {
if (context == null) {
setPayload(payload, defaultJaxbContext);
}
if (payload == null) {
lm = new EmptyLogicalMessageImpl();
} else {
lm = new JAXBLogicalMessageImpl(context, payload);
}
}
public boolean isPayloadModifed() {
return (lm != null);
}
/**
* This should be called by first checking if the payload is modified.
*
* @param headers
* @param attachments
* @param binding
* @return
*/
public Message getMessage(MessageHeaders headers, AttachmentSet attachments, WSBinding binding) {
assert isPayloadModifed();
if(isPayloadModifed()) {
return lm.getMessage(headers,attachments,binding);
} else {
return packet.getMessage();
}
}
private abstract class ImmutableLM {
public abstract Source getPayload();
public abstract Object getPayload(BindingContext context);
public abstract Object getPayload(JAXBContext context);
public abstract Message getMessage(MessageHeaders headers, AttachmentSet attachments, WSBinding binding);
}
private class DOMLogicalMessageImpl extends SourceLogicalMessageImpl {
private DOMSource dom;
public DOMLogicalMessageImpl(DOMSource dom) {
super(dom);
this.dom = dom;
}
@Override
public Source getPayload() {
return dom;
}
public Message getMessage(MessageHeaders headers, AttachmentSet attachments, WSBinding binding) {
Node n = dom.getNode();
if(n.getNodeType()== Node.DOCUMENT_NODE) {
n = ((Document)n).getDocumentElement();
}
return new DOMMessage(binding.getSOAPVersion(), headers, (Element)n, attachments);
}
}
private class EmptyLogicalMessageImpl extends ImmutableLM {
public EmptyLogicalMessageImpl() {
}
@Override
public Source getPayload() {
return null;
}
@Override
public Object getPayload(JAXBContext context) {
return null;
}
@Override
public Object getPayload(BindingContext context) {
return null;
}
public Message getMessage(MessageHeaders headers, AttachmentSet attachments, WSBinding binding) {
return new EmptyMessageImpl(headers,attachments,binding.getSOAPVersion());
}
}
private class JAXBLogicalMessageImpl extends ImmutableLM {
private JAXBContext ctxt;
private Object o;
public JAXBLogicalMessageImpl(JAXBContext ctxt, Object o) {
this.ctxt = ctxt;
this.o = o;
}
@Override
public Source getPayload() {
JAXBContext context = ctxt;
if (context == null) {
context = defaultJaxbContext.getJAXBContext();
}
try {
return new JAXBSource(context, o);
} catch (JAXBException e) {
throw new WebServiceException(e);
}
}
@Override
public Object getPayload(JAXBContext context) {
// if(context == ctxt) {
// return o;
// }
try {
Source payloadSrc = getPayload();
if (payloadSrc == null)
return null;
Unmarshaller unmarshaller = context.createUnmarshaller();
return unmarshaller.unmarshal(payloadSrc);
} catch (JAXBException e) {
throw new WebServiceException(e);
}
}
public Object getPayload(BindingContext context) {
// if(context == ctxt) {
// return o;
// }
try {
Source payloadSrc = getPayload();
if (payloadSrc == null)
return null;
Unmarshaller unmarshaller = context.createUnmarshaller();
return unmarshaller.unmarshal(payloadSrc);
} catch (JAXBException e) {
throw new WebServiceException(e);
}
}
public Message getMessage(MessageHeaders headers, AttachmentSet attachments, WSBinding binding) {
return JAXBMessage.create(BindingContextFactory.create(ctxt), o,binding.getSOAPVersion(), headers,attachments);
}
}
private class SourceLogicalMessageImpl extends ImmutableLM {
private Source payloadSrc;
public SourceLogicalMessageImpl(Source source) {
this.payloadSrc = source;
}
public Source getPayload() {
assert (!(payloadSrc instanceof DOMSource));
try {
Transformer transformer = XmlUtil.newTransformer();
DOMResult domResult = new DOMResult();
transformer.transform(payloadSrc, domResult);
DOMSource dom = new DOMSource(domResult.getNode());
lm = new DOMLogicalMessageImpl((DOMSource) dom);
payloadSrc = null;
return dom;
} catch (TransformerException te) {
throw new WebServiceException(te);
}
}
public Object getPayload(JAXBContext context) {
try {
Source payloadSrc = getPayload();
if (payloadSrc == null)
return null;
Unmarshaller unmarshaller = context.createUnmarshaller();
return unmarshaller.unmarshal(payloadSrc);
} catch (JAXBException e) {
throw new WebServiceException(e);
}
}
public Object getPayload(BindingContext context) {
try {
Source payloadSrc = getPayload();
if (payloadSrc == null)
return null;
Unmarshaller unmarshaller = context.createUnmarshaller();
return unmarshaller.unmarshal(payloadSrc);
} catch (JAXBException e) {
throw new WebServiceException(e);
}
}
public Message getMessage(MessageHeaders headers, AttachmentSet attachments, WSBinding binding) {
assert (payloadSrc!=null);
return new PayloadSourceMessage(headers, payloadSrc, attachments,binding.getSOAPVersion());
}
}
}