package PublicInbox::LeiAuth;
use strict;
use v5.10.1;
-use parent qw(PublicInbox::IPC);
use PublicInbox::PktOp qw(pkt_do);
-use PublicInbox::NetReader;
-sub nrd_merge {
- my ($lei, $nrd_new) = @_;
- if ($lei->{pkt_op_p}) { # from lei_convert worker
- pkt_do($lei->{pkt_op_p}, 'nrd_merge', $nrd_new);
- } else { # single lei-daemon consumer
- my $self = $lei->{auth} or return; # client disconnected
- my $nrd = $self->{nrd};
- %$nrd = (%$nrd, %$nrd_new);
- }
+sub do_auth_atfork { # used by IPC WQ workers
+ my ($self, $wq) = @_;
+ return if $wq->{-wq_worker_nr} != 0;
+ my $lei = $wq->{lei};
+ my $net = $lei->{net};
+ eval {
+ my $mics = $net->imap_common_init($lei);
+ my $nn = $net->nntp_common_init($lei);
+ pkt_do($lei->{pkt_op_p}, 'net_merge_continue', $net) or
+ die "pkt_do net_merge_continue: $!";
+ $net->{mics_cached} = $mics if $mics;
+ $net->{nn_cached} = $nn if $nn;
+ };
+ $lei->fail($@) if $@;
}
-sub do_auth { # called via wq_io_do
- my ($self) = @_;
- my ($lei, $nrd) = @$self{qw(lei nrd)};
- $nrd->imap_common_init($lei);
- nrd_merge($lei, $nrd); # tell lei-daemon updated auth info
+sub net_merge_done1 { # bump merge-count in top-level lei-daemon
+ my ($wq) = @_;
+ return if ++$wq->{nr_net_merge_done} != $wq->{-wq_nr_workers};
+ $wq->net_merge_all_done; # defined per wq-class (e.g. LeiImport)
}
-sub do_finish_auth { # dwaitpid callback
- my ($arg, $pid) = @_;
- my ($self, $lei, $post_auth_cb, @args) = @$arg;
- $? ? $lei->dclose : $post_auth_cb->(@args);
+sub net_merge_all { # called in wq worker via wq_broadcast
+ my ($wq, $net_new) = @_;
+ my $net = $wq->{lei}->{net};
+ %$net = (%$net, %$net_new);
+ pkt_do($wq->{lei}->{pkt_op_p}, 'net_merge_done1') or
+ die "pkt_op_do net_merge_done1: $!";
}
-sub auth_eof {
- my ($lei, $post_auth_cb, @args) = @_;
- my $self = delete $lei->{auth} or return;
- $self->wq_wait_old(\&do_finish_auth, $lei, $post_auth_cb, @args);
+# called by top-level lei-daemon when first worker is done with auth
+sub net_merge_continue {
+ my ($wq, $net_new) = @_;
+ $wq->wq_broadcast('net_merge_all', $net_new);
}
-sub auth_start {
- my ($self, $lei, $post_auth_cb, @args) = @_;
- $lei->_lei_cfg(1); # workers may need to read config
- my $op = $lei->workers_start($self, 'auth', 1, {
- 'nrd_merge' => [ \&nrd_merge, $lei ],
- '' => [ \&auth_eof, $lei, $post_auth_cb, @args ],
- });
- $self->wq_io_do('do_auth', []);
- $self->wq_close(1);
- while ($op && $op->{sock}) { $op->event_step }
+sub op_merge { # prepares PktOp->pair ops
+ my ($self, $ops, $wq) = @_;
+ $ops->{net_merge_continue} = [ \&net_merge_continue, $wq ];
+ $ops->{net_merge_done1} = [ \&net_merge_done1, $wq ];
}
-sub ipc_atfork_child {
- my ($self) = @_;
- # prevent {sock} from being closed in lei_atfork_child:
- my $s = delete $self->{lei}->{sock};
- delete $self->{lei}->{auth}; # drop circular ref
- $self->{lei}->lei_atfork_child;
- $self->{lei}->{sock} = $s if $s;
- $self->SUPER::ipc_atfork_child;
-}
-
-sub new {
- my ($cls, $nrd) = @_; # nrd may be NetReader or descendant (NetWriter)
- bless { nrd => $nrd }, $cls;
-}
+sub new { bless \(my $x), __PACKAGE__ }
1;