aoqi@0: /*
aoqi@0:  * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved.
aoqi@0:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0:  *
aoqi@0:  * This code is free software; you can redistribute it and/or modify it
aoqi@0:  * under the terms of the GNU General Public License version 2 only, as
aoqi@0:  * published by the Free Software Foundation.  Oracle designates this
aoqi@0:  * particular file as subject to the "Classpath" exception as provided
aoqi@0:  * by Oracle in the LICENSE file that accompanied this code.
aoqi@0:  *
aoqi@0:  * This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
aoqi@0:  * version 2 for more details (a copy is included in the LICENSE file that
aoqi@0:  * accompanied this code).
aoqi@0:  *
aoqi@0:  * You should have received a copy of the GNU General Public License version
aoqi@0:  * 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0:  *
aoqi@0:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0:  * or visit www.oracle.com if you need additional information or have any
aoqi@0:  * questions.
aoqi@0:  */
aoqi@0: 
aoqi@0: package com.sun.xml.internal.ws.client.dispatch;
aoqi@0: 
aoqi@0: import com.sun.xml.internal.bind.api.JAXBRIContext;
aoqi@0: import com.sun.xml.internal.ws.api.addressing.WSEndpointReference;
aoqi@0: import com.sun.xml.internal.ws.api.message.Header;
aoqi@0: import com.sun.xml.internal.ws.api.message.Headers;
aoqi@0: import com.sun.xml.internal.ws.api.message.Message;
aoqi@0: import com.sun.xml.internal.ws.api.message.Messages;
aoqi@0: import com.sun.xml.internal.ws.api.message.Packet;
aoqi@0: import com.sun.xml.internal.ws.api.pipe.Tube;
aoqi@0: import com.sun.xml.internal.ws.api.client.WSPortInfo;
aoqi@0: import com.sun.xml.internal.ws.binding.BindingImpl;
aoqi@0: import com.sun.xml.internal.ws.client.WSServiceDelegate;
aoqi@0: import com.sun.xml.internal.ws.message.jaxb.JAXBDispatchMessage;
aoqi@0: import com.sun.xml.internal.ws.spi.db.BindingContextFactory;
aoqi@0: 
aoqi@0: import javax.xml.bind.JAXBContext;
aoqi@0: import javax.xml.bind.JAXBException;
aoqi@0: import javax.xml.bind.Unmarshaller;
aoqi@0: import javax.xml.namespace.QName;
aoqi@0: import javax.xml.transform.Source;
aoqi@0: import javax.xml.ws.Service;
aoqi@0: import javax.xml.ws.WebServiceException;
aoqi@0: 
aoqi@0: /**
aoqi@0:  * The <code>JAXBDispatch</code> class provides support
aoqi@0:  * for the dynamic invocation of a service endpoint operation using
aoqi@0:  * JAXB objects. The <code>javax.xml.ws.Service</code>
aoqi@0:  * interface acts as a factory for the creation of <code>JAXBDispatch</code>
aoqi@0:  * instances.
aoqi@0:  *
aoqi@0:  * @author WS Development Team
aoqi@0:  * @version 1.0
aoqi@0:  */
aoqi@0: public class JAXBDispatch extends DispatchImpl<Object> {
aoqi@0: 
aoqi@0:     private final JAXBContext jaxbcontext;
aoqi@0: 
aoqi@0:     // We will support a JAXBContext parameter from an unknown JAXB
aoqi@0:     // implementation by marshaling and unmarshaling directly from the
aoqi@0:     // context object, as there is no Bond available.
aoqi@0:     private final boolean isContextSupported;
aoqi@0: 
aoqi@0:     @Deprecated
aoqi@0:     public JAXBDispatch(QName port, JAXBContext jc, Service.Mode mode, WSServiceDelegate service, Tube pipe, BindingImpl binding, WSEndpointReference epr) {
aoqi@0:         super(port, mode, service, pipe, binding, epr);
aoqi@0:         this.jaxbcontext = jc;
aoqi@0:         this.isContextSupported = BindingContextFactory.isContextSupported(jc);
aoqi@0:     }
aoqi@0: 
aoqi@0:     public JAXBDispatch(WSPortInfo portInfo, JAXBContext jc, Service.Mode mode, BindingImpl binding, WSEndpointReference epr) {
aoqi@0:         super(portInfo, mode, binding, epr);
aoqi@0:         this.jaxbcontext = jc;
aoqi@0:         this.isContextSupported = BindingContextFactory.isContextSupported(jc);
aoqi@0:     }
aoqi@0: 
aoqi@0:     Object toReturnValue(Packet response) {
aoqi@0:         try {
aoqi@0:             Unmarshaller unmarshaller = jaxbcontext.createUnmarshaller();
aoqi@0:             Message msg = response.getMessage();
aoqi@0:             switch (mode) {
aoqi@0:                 case PAYLOAD:
aoqi@0:                     return msg.<Object>readPayloadAsJAXB(unmarshaller);
aoqi@0:                 case MESSAGE:
aoqi@0:                     Source result = msg.readEnvelopeAsSource();
aoqi@0:                     return unmarshaller.unmarshal(result);
aoqi@0:                 default:
aoqi@0:                     throw new WebServiceException("Unrecognized dispatch mode");
aoqi@0:             }
aoqi@0:         } catch (JAXBException e) {
aoqi@0:             throw new WebServiceException(e);
aoqi@0:         }
aoqi@0:     }
aoqi@0: 
aoqi@0: 
aoqi@0:     Packet createPacket(Object msg) {
aoqi@0:         assert jaxbcontext != null;
aoqi@0: 
aoqi@0:         Message message;
aoqi@0:         if (mode == Service.Mode.MESSAGE) {
aoqi@0:             message = isContextSupported ?
aoqi@0:                     new JAXBDispatchMessage(BindingContextFactory.create(jaxbcontext), msg, soapVersion) :
aoqi@0:                     new JAXBDispatchMessage(jaxbcontext, msg, soapVersion);
aoqi@0:         } else {
aoqi@0:             if (msg == null) {
aoqi@0:                 message = Messages.createEmpty(soapVersion);
aoqi@0:             } else {
aoqi@0:                 message = isContextSupported ?
aoqi@0:                         Messages.create(jaxbcontext, msg, soapVersion) :
aoqi@0:                         Messages.createRaw(jaxbcontext, msg, soapVersion);
aoqi@0:             }
aoqi@0:         }
aoqi@0: 
aoqi@0:         return new Packet(message);
aoqi@0: 
aoqi@0:     }
aoqi@0: 
aoqi@0:     public void setOutboundHeaders(Object... headers) {
aoqi@0:         if (headers == null)
aoqi@0:             throw new IllegalArgumentException();
aoqi@0:         Header[] hl = new Header[headers.length];
aoqi@0:         for (int i = 0; i < hl.length; i++) {
aoqi@0:             if (headers[i] == null)
aoqi@0:                 throw new IllegalArgumentException();
aoqi@0:             // TODO: handle any JAXBContext.
aoqi@0:             hl[i] = Headers.create((JAXBRIContext) jaxbcontext, headers[i]);
aoqi@0:         }
aoqi@0:         super.setOutboundHeaders(hl);
aoqi@0:     }
aoqi@0: }