]> Sergey Matveev's repositories - public-inbox.git/commitdiff
httpd/async: support passing arg to callbacks
authorEric Wong <e@80x24.org>
Wed, 25 Dec 2019 07:50:36 +0000 (07:50 +0000)
committerEric Wong <e@80x24.org>
Thu, 26 Dec 2019 10:48:19 +0000 (10:48 +0000)
Another step towards removing anonymous subs to eliminate
a possible source of memory leaks and high memory use.

lib/PublicInbox/HTTPD/Async.pm
lib/PublicInbox/Qspawn.pm
lib/PublicInbox/SolverGit.pm

index d5628ee8d7ca8b457f9999ef118b18553dab61fd..d182c11886c5f4b1557d9e0828469fcdb7ae7b28 100644 (file)
@@ -10,7 +10,7 @@ package PublicInbox::HTTPD::Async;
 use strict;
 use warnings;
 use base qw(PublicInbox::DS);
-use fields qw(cb end);
+use fields qw(cb arg end end_arg);
 use Errno qw(EAGAIN);
 use PublicInbox::Syscall qw(EPOLLIN EPOLLET);
 
@@ -18,12 +18,12 @@ use PublicInbox::Syscall qw(EPOLLIN EPOLLET);
 # $io is a read-only pipe ($rpipe) for now, but may be a
 # bidirectional socket in the future.
 sub new {
-       my ($class, $io, $cb, $end) = @_;
+       my ($class, $io, $cb, $arg, $end, $end_arg) = @_;
 
        # no $io? call $cb at the top of the next event loop to
        # avoid recursion:
        unless (defined($io)) {
-               PublicInbox::DS::requeue($cb);
+               PublicInbox::DS::requeue($cb ? $cb : $arg);
                die '$end unsupported w/o $io' if $end;
                return;
        }
@@ -32,7 +32,9 @@ sub new {
        IO::Handle::blocking($io, 0);
        $self->SUPER::new($io, EPOLLIN | EPOLLET);
        $self->{cb} = $cb; # initial read callback, later replaced by main_cb
+       $self->{arg} = $arg; # arg for $cb
        $self->{end} = $end; # like END {}, but only for this object
+       $self->{end_arg} = $end_arg; # arg for $end
        $self;
 }
 
index 651fa39096d7144beb7d9bb6030fcb67c9873df0..c28566091304c6be0c5d780034f38e5da29ae7b9 100644 (file)
@@ -185,7 +185,7 @@ reread:
                ($rpipe) = @_; # popen_rd result
                if ($async) {
                # PublicInbox::HTTPD::Async->new($rpipe, $cb, $end)
-                       $async = $async->($rpipe, $cb, $end);
+                       $async = $async->($rpipe, $cb, undef, $end);
                        # $cb will call ->async_pass or ->close
                } else { # generic PSGI
                        $cb->() while $qx;
@@ -297,7 +297,7 @@ sub psgi_return {
                ($rpipe) = @_;
                if ($async) {
                        # PublicInbox::HTTPD::Async->new($rpipe, $cb, $end)
-                       $async = $async->($rpipe, $cb, $end);
+                       $async = $async->($rpipe, $cb, undef, $end);
                        # $cb will call ->async_pass or ->close
                } else { # generic PSGI
                        $cb->() while $rd_hdr;
index eea59b6ddbda047496fe4917517fe4b3c6c53bdb..b3fc5bef88c9ca24cb973a283b2ea3a5cde740c7 100644 (file)
@@ -327,7 +327,7 @@ sub do_finish ($$) {
        $user_cb->(undef);
 }
 
-sub do_step ($) {
+sub event_step ($) {
        my ($self) = @_;
        eval {
                # step 1: resolve blobs to patches in the todo queue
@@ -363,18 +363,13 @@ sub do_step ($) {
        }
 }
 
-sub step_cb ($) {
-       my ($self) = @_;
-       sub { do_step($self) };
-}
-
 sub next_step ($) {
        my ($self) = @_;
        # if outside of public-inbox-httpd, caller is expected to be
-       # looping step_cb, anyways
+       # looping event_step, anyways
        my $async = $self->{psgi_env}->{'pi-httpd.async'} or return;
        # PublicInbox::HTTPD::Async->new
-       $async->(undef, step_cb($self));
+       $async->(undef, undef, $self);
 }
 
 sub mark_found ($$$) {
@@ -598,12 +593,11 @@ sub solve ($$$$$) {
        $self->{tmp} = File::Temp->newdir("solver.$oid_want-XXXXXXXX", TMPDIR => 1);
 
        dbg($self, "solving $oid_want ...");
-       my $step_cb = step_cb($self);
        if (my $async = $env->{'pi-httpd.async'}) {
                # PublicInbox::HTTPD::Async->new
-               $async->(undef, $step_cb);
+               $async->(undef, undef, $self);
        } else {
-               $step_cb->() while $self->{user_cb};
+               event_step($self) while $self->{user_cb};
        }
 }