# 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;