repl works again

[scpubgit/Tak.git] / lib / Tak / Router.pm

diff --git a/lib/Tak/Router.pm b/lib/Tak/Router.pm
index f7e56dd..5c2502b 100644 (file)
--- a/lib/Tak/Router.pm
+++ b/lib/Tak/Router.pm
@@ -1,6 +1,7 @@
 package Tak::Router;
 
 use Tak::MetaService;
+use Scalar::Util qw(weaken);
 use Moo;
 
 has services => (is => 'ro', default => sub { {} });
@@ -12,9 +13,9 @@ sub BUILD {
 
 sub start_request {
   my ($self, $req, $target, @payload) = @_;
-  $req->mistake("Reached router with no target")
+  return $req->mistake("Reached router with no target")
     unless $target;
-  $req->failure("Reached router with invalid target ${target}")
+  return $req->failure("Reached router with invalid target ${target}")
     unless my $next = $self->services->{$target};
   $next->start_request($req, @payload);
 }
@@ -31,6 +32,11 @@ sub register {
   $self->services->{$name} = $service;
 }
 
+sub register_weak {
+  my ($self, $name, $service) = @_;
+  weaken($self->services->{$name} = $service);
+}
+
 sub deregister {
   my ($self, $name) = @_;
   delete $self->services->{$name}