mirrors/scylladb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: perf_load_balancing: Set node capacity

Browse Source 
Otherwise, load balancer will not make any plan once it becomes
capacity-aware.
This commit is contained in:
Tomasz Grabiec
2025-02-28 11:43:15 +01:00
parent 6169401dbc
commit dfc9101dfd
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
test/perf/tablet_load_balancing.cc
View File

@@ -263,10 +263,17 @@ future<results> test_load_balancing_with_many_tables(params p, bool tablet_aware
topology_builder topo(e);
std::vector<host_id> hosts;
locator::load_stats stats;
auto add_host = [&] {
hosts.push_back(topo.add_node(service::node_state::normal, shard_count));
testlog.info("Added new node: {}", hosts.back());
auto host = topo.add_node(service::node_state::normal, shard_count);
hosts.push_back(host);
stats.capacity[host] = default_target_tablet_size * shard_count;
testlog.info("Added new node: {}", host);
};
auto make_stats = [&] {
return make_lw_shared<locator::load_stats>(stats);
};
for (int i = 0; i < n_hosts; ++i) {
@@ -277,7 +284,7 @@ future<results> test_load_balancing_with_many_tables(params p, bool tablet_aware
auto bootstrap = [&] {
add_host();
global_res.stats += rebalance_tablets(e);
global_res.stats += rebalance_tablets(e, make_stats());
};
auto decommission = [&] (host_id host) {
@@ -286,7 +293,7 @@ future<results> test_load_balancing_with_many_tables(params p, bool tablet_aware
throw std::runtime_error(format("No such host: {}", host));
}
topo.set_node_state(host, service::node_state::decommissioning);
global_res.stats += rebalance_tablets(e);
global_res.stats += rebalance_tablets(e, make_stats());
if (stm.get()->tablets().has_replica_on(host)) {
throw std::runtime_error(format("Host {} still has replicas!", host));
}
@@ -364,7 +371,7 @@ future<results> test_load_balancing_with_many_tables(params p, bool tablet_aware
check_balance();
rebalance_tablets(e);
rebalance_tablets(e, make_stats());
global_res.init = global_res.worst = check_balance();