jdk8-mips64-public/jaxws: src/share/jaxws_classes/com/sun/tools/internal/ws/util/WSDLFetcher.java@9c07ef4934dd (annotated)

src/share/jaxws_classes/com/sun/tools/internal/ws/util/WSDLFetcher.java@9c07ef4934dd (annotated)

src/share/jaxws_classes/com/sun/tools/internal/ws/util/WSDLFetcher.java

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

author: aoqi
date: Thu, 31 Aug 2017 15:18:52 +0800
changeset 637: 9c07ef4934dd
parent 368: 0989ad8c0860
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.tools.internal.ws.util;
 import com.sun.istack.internal.NotNull;
 import com.sun.tools.internal.ws.resources.WscompileMessages;
 import com.sun.tools.internal.ws.wscompile.WsimportListener;
 import com.sun.tools.internal.ws.wscompile.WsimportOptions;
 import com.sun.tools.internal.ws.wsdl.parser.DOMForest;
 import com.sun.tools.internal.ws.wsdl.parser.MetadataFinder;
 import com.sun.xml.internal.txw2.output.IndentingXMLStreamWriter;
 import com.sun.xml.internal.ws.api.server.PortAddressResolver;
 import com.sun.xml.internal.ws.streaming.SourceReaderFactory;
 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 org.w3c.dom.Document;
 import org.w3c.dom.Element;
 import org.w3c.dom.Node;
 import org.w3c.dom.NodeList;
 import javax.xml.namespace.QName;
 import javax.xml.stream.XMLOutputFactory;
 import javax.xml.stream.XMLStreamException;
 import javax.xml.stream.XMLStreamReader;
 import javax.xml.stream.XMLStreamWriter;
 import javax.xml.transform.dom.DOMSource;
 import java.io.*;
 import java.net.MalformedURLException;
 import java.net.URL;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Set;
 /**
  * @author Rama Pulavarthi
  */
 public class WSDLFetcher {
     private WsimportOptions options;
     private WsimportListener listener;
     public WSDLFetcher(WsimportOptions options, WsimportListener listener) {
         this.options = options;
         this.listener = listener;
     }
     /**
      *  Fetches the wsdls in the DOMForest to the options.destDir
      * @param forest
      * @return location of fetched root WSDL document
      * @throws IOException
      * @throws XMLStreamException
      * @throws FileNotFoundException
      */
     public String fetchWsdls(MetadataFinder forest) throws IOException, XMLStreamException {
         String rootWsdl = null;
         for(String root: forest.getRootDocuments()) {
             rootWsdl = root;
         }
         Set<String> externalRefs = forest.getExternalReferences();
         Map<String,String> documentMap = createDocumentMap(forest, getWSDLDownloadDir(), rootWsdl, externalRefs);
         String rootWsdlName = fetchFile(rootWsdl,forest, documentMap,getWSDLDownloadDir());
         for(String reference: forest.getExternalReferences()) {
             fetchFile(reference,forest,documentMap,getWSDLDownloadDir());
         }
         return WSDL_PATH +"/" + rootWsdlName;
     }
     private String fetchFile(final String doc, DOMForest forest, final Map<String, String> documentMap, File destDir) throws IOException, XMLStreamException {
         DocumentLocationResolver docLocator = createDocResolver(doc, forest, documentMap);
         WSDLPatcher wsdlPatcher = new WSDLPatcher(new PortAddressResolver() {
             @Override
             public String getAddressFor(@NotNull QName serviceName, @NotNull String portName) {
                 return null;
             }
         }, docLocator);
         XMLStreamReader xsr = null;
         XMLStreamWriter xsw = null;
         OutputStream os = null;
         String resolvedRootWsdl = null;
         try {
             XMLOutputFactory writerfactory;
             xsr = SourceReaderFactory.createSourceReader(new DOMSource(forest.get(doc)), false);
             writerfactory = XMLOutputFactory.newInstance();
             resolvedRootWsdl = docLocator.getLocationFor(null, doc);
             File outFile = new File(destDir, resolvedRootWsdl);
             os = new FileOutputStream(outFile);
             if(options.verbose) {
                 listener.message(WscompileMessages.WSIMPORT_DOCUMENT_DOWNLOAD(doc,outFile));
             }
             xsw = writerfactory.createXMLStreamWriter(os);
             //DOMForest eats away the whitespace loosing all the indentation, so write it through
             // indenting writer for better readability of fetched documents
             IndentingXMLStreamWriter indentingWriter = new IndentingXMLStreamWriter(xsw);
             wsdlPatcher.bridge(xsr, indentingWriter);
             options.addGeneratedFile(outFile);
         } finally {
             try {
                 if (xsr != null) {xsr.close();}
                 if (xsw != null) {xsw.close();}
             } finally {
                 if (os != null) {os.close();}
             }
         }
         return resolvedRootWsdl;
     }
     private Map<String,String> createDocumentMap(MetadataFinder forest, File baseDir, final String rootWsdl, Set<String> externalReferences) {
         Map<String,String> map = new HashMap<String,String>();
         String rootWsdlFileName = rootWsdl;
         String rootWsdlName;
         int slashIndex = rootWsdl.lastIndexOf("/");
         if( slashIndex >= 0) {
             rootWsdlFileName = rootWsdl.substring(slashIndex+1);
         }
         if(!rootWsdlFileName.endsWith(WSDL_FILE_EXTENSION)) {
             Document rootWsdlDoc =  forest.get(rootWsdl);
             NodeList serviceNodes = rootWsdlDoc.getElementsByTagNameNS(WSDLConstants.QNAME_SERVICE.getNamespaceURI(),WSDLConstants.QNAME_SERVICE.getLocalPart());
             if (serviceNodes.getLength() == 0) {
                 rootWsdlName = "Service";
             } else {
                 Node serviceNode = serviceNodes.item(0);
                 String serviceName = ((Element)serviceNode).getAttribute( WSDLConstants.ATTR_NAME);
                 rootWsdlName = serviceName;
             }
             rootWsdlFileName = rootWsdlName+ WSDL_FILE_EXTENSION;
         } else {
             rootWsdlName = rootWsdlFileName.substring(0,rootWsdlFileName.length()-5);
         }
         map.put(rootWsdl,sanitize(rootWsdlFileName));
         int i =1;
         for(String ref: externalReferences) {
             Document refDoc =  forest.get(ref);
             Element rootEl = refDoc.getDocumentElement();
             String fileExtn;
             String fileName = null;
             int index = ref.lastIndexOf("/");
             if (index >= 0) {
                 fileName = ref.substring(index + 1);
             }
             if(rootEl.getLocalName().equals(WSDLConstants.QNAME_DEFINITIONS.getLocalPart()) && rootEl.getNamespaceURI().equals(WSDLConstants.NS_WSDL)) {
               fileExtn = WSDL_FILE_EXTENSION;
             } else if(rootEl.getLocalName().equals(WSDLConstants.QNAME_SCHEMA.getLocalPart()) && rootEl.getNamespaceURI().equals(WSDLConstants.NS_XMLNS)) {
               fileExtn = SCHEMA_FILE_EXTENSION;
             } else {
                 fileExtn = ".xml";
             }
             if(fileName != null && (fileName.endsWith(WSDL_FILE_EXTENSION) || fileName.endsWith(SCHEMA_FILE_EXTENSION))) {
                 map.put(ref, rootWsdlName+"_"+fileName);
             } else {
                 map.put(ref, rootWsdlName+"_metadata"+ (i++) + fileExtn);
             }
         }
         return map;
     }
     private DocumentLocationResolver createDocResolver(final String baseWsdl, final DOMForest forest, final Map<String,String> documentMap) {
         return new DocumentLocationResolver() {
             @Override
             public String getLocationFor(String namespaceURI, String systemId) {
                 try {
                     URL reference = new URL(new URL(baseWsdl),systemId);
                     systemId = reference.toExternalForm();
                 } catch (MalformedURLException e) {
                     throw new RuntimeException(e);
                 }
                 if(documentMap.get(systemId) != null) {
                     return documentMap.get(systemId);
                 } else {
                     String parsedEntity = forest.getReferencedEntityMap().get(systemId);
                     return documentMap.get(parsedEntity);
                 }
             }
         };
     }
     private String sanitize(String fileName) {
         fileName = fileName.replace('?', '.');
         StringBuilder sb = new StringBuilder(fileName);
         for (int i = 0; i < sb.length(); i++) {
             char c = sb.charAt(i);
             if (Character.isLetterOrDigit(c) ||
                     (c == '/') ||
                     (c == '.') ||
                     (c == '_') ||
                     (c == ' ') ||
                     (c == '-')) {
                 continue;
             } else {
                 sb.setCharAt(i, '_');
             }
         }
         return sb.toString();
     }
     private File getWSDLDownloadDir() {
         File wsdlDir = new File(options.destDir, WSDL_PATH);
         boolean created = wsdlDir.mkdirs();
         if (options.verbose && !created) {
             listener.message(WscompileMessages.WSCOMPILE_NO_SUCH_DIRECTORY(wsdlDir));
         }
         return wsdlDir;
     }
     private static String WSDL_PATH="META-INF/wsdl";
     private static String WSDL_FILE_EXTENSION=".wsdl";
     private static String SCHEMA_FILE_EXTENSION=".xsd";
 }

Mercurial > jdk8-mips64-public > jaxws / annotate

src/share/jaxws_classes/com/sun/tools/internal/ws/util/WSDLFetcher.java@9c07ef4934dd (annotated)

src/share/jaxws_classes/com/sun/tools/internal/ws/util/WSDLFetcher.java