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

Pre Merge pull request !79 from Srzou/1.2.0-beta1

Browse Source
This commit is contained in:
Srzou 2024-07-30 08:34:39 +00:00 committed by Gitee
commit 874ddddafe
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
5 changed files with 51 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceFirst.java Normal file
View File

@ -0,0 +1,17 @@
package com.aizuda.snailjob.server.common.allocate.client;
import com.aizuda.snailjob.server.common.ClientLoadBalance;
import java.util.TreeSet;
public class ClientLoadBalanceFirst implements ClientLoadBalance {
@Override
public String route(String key, TreeSet<String> clientAllAddressSet) {
return clientAllAddressSet.first();
}
@Override
public int routeType() {
return ClientLoadBalanceManager.AllocationAlgorithmEnum.FIRST.getType();
}
}

7
snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceLRU.java
View File

@ -21,9 +21,16 @@ public class ClientLoadBalanceLRU implements ClientLoadBalance {
}
private ConcurrentHashMap<String, LinkedHashMap<String, String>> LRU_CACHE = new ConcurrentHashMap<>();
private static long CACHE_VALID_TIME = 0;
@Override
public String route(String allocKey, TreeSet<String> clientAllAddressSet) {
// cache clear
if (System.currentTimeMillis() > CACHE_VALID_TIME) {
LRU_CACHE.clear();
// 每12个小时定时清理一次数据
CACHE_VALID_TIME = System.currentTimeMillis() + 1000*60*60*12;
}
LinkedHashMap<String, String> lruItem = LRU_CACHE.get(allocKey);
if (Objects.isNull(lruItem)) {
lruItem = new LinkedHashMap<String, String>(16, 0.75f, true) {

17
snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceLast.java Normal file
View File

@ -0,0 +1,17 @@
package com.aizuda.snailjob.server.common.allocate.client;
import com.aizuda.snailjob.server.common.ClientLoadBalance;
import java.util.TreeSet;
public class ClientLoadBalanceLast implements ClientLoadBalance {
@Override
public String route(String key, TreeSet<String> clientAllAddressSet) {
return clientAllAddressSet.last();
}
@Override
public int routeType() {
return ClientLoadBalanceManager.AllocationAlgorithmEnum.LAST.getType();
}
}

4
snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceManager.java
View File

@ -29,7 +29,9 @@ public class ClientLoadBalanceManager {
CONSISTENT_HASH(1, new ClientLoadBalanceConsistentHash(100)),
RANDOM(2, new ClientLoadBalanceRandom()),
LRU(3, new ClientLoadBalanceLRU(100)),
ROUND(4, new ClientLoadBalanceRound());
ROUND(4, new ClientLoadBalanceRound()),
FIRST(5, new ClientLoadBalanceFirst()),
LAST(6, new ClientLoadBalanceLast());
private final int type;
private final ClientLoadBalance clientLoadBalance;

7
snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceRound.java
View File

@ -16,9 +16,16 @@ public class ClientLoadBalanceRound implements ClientLoadBalance {
private static final ConcurrentHashMap<String, AtomicInteger> COUNTER = new ConcurrentHashMap<>();
private static final int THRESHOLD = Integer.MAX_VALUE - 10000;
private static long CACHE_VALID_TIME = 0;
@Override
public String route(final String allocKey, final TreeSet<String> clientAllAddressSet) {
// cache clear
if (System.currentTimeMillis() > CACHE_VALID_TIME) {
COUNTER.clear();
// 每12个小时定时清理一次数据
CACHE_VALID_TIME = System.currentTimeMillis() + 1000*60*60*12;
}
String[] addressArr = clientAllAddressSet.toArray(new String[0]);
AtomicInteger next = COUNTER.getOrDefault(allocKey, new AtomicInteger(1));
String nextClientId = addressArr[next.get() % clientAllAddressSet.size()];