diff options
-rw-r--r-- | lib/presque/worker/Role/Management.pm | 15 |
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 { |