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 }