lib/PublicInbox/EvCleanup.pm

   1 # Copyright (C) 2016-2018 all contributors <meta@public-inbox.org>
   2 # License: AGPL-3.0+ <https://www.gnu.org/licenses/agpl-3.0.txt>
   3
   4 # event cleanups (currently for PublicInbox::DS)
   5 package PublicInbox::EvCleanup;
   6 use strict;
   7 use warnings;
   8 use base qw(PublicInbox::DS);
   9
  10 my $ENABLED;
  11 sub enabled { $ENABLED }
  12 sub enable { $ENABLED = 1 }
  13 my $singleton;
  14 my $asapq = [ [], undef ];
  15 my $nextq = [ [], undef ];
  16 my $laterq = [ [], undef ];
  17
  18 sub once_init () {
  19         my $self = fields::new('PublicInbox::EvCleanup');
  20         my ($r, $w);
  21
  22         # This is a dummy pipe which is always writable so it can always
  23         # fires in the next event loop iteration.
  24         pipe($r, $w) or die "pipe: $!";
  25         fcntl($w, 1031, 4096) if $^O eq 'linux'; # 1031: F_SETPIPE_SZ
  26         $self->SUPER::new($w, 0);
  27
  28         # always writable, since PublicInbox::EvCleanup::event_step
  29         # never drains wbuf.  We can avoid wasting a hash slot by
  30         # stuffing the read-end of the pipe into the never-to-be-touched
  31         # wbuf
  32         $self->{wbuf} = $r;
  33         $self;
  34 }
  35
  36 sub _run_all ($) {
  37         my ($q) = @_;
  38
  39         my $run = $q->[0];
  40         $q->[0] = [];
  41         $q->[1] = undef;
  42         $_->() foreach @$run;
  43 }
  44
  45 # ensure PublicInbox::DS::ToClose processing after timers fire
  46 sub _asap_close () { $asapq->[1] ||= _asap_timer() }
  47
  48 sub _run_asap () { _run_all($asapq) }
  49 sub _run_next () {
  50         _run_all($nextq);
  51         _asap_close();
  52 }
  53
  54 sub _run_later () {
  55         _run_all($laterq);
  56         _asap_close();
  57 }
  58
  59 # Called by PublicInbox::DS
  60 sub event_step {
  61         my ($self) = @_;
  62         $self->watch_write(0);
  63         _run_asap();
  64 }
  65
  66 sub _asap_timer () {
  67         $singleton ||= once_init();
  68         $singleton->watch_write(1);
  69         1;
  70 }
  71
  72 sub asap ($) {
  73         my ($cb) = @_;
  74         push @{$asapq->[0]}, $cb;
  75         $asapq->[1] ||= _asap_timer();
  76 }
  77
  78 sub next_tick ($) {
  79         my ($cb) = @_;
  80         push @{$nextq->[0]}, $cb;
  81         $nextq->[1] ||= PublicInbox::DS->AddTimer(0, *_run_next);
  82 }
  83
  84 sub later ($) {
  85         my ($cb) = @_;
  86         push @{$laterq->[0]}, $cb;
  87         $laterq->[1] ||= PublicInbox::DS->AddTimer(60, *_run_later);
  88 }
  89
  90 END {
  91         _run_asap();
  92         _run_all($nextq);
  93         _run_all($laterq);
  94 }
  95
  96 1;