Mercurial > jdk8-mips64-public > jaxws / file revision

 /*

  * Copyright (c) 1997, 2012, 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.processor.util;

 import com.sun.tools.internal.ws.processor.model.*;

 import com.sun.tools.internal.ws.processor.model.java.JavaInterface;

 import com.sun.tools.internal.ws.processor.model.jaxb.JAXBType;

 import com.sun.tools.internal.ws.processor.model.jaxb.JAXBTypeVisitor;

 import com.sun.tools.internal.ws.processor.model.jaxb.RpcLitStructure;

 import javax.xml.namespace.QName;

 import java.util.HashSet;

 import java.util.Iterator;

 import java.util.Set;

 /**

  * This class writes out a Model as an XML document.

  *

  * @author WS Development Team

  */

 public class ClassNameCollector extends ExtendedModelVisitor

     implements JAXBTypeVisitor {

     public ClassNameCollector() {

     }

     public void process(Model model) {

         try {

             _allClassNames = new HashSet();

             _exceptions = new HashSet();

             _wsdlBindingNames = new HashSet();

             _conflictingClassNames = new HashSet();

             _seiClassNames = new HashSet<String>();

             _jaxbGeneratedClassNames = new HashSet<String>();

             _exceptionClassNames = new HashSet<String>();

             _portTypeNames = new HashSet<QName>();

             visit(model);

         } catch (Exception e) {

             e.printStackTrace();

             // fail silently

         } finally {

             _allClassNames = null;

             _exceptions = null;

         }

     }

     public Set getConflictingClassNames() {

         return _conflictingClassNames;

     }

     protected void postVisit(Model model) throws Exception {

         for (Iterator iter = model.getExtraTypes(); iter.hasNext();) {

             visitType((AbstractType)iter.next());

         }

     }

     protected void preVisit(Service service) throws Exception {

         registerClassName(

             ((JavaInterface)service.getJavaInterface()).getName());

         // We don't generate Impl classes, commenting it out.

         // Otherwise, it would cause naming conflicts

         //registerClassName(

         //    ((JavaInterface)service.getJavaInterface()).getImpl());

     }

     protected void processPort11x(Port port){

         QName wsdlBindingName = (QName) port.getProperty(

             ModelProperties.PROPERTY_WSDL_BINDING_NAME);

         if (!_wsdlBindingNames.contains(wsdlBindingName)) {

             // multiple ports can share a binding without causing a conflict

             registerClassName(port.getJavaInterface().getName());

         }

         registerClassName((String) port.getProperty(

             ModelProperties.PROPERTY_STUB_CLASS_NAME));

         registerClassName((String) port.getProperty(

             ModelProperties.PROPERTY_TIE_CLASS_NAME));

     }

     protected void preVisit(Port port) throws Exception {

         QName portTypeName = (QName)port.getProperty(ModelProperties.PROPERTY_WSDL_PORT_TYPE_NAME);

         if(_portTypeNames.contains(portTypeName))

             return;

         //in 2.0, stub/tie class are binding agnostic so they should be per port, that is multiple

         // bindings can share the same port

         addSEIClassName(port.getJavaInterface().getName());

     }

     private void addSEIClassName(String s) {

         _seiClassNames.add(s);

         registerClassName(s);

     }

     protected void postVisit(Port port) throws Exception {

         QName wsdlBindingName = (QName) port.getProperty(

             ModelProperties.PROPERTY_WSDL_BINDING_NAME);

         if (!_wsdlBindingNames.contains(wsdlBindingName)) {

             _wsdlBindingNames.add(wsdlBindingName);

         }

         QName portTypeName = (QName)port.getProperty(ModelProperties.PROPERTY_WSDL_PORT_TYPE_NAME);

         if(!_portTypeNames.contains(portTypeName)){

             _portTypeNames.add(portTypeName);

         }

     }

     protected boolean shouldVisit(Port port) {

         QName wsdlBindingName = (QName) port.getProperty(

             ModelProperties.PROPERTY_WSDL_BINDING_NAME);

         return !_wsdlBindingNames.contains(wsdlBindingName);

     }

     protected void preVisit(Fault fault) throws Exception {

         if (!_exceptions.contains(fault.getJavaException())) {

             /* the same exception can be used in several faults, but that

              * doesn't mean that there is a conflict

              */

             _exceptions.add(fault.getJavaException());

             addExceptionClassName(fault.getJavaException().getName());

             for (Iterator iter = fault.getSubfaults();

                 iter != null && iter.hasNext();) {

                 Fault subfault = (Fault) iter.next();

                 preVisit(subfault);

             }

         }

     }

     private void addExceptionClassName(String name) {

         if(_allClassNames.contains(name))

             _exceptionClassNames.add(name);

         registerClassName(name);

         //To change body of created methods use File | Settings | File Templates.

     }

     protected void visitBodyBlock(Block block) throws Exception {

         visitBlock(block);

     }

     protected void visitHeaderBlock(Block block) throws Exception {

         visitBlock(block);

     }

     protected void visitFaultBlock(Block block) throws Exception {

     }

     protected void visitBlock(Block block) throws Exception {

         visitType(block.getType());

     }

     protected void visit(Parameter parameter) throws Exception {

         visitType(parameter.getType());

     }

     private void visitType(AbstractType type) throws Exception {

         if (type != null) {

             if (type instanceof JAXBType)

                 visitType((JAXBType)type);

             else if (type instanceof RpcLitStructure)

                 visitType((RpcLitStructure)type);

         }

     }

     private void visitType(JAXBType type) throws Exception {

         type.accept(this);

     }

     private void visitType(RpcLitStructure type) throws Exception {

         type.accept(this);

     }

     private void registerClassName(String name) {

         if (name == null || name.equals("")) {

             return;

         }

         if (_allClassNames.contains(name)) {

             _conflictingClassNames.add(name);

         } else {

             _allClassNames.add(name);

         }

     }

     public Set<String> getSeiClassNames() {

         return _seiClassNames;

     }

     private Set<String> _seiClassNames;

     public Set<String> getJaxbGeneratedClassNames() {

         return _jaxbGeneratedClassNames;

     }

     private Set<String> _jaxbGeneratedClassNames;

     public Set<String> getExceptionClassNames() {

         return _exceptionClassNames;

     }

     private Set<String> _exceptionClassNames;

     boolean doneVisitingJAXBModel = false;

     public void visit(JAXBType type) throws Exception {

         if(!doneVisitingJAXBModel && type.getJaxbModel() != null){

             Set<String> classNames = type.getJaxbModel().getGeneratedClassNames();

             for(String className : classNames){

                 addJAXBGeneratedClassName(className);

             }

             doneVisitingJAXBModel = true;

         }

     }

     public void visit(RpcLitStructure type) throws Exception {

         if(!doneVisitingJAXBModel){

             Set<String> classNames = type.getJaxbModel().getGeneratedClassNames();

             for(String className : classNames){

                 addJAXBGeneratedClassName(className);

             }

             doneVisitingJAXBModel = true;

         }

     }

     private void addJAXBGeneratedClassName(String name) {

         _jaxbGeneratedClassNames.add(name);

         registerClassName(name);

     }

     private Set _allClassNames;

     private Set _exceptions;

     private Set _wsdlBindingNames;

     private Set _conflictingClassNames;

     private Set<QName> _portTypeNames;

 }