tiennm99/zfoo
1
0 Fork 0
mirror of https://github.com/tiennm99/zfoo.git synced 2026-10-13 10:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

ref[tree]: fast tree map

Browse Source
This commit is contained in:
godotg committed 2023-05-02 07:44:28 +08:00
1 parent b4705a125e
commit a4538f4719
1 file changed
+8 -7
Download Patch File Download Diff File Expand all files Collapse all files
net/src/main/java/com/zfoo/net/consumer/balancer/ConsistentHashConsumerLoadBalancer.java
+8 -7
View File
@@ -80,18 +80,18 @@ public class ConsistentHashConsumerLoadBalancer extends AbstractConsumerLoadBala
}
var module = ProtocolManager.moduleByProtocolId(packet.protocolId());
var treeMap = consistentHashMap.get(module.getId());
if (treeMap == null) {
treeMap = updateModuleToConsistentHash(module);
var fastTreeMap = consistentHashMap.get(module.getId());
if (fastTreeMap == null) {
fastTreeMap = updateModuleToConsistentHash(module);
}
if (treeMap == null) {
if (fastTreeMap == null) {
throw new RunException("ConsistentHashLoadBalancer [protocolId:{}][argument:{}], no service provides the [module:{}]", packet.protocolId(), argument, module);
}
var index = treeMap.indexOfNearestCeilingKey(argument.hashCode());
if (index < 0) {
var nearestIndex = fastTreeMap.indexOfNearestCeilingKey(argument.hashCode());
if (nearestIndex < 0) {
throw new RunException("no service provides the [module:{}]", packet.protocolId(), argument, module);
}
var sid = treeMap.getByIndex(index);
var sid = fastTreeMap.getByIndex(nearestIndex);
var session = NetContext.getSessionManager().getClientSession(sid);
if (session == null) {
throw new RunException("unknown no service provides the [module:{}]", packet.protocolId(), argument, module);
@@ -117,6 +117,7 @@ public class ConsistentHashConsumerLoadBalancer extends AbstractConsumerLoadBala
for (var entry : virtualNodeTreeMap.entrySet()) {
virtualTreeMap.put(entry.getKey(), entry.getValue().getValue());
}
// 使用更高性能的tree map
var fastTreeMap = new FastTreeMapIntLong(virtualTreeMap);
consistentHashMap.set(module.getId(), fastTreeMap);
return fastTreeMap;