This is done in preparation to the next patches, where the helper will be used in different packages.
As the helper is not dependent on resource affinity rules at all, rename it to find_disjoint_rules_resource_sets(...) and adapt the documentation accordingly. Signed-off-by: Daniel Kral <[email protected]> Reviewed-by: Michael Köppl <[email protected]> Tested-by: Michael Köppl <[email protected]> --- changes since v1: - remove sort_by_lowest_resource_id and find_disjoint_rules_resource_sets here already from PVE::HA::Rules::ResourceAffinity instead of the next patch (was a squashing error while rebasing the v1) debian/pve-ha-manager.install | 1 + src/PVE/HA/Rules/Helpers.pm | 61 ++++++++++++++++++++++++++++ src/PVE/HA/Rules/Makefile | 2 +- src/PVE/HA/Rules/ResourceAffinity.pm | 58 ++------------------------ 4 files changed, 67 insertions(+), 55 deletions(-) create mode 100644 src/PVE/HA/Rules/Helpers.pm diff --git a/debian/pve-ha-manager.install b/debian/pve-ha-manager.install index 2e6b7d55..38d5d60b 100644 --- a/debian/pve-ha-manager.install +++ b/debian/pve-ha-manager.install @@ -35,6 +35,7 @@ /usr/share/perl5/PVE/HA/Resources/PVECT.pm /usr/share/perl5/PVE/HA/Resources/PVEVM.pm /usr/share/perl5/PVE/HA/Rules.pm +/usr/share/perl5/PVE/HA/Rules/Helpers.pm /usr/share/perl5/PVE/HA/Rules/NodeAffinity.pm /usr/share/perl5/PVE/HA/Rules/ResourceAffinity.pm /usr/share/perl5/PVE/HA/Tools.pm diff --git a/src/PVE/HA/Rules/Helpers.pm b/src/PVE/HA/Rules/Helpers.pm new file mode 100644 index 00000000..2e6ea3f8 --- /dev/null +++ b/src/PVE/HA/Rules/Helpers.pm @@ -0,0 +1,61 @@ +package PVE::HA::Rules::Helpers; + +use strict; +use warnings; + +use PVE::HA::HashTools qw(sets_are_disjoint); + +my $sort_by_lowest_resource_id = sub { + my ($rules) = @_; + + my $lowest_rule_resource_id = {}; + for my $ruleid (keys %$rules) { + my @rule_resources = sort keys $rules->{$ruleid}->{resources}->%*; + $lowest_rule_resource_id->{$ruleid} = $rule_resources[0]; + } + + # sort rules such that rules with the lowest numbered resource come first + my @sorted_ruleids = sort { + $lowest_rule_resource_id->{$a} cmp $lowest_rule_resource_id->{$b} + } sort keys %$rules; + + return @sorted_ruleids; +}; + +# returns a list of hashes, where each hash item contains a resource set +# disjoint from the other items and the ruleids that were used to form that +# resource set. +sub find_disjoint_rules_resource_sets { + my ($rules) = @_; + + my @disjoint_rules = (); + + # order needed so that it is easier to check whether there is an overlap + my @sorted_ruleids = $sort_by_lowest_resource_id->($rules); + + for my $ruleid (@sorted_ruleids) { + my $rule = $rules->{$ruleid}; + + my $found = 0; + for my $entry (@disjoint_rules) { + next if sets_are_disjoint($rule->{resources}, $entry->{resources}); + + $found = 1; + push @{ $entry->{ruleids} }, $ruleid; + $entry->{resources}->{$_} = 1 for keys $rule->{resources}->%*; + + last; + } + if (!$found) { + push @disjoint_rules, + { + ruleids => [$ruleid], + resources => { $rule->{resources}->%* }, + }; + } + } + + return @disjoint_rules; +} + +1; diff --git a/src/PVE/HA/Rules/Makefile b/src/PVE/HA/Rules/Makefile index 64119257..ce030d1f 100644 --- a/src/PVE/HA/Rules/Makefile +++ b/src/PVE/HA/Rules/Makefile @@ -1,4 +1,4 @@ -SOURCES=NodeAffinity.pm ResourceAffinity.pm +SOURCES=Helpers.pm NodeAffinity.pm ResourceAffinity.pm .PHONY: install install: diff --git a/src/PVE/HA/Rules/ResourceAffinity.pm b/src/PVE/HA/Rules/ResourceAffinity.pm index 9bc039ba..8a2bee5d 100644 --- a/src/PVE/HA/Rules/ResourceAffinity.pm +++ b/src/PVE/HA/Rules/ResourceAffinity.pm @@ -3,8 +3,9 @@ package PVE::HA::Rules::ResourceAffinity; use strict; use warnings; -use PVE::HA::HashTools qw(set_intersect sets_are_disjoint); +use PVE::HA::HashTools qw(set_intersect); use PVE::HA::Rules; +use PVE::HA::Rules::Helpers; use base qw(Exporter); use base qw(PVE::HA::Rules); @@ -247,58 +248,6 @@ __PACKAGE__->register_check( =cut -my $sort_by_lowest_resource_id = sub { - my ($rules) = @_; - - my $lowest_rule_resource_id = {}; - for my $ruleid (keys %$rules) { - my @rule_resources = sort keys $rules->{$ruleid}->{resources}->%*; - $lowest_rule_resource_id->{$ruleid} = $rule_resources[0]; - } - - # sort rules such that rules with the lowest numbered resource come first - my @sorted_ruleids = sort { - $lowest_rule_resource_id->{$a} cmp $lowest_rule_resource_id->{$b} - } sort keys %$rules; - - return @sorted_ruleids; -}; - -# returns a list of hashes, which contain disjoint resource affinity rules, i.e., -# put resource affinity constraints on disjoint sets of resources -my $find_disjoint_resource_affinity_rules = sub { - my ($rules) = @_; - - my @disjoint_rules = (); - - # order needed so that it is easier to check whether there is an overlap - my @sorted_ruleids = $sort_by_lowest_resource_id->($rules); - - for my $ruleid (@sorted_ruleids) { - my $rule = $rules->{$ruleid}; - - my $found = 0; - for my $entry (@disjoint_rules) { - next if sets_are_disjoint($rule->{resources}, $entry->{resources}); - - $found = 1; - push @{ $entry->{ruleids} }, $ruleid; - $entry->{resources}->{$_} = 1 for keys $rule->{resources}->%*; - - last; - } - if (!$found) { - push @disjoint_rules, - { - ruleids => [$ruleid], - resources => { $rule->{resources}->%* }, - }; - } - } - - return @disjoint_rules; -}; - =head3 merge_connected_positive_resource_affinity_rules($rules, $positive_rules) Modifies C<$rules> to contain only disjoint positive resource affinity rules @@ -319,7 +268,8 @@ a resource, in C<$rules> at a later point in time. sub merge_connected_positive_resource_affinity_rules { my ($rules, $positive_rules) = @_; - my @disjoint_positive_rules = $find_disjoint_resource_affinity_rules->($positive_rules); + my @disjoint_positive_rules = + PVE::HA::Rules::Helpers::find_disjoint_rules_resource_sets($positive_rules); for my $entry (@disjoint_positive_rules) { next if @{ $entry->{ruleids} } < 2; -- 2.47.3 _______________________________________________ pve-devel mailing list [email protected] https://lists.proxmox.com/cgi-bin/mailman/listinfo/pve-devel
