可能有一种简单的方法可以让您的数据库为您处理这些问题。诚然,我在数据库方面的知识很薄弱。取而代之的是,这里有一种方法,它涉及为每个键名创建一个单独的锁。有一个单独的存储库管理单个锁的创建/销毁,需要一个针对整个应用程序的锁,但它仅在找到、创建或销毁单个键名锁时才持有该锁.为实际数据库操作持有的锁是该操作中使用的键名专有的。
KeyLock 类用于防止对单个键名同时进行数据库操作。
package KeyLocks;
import java.util.concurrent.locks.Lock;
public class KeyLock
{
private final KeyLockManager keyLockManager;
private final String keyName;
private final Lock lock;
KeyLock(KeyLockManager keyLockManager, String keyName, Lock lock)
{
this.keyLockManager = keyLockManager;
this.keyName = keyName;
this.lock = lock;
}
@Override
protected void finalize()
{
release();
}
public void release()
{
keyLockManager.releaseLock(keyName);
}
public void lock()
{
lock.lock();
}
public void unlock()
{
lock.unlock();
}
}
KeyLockManager 类是负责密钥锁生命周期的存储库。
package KeyLocks;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;
public class KeyLockManager
{
private class LockEntry
{
int acquisitionCount = 0;
final Lock lock = new ReentrantLock();
}
private final Map<String, LockEntry> locks = new HashMap<String, LockEntry>();
private final Object mutex = new Object();
public KeyLock getLock(String keyName)
{
synchronized (mutex)
{
LockEntry lockEntry = locks.get(keyName);
if (lockEntry == null)
{
lockEntry = new LockEntry();
locks.put(keyName, lockEntry);
}
lockEntry.acquisitionCount++;
return new KeyLock(this, keyName, lockEntry.lock);
}
}
void releaseLock(String keyName)
{
synchronized (mutex)
{
LockEntry lockEntry = locks.get(keyName);
lockEntry.acquisitionCount--;
if (lockEntry.acquisitionCount == 0)
{
locks.remove(keyName);
}
}
}
}
以下是如何使用钥匙锁的示例。
package test;
import KeyLocks.KeyLock;
import KeyLocks.KeyLockManager;
public class Main
{
private static final String KEY_NAME = "TEST_KEY";
public static void main(String[] args)
{
final KeyLockManager keyLockManager = new KeyLockManager();
KeyLock keyLock = null;
try
{
keyLock = keyLockManager.getLock(KEY_NAME);
keyLock.lock();
try
{
// Do database operation on the data with the specified key name
}
finally
{
keyLock.unlock();
}
}
finally
{
if (keyLock != null)
{
keyLock.release();
}
}
}
}