com.sun.tools.xjc.model.AutoClassNameAllocator Maven / Gradle / Ivy
/*
* Copyright (c) 1997, 2022 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.xjc.model;
import java.util.Set;
import java.util.Map;
import java.util.HashMap;
import java.util.HashSet;
import com.sun.tools.xjc.api.ClassNameAllocator;
/**
* {@link ClassNameAllocator} filter that performs automatic name conflict resolution.
*
* @author Kohsuke Kawaguchi
*/
public class AutoClassNameAllocator implements ClassNameAllocator {
private final ClassNameAllocator core;
private final Map> names = new HashMap<>();
public AutoClassNameAllocator(ClassNameAllocator core) {
this.core = core;
}
@Override
public String assignClassName(String packageName, String className) {
className = determineName(packageName, className);
if(core!=null)
className = core.assignClassName(packageName,className);
return className;
}
private String determineName(String packageName, String className) {
Set s = names.computeIfAbsent(packageName, k -> new HashSet<>());
if(s.add(className))
return className;
for(int i=2;true;i++) {
if(s.add(className+i))
return className+i;
}
}
}