jdk8-mips64-public/jaxws: src/share/jaxws_classes/com/sun/xml/internal/ws/server/SDDocumentImpl.java@9c07ef4934dd (annotated)

src/share/jaxws_classes/com/sun/xml/internal/ws/server/SDDocumentImpl.java@9c07ef4934dd (annotated)

src/share/jaxws_classes/com/sun/xml/internal/ws/server/SDDocumentImpl.java

Thu, 31 Aug 2017 15:18:52 +0800

author: aoqi
date: Thu, 31 Aug 2017 15:18:52 +0800
changeset 637: 9c07ef4934dd
parent 397: b99d7e355d4b
parent 0: 373ffda63c9a
permissions: -rw-r--r--

merge

 /*
  * Copyright (c) 1997, 2013, 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.server;
 import com.sun.istack.internal.Nullable;
 import com.sun.xml.internal.ws.api.server.*;
 import com.sun.xml.internal.ws.api.streaming.XMLStreamWriterFactory;
 import com.sun.xml.internal.ws.streaming.XMLStreamReaderUtil;
 import com.sun.xml.internal.ws.wsdl.SDDocumentResolver;
 import com.sun.xml.internal.ws.util.RuntimeVersion;
 import com.sun.xml.internal.ws.util.xml.XMLStreamReaderToXMLStreamWriter;
 import com.sun.xml.internal.ws.wsdl.parser.ParserUtil;
 import com.sun.xml.internal.ws.wsdl.parser.WSDLConstants;
 import com.sun.xml.internal.ws.wsdl.writer.DocumentLocationResolver;
 import com.sun.xml.internal.ws.wsdl.writer.WSDLPatcher;
 import javax.xml.namespace.QName;
 import javax.xml.stream.*;
 import javax.xml.ws.WebServiceException;
 import java.io.IOException;
 import java.io.OutputStream;
 import java.net.MalformedURLException;
 import java.net.URL;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Set;
 /**
  * {@link SDDocument} implmentation.
  *
  * <p>
  * This extends from {@link SDDocumentSource} so that
  * JAX-WS server runtime code can use {@link SDDocument}
  * as {@link SDDocumentSource}.
  *
  * @author Kohsuke Kawaguchi
  * @author Jitendra Kotamraju
  */
 public class SDDocumentImpl extends SDDocumentSource implements SDDocument {
     private static final String NS_XSD = "http://www.w3.org/2001/XMLSchema";
     private static final QName SCHEMA_INCLUDE_QNAME = new QName(NS_XSD, "include");
     private static final QName SCHEMA_IMPORT_QNAME = new QName(NS_XSD, "import");
     private static final QName SCHEMA_REDEFINE_QNAME = new QName(NS_XSD, "redefine");
     private static final String VERSION_COMMENT =
         " Published by JAX-WS RI (http://jax-ws.java.net). RI's version is "+RuntimeVersion.VERSION+". ";
     private final QName rootName;
     private final SDDocumentSource source;
     /**
      * Set when {@link ServiceDefinitionImpl} is constructed.
      */
     @Nullable List<SDDocumentFilter> filters;
     @Nullable SDDocumentResolver sddocResolver;
     /**
      * The original system ID of this document.
      *
      * When this document contains relative references to other resources,
      * this field is used to find which {@link com.sun.xml.internal.ws.server.SDDocumentImpl} it refers to.
      *
      * Must not be null.
      */
     private final URL url;
     private final Set<String> imports;
     /**
      * Creates {@link SDDocument} from {@link SDDocumentSource}.
      * @param src WSDL document infoset
      * @param serviceName wsdl:service name
      * @param portTypeName
      *      The information about the port of {@link WSEndpoint} to which this document is built for.
      *      These values are used to determine which document is the concrete and abstract WSDLs
      *      for this endpoint.
      *
      * @return null
      *      Always non-null.
      */
     public static SDDocumentImpl create(SDDocumentSource src, QName serviceName, QName portTypeName) {
         URL systemId = src.getSystemId();
         try {
             // RuntimeWSDLParser parser = new RuntimeWSDLParser(null);
             XMLStreamReader reader = src.read();
             try {
                 XMLStreamReaderUtil.nextElementContent(reader);
                 QName rootName = reader.getName();
                 if(rootName.equals(WSDLConstants.QNAME_SCHEMA)) {
                     String tns = ParserUtil.getMandatoryNonEmptyAttribute(reader, WSDLConstants.ATTR_TNS);
                     Set<String> importedDocs = new HashSet<String>();
                     while (XMLStreamReaderUtil.nextContent(reader) != XMLStreamConstants.END_DOCUMENT) {
                          if (reader.getEventType() != XMLStreamConstants.START_ELEMENT)
                             continue;
                         QName name = reader.getName();
                         if (SCHEMA_INCLUDE_QNAME.equals(name) || SCHEMA_IMPORT_QNAME.equals(name) ||
                                 SCHEMA_REDEFINE_QNAME.equals(name)) {
                             String importedDoc = reader.getAttributeValue(null, "schemaLocation");
                             if (importedDoc != null) {
                                 importedDocs.add(new URL(src.getSystemId(), importedDoc).toString());
                             }
                         }
                     }
                     return new SchemaImpl(rootName,systemId,src,tns,importedDocs);
                 } else if (rootName.equals(WSDLConstants.QNAME_DEFINITIONS)) {
                     String tns = ParserUtil.getMandatoryNonEmptyAttribute(reader, WSDLConstants.ATTR_TNS);
                     boolean hasPortType = false;
                     boolean hasService = false;
                     Set<String> importedDocs = new HashSet<String>();
                     Set<QName> allServices = new HashSet<QName>();
                     // if WSDL, parse more
                     while (XMLStreamReaderUtil.nextContent(reader) != XMLStreamConstants.END_DOCUMENT) {
                          if(reader.getEventType() != XMLStreamConstants.START_ELEMENT)
                             continue;
                         QName name = reader.getName();
                         if (WSDLConstants.QNAME_PORT_TYPE.equals(name)) {
                             String pn = ParserUtil.getMandatoryNonEmptyAttribute(reader, WSDLConstants.ATTR_NAME);
                             if (portTypeName != null) {
                                 if(portTypeName.getLocalPart().equals(pn)&&portTypeName.getNamespaceURI().equals(tns)) {
                                     hasPortType = true;
                                 }
                             }
                         } else if (WSDLConstants.QNAME_SERVICE.equals(name)) {
                             String sn = ParserUtil.getMandatoryNonEmptyAttribute(reader, WSDLConstants.ATTR_NAME);
                             QName sqn = new QName(tns,sn);
                             allServices.add(sqn);
                             if(serviceName.equals(sqn)) {
                                 hasService = true;
                             }
                         } else if (WSDLConstants.QNAME_IMPORT.equals(name)) {
                             String importedDoc = reader.getAttributeValue(null, "location");
                             if (importedDoc != null) {
                                 importedDocs.add(new URL(src.getSystemId(), importedDoc).toString());
                             }
                         } else if (SCHEMA_INCLUDE_QNAME.equals(name) || SCHEMA_IMPORT_QNAME.equals(name) ||
                                 SCHEMA_REDEFINE_QNAME.equals(name)) {
                             String importedDoc = reader.getAttributeValue(null, "schemaLocation");
                             if (importedDoc != null) {
                                 importedDocs.add(new URL(src.getSystemId(), importedDoc).toString());
                             }
                         }
                     }
                     return new WSDLImpl(
                         rootName,systemId,src,tns,hasPortType,hasService,importedDocs,allServices);
                 } else {
                     return new SDDocumentImpl(rootName,systemId,src);
                 }
             } finally {
                 reader.close();
             }
         } catch (WebServiceException e) {
             throw new ServerRtException("runtime.parser.wsdl", systemId,e);
         } catch (IOException e) {
             throw new ServerRtException("runtime.parser.wsdl", systemId,e);
         } catch (XMLStreamException e) {
             throw new ServerRtException("runtime.parser.wsdl", systemId,e);
         }
     }
     protected SDDocumentImpl(QName rootName, URL url, SDDocumentSource source) {
         this(rootName, url, source, new HashSet<String>());
     }
     protected SDDocumentImpl(QName rootName, URL url, SDDocumentSource source, Set<String> imports) {
         if (url == null) {
             throw new IllegalArgumentException("Cannot construct SDDocument with null URL.");
         }
         this.rootName = rootName;
         this.source = source;
         this.url = url;
         this.imports = imports;
     }
     void setFilters(List<SDDocumentFilter> filters) {
         this.filters = filters;
     }
     void setResolver(SDDocumentResolver sddocResolver) {
         this.sddocResolver = sddocResolver;
     }
     public QName getRootName() {
         return rootName;
     }
     public boolean isWSDL() {
         return false;
     }
     public boolean isSchema() {
         return false;
     }
     public URL getURL() {
         return url;
     }
     public XMLStreamReader read(XMLInputFactory xif) throws IOException, XMLStreamException {
         return source.read(xif);
     }
     public XMLStreamReader read() throws IOException, XMLStreamException {
         return source.read();
     }
     public URL getSystemId() {
         return url;
     }
     public Set<String> getImports() {
         return imports;
     }
     public void writeTo(OutputStream os) throws IOException {
         XMLStreamWriter w = null;
         try {
             //generate the WSDL with utf-8 encoding and XML version 1.0
             w = XMLStreamWriterFactory.create(os, "UTF-8");
             w.writeStartDocument("UTF-8", "1.0");
             new XMLStreamReaderToXMLStreamWriter().bridge(source.read(), w);
             w.writeEndDocument();
         } catch (XMLStreamException e) {
             IOException ioe = new IOException(e.getMessage());
             ioe.initCause(e);
             throw ioe;
         } finally {
             try {
                 if (w != null)
                     w.close();
             } catch (XMLStreamException e) {
                 IOException ioe = new IOException(e.getMessage());
                 ioe.initCause(e);
                 throw ioe;
             }
         }
     }
     public void writeTo(PortAddressResolver portAddressResolver, DocumentAddressResolver resolver, OutputStream os) throws IOException {
         XMLStreamWriter w = null;
         try {
             //generate the WSDL with utf-8 encoding and XML version 1.0
             w = XMLStreamWriterFactory.create(os, "UTF-8");
             w.writeStartDocument("UTF-8", "1.0");
             writeTo(portAddressResolver,resolver,w);
             w.writeEndDocument();
         } catch (XMLStreamException e) {
             IOException ioe = new IOException(e.getMessage());
             ioe.initCause(e);
             throw ioe;
         } finally {
             try {
                 if (w != null)
                     w.close();
             } catch (XMLStreamException e) {
                 IOException ioe = new IOException(e.getMessage());
                 ioe.initCause(e);
                 throw ioe;
             }
         }
     }
     public void writeTo(PortAddressResolver portAddressResolver, DocumentAddressResolver resolver, XMLStreamWriter out) throws XMLStreamException, IOException {
         if (filters != null) {
             for (SDDocumentFilter f : filters) {
                 out = f.filter(this,out);
             }
         }
         XMLStreamReader xsr = source.read();
         try {
             out.writeComment(VERSION_COMMENT);
             new WSDLPatcher(portAddressResolver, new DocumentLocationResolverImpl(resolver)).bridge(xsr,out);
         } finally {
             xsr.close();
         }
     }
     /**
      * {@link SDDocument.Schema} implementation.
      *
      * @author Kohsuke Kawaguchi
      */
     private static final class SchemaImpl extends SDDocumentImpl implements SDDocument.Schema {
         private final String targetNamespace;
         public SchemaImpl(QName rootName, URL url, SDDocumentSource source, String targetNamespace,
                           Set<String> imports) {
             super(rootName, url, source, imports);
             this.targetNamespace = targetNamespace;
         }
         public String getTargetNamespace() {
             return targetNamespace;
         }
         public boolean isSchema() {
             return true;
         }
     }
     private static final class WSDLImpl extends SDDocumentImpl implements SDDocument.WSDL {
         private final String targetNamespace;
         private final boolean hasPortType;
         private final boolean hasService;
         private final Set<QName> allServices;
         public WSDLImpl(QName rootName, URL url, SDDocumentSource source, String targetNamespace, boolean hasPortType,
                         boolean hasService, Set<String> imports,Set<QName> allServices) {
             super(rootName, url, source, imports);
             this.targetNamespace = targetNamespace;
             this.hasPortType = hasPortType;
             this.hasService = hasService;
             this.allServices = allServices;
         }
         public String getTargetNamespace() {
             return targetNamespace;
         }
         public boolean hasPortType() {
             return hasPortType;
         }
         public boolean hasService() {
             return hasService;
         }
         public Set<QName> getAllServices() {
             return allServices;
         }
         public boolean isWSDL() {
             return true;
         }
     }
     private class DocumentLocationResolverImpl implements DocumentLocationResolver {
         private DocumentAddressResolver delegate;
         DocumentLocationResolverImpl(DocumentAddressResolver delegate) {
             this.delegate = delegate;
         }
         public String getLocationFor(String namespaceURI, String systemId) {
             if (sddocResolver == null) {
                 return systemId;
             }
             try {
                 URL ref = new URL(getURL(), systemId);
                 SDDocument refDoc = sddocResolver.resolve(ref.toExternalForm());
                 if (refDoc == null)
                     return systemId;  // not something we know. just leave it as is.
                 return delegate.getRelativeAddressFor(SDDocumentImpl.this, refDoc);
             } catch (MalformedURLException mue) {
                 return null;
             }
         }
     }
 }

Mercurial > jdk8-mips64-public > jaxws / annotate

src/share/jaxws_classes/com/sun/xml/internal/ws/server/SDDocumentImpl.java@9c07ef4934dd (annotated)

src/share/jaxws_classes/com/sun/xml/internal/ws/server/SDDocumentImpl.java