X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?p=scpubgit%2FObject-Remote.git;a=blobdiff_plain;f=lib%2FObject%2FRemote%2FRole%2FConnector%2FPerlInterpreter.pm;h=c430fa6e48dafdcd76a8a3bb5986056a9b51cecf;hp=933924e4c64863c39d476047601fc237e374371f;hb=55c0d0209fa9d9265ff178f54ae9fe5fdddef3c1;hpb=eb351344773af82344bd550ffd5a8814b816cfeb diff --git a/lib/Object/Remote/Role/Connector/PerlInterpreter.pm b/lib/Object/Remote/Role/Connector/PerlInterpreter.pm index 933924e..c430fa6 100644 --- a/lib/Object/Remote/Role/Connector/PerlInterpreter.pm +++ b/lib/Object/Remote/Role/Connector/PerlInterpreter.pm @@ -1,8 +1,8 @@ package Object::Remote::Role::Connector::PerlInterpreter; -use IPC::Open3; +use IPC::Open3; use IO::Handle; -use Symbol; +use Symbol; use Object::Remote::Logging qw(:log :dlog router); use Object::Remote::ModuleSender; use Object::Remote::Handle; @@ -13,9 +13,8 @@ use Moo::Role; with 'Object::Remote::Role::Connector'; has module_sender => (is => 'lazy'); -has ulimit => ( is => 'ro'); -has nice => ( is => 'ro'); has watchdog_timeout => ( is => 'ro', required => 1, default => sub { undef }); +has forward_env => (is => 'ro', required => 1, builder => 1); has perl_command => (is => 'lazy'); has pid => (is => 'rwp'); has connection_id => (is => 'rwp'); @@ -33,30 +32,22 @@ sub _build_module_sender { return $hook ? $hook->sender : Object::Remote::ModuleSender->new; } -#FIXME by policy object-remote does not invoke a shell +#By policy object-remote does not invoke a shell sub _build_perl_command { - my ($self) = @_; - my $nice = $self->nice; - my $ulimit = $self->ulimit; - my $perl_path = 'perl'; - my $shell_code = ''; - - if (defined($ulimit)) { - $shell_code .= "ulimit $ulimit || exit 1; "; - } - - if (defined($nice)) { - $shell_code .= "nice -n $nice "; - } + my $perl_bin = 'perl'; - if (defined($ENV{OBJECT_REMOTE_PERL_BIN})) { - log_debug { "Using OBJECT_REMOTE_PERL_BIN environment variable as perl path" }; - $perl_path = $ENV{OBJECT_REMOTE_PERL_BIN}; + if (exists $ENV{OBJECT_REMOTE_PERL_BIN}) { + $perl_bin = $ENV{OBJECT_REMOTE_PERL_BIN}; } + return [qw(perl -)]; +} - $shell_code .= $perl_path . ' -'; - - return [ 'bash', '-c', $shell_code ]; +sub _build_forward_env { + return [qw( + OBJECT_REMOTE_PERL_BIN + OBJECT_REMOTE_LOG_LEVEL OBJECT_REMOTE_LOG_FORMAT OBJECT_REMOTE_LOG_SELECTIONS + OBJECT_REMOTE_LOG_FORWARDING + )]; } around connect => sub { @@ -86,11 +77,11 @@ sub _start_perl { my $self = shift; my $given_stderr = $self->stderr; my $foreign_stderr; - + Dlog_verbose { s/\n/ /g; "invoking connection to perl interpreter using command line: $_" } @{$self->final_perl_command}; - + if (defined($given_stderr)) { #if the stderr data goes to an existing file handle #an anonymous file handle is required @@ -103,24 +94,24 @@ sub _start_perl { #the child stderr to the parent stderr $foreign_stderr = ">&STDERR"; } - + my $pid = open3( my $foreign_stdin, my $foreign_stdout, $foreign_stderr, @{$self->final_perl_command}, ) or die "Failed to run perl at '$_[0]': $!"; - + $self->_set_pid($pid); - - if (defined($given_stderr)) { + + if (defined($given_stderr)) { Dlog_debug { "Child process STDERR is being handled via run loop" }; - + Object::Remote->current_loop ->watch_io( handle => $foreign_stderr, on_read_ready => sub { - my $buf = ''; + my $buf = ''; my $len = sysread($foreign_stderr, $buf, 32768); if (!defined($len) or $len == 0) { log_trace { "Got EOF or error on child stderr, removing from watcher" }; @@ -133,10 +124,10 @@ sub _start_perl { Dlog_trace { "got $len characters of stderr data for connection" }; print $given_stderr $buf or die "could not send stderr data: $!"; } - } - ); + } + ); } - + return ($foreign_stdin, $foreign_stdout, $pid); } @@ -172,14 +163,14 @@ sub _open2_for { sub _setup_watchdog_reset { my ($self, $conn) = @_; - my $timer_id; - + my $timer_id; + return unless $self->watchdog_timeout; - + Dlog_trace { "Creating Watchdog management timer for connection id $_" } $conn->_id; - + weaken($conn); - + $timer_id = Object::Remote->current_loop->watch_time( every => $self->watchdog_timeout / 3, code => sub { @@ -188,13 +179,13 @@ sub _setup_watchdog_reset { Object::Remote->current_loop->unwatch_time($timer_id); return; } - + unless($conn->is_valid) { log_warn { "Watchdog timer found an invalid connection, removing the timer" }; Object::Remote->current_loop->unwatch_time($timer_id); return; } - + Dlog_trace { "Reseting Watchdog for connection id $_" } $conn->_id; #we do not want to block in the run loop so send the #update off and ignore any result, we don't need it @@ -202,7 +193,7 @@ sub _setup_watchdog_reset { $conn->send_class_call(0, 'Object::Remote::WatchDog', 'reset'); } ); - + $conn->on_close->on_ready(sub { log_debug { "Removing watchdog for connection that is now closed" }; Object::Remote->current_loop->unwatch_time($timer_id); @@ -216,19 +207,26 @@ sub fatnode_text { my $text = ''; require Object::Remote::FatNode; - + if (defined($connection_timeout)) { $text .= "alarm($connection_timeout);\n"; } - + if (defined($watchdog_timeout)) { $text .= "my \$WATCHDOG_TIMEOUT = $watchdog_timeout;\n"; } else { $text .= "my \$WATCHDOG_TIMEOUT = undef;\n"; } - + + $text .= $self->_create_env_forward(@{$self->forward_env}); + + #Action at a distance but at least it's not spooky - the logging + #system needs to know if a node is remote but there is a period + #during init where the remote connection information has not been + #setup on the remote side yet so this flag allows a graceful + #degredation to happen $text .= '$Object::Remote::FatNode::REMOTE_NODE = "1";' . "\n"; - + $text .= <<'END'; $INC{'Object/Remote/FatNode.pm'} = __FILE__; $Object::Remote::FatNode::DATA = <<'ENDFAT'; @@ -239,9 +237,29 @@ END eval $Object::Remote::FatNode::DATA; die $@ if $@; END - + $text .= "__END__\n"; return $text; } +sub _create_env_forward { + my ($self, @env_names) = @_; + my $code = ''; + + foreach my $name (@env_names) { + next unless exists $ENV{$name}; + my $value = $ENV{$name}; + $name =~ s/'/\\'/g; + if(defined($value)) { + $value =~ s/'/\\'/g; + $value = "'$value'"; + } else { + $value = 'undef'; + } + $code .= "\$ENV{'$name'} = $value;\n"; + } + + return $code; +} + 1;