mirrors/scylladb
1
0
Fork 0
mirror of https://github.com/scylladb/scylladb.git synced 2026-05-29 19:21:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

tablets: load_balancer: Extract nodes_by_load_cmp

Browse Source 
Will be reused in a different place.
This commit is contained in:
Tomasz Grabiec
2024-07-16 20:44:42 +02:00
parent 8a7257971d
commit 41e643ddb9
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
service/tablet_allocator.cc
View File

@@ -311,6 +311,15 @@ class load_balancer {
// Data structure used for making load-balancing decisions over a set of nodes.
using node_load_map = std::unordered_map<host_id, node_load>;
// Less-comparator which orders nodes by load.
struct nodes_by_load_cmp {
node_load_map& nodes;
bool operator()(host_id a, host_id b) const {
return nodes[a].avg_load < nodes[b].avg_load;
}
};
// We have split and merge thresholds, which work respectively as (target) upper and lower
// bound for average size of tablets.
//
@@ -884,9 +893,7 @@ public:
std::vector<host_id> nodes_by_load_dst;
nodes_by_load_dst.reserve(nodes.size());
auto nodes_cmp = [&] (const host_id& a, const host_id& b) {
return nodes[a].avg_load < nodes[b].avg_load;
};
auto nodes_cmp = nodes_by_load_cmp(nodes);
auto nodes_dst_cmp = [&] (const host_id& a, const host_id& b) {
return nodes_cmp(b, a);
};