cleanup trailing whitespace ugliness

[scpubgit/Object-Remote.git] / lib / Object / Remote / Role / LogForwarder.pm

diff --git a/lib/Object/Remote/Role/LogForwarder.pm b/lib/Object/Remote/Role/LogForwarder.pm
index 495a9c2..ce5bbc2 100644 (file)
--- a/lib/Object/Remote/Role/LogForwarder.pm
+++ b/lib/Object/Remote/Role/LogForwarder.pm
@@ -1,36 +1,48 @@
 package Object::Remote::Role::LogForwarder;
 
-use Moo::Role; 
+use Moo::Role;
 
-with 'Log::Contextual::Role::Router';
+has enable_forward => ( is => 'rw', default => sub { 1 } );
+has _forward_destination => ( is => 'rw' );
+#lookup table for package names that should not
+#be forwarded across Object::Remote connections
+has _forward_stop => ( is => 'ro', required => 1, default => sub { {} } );
 
-#TODO re-weaken router references when object::remote
-#weak reference operation is figured out
+after _deliver_message => sub {
+#  my ($self, $level, $generator, $args, $metadata) = @_;
+  my ($self, %message_info) = @_;
+  my $package = $message_info{caller_package};
+  my $destination = $self->_forward_destination;
+  our $reentrant;
 
-has child_routers => ( is => 'ro', required => 1, default => sub { {} } );
-has parent_router => ( is => 'rw', );#weak_ref => 1 );
+  if (defined $message_info{object_remote}) {
+    $message_info{object_remote} = { %{$message_info{object_remote}} };
+  }
 
-#adds a child router to this router and gives it
-#a friendly display name
-sub add_child_router {
-   my ($self, $description, $router) = @_;
-   $self->child_routers->{$description} = $router;
-   #weaken($self->child_routers->{$class});
-   $router->parent_router($self);
-   return; 
-}
+  $message_info{object_remote}->{forwarded} = 1;
 
-sub remove_child_router {
-   my ($self, $description) = @_;
-   return delete $self->child_routers->{$description};
-}
+  return unless $self->enable_forward;
+  return unless defined $destination;
+  return if $self->_forward_stop->{$package};
+
+  if (defined $reentrant) {
+    warn "log forwarding went reentrant. bottom: '$reentrant' top: '$package'";
+    return;
+  }
+
+  local $reentrant = $package;
 
-after handle_log_message => sub {
-   my ($self, @args) = @_;
-   my $parent = $self->parent_router;
-      
-   return unless defined $parent;
-   $parent->handle_log_message(@args);
+  eval { $destination->_deliver_message(%message_info) };
+
+  if ($@ && $@ !~ /^Attempt to use Object::Remote::Proxy backed by an invalid handle/) {
+    die $@;
+  }
 };
 
+sub exclude_forwarding {
+  my ($self, $package) = @_;
+  $package = caller unless defined $package;
+  $self->_forward_stop->{$package} = 1;
+}
+
 1;