1 /* 2 * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 26 package com.sun.tools.internal.ws.processor.modeler.wsdl; 27 28 import com.sun.tools.internal.ws.processor.util.ClassNameCollector; 29 import com.sun.tools.internal.xjc.api.ClassNameAllocator; 30 31 import java.util.HashSet; 32 import java.util.Set; 33 34 /** 35 * @author Vivek Pandey 36 * <p/> 37 * Implementation of Callback interface that allows the driver of the XJC API to rename JAXB-generated classes/interfaces/enums. 38 */ 39 public class ClassNameAllocatorImpl implements ClassNameAllocator { 40 public ClassNameAllocatorImpl(ClassNameCollector classNameCollector) { 41 this.classNameCollector = classNameCollector; 42 this.jaxbClasses = new HashSet<String>(); 43 } 44 45 public String assignClassName(String packageName, String className) { 46 if(packageName== null || className == null){ 47 //TODO: throw Exception 48 return className; 49 } 50 51 //if either of the values are empty string return the default className 52 if(packageName.equals("") || className.equals("")) 53 return className; 54 55 String fullClassName = packageName+"."+className; 56 57 // Check if there is any conflict with jaxws generated classes 58 Set<String> seiClassNames = classNameCollector.getSeiClassNames(); 59 if(seiClassNames != null && seiClassNames.contains(fullClassName)){ 60 className += TYPE_SUFFIX; 61 } 62 63 jaxbClasses.add(packageName+"."+className); 64 return className; 65 } 66 67 /** 68 * 69 * @return jaxbGenerated classNames 70 */ 71 public Set<String> getJaxbGeneratedClasses() { 72 return jaxbClasses; 73 } 74 75 private final static String TYPE_SUFFIX = "_Type"; 76 private ClassNameCollector classNameCollector; 77 private Set<String> jaxbClasses; 78 }