package com.product.data.center.utils;
|
|
import cn.hutool.core.collection.ConcurrentHashSet;
|
import cn.hutool.core.date.DateUtil;
|
import cn.hutool.core.thread.ThreadUtil;
|
import cn.hutool.core.util.ArrayUtil;
|
import com.product.core.spring.context.SpringMVCContextHolder;
|
import org.apache.commons.lang3.StringUtils;
|
|
import javax.swing.*;
|
import java.text.DateFormat;
|
import java.util.Date;
|
import java.util.Map;
|
import java.util.Set;
|
import java.util.concurrent.ConcurrentHashMap;
|
|
/**
|
* @Author cheng
|
* @Date 2022/11/14 11:30
|
* @Desc
|
*/
|
public class CustomLock {
|
|
|
private Map<String, Long> obj = new ConcurrentHashMap<>();
|
private Map<String, Long> threadObj = new ConcurrentHashMap<>();
|
|
|
private long detectionWaitTime = 1000L;
|
|
|
/**
|
* 无参构造时
|
*/
|
public CustomLock() {
|
}
|
|
/**
|
* @param detectionWaitTime 检测等待时间，单位毫秒
|
*/
|
public CustomLock(long detectionWaitTime) {
|
this.detectionWaitTime = detectionWaitTime;
|
}
|
|
public boolean tryLock(String[] key) {
|
return this.tryLock(ArrayUtil.join(key, ","));
|
}
|
|
public static void main(String[] args) {
|
long timemillis = 1709654400039L;
|
//转换为年月日时分秒
|
String format = DateUtil.format(DateUtil.date(timemillis), "yyyy-MM-dd HH:mm:ss");
|
|
System.out.println(format);
|
}
|
|
public boolean tryLock(String key, long timeoutMinute) {
|
if (!StringUtils.isEmpty(key)) {
|
key = key.toUpperCase();
|
synchronized (key.intern()) {
|
if (!this.obj.containsKey(key)) {
|
lock(key);
|
//获取当前线程
|
long id = Thread.currentThread().getId();
|
threadObj.put(key, id);
|
return true;
|
} else if (timeoutMinute > 0) {
|
//获取锁进入开始时间
|
long startTime = obj.get(key);
|
//获取当前时间
|
long currentTime = System.currentTimeMillis();
|
|
//将分钟转换为毫秒
|
long timeout = timeoutMinute * 60 * 1000;
|
//如果当前时间减去开始时间大于等于超时时间
|
if (currentTime - startTime >= timeout) {
|
long threadId = threadObj.get(key);
|
//根据线程id获取线程
|
Thread[] threads = ThreadUtil.getThreads();
|
for (Thread thread : threads) {
|
if (thread.getId() == threadId) {
|
//中断线程
|
ThreadUtil.interrupt(thread, true);
|
SpringMVCContextHolder.getSystemLogger().error("线程：" + thread.getName() + "超时被中断");
|
return tryLock(key, timeoutMinute);
|
}
|
}
|
return true;
|
}
|
|
}
|
}
|
}
|
return false;
|
}
|
|
|
public boolean tryLock(String key) {
|
if (!StringUtils.isEmpty(key)) {
|
key = key.toUpperCase();
|
synchronized (key.intern()) {
|
if (!this.obj.containsKey(key)) {
|
lock(key);
|
//获取当前线程
|
long id = Thread.currentThread().getId();
|
threadObj.put(key, id);
|
return true;
|
}
|
}
|
}
|
return false;
|
}
|
|
|
public void lock(String... key) {
|
this.lock(ArrayUtil.join(key, ","));
|
}
|
|
public void lock(String key) {
|
String obj = key;
|
if (StringUtils.isEmpty(obj)) {
|
return;
|
}
|
obj = obj.toUpperCase();
|
// boolean enterLock = false;
|
long starTime = System.currentTimeMillis();
|
Long outLogTime = null;
|
while (this.obj.containsKey(obj)) {
|
if (System.currentTimeMillis() - starTime >= (1000 * 60 * 5)) {
|
if (outLogTime == null || System.currentTimeMillis() - outLogTime >= (1000 * 60)) {
|
outLogTime = System.currentTimeMillis();
|
SpringMVCContextHolder.getSystemLogger().error("等待释放锁时间超过5分钟,等待时间：" + (System.currentTimeMillis() - starTime) + "ms,lockKey:" + obj);
|
}
|
}
|
// enterLock = true;
|
// SpringMVCContextHolder.getSystemLogger().info("锁释放等待,lockKey:" + obj);
|
ThreadUtil.sleep(this.detectionWaitTime);
|
}
|
// if (enterLock) {
|
// SpringMVCContextHolder.getSystemLogger().info("等待定时间：" + (System.currentTimeMillis() - starTime) + " ms,lockKey:" + obj);
|
// }
|
// SpringMVCContextHolder.getSystemLogger().info("上锁成功,lockKey:" + obj);
|
this.obj.put(obj, System.currentTimeMillis());
|
}
|
|
public void unLock(String... key) {
|
String obj = ArrayUtil.join(key, ",");
|
unLock(obj);
|
}
|
|
public void unLock(String key) {
|
|
if (StringUtils.isEmpty(key)) {
|
return;
|
}
|
this.obj.remove(key.toUpperCase());
|
// SpringMVCContextHolder.getSystemLogger().info("释放锁成功,lockKey:" + key);
|
}
|
|
|
}
|
