]> git.proxmox.com Git - pve-ha-manager.git/blame - src/PVE/HA/LRM.pm
write lrm mode into lrm status file
[pve-ha-manager.git] / src / PVE / HA / LRM.pm
CommitLineData
5f095798
DM
1package PVE::HA::LRM;
2
3# Local Resource Manager
4
5use strict;
6use warnings;
c4a221bc
DM
7use Data::Dumper;
8use POSIX qw(:sys_wait_h);
5f095798
DM
9
10use PVE::SafeSyslog;
11use PVE::Tools;
12use PVE::HA::Tools;
13
14# Server can have several states:
15
16my $valid_states = {
ec911edd 17 wait_for_agent_lock => "waiting for agent lock",
0bba8f60 18 active => "got agent_lock",
5f095798
DM
19 lost_agent_lock => "lost agent_lock",
20};
21
22sub new {
23 my ($this, $haenv) = @_;
24
25 my $class = ref($this) || $this;
26
27 my $self = bless {
28 haenv => $haenv,
29 status => { state => 'startup' },
c4a221bc
DM
30 workers => {},
31 results => {},
067cdf33 32 shutdown_request => 0,
9c7d068b
DM
33 # mode can be: active, reboot, shutdown, restart
34 mode => 'active',
5f095798
DM
35 }, $class;
36
b0bf08a9 37 $self->set_local_status({ state => 'wait_for_agent_lock' });
9c7d068b
DM
38
39 $self->update_lrm_status();
5f095798
DM
40
41 return $self;
42}
43
44sub shutdown_request {
45 my ($self) = @_;
46
47 $self->{shutdown_request} = 1;
9c7d068b
DM
48
49 $self->{mode} = 'restart'; # fixme: detect shutdown/reboot
50
51 eval { $self->update_lrm_status(); };
52 if (my $err = $@) {
53 $self->log('err', "unable to update lrm status file");
54 }
5f095798
DM
55}
56
57sub get_local_status {
58 my ($self) = @_;
59
60 return $self->{status};
61}
62
63sub set_local_status {
64 my ($self, $new) = @_;
65
66 die "invalid state '$new->{state}'" if !$valid_states->{$new->{state}};
67
68 my $haenv = $self->{haenv};
69
70 my $old = $self->{status};
71
72 # important: only update if if really changed
73 return if $old->{state} eq $new->{state};
74
0bba8f60 75 $haenv->log('info', "status change $old->{state} => $new->{state}");
5f095798
DM
76
77 $new->{state_change_time} = $haenv->get_time();
78
79 $self->{status} = $new;
80}
81
9c7d068b
DM
82sub update_lrm_status {
83 my ($self) = @_;
84
85 my $lrm_status = {
86 mode => $self->{mode},
87 results => $self->{results},
88 };
89
90 $self->{haenv}->write_lrm_status($lrm_status);
91}
92
5f095798
DM
93sub get_protected_ha_agent_lock {
94 my ($self) = @_;
95
96 my $haenv = $self->{haenv};
97
98 my $count = 0;
99 my $starttime = $haenv->get_time();
100
101 for (;;) {
102
103 if ($haenv->get_ha_agent_lock()) {
104 if ($self->{ha_agent_wd}) {
105 $haenv->watchdog_update($self->{ha_agent_wd});
106 } else {
107 my $wfh = $haenv->watchdog_open();
108 $self->{ha_agent_wd} = $wfh;
109 }
110 return 1;
111 }
112
113 last if ++$count > 5; # try max 5 time
114
115 my $delay = $haenv->get_time() - $starttime;
116 last if $delay > 5; # for max 5 seconds
117
118 $haenv->sleep(1);
119 }
120
121 return 0;
122}
123
546e2f1f
DM
124sub active_service_count {
125 my ($self) = @_;
126
127 my $haenv = $self->{haenv};
128
129 my $nodename = $haenv->nodename();
130
131 my $ss = $self->{service_status};
132
133 my $count = 0;
134
135 foreach my $sid (keys %$ss) {
136 my $sd = $ss->{$sid};
137 next if !$sd->{node};
138 next if $sd->{node} ne $nodename;
139 my $req_state = $sd->{state};
140 next if !defined($req_state);
141 next if $req_state eq 'stopped';
9c7d068b 142 next if $req_state eq 'freeze';
546e2f1f
DM
143
144 $count++;
145 }
146
147 return $count;
148}
067cdf33 149
5f095798
DM
150sub do_one_iteration {
151 my ($self) = @_;
152
153 my $haenv = $self->{haenv};
154
155 my $status = $self->get_local_status();
156 my $state = $status->{state};
157
067cdf33
DM
158 my $ms = $haenv->read_manager_status();
159 $self->{service_status} = $ms->{service_status} || {};
160
49777d09 161 my $fence_request = PVE::HA::Tools::count_fenced_services($self->{service_status}, $haenv->nodename());
067cdf33 162
5f095798
DM
163 # do state changes first
164
165 my $ctime = $haenv->get_time();
166
b0bf08a9 167 if ($state eq 'wait_for_agent_lock') {
5f095798 168
546e2f1f 169 my $service_count = $self->active_service_count();
5f095798 170
067cdf33 171 if (!$fence_request && $service_count && $haenv->quorate()) {
0bba8f60
DM
172 if ($self->get_protected_ha_agent_lock()) {
173 $self->set_local_status({ state => 'active' });
5f095798
DM
174 }
175 }
176
177 } elsif ($state eq 'lost_agent_lock') {
178
067cdf33 179 if (!$fence_request && $haenv->quorate()) {
0bba8f60
DM
180 if ($self->get_protected_ha_agent_lock()) {
181 $self->set_local_status({ state => 'active' });
5f095798
DM
182 }
183 }
184
0bba8f60 185 } elsif ($state eq 'active') {
5f095798 186
067cdf33
DM
187 if ($fence_request) {
188 $haenv->log('err', "node need to be fenced - releasing agent_lock\n");
189 $self->set_local_status({ state => 'lost_agent_lock'});
190 } elsif (!$self->get_protected_ha_agent_lock()) {
5f095798
DM
191 $self->set_local_status({ state => 'lost_agent_lock'});
192 }
193 }
194
195 $status = $self->get_local_status();
196 $state = $status->{state};
197
198 # do work
199
200 if ($state eq 'wait_for_agent_lock') {
201
202 return 0 if $self->{shutdown_request};
203
204 $haenv->sleep(5);
205
0bba8f60 206 } elsif ($state eq 'active') {
5f095798
DM
207
208 my $startime = $haenv->get_time();
209
210 my $max_time = 10;
211
212 my $shutdown = 0;
213
214 # do work (max_time seconds)
215 eval {
216 # fixme: set alert timer
217
218 if ($self->{shutdown_request}) {
219
220 # fixme: request service stop or relocate ?
221
546e2f1f 222 my $service_count = $self->active_service_count();
5f095798
DM
223
224 if ($service_count == 0) {
225
226 if ($self->{ha_agent_wd}) {
227 $haenv->watchdog_close($self->{ha_agent_wd});
228 delete $self->{ha_agent_wd};
229 }
230
231 $shutdown = 1;
232 }
c4a221bc 233 } else {
c4a221bc
DM
234
235 $self->manage_resources();
067cdf33 236
5f095798
DM
237 }
238 };
239 if (my $err = $@) {
240 $haenv->log('err', "got unexpected error - $err");
241 }
242
243 return 0 if $shutdown;
244
245 $haenv->sleep_until($startime + $max_time);
246
247 } elsif ($state eq 'lost_agent_lock') {
248
249 # Note: watchdog is active an will triger soon!
250
251 # so we hope to get the lock back soon!
252
253 if ($self->{shutdown_request}) {
254
546e2f1f 255 my $service_count = $self->active_service_count();
5f095798 256
546e2f1f 257 if ($service_count > 0) {
5f095798 258 $haenv->log('err', "get shutdown request in state 'lost_agent_lock' - " .
546e2f1f 259 "detected $service_count running services");
5f095798 260
546e2f1f 261 } else {
5f095798 262
546e2f1f 263 # all services are stopped, so we can close the watchdog
5f095798 264
546e2f1f
DM
265 if ($self->{ha_agent_wd}) {
266 $haenv->watchdog_close($self->{ha_agent_wd});
267 delete $self->{ha_agent_wd};
268 }
269
270 return 0;
5f095798 271 }
5f095798
DM
272 }
273
b0bf08a9
DM
274 $haenv->sleep(5);
275
5f095798
DM
276 } else {
277
278 die "got unexpected status '$state'\n";
279
280 }
281
282 return 1;
283}
284
c4a221bc
DM
285sub manage_resources {
286 my ($self) = @_;
287
288 my $haenv = $self->{haenv};
289
290 my $nodename = $haenv->nodename();
291
c4a221bc
DM
292 my $ss = $self->{service_status};
293
294 foreach my $sid (keys %$ss) {
295 my $sd = $ss->{$sid};
296 next if !$sd->{node};
297 next if !$sd->{uid};
298 next if $sd->{node} ne $nodename;
299 my $req_state = $sd->{state};
300 next if !defined($req_state);
9c7d068b 301 next if $req_state eq 'freeze';
c4a221bc 302 eval {
e88469ba 303 $self->queue_resource_command($sid, $sd->{uid}, $req_state, $sd->{target});
c4a221bc
DM
304 };
305 if (my $err = $@) {
f31b7e94 306 $haenv->log('err', "unable to run resource agent for '$sid' - $err"); # fixme
c4a221bc
DM
307 }
308 }
309
f31b7e94 310 my $starttime = $haenv->get_time();
c4a221bc
DM
311
312 # start workers
313 my $max_workers = 4;
314
6dbf93a0 315 my $sc = $haenv->read_service_config();
f31b7e94
DM
316
317 while (($haenv->get_time() - $starttime) < 5) {
c4a221bc
DM
318 my $count = $self->check_active_workers();
319
320 foreach my $sid (keys %{$self->{workers}}) {
321 last if $count >= $max_workers;
322 my $w = $self->{workers}->{$sid};
6dbf93a0
DM
323 my $cd = $sc->{$sid};
324 if (!$cd) {
f31b7e94 325 $haenv->log('err', "missing resource configuration for '$sid'");
6dbf93a0
DM
326 next;
327 }
c4a221bc 328 if (!$w->{pid}) {
f31b7e94
DM
329 if ($haenv->can_fork()) {
330 my $pid = fork();
331 if (!defined($pid)) {
332 $haenv->log('err', "fork worker failed");
333 $count = 0; last; # abort, try later
334 } elsif ($pid == 0) {
335 # do work
336 my $res = -1;
337 eval {
338 $res = $haenv->exec_resource_agent($sid, $cd, $w->{state}, $w->{target});
339 };
340 if (my $err = $@) {
341 $haenv->log('err', $err);
342 POSIX::_exit(-1);
343 }
344 POSIX::_exit($res);
345 } else {
346 $count++;
347 $w->{pid} = $pid;
348 }
349 } else {
c4a221bc
DM
350 my $res = -1;
351 eval {
6dbf93a0 352 $res = $haenv->exec_resource_agent($sid, $cd, $w->{state}, $w->{target});
c4a221bc
DM
353 };
354 if (my $err = $@) {
f31b7e94
DM
355 $haenv->log('err', $err);
356 }
357 $self->resource_command_finished($sid, $w->{uid}, $res);
c4a221bc
DM
358 }
359 }
360 }
361
362 last if !$count;
363
f31b7e94 364 $haenv->sleep(1);
c4a221bc
DM
365 }
366}
367
368# fixme: use a queue an limit number of parallel workers?
369sub queue_resource_command {
e88469ba 370 my ($self, $sid, $uid, $state, $target) = @_;
c4a221bc
DM
371
372 if (my $w = $self->{workers}->{$sid}) {
373 return if $w->{pid}; # already started
374 # else, delete and overwrite queue entry with new command
375 delete $self->{workers}->{$sid};
376 }
377
378 $self->{workers}->{$sid} = {
379 sid => $sid,
380 uid => $uid,
381 state => $state,
382 };
e88469ba
DM
383
384 $self->{workers}->{$sid}->{target} = $target if $target;
c4a221bc
DM
385}
386
387sub check_active_workers {
388 my ($self) = @_;
389
390 # finish/count workers
391 my $count = 0;
392 foreach my $sid (keys %{$self->{workers}}) {
393 my $w = $self->{workers}->{$sid};
394 if (my $pid = $w->{pid}) {
395 # check status
396 my $waitpid = waitpid($pid, WNOHANG);
397 if (defined($waitpid) && ($waitpid == $pid)) {
398 $self->resource_command_finished($sid, $w->{uid}, $?);
399 } else {
400 $count++;
401 }
402 }
403 }
404
405 return $count;
406}
407
408sub resource_command_finished {
409 my ($self, $sid, $uid, $status) = @_;
410
411 my $haenv = $self->{haenv};
412
413 my $w = delete $self->{workers}->{$sid};
414 return if !$w; # should not happen
415
416 my $exit_code = -1;
417
418 if ($status == -1) {
0f70400d 419 $haenv->log('err', "resource agent $sid finished - failed to execute");
c4a221bc 420 } elsif (my $sig = ($status & 127)) {
0f70400d 421 $haenv->log('err', "resource agent $sid finished - got signal $sig");
c4a221bc
DM
422 } else {
423 $exit_code = ($status >> 8);
c4a221bc
DM
424 }
425
426 $self->{results}->{$uid} = {
427 sid => $w->{sid},
428 state => $w->{state},
429 exit_code => $exit_code,
430 };
431
432 my $ss = $self->{service_status};
433
434 # compute hash of valid/existing uids
435 my $valid_uids = {};
436 foreach my $sid (keys %$ss) {
437 my $sd = $ss->{$sid};
438 next if !$sd->{uid};
439 $valid_uids->{$sd->{uid}} = 1;
440 }
441
442 my $results = {};
443 foreach my $id (keys %{$self->{results}}) {
444 next if !$valid_uids->{$id};
445 $results->{$id} = $self->{results}->{$id};
446 }
447 $self->{results} = $results;
448
9c7d068b 449 $self->update_lrm_status();
c4a221bc
DM
450}
451
5f095798 4521;