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 Danga::Socket)
   5 package PublicInbox::EvCleanup;
   6 use strict;
   7 use warnings;
   8 use base qw(Danga::Socket);
   9 use fields qw(rd);
  10
  11 our $ENABLED;
  12 my $singleton;
  13 my $asapq = [ [], undef ];
  14 my $nextq = [ [], undef ];
  15 my $laterq = [ [], undef ];
  16
  17 sub once_init () {
  18         my $self = fields::new('PublicInbox::EvCleanup');
  19         my ($r, $w);
  20         pipe($r, $w) or die "pipe: $!";
  21         $self->SUPER::new($w);
  22         $self->{rd} = $r; # never read, since we never write..
  23         $self;
  24 }
  25
  26 sub _run_all ($) {
  27         my ($q) = @_;
  28
  29         my $run = $q->[0];
  30         $q->[0] = [];
  31         $q->[1] = undef;
  32         $_->() foreach @$run;
  33 }
  34
  35 # ensure Danga::Socket::ToClose fires after timers fire
  36 sub _asap_close () { $asapq->[1] ||= _asap_timer() }
  37
  38 sub _run_asap () { _run_all($asapq) }
  39 sub _run_next () {
  40         _run_all($nextq);
  41         _asap_close();
  42 }
  43
  44 sub _run_later () {
  45         _run_all($laterq);
  46         _asap_close();
  47 }
  48
  49 # Called by Danga::Socket
  50 sub event_write {
  51         my ($self) = @_;
  52         $self->watch_write(0);
  53         _run_asap();
  54 }
  55
  56 sub _asap_timer () {
  57         $singleton ||= once_init();
  58         $singleton->watch_write(1);
  59         1;
  60 }
  61
  62 sub asap ($) {
  63         my ($cb) = @_;
  64         push @{$asapq->[0]}, $cb;
  65         $asapq->[1] ||= _asap_timer();
  66 }
  67
  68 sub next_tick ($) {
  69         my ($cb) = @_;
  70         push @{$nextq->[0]}, $cb;
  71         $nextq->[1] ||= Danga::Socket->AddTimer(0, *_run_next);
  72 }
  73
  74 sub later ($) {
  75         my ($cb) = @_;
  76         push @{$laterq->[0]}, $cb;
  77         $laterq->[1] ||= Danga::Socket->AddTimer(60, *_run_later);
  78 }
  79
  80 END {
  81         _run_asap();
  82         _run_next();
  83         _run_later();
  84 }
  85
  86 1;