Commit | Line | Data |
a4952679 |
1 | package MooseX::Daemonize; |
a392fa53 |
2 | use strict; # because Kwalitee is pedantic |
a4952679 |
3 | use Moose::Role; |
8ac4733f |
4 | use MooseX::Types::Path::Class; |
a392fa53 |
5 | |
d9e417f4 |
6 | our $VERSION = 0.05; |
a4952679 |
7 | |
4327fe98 |
8 | with 'MooseX::Daemonize::WithPidFile', |
9 | 'MooseX::Getopt'; |
92cf56b7 |
10 | |
11 | use constant OK => 0; |
12 | use constant ERROR => 1; |
a4952679 |
13 | |
14 | has progname => ( |
5b9ebe08 |
15 | metaclass => 'Getopt', |
4327fe98 |
16 | isa => 'Str', |
17 | is => 'ro', |
18 | lazy => 1, |
19 | required => 1, |
20 | default => sub { |
a392fa53 |
21 | ( my $name = lc $_[0]->meta->name ) =~ s/::/_/g; |
22 | return $name; |
23 | }, |
a4952679 |
24 | ); |
25 | |
d9e417f4 |
26 | has pidbase => ( |
5b9ebe08 |
27 | metaclass => 'Getopt', |
4327fe98 |
28 | isa => 'Path::Class::Dir', |
29 | is => 'ro', |
30 | coerce => 1, |
5b9ebe08 |
31 | required => 1, |
4327fe98 |
32 | lazy => 1, |
33 | default => sub { Path::Class::Dir->new('var', 'run') }, |
24a6a660 |
34 | ); |
35 | |
d9e417f4 |
36 | has basedir => ( |
5b9ebe08 |
37 | metaclass => 'Getopt', |
4327fe98 |
38 | isa => 'Path::Class::Dir', |
39 | is => 'ro', |
40 | coerce => 1, |
41 | required => 1, |
42 | lazy => 1, |
43 | default => sub { Path::Class::Dir->new('/') }, |
a4952679 |
44 | ); |
45 | |
46 | has foreground => ( |
2eced271 |
47 | metaclass => 'Getopt', |
cbff8e52 |
48 | cmd_aliases => 'f', |
a4952679 |
49 | isa => 'Bool', |
50 | is => 'ro', |
51 | default => sub { 0 }, |
52 | ); |
53 | |
b916501e |
54 | has stop_timeout => ( |
5b9ebe08 |
55 | metaclass => 'Getopt', |
4327fe98 |
56 | isa => 'Int', |
57 | is => 'rw', |
58 | default => sub { 2 } |
b916501e |
59 | ); |
60 | |
5b9ebe08 |
61 | # internal book-keeping |
62 | |
63 | has status_message => ( |
64 | metaclass => 'NoGetopt', |
65 | isa => 'Str', |
66 | is => 'rw', |
67 | clearer => 'clear_status_message', |
68 | ); |
69 | |
70 | has exit_code => ( |
71 | metaclass => 'NoGetopt', |
72 | isa => 'Int', |
73 | is => 'rw', |
74 | clearer => 'clear_exit_code', |
75 | ); |
76 | |
4327fe98 |
77 | # methods ... |
78 | |
79 | ## PID file related stuff ... |
80 | |
d9e417f4 |
81 | sub init_pidfile { |
82 | my $self = shift; |
83 | my $file = $self->pidbase . '/' . $self->progname . '.pid'; |
84 | confess "Cannot write to $file" unless (-e $file ? -w $file : -w $self->pidbase); |
d8985b7d |
85 | MooseX::Daemonize::Pid::File->new( file => $file ); |
d9e417f4 |
86 | } |
87 | |
5b9ebe08 |
88 | # backwards compat, |
4327fe98 |
89 | sub check { (shift)->pidfile->is_running } |
90 | sub save_pid { (shift)->pidfile->write } |
91 | sub remove_pid { (shift)->pidfile->remove } |
92 | sub get_pid { (shift)->pidfile->pid } |
93 | |
94 | ## signal handling ... |
95 | |
96 | sub setup_signals { |
97 | my $self = shift; |
4a24225a |
98 | $SIG{'INT'} = sub { $self->shutdown }; |
99 | # I can't think of a sane default here really ... |
100 | # $SIG{'HUP'} = sub { $self->handle_sighup }; |
4327fe98 |
101 | } |
102 | |
4a24225a |
103 | sub shutdown { |
104 | my $self = shift; |
105 | $self->pidfile->remove if $self->pidfile->pid == $$; |
106 | exit(0); |
107 | } |
4327fe98 |
108 | |
109 | ## daemon control methods ... |
110 | |
a4952679 |
111 | sub start { |
5b9ebe08 |
112 | my $self = shift; |
113 | |
114 | $self->clear_status_message; |
115 | $self->clear_exit_code; |
116 | |
117 | if ($self->pidfile->is_running) { |
92cf56b7 |
118 | $self->exit_code(OK); |
119 | $self->status_message('Daemon is already running with pid (' . $self->pidfile->pid . ')'); |
5b9ebe08 |
120 | return !($self->exit_code); |
121 | } |
8ac4733f |
122 | |
5b9ebe08 |
123 | if ($self->foreground) { |
124 | $self->is_daemon(1); |
125 | } |
126 | else { |
127 | eval { $self->daemonize }; |
128 | if ($@) { |
92cf56b7 |
129 | $self->exit_code(ERROR); |
5b9ebe08 |
130 | $self->status_message('Start failed : ' . $@); |
131 | return !($self->exit_code); |
132 | } |
133 | } |
cbff8e52 |
134 | |
5b9ebe08 |
135 | unless ($self->is_daemon) { |
92cf56b7 |
136 | $self->exit_code(OK); |
5b9ebe08 |
137 | $self->status_message('Start succeeded'); |
138 | return !($self->exit_code); |
139 | } |
140 | |
141 | $self->pidfile->pid($$); |
cbff8e52 |
142 | |
24a6a660 |
143 | # Change to basedir |
144 | chdir $self->basedir; |
fa2b72a4 |
145 | |
ff5cee29 |
146 | $self->pidfile->write; |
3543c999 |
147 | $self->setup_signals; |
148 | return $$; |
149 | } |
150 | |
5b9ebe08 |
151 | sub status { |
152 | my $self = shift; |
153 | |
154 | $self->clear_status_message; |
155 | $self->clear_exit_code; |
156 | |
157 | if ($self->pidfile->is_running) { |
92cf56b7 |
158 | $self->exit_code(OK); |
5b9ebe08 |
159 | $self->status_message('Daemon is running with pid (' . $self->pidfile->pid . ')'); |
160 | } |
161 | else { |
92cf56b7 |
162 | $self->exit_code(ERROR); |
5b9ebe08 |
163 | $self->status_message('Daemon is not running with pid (' . $self->pidfile->pid . ')'); |
164 | } |
165 | |
166 | return !($self->exit_code); |
167 | } |
168 | |
4327fe98 |
169 | sub restart { |
5b9ebe08 |
170 | my $self = shift; |
171 | |
172 | $self->clear_status_message; |
173 | $self->clear_exit_code; |
174 | |
175 | unless ($self->stop) { |
92cf56b7 |
176 | $self->exit_code(ERROR); |
5b9ebe08 |
177 | $self->status_message('Restart (Stop) failed : ' . $@); |
178 | } |
179 | |
180 | unless ($self->start) { |
92cf56b7 |
181 | $self->exit_code(ERROR); |
5b9ebe08 |
182 | $self->status_message('Restart (Start) failed : ' . $@); |
183 | } |
184 | |
92cf56b7 |
185 | if ($self->exit_code == OK) { |
186 | $self->exit_code(OK); |
187 | $self->status_message("Restart successful"); |
188 | } |
5b9ebe08 |
189 | |
190 | return !($self->exit_code); |
4327fe98 |
191 | } |
192 | |
b916501e |
193 | # Make _kill *really* private |
194 | my $_kill; |
195 | |
3543c999 |
196 | sub stop { |
5b9ebe08 |
197 | my $self = shift; |
198 | |
199 | $self->clear_status_message; |
200 | $self->clear_exit_code; |
201 | |
202 | # if the pid is not running |
203 | # then we dont need to stop |
204 | # anything ... |
205 | if ($self->pidfile->is_running) { |
206 | |
207 | # if we are foreground, then |
208 | # no need to try and kill |
209 | # ourselves |
210 | unless ($self->foreground) { |
211 | |
212 | # kill the process ... |
213 | eval { $self->$_kill($self->pidfile->pid) }; |
214 | # and complain if we can't ... |
215 | if ($@) { |
92cf56b7 |
216 | $self->exit_code(ERROR); |
5b9ebe08 |
217 | $self->status_message('Stop failed : ' . $@); |
218 | } |
219 | # or gloat if we succeed .. |
220 | else { |
92cf56b7 |
221 | $self->exit_code(OK); |
5b9ebe08 |
222 | $self->status_message('Stop succeeded'); |
223 | } |
224 | |
225 | } |
5b9ebe08 |
226 | } |
227 | else { |
228 | # this just returns the OK |
229 | # exit code for now, but |
230 | # we should make this overridable |
92cf56b7 |
231 | $self->exit_code(OK); |
5b9ebe08 |
232 | $self->status_message("Not running"); |
233 | } |
234 | |
235 | # if we are returning to our script |
236 | # then we actually need the opposite |
237 | # of what the system/OS expects |
238 | return !($self->exit_code); |
3543c999 |
239 | } |
240 | |
b916501e |
241 | $_kill = sub { |
a4952679 |
242 | my ( $self, $pid ) = @_; |
2361a590 |
243 | return unless $pid; |
3543c999 |
244 | unless ( CORE::kill 0 => $pid ) { |
3543c999 |
245 | # warn "$pid already appears dead."; |
246 | return; |
247 | } |
248 | |
249 | if ( $pid eq $$ ) { |
4a24225a |
250 | die "$pid is us! Can't commit suicide."; |
a4952679 |
251 | } |
252 | |
b916501e |
253 | my $timeout = $self->stop_timeout; |
a4952679 |
254 | |
b916501e |
255 | # kill 0 => $pid returns 0 if the process is dead |
256 | # $!{EPERM} could also be true if we cant kill it (permission error) |
a4952679 |
257 | |
b916501e |
258 | # Try SIGINT ... 2s ... SIGTERM ... 2s ... SIGKILL ... 3s ... UNDEAD! |
b6b69aca |
259 | my $terminating_signal; |
b916501e |
260 | for ( [ 2, $timeout ], [15, $timeout], [9, $timeout * 1.5] ) { |
ea9485d8 |
261 | my ($signal, $timeout) = @$_; |
262 | $timeout = int $timeout; |
5b9ebe08 |
263 | |
ea9485d8 |
264 | CORE::kill($signal, $pid); |
5b9ebe08 |
265 | |
ea9485d8 |
266 | while ($timeout) { |
b6b69aca |
267 | unless(CORE::kill 0 => $pid or $!{EPERM}) { |
268 | $terminating_signal = $signal; |
269 | last; |
270 | } |
ea9485d8 |
271 | $timeout--; |
b6b69aca |
272 | sleep(1) if $timeout; |
ea9485d8 |
273 | } |
b6b69aca |
274 | |
275 | last if $terminating_signal; |
a4952679 |
276 | } |
277 | |
b6b69aca |
278 | if($terminating_signal) { |
279 | if($terminating_signal == 9) { |
280 | # clean up the pidfile ourselves iff we used -9 and it worked |
281 | warn "Had to resort to 'kill -9' and it worked, wiping pidfile"; |
282 | eval { $self->pidfile->remove }; |
283 | if ($@) { |
284 | warn "Could not remove pidfile (" |
285 | . $self->pidfile->file |
286 | . ") because : $!"; |
287 | } |
288 | } |
289 | return; |
290 | } |
b916501e |
291 | |
292 | # IF it is still running |
d9e417f4 |
293 | Carp::carp "$pid doesn't seem to want to die."; # AHH EVIL DEAD! |
b916501e |
294 | }; |
a4952679 |
295 | |
296 | 1; |
297 | __END__ |
298 | |
8ac4733f |
299 | =pod |
300 | |
a4952679 |
301 | =head1 NAME |
302 | |
892a51e6 |
303 | MooseX::Daemonize - Role for daemonizing your Moose based application |
a4952679 |
304 | |
305 | =head1 VERSION |
306 | |
4327fe98 |
307 | This document describes MooseX::Daemonize version 0.05 |
a4952679 |
308 | |
309 | =head1 SYNOPSIS |
310 | |
4327fe98 |
311 | package My::Daemon; |
a4952679 |
312 | use Moose; |
5b9ebe08 |
313 | |
a4952679 |
314 | with qw(MooseX::Daemonize); |
5b9ebe08 |
315 | |
4327fe98 |
316 | # ... define your class .... |
5b9ebe08 |
317 | |
318 | after start => sub { |
4327fe98 |
319 | my $self = shift; |
320 | return unless $self->is_daemon; |
321 | # your daemon code here ... |
322 | }; |
a4952679 |
323 | |
5b9ebe08 |
324 | # then in your script ... |
325 | |
4327fe98 |
326 | my $daemon = My::Daemon->new_with_options(); |
5b9ebe08 |
327 | |
4327fe98 |
328 | my ($command) = @{$daemon->extra_argv} |
329 | defined $command || die "No command specified"; |
5b9ebe08 |
330 | |
331 | $daemon->start if $command eq 'start'; |
332 | $daemon->status if $command eq 'status'; |
333 | $daemon->restart if $command eq 'restart'; |
334 | $daemon->stop if $command eq 'stop'; |
335 | |
92cf56b7 |
336 | warn($daemon->status_message); |
5b9ebe08 |
337 | exit($daemon->exit_code); |
338 | |
a4952679 |
339 | =head1 DESCRIPTION |
340 | |
b916501e |
341 | Often you want to write a persistant daemon that has a pid file, and responds |
5b9ebe08 |
342 | appropriately to Signals. This module provides a set of basic roles as an |
4327fe98 |
343 | infrastructure to do that. |
a4952679 |
344 | |
345 | =head1 ATTRIBUTES |
346 | |
4327fe98 |
347 | This list includes attributes brought in from other roles as well |
348 | we include them here for ease of documentation. All of these attributes |
5b9ebe08 |
349 | are settable though L<MooseX::Getopt>'s command line handling, with the |
4327fe98 |
350 | exception of C<is_daemon>. |
351 | |
a4952679 |
352 | =over |
353 | |
4327fe98 |
354 | =item I<progname Path::Class::Dir | Str> |
a4952679 |
355 | |
4327fe98 |
356 | The name of our daemon, defaults to C<$package_name =~ s/::/_/>; |
a4952679 |
357 | |
4327fe98 |
358 | =item I<pidbase Path::Class::Dir | Str> |
a4952679 |
359 | |
4327fe98 |
360 | The base for our bid, defaults to C</var/run/$progname> |
a4952679 |
361 | |
4327fe98 |
362 | =item I<pidfile MooseX::Daemonize::Pid::File | Str> |
a4952679 |
363 | |
4327fe98 |
364 | The file we store our PID in, defaults to C</var/run/$progname> |
a4952679 |
365 | |
4327fe98 |
366 | =item I<foreground Bool> |
a4952679 |
367 | |
5b9ebe08 |
368 | If true, the process won't background. Useful for debugging. This option can |
b916501e |
369 | be set via Getopt's -f. |
a4952679 |
370 | |
4327fe98 |
371 | =item I<is_daemon Bool> |
372 | |
5b9ebe08 |
373 | If true, the process is the backgrounded daemon process, if false it is the |
374 | parent process. This is useful for example in an C<after 'start' => sub { }> |
375 | block. |
e7a196e7 |
376 | |
4327fe98 |
377 | B<NOTE:> This option is explicitly B<not> available through L<MooseX::Getopt>. |
b916501e |
378 | |
4327fe98 |
379 | =item I<stop_timeout> |
b916501e |
380 | |
381 | Number of seconds to wait for the process to stop, before trying harder to kill |
4327fe98 |
382 | it. Defaults to 2 seconds. |
e7a196e7 |
383 | |
a4952679 |
384 | =back |
385 | |
5b9ebe08 |
386 | These are the internal attributes, which are not available through MooseX::Getopt. |
387 | |
388 | =over 4 |
389 | |
390 | =item I<exit_code Int> |
391 | |
92cf56b7 |
392 | =item I<status_message Str> |
5b9ebe08 |
393 | |
394 | =back |
395 | |
396 | =head1 METHODS |
a4952679 |
397 | |
4327fe98 |
398 | =head2 Daemon Control Methods |
399 | |
5b9ebe08 |
400 | These methods can be used to control the daemon behavior. Every effort |
401 | has been made to have these methods DWIM (Do What I Mean), so that you |
402 | can focus on just writing the code for your daemon. |
a4952679 |
403 | |
5b9ebe08 |
404 | Extending these methods is best done with the L<Moose> method modifiers, |
4327fe98 |
405 | such as C<before>, C<after> and C<around>. |
406 | |
407 | =over 4 |
408 | |
409 | =item B<start> |
a4952679 |
410 | |
411 | Setup a pidfile, fork, then setup the signal handlers. |
412 | |
4327fe98 |
413 | =item B<stop> |
a4952679 |
414 | |
415 | Stop the process matching the pidfile, and unlinks the pidfile. |
416 | |
4327fe98 |
417 | =item B<restart> |
a4952679 |
418 | |
4327fe98 |
419 | Literally this is: |
a4952679 |
420 | |
421 | $self->stop(); |
422 | $self->start(); |
423 | |
5b9ebe08 |
424 | =item B<status> |
425 | |
92cf56b7 |
426 | =item B<shutdown> |
427 | |
4327fe98 |
428 | =back |
429 | |
5b9ebe08 |
430 | |
4327fe98 |
431 | =head2 Pidfile Handling Methods |
432 | |
433 | =over 4 |
434 | |
435 | =item B<init_pidfile> |
436 | |
437 | This method will create a L<MooseX::Daemonize::Pid::File> object and tell |
438 | it to store the PID in the file C<$pidbase/$progname.pid>. |
439 | |
440 | =item B<check> |
441 | |
5b9ebe08 |
442 | This checks to see if the daemon process is currently running by checking |
4327fe98 |
443 | the pidfile. |
a4952679 |
444 | |
4327fe98 |
445 | =item B<get_pid> |
a4952679 |
446 | |
4327fe98 |
447 | Returns the PID of the daemon process. |
a4952679 |
448 | |
4327fe98 |
449 | =item B<save_pid> |
a4952679 |
450 | |
4327fe98 |
451 | Write the pidfile. |
452 | |
453 | =item B<remove_pid> |
454 | |
455 | Removes the pidfile. |
456 | |
457 | =back |
458 | |
459 | =head2 Signal Handling Methods |
460 | |
461 | =over 4 |
462 | |
463 | =item B<setup_signals> |
464 | |
5b9ebe08 |
465 | Setup the signal handlers, by default it only sets up handlers for SIGINT and |
4327fe98 |
466 | SIGHUP. If you wish to add more signals just use the C<after> method modifier |
467 | and add them. |
468 | |
469 | =item B<handle_sigint> |
a4952679 |
470 | |
cecbee2d |
471 | Handle a INT signal, by default calls C<$self->stop()> |
a4952679 |
472 | |
4327fe98 |
473 | =item B<handle_sighup> |
a4952679 |
474 | |
cecbee2d |
475 | Handle a HUP signal. By default calls C<$self->restart()> |
a4952679 |
476 | |
4327fe98 |
477 | =back |
478 | |
479 | =head2 Introspection |
480 | |
481 | =over 4 |
482 | |
a4952679 |
483 | =item meta() |
484 | |
cecbee2d |
485 | The C<meta()> method from L<Class::MOP::Class> |
a4952679 |
486 | |
487 | =back |
488 | |
489 | =head1 DEPENDENCIES |
490 | |
4327fe98 |
491 | L<Moose>, L<MooseX::Getopt>, L<MooseX::Types::Path::Class> and L<POSIX> |
a4952679 |
492 | |
493 | =head1 INCOMPATIBILITIES |
494 | |
4327fe98 |
495 | None reported. Although obviously this will not work on Windows. |
a4952679 |
496 | |
497 | =head1 BUGS AND LIMITATIONS |
498 | |
a4952679 |
499 | No bugs have been reported. |
500 | |
501 | Please report any bugs or feature requests to |
502 | C<bug-acme-dahut-call@rt.cpan.org>, or through the web interface at |
503 | L<http://rt.cpan.org>. |
504 | |
505 | =head1 SEE ALSO |
506 | |
4327fe98 |
507 | L<Proc::Daemon>, L<Daemon::Generic> |
a4952679 |
508 | |
92cf56b7 |
509 | =head1 AUTHORS |
a4952679 |
510 | |
511 | Chris Prather C<< <perigrin@cpan.org> >> |
512 | |
92cf56b7 |
513 | Stevan Little C<< <stevan.little@iinteractive.com> >> |
514 | |
7ada91b8 |
515 | =head1 THANKS |
516 | |
5b9ebe08 |
517 | Mike Boyko, Matt S. Trout, Stevan Little, Brandon Black, Ash Berlin and the |
b916501e |
518 | #moose denzians |
a4952679 |
519 | |
637573c4 |
520 | Some bug fixes sponsored by Takkle Inc. |
521 | |
a4952679 |
522 | =head1 LICENCE AND COPYRIGHT |
523 | |
5b9ebe08 |
524 | Copyright (c) 2007, Chris Prather C<< <perigrin@cpan.org> >>. All rights |
b916501e |
525 | reserved. |
a4952679 |
526 | |
527 | This module is free software; you can redistribute it and/or |
528 | modify it under the same terms as Perl itself. See L<perlartistic>. |
529 | |
a4952679 |
530 | =head1 DISCLAIMER OF WARRANTY |
531 | |
532 | BECAUSE THIS SOFTWARE IS LICENSED FREE OF CHARGE, THERE IS NO WARRANTY |
533 | FOR THE SOFTWARE, TO THE EXTENT PERMITTED BY APPLICABLE LAW. EXCEPT WHEN |
534 | OTHERWISE STATED IN WRITING THE COPYRIGHT HOLDERS AND/OR OTHER PARTIES |
535 | PROVIDE THE SOFTWARE "AS IS" WITHOUT WARRANTY OF ANY KIND, EITHER |
536 | EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
537 | WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE. THE |
538 | ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE SOFTWARE IS WITH |
539 | YOU. SHOULD THE SOFTWARE PROVE DEFECTIVE, YOU ASSUME THE COST OF ALL |
540 | NECESSARY SERVICING, REPAIR, OR CORRECTION. |
541 | |
542 | IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING |
543 | WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MAY MODIFY AND/OR |
544 | REDISTRIBUTE THE SOFTWARE AS PERMITTED BY THE ABOVE LICENCE, BE |
545 | LIABLE TO YOU FOR DAMAGES, INCLUDING ANY GENERAL, SPECIAL, INCIDENTAL, |
546 | OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE USE OR INABILITY TO USE |
547 | THE SOFTWARE (INCLUDING BUT NOT LIMITED TO LOSS OF DATA OR DATA BEING |
548 | RENDERED INACCURATE OR LOSSES SUSTAINED BY YOU OR THIRD PARTIES OR A |
549 | FAILURE OF THE SOFTWARE TO OPERATE WITH ANY OTHER SOFTWARE), EVEN IF |
550 | SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF |
551 | SUCH DAMAGES. |
8ac4733f |
552 | |
553 | =cut |