功能主要是用Map来保存id,首先判断当前的key中有没有保存id,有则获得id并自增,如果没有则创建一个AtomicInteger对象来保存id。创建过程中使用了同步锁,然而在初始化过程中依然会出现相同的id。代码如下:
public class IIDGenerator {
private static Map<String, AtomicInteger> map;
private static final Object lock = new Object();
public IIDGenerator() {
map = new ConcurrentHashMap<>();
}
public Integer get(String key, int init) {
if (!map.containsKey(key)) {
synchronized (lock) {
map.put(key, new AtomicInteger(init));
}
}
return map.get(key).getAndIncrement();
}
public static void main(String[] args) {
IIDGenerator generator = new IIDGenerator();
for (int i = 0; i < 10000; i++) {
new Thread(() -> {
int iid = generator.get("id1", 1);
System.out.println("id1:" + iid);
}).start();
}
for (int i = 0; i < 10000; i++) {
new Thread(() -> {
int iid = generator.get("id2", 1);
System.out.println("id2:" + iid);
}).start();
}
}
}
假设启动时有两个线程同时通过了get方法中的if判断,接下来第一个线程执行同步块中的方法,创建一个新的对象,然后返回id值并自增。第二个线程已经通过了if判断,在第一个线程执行完同步块之后也会进入同步块,再执行一次创建AtomicInteger对象,因此会产生相同的id。
所以在同步块中需要进行一次判断,判断是否已经初始化。以下为修改之后的get方法
public Integer get(String key, int init) {
if (!map.containsKey(key)) {
synchronized (lock) {
if (!map.containsKey(key)) {
map.put(key, new AtomicInteger(init));
}
}
}
return map.get(key).getAndIncrement();
}