001: /*
002: * Copyright (c) 1998-2007 Caucho Technology -- all rights reserved
003: *
004: * This file is part of Resin(R) Open Source
005: *
006: * Each copy or derived work must preserve the copyright notice and this
007: * notice unmodified.
008: *
009: * Resin Open Source is free software; you can redistribute it and/or modify
010: * it under the terms of the GNU General Public License as published by
011: * the Free Software Foundation; either version 2 of the License, or
012: * (at your option) any later version.
013: *
014: * Resin Open Source is distributed in the hope that it will be useful,
015: * but WITHOUT ANY WARRANTY; without even the implied warranty of
016: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
017: * of NON-INFRINGEMENT. See the GNU General Public License for more
018: * details.
019: *
020: * You should have received a copy of the GNU General Public License
021: * along with Resin Open Source; if not, write to the
022: *
023: * Free Software Foundation, Inc.
024: * 59 Temple Place, Suite 330
025: * Boston, MA 02111-1307 USA
026: *
027: * @author Emil Ong
028: */
029:
030: package com.caucho.soap.jaxws;
031:
032: import java.io.ByteArrayInputStream;
033: import java.io.ByteArrayOutputStream;
034:
035: import java.util.AbstractMap;
036: import java.util.HashMap;
037: import java.util.List;
038: import java.util.Map;
039: import java.util.Set;
040:
041: import java.util.logging.Logger;
042: import java.util.logging.Level;
043:
044: import javax.xml.bind.JAXBContext;
045: import javax.xml.bind.Marshaller;
046: import javax.xml.bind.Unmarshaller;
047:
048: import javax.xml.namespace.QName;
049: import javax.xml.soap.SOAPMessage;
050: import javax.xml.transform.Source;
051: import javax.xml.transform.Transformer;
052: import javax.xml.transform.TransformerException;
053: import javax.xml.transform.TransformerFactory;
054: import javax.xml.transform.dom.DOMResult;
055: import javax.xml.transform.dom.DOMSource;
056: import javax.xml.transform.sax.SAXSource;
057: import javax.xml.transform.stream.StreamResult;
058: import javax.xml.transform.stream.StreamSource;
059:
060: import javax.xml.ws.*;
061: import javax.xml.ws.handler.*;
062:
063: import com.caucho.util.L10N;
064:
065: public class LogicalMessageImpl implements LogicalMessage {
066: private static Transformer _transformer;
067:
068: private Source _source;
069:
070: public Source getPayload() {
071: return _source;
072: }
073:
074: public void setPayload(Source source) throws WebServiceException {
075: try {
076: // The payload needs to be stable, so transform stream-based
077: // sources to DOM
078: if ((source instanceof StreamSource)
079: || (source instanceof SAXSource)) {
080: DOMResult dom = new DOMResult();
081: getTransformer().transform(source, dom);
082:
083: _source = new DOMSource(dom.getNode());
084: } else
085: _source = source;
086: } catch (Exception e) {
087: throw new WebServiceException(e);
088: }
089: }
090:
091: public void setPayload(Object payload, JAXBContext context)
092: throws WebServiceException {
093: try {
094: ByteArrayOutputStream baos = new ByteArrayOutputStream();
095:
096: Marshaller marshaller = context.createMarshaller();
097: marshaller.marshal(payload, baos);
098:
099: ByteArrayInputStream bais = new ByteArrayInputStream(baos
100: .toByteArray());
101: _source = new StreamSource(bais);
102: } catch (Exception e) {
103: throw new WebServiceException(e);
104: }
105: }
106:
107: public Object getPayload(JAXBContext context)
108: throws WebServiceException {
109: try {
110: Unmarshaller unmarshaller = context.createUnmarshaller();
111: return unmarshaller.unmarshal(_source);
112: } catch (Exception e) {
113: throw new WebServiceException(e);
114: }
115: }
116:
117: private static Transformer getTransformer()
118: throws TransformerException {
119: if (_transformer == null) {
120: TransformerFactory factory = TransformerFactory
121: .newInstance();
122: _transformer = factory.newTransformer();
123: }
124:
125: return _transformer;
126: }
127: }
|