1 /*

     2  * Copyright (c) 2014, Oracle and/or its affiliates. All rights reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.  Oracle designates this

     8  * particular file as subject to the "Classpath" exception as provided

     9  * by Oracle in the LICENSE file that accompanied this code.

    10  *

    11  * This code is distributed in the hope that it will be useful, but WITHOUT

    12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    14  * version 2 for more details (a copy is included in the LICENSE file that

    15  * accompanied this code).

    16  *

    17  * You should have received a copy of the GNU General Public License version

    18  * 2 along with this work; if not, write to the Free Software Foundation,

    19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    20  *

    21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    22  * or visit www.oracle.com if you need additional information or have any

    23  * questions.

    24  */

    25 

    26 package com.sun.xml.internal.messaging.saaj.soap;

    27 

    28 import javax.xml.namespace.QName;

    29 import javax.xml.soap.SOAPException;

    30 import javax.xml.stream.XMLStreamException;

    31 import javax.xml.stream.XMLStreamReader;

    32 import javax.xml.stream.XMLStreamWriter;

    33 

    34 import com.sun.xml.internal.messaging.saaj.LazyEnvelopeSource;

    35 import com.sun.xml.internal.org.jvnet.staxex.util.XMLStreamReaderToXMLStreamWriter;

    36 

    37 

    38 /**

    39  * StaxBridge builds Envelope from LazyEnvelopeSource

    40  *

    41  * @author shih-chang.chen@oracle.com

    42  */

    43 public class StaxLazySourceBridge extends StaxBridge {

    44         private LazyEnvelopeSource lazySource;

    45 

    46         public StaxLazySourceBridge(LazyEnvelopeSource src, SOAPPartImpl soapPart) throws SOAPException {

    47                 super(soapPart);

    48                 lazySource = src;

    49                 final String soapEnvNS = soapPart.getSOAPNamespace();

    50                 try {

    51                         breakpoint = new XMLStreamReaderToXMLStreamWriter.Breakpoint(src.readToBodyStarTag(), saajWriter) {

    52                                         public boolean proceedAfterStartElement()  {

    53                                                 if ("Body".equals(reader.getLocalName()) && soapEnvNS.equals(reader.getNamespaceURI()) ){

    54                                                         return false;

    55                                                 } else

    56                                                         return true;

    57                                         }

    58                                 };

    59                 } catch (XMLStreamException e) {

    60                         throw new SOAPException(e);

    61                 }

    62         }

    63 

    64         @Override

    65     public XMLStreamReader getPayloadReader() {

    66         return lazySource.readPayload();

    67 //              throw new UnsupportedOperationException();

    68     }

    69 

    70         @Override

    71     public QName getPayloadQName() {

    72         return lazySource.getPayloadQName();

    73     }

    74 

    75         @Override

    76     public String getPayloadAttributeValue(String attName) {

    77         if (lazySource.isPayloadStreamReader()) {

    78             XMLStreamReader reader = lazySource.readPayload();

    79             if (reader.getEventType() == reader.START_ELEMENT) {

    80                 return reader.getAttributeValue(null, attName);

    81             }

    82         }

    83         return null;

    84     }

    85 

    86         @Override

    87     public String getPayloadAttributeValue(QName attName) {

    88         if (lazySource.isPayloadStreamReader()) {

    89             XMLStreamReader reader = lazySource.readPayload();

    90             if (reader.getEventType() == reader.START_ELEMENT) {

    91                 return reader.getAttributeValue(attName.getNamespaceURI(), attName.getLocalPart());

    92             }

    93         }

    94         return null;

    95     }

    96 

    97         public void bridgePayload() throws XMLStreamException {

    98                 //Assuming out is at Body

    99                 writePayloadTo(saajWriter);

   100         }

   101 

   102         public void writePayloadTo(XMLStreamWriter writer) throws XMLStreamException {

   103         lazySource.writePayloadTo(writer);

   104     }

   105 }