ZhouXY108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-05-09 23:51:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 时间轮添加任务线程安全问题

Browse Source
This commit is contained in:
BeauFang 2022-11-05 12:37:26 +08:00
parent 4b2ec7d7a5
commit 25e1ee4d62
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
hutool-cron/src/main/java/cn/hutool/cron/timingwheel/TimingWheel.java
View File

@ -72,6 +72,7 @@ public class TimingWheel {
this.wheelSize = wheelSize; this.wheelSize = wheelSize;
this.interval = tickMs * wheelSize; this.interval = tickMs * wheelSize;
this.timerTaskLists = new TimerTaskList[wheelSize]; this.timerTaskLists = new TimerTaskList[wheelSize];
initTimerTaskList();
//currentTime为tickMs的整数倍 这里做取整操作 //currentTime为tickMs的整数倍 这里做取整操作
this.currentTime = currentTime - (currentTime % tickMs); this.currentTime = currentTime - (currentTime % tickMs);
this.consumer = consumer; this.consumer = consumer;
@ -93,12 +94,7 @@ public class TimingWheel {
long virtualId = expiration / tickMs; long virtualId = expiration / tickMs;
int index = (int) (virtualId % wheelSize); int index = (int) (virtualId % wheelSize);
StaticLog.debug("tickMs: {} ------index: {} ------expiration: {}", tickMs, index, expiration); StaticLog.debug("tickMs: {} ------index: {} ------expiration: {}", tickMs, index, expiration);
TimerTaskList timerTaskList = timerTaskLists[index]; TimerTaskList timerTaskList = timerTaskLists[index];
if (null == timerTaskList) {
timerTaskList = new TimerTaskList();
timerTaskLists[index] = timerTaskList;
}
timerTaskList.addTask(timerTask); timerTaskList.addTask(timerTask);
if (timerTaskList.setExpiration(virtualId * tickMs)) { if (timerTaskList.setExpiration(virtualId * tickMs)) {
//添加到delayQueue中 //添加到delayQueue中
@ -140,4 +136,13 @@ public class TimingWheel {
} }
return overflowWheel; return overflowWheel;
} }
/**
* 初始 timerTaskLists
*/
private void initTimerTaskList() {
for (int i = 0; i < this.timerTaskLists.length; i++) {
this.timerTaskLists[i] = new TimerTaskList();
}
}
} }