| 1 | #! perl |
1 | #! perl |
| 2 | |
2 | |
| 3 | # restart server when certain resource limits are exceeded |
3 | # restart server when certain resource limits are exceeded |
| 4 | |
4 | |
|
|
5 | # TODO, CONF framework? |
| 5 | our $MAX_VMSIZE = $cf::CFG{checkrusage}{vmsize} || 1_000_000_000; |
6 | our $MAX_VMSIZE = $cf::CFG{checkrusage}{vmsize} || 1_000_000_000; |
| 6 | |
7 | |
| 7 | $^O eq "linux" or die "only linux supported right now.\n"; |
8 | $^O eq "linux" or die "only linux supported right now.\n"; |
| 8 | |
9 | |
| 9 | our $STAT_FH; |
10 | our $STAT_FH; |
| 10 | our $TIMER; |
11 | our $TIMER; |
| 11 | |
12 | |
| 12 | cf::post_init { |
13 | cf::post_init { |
|
|
14 | cf::async_ext { |
| 13 | $STAT_FH = Coro::AIO::aio_open "/proc/self/stat", IO::AIO::O_RDONLY, 0 |
15 | $STAT_FH = Coro::AIO::aio_open "/proc/self/stat", IO::AIO::O_RDONLY, 0 |
| 14 | or return; |
16 | or return; |
| 15 | |
17 | |
| 16 | $TIMER = cf::periodic 10, sub { |
18 | $TIMER = cf::periodic 10, sub { |
| 17 | sysseek $STAT_FH, 0, 0; |
19 | sysseek $STAT_FH, 0, 0; |
| 18 | sysread $STAT_FH, my $stat, 4096; # only a single line |
20 | sysread $STAT_FH, my $stat, 4096; # only a single line |
| 19 | my $vmsize = (split / /, $stat)[22]; |
21 | my $vmsize = (split / /, $stat)[22]; |
| 20 | |
22 | |
| 21 | cf::cleanup "memory usage growing high, server would run out of memory soon", 0 |
23 | cf::cleanup "memory usage growing high ($vmsize > $MAX_VMSIZE), server would run out of memory soon", 0 |
| 22 | if $vmsize >= $MAX_VMSIZE; |
24 | if $vmsize >= $MAX_VMSIZE; |
|
|
25 | }; |
| 23 | }; |
26 | }; |
| 24 | }; |
27 | }; |
