src/share/jaxws_classes/com/sun/tools/internal/ws/processor/modeler/wsdl/ClassNameAllocatorImpl.java

changeset 286
f50545b5e2f1
child 368
0989ad8c0860
     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	Tue Mar 06 16:09:35 2012 -0800
     1.3 @@ -0,0 +1,78 @@
     1.4 +/*
     1.5 + * Copyright (c) 1997, 2010, 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 +}

mercurial