com.sun.tools.ws.processor.util.ClassNameCollector Maven / Gradle / Ivy
/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 1997-2017 Oracle and/or its affiliates. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in compliance with the License. You can
* obtain a copy of the License at
* https://oss.oracle.com/licenses/CDDL+GPL-1.1
* or LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at LICENSE.txt.
*
* GPL Classpath Exception:
* Oracle designates this particular file as subject to the "Classpath"
* exception as provided by Oracle in the GPL Version 2 section of the License
* file that accompanied this code.
*
* Modifications:
* If applicable, add the following below the License Header, with the fields
* enclosed by brackets [] replaced by your own identifying information:
* "Portions Copyright [year] [name of copyright owner]"
*
* Contributor(s):
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your decision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*/
package com.sun.tools.ws.processor.util;
import com.sun.tools.ws.processor.model.*;
import com.sun.tools.ws.processor.model.java.JavaInterface;
import com.sun.tools.ws.processor.model.jaxb.JAXBType;
import com.sun.tools.ws.processor.model.jaxb.JAXBTypeVisitor;
import com.sun.tools.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();
_jaxbGeneratedClassNames = new HashSet();
_exceptionClassNames = new HashSet();
_portTypeNames = new HashSet();
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 getSeiClassNames() {
return _seiClassNames;
}
private Set _seiClassNames;
public Set getJaxbGeneratedClassNames() {
return _jaxbGeneratedClassNames;
}
private Set _jaxbGeneratedClassNames;
public Set getExceptionClassNames() {
return _exceptionClassNames;
}
private Set _exceptionClassNames;
boolean doneVisitingJAXBModel = false;
public void visit(JAXBType type) throws Exception {
if(!doneVisitingJAXBModel && type.getJaxbModel() != null){
Set classNames = type.getJaxbModel().getGeneratedClassNames();
for(String className : classNames){
addJAXBGeneratedClassName(className);
}
doneVisitingJAXBModel = true;
}
}
public void visit(RpcLitStructure type) throws Exception {
if(!doneVisitingJAXBModel){
Set 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 _portTypeNames;
}