logger read from context; more net::presque

1 files changed, 13 insertions, 4 deletions

diff --git a/lib/presque/worker/Role/Management.pm b/lib/presque/worker/Role/Management.pm
index cceea4e..c2a13fd 100644
--- a/lib/presque/worker/Role/Management.pm
+++ b/lib/presque/worker/Role/Management.pm
@@ -6,7 +6,7 @@ has shut_down => (is => 'rw', isa => 'Bool', default => 0,);
 
 before start => sub {
     my $self = shift;
-    $self->rest_register_worker;
+    $self->register_worker(worker_id => $self->worker_id);
     $SIG{INT}  = sub { $self->_shutdown };
     $SIG{TERM} = sub { $self->_shutdown };
     $SIG{QUIT} = sub { $self->_graceful_shutdown };
@@ -14,9 +14,18 @@ before start => sub {
     $SIG{CHLD} = 'IGNORE';
 };
 
-after start              => sub { (shift)->rest_unregister_worker; };
-after _graceful_shutdown => sub { (shift)->rest_unregister_worker; };
-after _shutdown          => sub { (shift)->rest_unregister_worker; };
+after start => sub {
+    my $self = shift;
+    $self->unregister_worker(worker_id => $self->worker_id);
+};
+after _graceful_shutdown => sub {
+    my $self = shift;
+    $self->unregister_worker(worker_id => $self->worker_id);
+};
+after _shutdown => sub {
+    my $self = shift;
+    $self->unregister_worker(worker_id => $self->worker_id);
+};
 
 sub _shutdown {
     my $self = shift;