add:添加锁工具
This commit is contained in:
@@ -0,0 +1,64 @@
|
|||||||
|
package org.nl.acs.utils;
|
||||||
|
|
||||||
|
import org.nl.common.exception.BadRequestException;
|
||||||
|
|
||||||
|
import java.util.Random;
|
||||||
|
import java.util.concurrent.ConcurrentHashMap;
|
||||||
|
import java.util.concurrent.TimeUnit;
|
||||||
|
import java.util.concurrent.locks.ReentrantLock;
|
||||||
|
import java.util.function.Supplier;
|
||||||
|
|
||||||
|
public class LockUtils {
|
||||||
|
private static ConcurrentHashMap<String, ReentrantLock> locks = new ConcurrentHashMap<>();
|
||||||
|
public static <T> T lock(String key, Supplier<T> supplier,Integer second) {
|
||||||
|
ReentrantLock lock;
|
||||||
|
synchronized (key.intern()){
|
||||||
|
lock = locks.get(key);
|
||||||
|
if (lock==null){
|
||||||
|
lock = new ReentrantLock();
|
||||||
|
locks.put(key, lock) ;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
boolean isLock = false;
|
||||||
|
if (second !=null){
|
||||||
|
try {
|
||||||
|
isLock = lock.tryLock(second, TimeUnit.SECONDS);
|
||||||
|
}catch (Exception ex){
|
||||||
|
isLock = false;
|
||||||
|
}
|
||||||
|
}else {
|
||||||
|
isLock = lock.tryLock();
|
||||||
|
}
|
||||||
|
try {
|
||||||
|
if (isLock){
|
||||||
|
return supplier.get();
|
||||||
|
}else {
|
||||||
|
throw new BadRequestException(Thread.currentThread().getName()+"当前业务"+key+"正在执行");
|
||||||
|
}
|
||||||
|
}catch (Exception ex){
|
||||||
|
throw new BadRequestException(Thread.currentThread().getName()+"当前业务"+key+"正在执行");
|
||||||
|
}finally {
|
||||||
|
if (lock.isLocked() && lock.isHeldByCurrentThread()){
|
||||||
|
lock.unlock();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static void main(String[] args) {
|
||||||
|
for (int i = 0; i < 100; i++) {
|
||||||
|
new Thread(()->{
|
||||||
|
String key = String.valueOf(new Random().nextInt(5));
|
||||||
|
String result = LockUtils.lock(key, () -> {
|
||||||
|
try {
|
||||||
|
System.out.println(Thread.currentThread().getName() + "正在执行");
|
||||||
|
Thread.sleep(2000);
|
||||||
|
} catch (Exception ex) {
|
||||||
|
}
|
||||||
|
return key;
|
||||||
|
}, null);
|
||||||
|
System.out.println("执行结果"+result);
|
||||||
|
}).start();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
Reference in New Issue
Block a user