# Copyright (C) 2021 all contributors
# License: AGPL-3.0+
# op dispatch socket, reads a message, runs a sub
# There may be multiple producers, but (for now) only one consumer
# Used for lei_xsearch and maybe other things
# "command" => [ $sub, @fixed_operands ]
package PublicInbox::PktOp;
use strict;
use v5.10.1;
use parent qw(PublicInbox::DS);
use Errno qw(EAGAIN ECONNRESET);
use PublicInbox::Syscall qw(EPOLLIN);
use Socket qw(AF_UNIX MSG_EOR SOCK_SEQPACKET);
use PublicInbox::IPC qw(ipc_freeze ipc_thaw);
use Scalar::Util qw(blessed);
sub new {
my ($cls, $r) = @_;
my $self = bless { sock => $r }, $cls;
$r->blocking(0);
$self->SUPER::new($r, EPOLLIN);
}
# returns a blessed objects as the consumer and producer
sub pair {
my ($cls) = @_;
my ($c, $p);
socketpair($c, $p, AF_UNIX, SOCK_SEQPACKET, 0) or die "socketpair: $!";
(new($cls, $c), bless { op_p => $p }, $cls);
}
sub pkt_do { # for the producer to trigger event_step in consumer
my ($self, $cmd, @args) = @_;
send($self->{op_p}, @args ? "$cmd\0".ipc_freeze(\@args) : $cmd, MSG_EOR)
}
sub event_step {
my ($self) = @_;
my $c = $self->{sock};
my $n = recv($c, my $msg, 4096, 0);
unless (defined $n) {
return if $! == EAGAIN;
die "recv: $!" if $! != ECONNRESET; # we may be bidirectional
}
my ($cmd, @pargs);
if (index($msg, "\0") > 0) {
($cmd, my $pargs) = split(/\0/, $msg, 2);
@pargs = @{ipc_thaw($pargs)};
} else {
# for compatibility with the script/lei in client mode,
# it doesn't load Sereal||Storable for startup speed
($cmd, @pargs) = split(/ /, $msg);
}
my $op = $self->{ops}->{$cmd //= $msg};
if ($op) {
my ($obj, @args) = (@$op, @pargs);
blessed($obj) ? $obj->$cmd(@args) : $obj->(@args);
} elsif ($msg ne '') {
die "BUG: unknown message: `$cmd'";
}
$self->close if $msg eq ''; # close on EOF
}
1;