gtsoft/gtsoft-snail-job-server
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects 1 Releases Wiki Activity

fix(sj_1.1.0): 优化负载均衡异常问题

Browse Source
This commit is contained in:
opensnail 2024-06-13 17:12:36 +08:00
parent ae31c5ef87
commit 7894cfb284
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/server/AllocateMessageQueueAveragely.java
View File

@ -1,6 +1,9 @@
package com.aizuda.snailjob.server.common.allocate.server; package com.aizuda.snailjob.server.common.allocate.server;
import cn.hutool.core.collection.CollUtil;
import com.aizuda.snailjob.common.log.SnailJobLog;
import com.aizuda.snailjob.server.common.ServerLoadBalance; import com.aizuda.snailjob.server.common.ServerLoadBalance;
import com.google.common.collect.Lists;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.List; import java.util.List;
@ -17,10 +20,18 @@ public class AllocateMessageQueueAveragely implements ServerLoadBalance<Integer,
@Override @Override
public List<Integer> allocate(String currentCID, List<Integer> bucketList, List<String> serverList) { public List<Integer> allocate(String currentCID, List<Integer> bucketList, List<String> serverList) {
if (CollUtil.isEmpty(serverList)) {
return Lists.newArrayList();
}
List<Integer> consumerBucket = new ArrayList<>(); List<Integer> consumerBucket = new ArrayList<>();
// 找到当前消费者在消费者队列里面的下标 // 找到当前消费者在消费者队列里面的下标
int index = serverList.indexOf(currentCID); int index = serverList.indexOf(currentCID);
if (index < 0) {
SnailJobLog.LOCAL.warn("currentCID: [{}] not in serverList:[{}]", currentCID, serverList);
return Lists.newArrayList();
}
// 此处取余是为了判断队列与总消费者数的是否是整数倍 mod=0则是整数倍，否则不是 // 此处取余是为了判断队列与总消费者数的是否是整数倍 mod=0则是整数倍，否则不是
int mod = bucketList.size() % serverList.size(); int mod = bucketList.size() % serverList.size();