diff --git a/snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceFirst.java b/snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceFirst.java new file mode 100644 index 000000000..57f3d72df --- /dev/null +++ b/snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceFirst.java @@ -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 clientAllAddressSet) { + return clientAllAddressSet.first(); + } + + @Override + public int routeType() { + return ClientLoadBalanceManager.AllocationAlgorithmEnum.FIRST.getType(); + } +} diff --git a/snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceLast.java b/snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceLast.java new file mode 100644 index 000000000..1e978cc21 --- /dev/null +++ b/snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceLast.java @@ -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 clientAllAddressSet) { + return clientAllAddressSet.last(); + } + + @Override + public int routeType() { + return ClientLoadBalanceManager.AllocationAlgorithmEnum.LAST.getType(); + } +} diff --git a/snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceManager.java b/snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceManager.java index a18183aba..b8fd40d70 100644 --- a/snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceManager.java +++ b/snail-job-server/snail-job-server-common/src/main/java/com/aizuda/snailjob/server/common/allocate/client/ClientLoadBalanceManager.java @@ -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;