com.sun.tools.ws.processor.util.ClassNameCollector Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jaxws-tools Show documentation
Show all versions of jaxws-tools Show documentation
Open source Reference Implementation of JSR-224: Java API for XML Web Services
/*
* Copyright (c) 1997, 2018 Oracle and/or its affiliates. All rights reserved.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Distribution License v. 1.0, which is available at
* http://www.eclipse.org/org/documents/edl-v10.php.
*
* SPDX-License-Identifier: BSD-3-Clause
*/
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;
}