summary refs log tree commit diff
path: root/lib/presque/worker
diff options
context:
space:
mode:
Diffstat (limited to 'lib/presque/worker')
-rw-r--r--lib/presque/worker/Role/Management.pm15
1 files changed, 12 insertions, 3 deletions
diff --git a/lib/presque/worker/Role/Management.pm b/lib/presque/worker/Role/Management.pm
index c2a13fd..857da91 100644
--- a/lib/presque/worker/Role/Management.pm
+++ b/lib/presque/worker/Role/Management.pm
@@ -16,15 +16,24 @@ before start => sub {
 
 after start => sub {
     my $self = shift;
-    $self->unregister_worker(worker_id => $self->worker_id);
+    $self->register_worker(
+        queue_name => $self->queue_name,
+        worker_id  => $self->worker_id
+    );
 };
 after _graceful_shutdown => sub {
     my $self = shift;
-    $self->unregister_worker(worker_id => $self->worker_id);
+    $self->unregister_worker(
+        queue_name => $self->queue_name,
+        worker_id  => $self->worker_id
+    );
 };
 after _shutdown => sub {
     my $self = shift;
-    $self->unregister_worker(worker_id => $self->worker_id);
+    $self->unregister_worker(
+        queue_name => $self->queue_name,
+        worker_id  => $self->worker_id
+    );
 };
 
 sub _shutdown {