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