1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.2 +++ b/src/share/jaxws_classes/com/sun/tools/internal/ws/processor/modeler/wsdl/ClassNameAllocatorImpl.java Wed Apr 27 01:27:09 2016 +0800 1.3 @@ -0,0 +1,78 @@ 1.4 +/* 1.5 + * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved. 1.6 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 1.7 + * 1.8 + * This code is free software; you can redistribute it and/or modify it 1.9 + * under the terms of the GNU General Public License version 2 only, as 1.10 + * published by the Free Software Foundation. Oracle designates this 1.11 + * particular file as subject to the "Classpath" exception as provided 1.12 + * by Oracle in the LICENSE file that accompanied this code. 1.13 + * 1.14 + * This code is distributed in the hope that it will be useful, but WITHOUT 1.15 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 1.16 + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 1.17 + * version 2 for more details (a copy is included in the LICENSE file that 1.18 + * accompanied this code). 1.19 + * 1.20 + * You should have received a copy of the GNU General Public License version 1.21 + * 2 along with this work; if not, write to the Free Software Foundation, 1.22 + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 1.23 + * 1.24 + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 1.25 + * or visit www.oracle.com if you need additional information or have any 1.26 + * questions. 1.27 + */ 1.28 + 1.29 +package com.sun.tools.internal.ws.processor.modeler.wsdl; 1.30 + 1.31 +import com.sun.tools.internal.ws.processor.util.ClassNameCollector; 1.32 +import com.sun.tools.internal.xjc.api.ClassNameAllocator; 1.33 + 1.34 +import java.util.HashSet; 1.35 +import java.util.Set; 1.36 + 1.37 +/** 1.38 + * @author Vivek Pandey 1.39 + * <p/> 1.40 + * Implementation of Callback interface that allows the driver of the XJC API to rename JAXB-generated classes/interfaces/enums. 1.41 + */ 1.42 +public class ClassNameAllocatorImpl implements ClassNameAllocator { 1.43 + public ClassNameAllocatorImpl(ClassNameCollector classNameCollector) { 1.44 + this.classNameCollector = classNameCollector; 1.45 + this.jaxbClasses = new HashSet<String>(); 1.46 + } 1.47 + 1.48 + public String assignClassName(String packageName, String className) { 1.49 + if(packageName== null || className == null){ 1.50 + //TODO: throw Exception 1.51 + return className; 1.52 + } 1.53 + 1.54 + //if either of the values are empty string return the default className 1.55 + if(packageName.equals("") || className.equals("")) 1.56 + return className; 1.57 + 1.58 + String fullClassName = packageName+"."+className; 1.59 + 1.60 + // Check if there is any conflict with jaxws generated classes 1.61 + Set<String> seiClassNames = classNameCollector.getSeiClassNames(); 1.62 + if(seiClassNames != null && seiClassNames.contains(fullClassName)){ 1.63 + className += TYPE_SUFFIX; 1.64 + } 1.65 + 1.66 + jaxbClasses.add(packageName+"."+className); 1.67 + return className; 1.68 + } 1.69 + 1.70 + /** 1.71 + * 1.72 + * @return jaxbGenerated classNames 1.73 + */ 1.74 + public Set<String> getJaxbGeneratedClasses() { 1.75 + return jaxbClasses; 1.76 + } 1.77 + 1.78 + private final static String TYPE_SUFFIX = "_Type"; 1.79 + private ClassNameCollector classNameCollector; 1.80 + private Set<String> jaxbClasses; 1.81 +}