]> git.proxmox.com Git - pve-ha-manager.git/blobdiff - src/PVE/HA/Tools.pm
Tools/Config: refactor lrm status json reading
[pve-ha-manager.git] / src / PVE / HA / Tools.pm
index 94613de8dc0850d45ad2db649891c6c599c64831..2bdd6eaff65e05dabaef2561b052b1f880bb793e 100644 (file)
@@ -5,6 +5,27 @@ use warnings;
 use JSON;
 use PVE::JSONSchema;
 use PVE::Tools;
+use PVE::Cluster;
+use PVE::ProcFSTools;
+
+# return codes used in the ha environment
+# mainly by the resource agents
+use constant {
+    SUCCESS => 0, # action finished as expected
+    ERROR => 1, # action was erroneous
+    ETRY_AGAIN => 2, # action was erroneous and needs to be repeated
+    EWRONG_NODE => 3, # needs to fixup the service location
+    EUNKNOWN_SERVICE_TYPE => 4, # no plugin for this type service found
+    EUNKNOWN_COMMAND => 5,
+    EINVALID_PARAMETER => 6,
+    EUNKNOWN_SERVICE => 7, # service not found
+};
+
+# get constants out of package in a somewhat easy way
+use base 'Exporter';
+our @EXPORT_OK = qw(SUCCESS ERROR EWRONG_NODE EUNKNOWN_SERVICE_TYPE
+ EUNKNOWN_COMMAND EINVALID_PARAMETER ETRY_AGAIN EUNKNOWN_SERVICE);
+our %EXPORT_TAGS = ( 'exit_codes' => [@EXPORT_OK] );
 
 PVE::JSONSchema::register_format('pve-ha-resource-id', \&pve_verify_ha_resource_id);
 sub pve_verify_ha_resource_id {
@@ -18,7 +39,7 @@ sub pve_verify_ha_resource_id {
 }
 
 PVE::JSONSchema::register_standard_option('pve-ha-resource-id', {
-    description => "HA resource ID. This consists of a resource type followed by a resource specific name, separated with collon (example: vm:100).",
+    description => "HA resource ID. This consists of a resource type followed by a resource specific name, separated with colon (example: vm:100 / ct:100).",
     typetext => "<type>:<name>",
     type => 'string', format => 'pve-ha-resource-id',                                   
 });
@@ -35,7 +56,7 @@ sub pve_verify_ha_resource_or_vm_id {
 }
 
 PVE::JSONSchema::register_standard_option('pve-ha-resource-or-vm-id', {
-    description => "HA resource ID. This consists of a resource type followed by a resource specific name, separated with collon (example: vm:100). For virtual machines, you can simply use the VM id as shortcut (example: 100).",
+    description => "HA resource ID. This consists of a resource type followed by a resource specific name, separated with colon (example: vm:100 / ct:100). For virtual machines and containers, you can simply use the VM or CT id as a shortcut (example: 100).",
     typetext => "<type>:<name>",
     type => 'string', format => 'pve-ha-resource-or-vm-id',                                     
 });
@@ -52,7 +73,8 @@ sub pve_verify_ha_group_node {
 }
 
 PVE::JSONSchema::register_standard_option('pve-ha-group-node-list', {
-    description => "List of cluster node names with optional priority. We use priority '0' as default. The CRM tries to run services on the node with higest priority (also see option 'nofailback').",
+    description => "List of cluster node names with optional priority.",
+    verbose_description => "List of cluster node members, where a priority can be given to each node. A resource bound to a group will run on the available nodes with the highest priority. If there are more nodes in the highest priority class, the services will get distributed to those nodes. The priorities have a relative meaning only.",
     type => 'string', format => 'pve-ha-group-node-list',
     typetext => '<node>[:<pri>]{,<node>[:<pri>]}*',
 });
@@ -69,8 +91,21 @@ sub parse_sid {
 
     if ($sid =~ m/^(\d+)$/) {
        $name = $1;
-       $type ='vm';
-       $sid = "vm:$name";
+       my $vmlist = PVE::Cluster::get_vmlist();
+       if (defined($vmlist->{ids}->{$name})) {
+           my $vm_type = $vmlist->{ids}->{$name}->{type};
+           if ($vm_type eq 'lxc') {
+               $type = 'ct';
+           } elsif ($vm_type eq 'qemu') {
+               $type = 'vm';
+           } else {
+               die "internal error";
+           }
+           $sid = "$type:$name";
+       }
+       else {
+           die "unable do add resource - VM/CT $1 does not exist\n";
+       }
     } elsif  ($sid =~m/^(\S+):(\S+)$/) {
        $name = $2;
        $type = $1;
@@ -124,4 +159,135 @@ sub count_fenced_services {
     return $count;
 }
 
+sub get_verbose_service_state {
+    my ($service_state, $service_conf) = @_;
+
+    my $req = $service_conf->{state} // 'ignored';
+    return 'ignored' if $req eq 'ignored';
+
+    # service not yet processed by manager
+    return 'queued' if !defined($service_state);
+    my $cur = $service_state->{state};
+
+    # give fast feedback to the user
+    my $state = $cur;
+    if (!defined($cur)) {
+       $state = 'queued';
+    } elsif ($cur eq 'stopped') {
+       if ($req eq 'started') {
+           $state = 'starting';
+       } elsif ($req eq 'disabled') {
+           $state = 'disabled';
+       }
+    } elsif ($cur eq 'started') {
+       if ($req eq 'stopped' || $req eq 'disabled') {
+           $state = 'stopping';
+       }
+       $state = 'starting' if !$service_state->{running};
+    } elsif ($cur eq 'error') {
+       if ($req eq 'disabled') {
+           $state = 'clearing error flag';
+       }
+    }
+
+    return $state;
+}
+
+sub upid_wait {
+    my ($upid, $haenv) = @_;
+
+    my $waitfunc = sub {
+       my $task = PVE::Tools::upid_encode(shift);
+       $haenv->log('info', "Task '$task' still active, waiting");
+    };
+
+    PVE::ProcFSTools::upid_wait($upid, $waitfunc, 5);
+}
+
+# bash auto completion helper
+
+sub complete_sid {
+    my ($cmd, $pname, $cur) = @_;
+
+    my $cfg = PVE::HA::Config::read_resources_config();
+
+    my $res = [];
+
+    if ($cmd eq 'add') {
+
+       my $vmlist = PVE::Cluster::get_vmlist();
+
+       while (my ($vmid, $info) = each %{$vmlist->{ids}}) {
+
+           my $sid;
+
+           if ($info->{type} eq 'lxc') {
+               $sid = "ct:$vmid";
+           } elsif ($info->{type} eq 'qemu') {
+               $sid = "vm:$vmid";
+           } else {
+               next; # should not happen
+           }
+
+           next if $cfg->{ids}->{$sid};
+
+           push @$res, $sid;
+       }
+
+    } else {
+
+       foreach my $sid (keys %{$cfg->{ids}}) {
+           push @$res, $sid;
+       }
+    }
+
+    return $res;
+}
+
+sub complete_enabled_sid {
+
+    my $cfg = PVE::HA::Config::read_resources_config();
+
+    my $res = [];
+    foreach my $sid (keys %{$cfg->{ids}}) {
+       my $state = $cfg->{ids}->{$sid}->{state} // 'started';
+       next if $state ne 'started';
+       push @$res, $sid;
+    }
+
+    return $res;
+}
+
+sub complete_disabled_sid {
+
+    my $cfg = PVE::HA::Config::read_resources_config();
+
+    my $res = [];
+    foreach my $sid (keys %{$cfg->{ids}}) {
+       my $state = $cfg->{ids}->{$sid}->{state} // 'started';
+       next if $state eq 'started';
+       push @$res, $sid;
+    }
+
+    return $res;
+}
+
+sub complete_group {
+    my ($cmd, $pname, $cur) = @_;
+
+    my $cfg = PVE::HA::Config::read_group_config();
+
+    my $res = [];
+    if ($cmd ne 'groupadd') {
+
+       foreach my $group (keys %{$cfg->{ids}}) {
+           push @$res, $group;
+       }
+
+    }
+
+    return $res;
+}
+
+
 1;