java - 基于函数输入的线程 block

Question

我遇到了必须根据输入同步功能的情况。例如。有功能。func(int a)--- 这个函数可以被许多线程调用 我想阻止线程，如果他们试图用相同的输入整数调用这个函数。否则让他们继续不继续。我写了函数

HashTable<Integer> check=new Hashtable<Integer>();

func(int a){

Integer lock=check.get(a);
if(lock==null){
check.put(a);
}else{
check.put(a);
}
synchronized(lock){
//do something....

}

请告诉我这是否正确。另外，如果除此之外还有其他解决方案。我想弄清楚。如果我找到答案，我会发布它。

Answer 1

1) 获取锁也应该在同步块(synchronized block)中

2) 存在内存泄漏 - 你不删除未使用的锁

我会这样做

Map<Integer, Lock> map = new HashMap<>();

static class Lock {
    int count = 1;
}

void func(int i) {
    Object lock = getLock(i);
    try {
        synchronized (lock) {
            //
        }
    } finally {
        releaseLock(i);
    }
}

private synchronized Object getLock(int i) {
    Lock lock = map.get(i);
    if (lock != null) {
        lock.count++;
    } else {
        lock = new Lock();
        map.put(i, lock);
    }
    return lock;
}

private synchronized void releaseLock(int i) {
    Lock lock = map.get(i);
    if (--lock.count == 0) {
        map.remove(i);   
    }
}