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