commit 713bbb175811f1919737fb22f221938f4c7b12eb
author Yazawa Nico <37871611+FancyJan@users.noreply.github.com> Mon Nov 27 17:19:14 2023 +0800
committer GitHub <noreply@github.com> Mon Nov 27 17:19:14 2023 +0800
tree 3087f432de846e1e43b7b95150a2dd8062b1607e
parent 806f4d46c9393f6aeb92aa558d95fe95d1784355

lalb try left server nodes (#2428)

* lalb try left server nodes

* beautify

* add comment

* typo

---------

Co-authored-by: zhoushan <zhoushan@pinduoduo.com>

src/brpc/policy/locality_aware_load_balancer.cpp

diff --git a/src/brpc/policy/locality_aware_load_balancer.cpp b/src/brpc/policy/locality_aware_load_balancer.cpp
index 6f0c163..68d85ad 100644
--- a/src/brpc/policy/locality_aware_load_balancer.cpp
+++ b/src/brpc/policy/locality_aware_load_balancer.cpp
@@ -334,9 +334,13 @@
             if (dice >= left + self + diff) {
                 dice -= left + self + diff;
                 index = index * 2 + 2;
-                if (index < n) {
-                    continue;
-                }
+            } else {
+		// left child may contain available nodes
+		dice = butil::fast_rand_less_than(left);
+		index = index * 2 + 1;
+	    }
+	    if (index < n) {
+		continue;
             }
             if (++ntry >= n) {
                 break;