From c3fa7ddfb6fb0dd128963b57c0af73ebf71484d5 Mon Sep 17 00:00:00 2001
From: franck cuny <franck@lumberjaph.net>
Date: Tue, 6 Jul 2010 10:49:03 +0200
Subject: fix hook to add worker_id

---
 lib/Net/Presque.pm | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

(limited to 'lib/Net')

diff --git a/lib/Net/Presque.pm b/lib/Net/Presque.pm
index bc4c9c0..1a7c976 100644
--- a/lib/Net/Presque.pm
+++ b/lib/Net/Presque.pm
@@ -117,14 +117,16 @@ net_api_method unregister_worker => (
     required => [qw/queue_name/],
 );
 
-after 'new' => sub {
+sub BUILD {
     my $self = shift;
 
     if ($self->has_worker_id) {
-        $self->useragent->add_handler('request_prepare' => sub {
-            my ($request, ) = @_;
-            $request->header('X-presque-workerid' => $self->worker_id);
-        });
+        $self->api_useragent->add_handler(
+            'request_prepare' => sub {
+                my ($request,) = @_;
+                $request->header('X-presque-workerid' => $self->worker_id);
+            }
+        );
     }
 };
 
-- 
cgit v1.2.3