jdk8-mips64-public/jaxws: src/share/jaxws_classes/com/sun/xml/internal/ws/wsdl/writer/WSDLPatcher.java@f50545b5e2f1 (annotated)

src/share/jaxws_classes/com/sun/xml/internal/ws/wsdl/writer/WSDLPatcher.java@f50545b5e2f1 (annotated)

src/share/jaxws_classes/com/sun/xml/internal/ws/wsdl/writer/WSDLPatcher.java

Tue, 06 Mar 2012 16:09:35 -0800

author: ohair
date: Tue, 06 Mar 2012 16:09:35 -0800
changeset 286: f50545b5e2f1
child 368: 0989ad8c0860
permissions: -rw-r--r--

7150322: Stop using drop source bundles in jaxws
Reviewed-by: darcy, ohrstrom

 /*
  * Copyright (c) 1997, 2010, 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.wsdl.writer;
 import com.sun.istack.internal.NotNull;
 import com.sun.xml.internal.ws.api.server.PortAddressResolver;
 import com.sun.xml.internal.ws.util.xml.XMLStreamReaderToXMLStreamWriter;
 import com.sun.xml.internal.ws.wsdl.parser.WSDLConstants;
 import com.sun.xml.internal.ws.addressing.W3CAddressingConstants;
 import com.sun.istack.internal.Nullable;
 import javax.xml.namespace.QName;
 import javax.xml.stream.XMLStreamException;
 import java.util.logging.Logger;
 /**
  * Patches WSDL with the correct endpoint address and the relative paths
  * to other documents.
  *
  * @author Jitendra Kotamraju
  * @author Kohsuke Kawaguchi
  */
 public final class WSDLPatcher extends XMLStreamReaderToXMLStreamWriter {
     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 Logger logger = Logger.getLogger(
             com.sun.xml.internal.ws.util.Constants.LoggingDomain + ".wsdl.patcher");
     private final DocumentLocationResolver docResolver;
     private final PortAddressResolver portAddressResolver;
     //
     // fields accumulated as we parse through documents
     //
     private String targetNamespace;
     private QName serviceName;
     private QName portName;
     private String portAddress;
     // true inside <wsdl:service>/<wsdl:part>/<wsa:EndpointReference>
     private boolean inEpr;
     // true inside <wsdl:service>/<wsdl:part>/<wsa:EndpointReference>/<wsa:Address>
     private boolean inEprAddress;
     /**
      * Creates a {@link WSDLPatcher} for patching WSDL.
      *
      * @param portAddressResolver
      *      address of the endpoint is resolved using this docResolver.
      * @param docResolver
      *      Consulted to get the import/include document locations.
      *      Must not be null.
      */
     public WSDLPatcher(@NotNull PortAddressResolver portAddressResolver,
             @NotNull DocumentLocationResolver docResolver) {
         this.portAddressResolver = portAddressResolver;
         this.docResolver = docResolver;
     }
     @Override
     protected void handleAttribute(int i) throws XMLStreamException {
         QName name = in.getName();
         String attLocalName = in.getAttributeLocalName(i);
         if((name.equals(SCHEMA_INCLUDE_QNAME) && attLocalName.equals("schemaLocation"))
         || (name.equals(SCHEMA_IMPORT_QNAME)  && attLocalName.equals("schemaLocation"))
         || (name.equals(SCHEMA_REDEFINE_QNAME)  && attLocalName.equals("schemaLocation"))
         || (name.equals(WSDLConstants.QNAME_IMPORT)  && attLocalName.equals("location"))) {
             // patch this attribute value.
             String relPath = in.getAttributeValue(i);
             String actualPath = getPatchedImportLocation(relPath);
             if (actualPath == null) {
                 return; // skip this attribute to leave it up to "implicit reference".
             }
             logger.fine("Fixing the relative location:"+relPath
                     +" with absolute location:"+actualPath);
             writeAttribute(i, actualPath);
             return;
         }
         if (name.equals(WSDLConstants.NS_SOAP_BINDING_ADDRESS) ||
             name.equals(WSDLConstants.NS_SOAP12_BINDING_ADDRESS)) {
             if(attLocalName.equals("location")) {
                 portAddress = in.getAttributeValue(i);
                 String value = getAddressLocation();
                 if (value != null) {
                     logger.fine("Service:"+serviceName+ " port:"+portName
                             + " current address "+portAddress+" Patching it with "+value);
                     writeAttribute(i, value);
                     return;
                 }
             }
         }
         super.handleAttribute(i);
     }
     /**
      * Writes out an {@code i}-th attribute but with a different value.
      * @param i attribute index
      * @param value attribute value
      * @throws XMLStreamException when an error encountered while writing attribute
      */
     private void writeAttribute(int i, String value) throws XMLStreamException {
         String nsUri = in.getAttributeNamespace(i);
         if(nsUri!=null)
             out.writeAttribute( in.getAttributePrefix(i), nsUri, in.getAttributeLocalName(i), value );
         else
             out.writeAttribute( in.getAttributeLocalName(i), value );
     }
     @Override
     protected void handleStartElement() throws XMLStreamException {
         QName name = in.getName();
         if (name.equals(WSDLConstants.QNAME_DEFINITIONS)) {
             String value = in.getAttributeValue(null,"targetNamespace");
             if (value != null) {
                 targetNamespace = value;
             }
         } else if (name.equals(WSDLConstants.QNAME_SERVICE)) {
             String value = in.getAttributeValue(null,"name");
             if (value != null) {
                 serviceName = new QName(targetNamespace, value);
             }
         } else if (name.equals(WSDLConstants.QNAME_PORT)) {
             String value = in.getAttributeValue(null,"name");
             if (value != null) {
                 portName = new QName(targetNamespace,value);
             }
         } else if (name.equals(W3CAddressingConstants.WSA_EPR_QNAME)) {
             if (serviceName != null && portName != null) {
                 inEpr = true;
             }
         } else if (name.equals(W3CAddressingConstants.WSA_ADDRESS_QNAME)) {
             if (inEpr) {
                 inEprAddress = true;
             }
         }
         super.handleStartElement();
     }
     @Override
     protected void handleEndElement() throws XMLStreamException {
         QName name = in.getName();
         if (name.equals(WSDLConstants.QNAME_SERVICE)) {
             serviceName = null;
         } else if (name.equals(WSDLConstants.QNAME_PORT)) {
             portName = null;
         } else if (name.equals(W3CAddressingConstants.WSA_EPR_QNAME)) {
             if (inEpr) {
                 inEpr = false;
             }
         } else if (name.equals(W3CAddressingConstants.WSA_ADDRESS_QNAME)) {
             if (inEprAddress) {
                 String value = getAddressLocation();
                 if (value != null) {
                     logger.fine("Fixing EPR Address for service:"+serviceName+ " port:"+portName
                                 + " address with "+value);
                     out.writeCharacters(value);
                 }
                 inEprAddress = false;
             }
         }
         super.handleEndElement();
     }
     @Override
     protected void handleCharacters() throws XMLStreamException {
         // handleCharacters() may be called multiple times.
         if (inEprAddress) {
             String value = getAddressLocation();
             if (value != null) {
                 // will write the address with <wsa:Address> end element
                 return;
             }
         }
         super.handleCharacters();
     }
     /**
      * Returns the location to be placed into the generated document.
      *
      * @param relPath relative URI to be resolved
      * @return
      *      null to leave it to the "implicit reference".
      */
     private @Nullable String getPatchedImportLocation(String relPath) {
         return docResolver.getLocationFor(null, relPath);
     }
     /**
      * For the given service, port names it matches the correct endpoint and
      * reutrns its endpoint address
      *
      * @return returns the resolved endpoint address
      */
     private String getAddressLocation() {
         return (portAddressResolver == null || portName == null)
                 ? null : portAddressResolver.getAddressFor(serviceName, portName.getLocalPart(), portAddress);
     }
 }

Mercurial > jdk8-mips64-public > jaxws / annotate

src/share/jaxws_classes/com/sun/xml/internal/ws/wsdl/writer/WSDLPatcher.java@f50545b5e2f1 (annotated)

src/share/jaxws_classes/com/sun/xml/internal/ws/wsdl/writer/WSDLPatcher.java