1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
package presque::worker::Role::RESTClient;
use Moose::Role;
use LWP::UserAgent;
use HTTP::Request;
use MooseX::Types::URI qw/Uri/;
has base_uri => (is => 'ro', isa => Uri, coerce => 1, required => 1);
has ua => (
is => 'rw',
isa => 'LWP::UserAgent',
lazy => 1,
default => sub { my $ua = LWP::UserAgent->new; $ua }
);
sub _job_uri {
my $self = shift;
my $uri = $self->base_uri->clone;
$uri->path_segments($uri->path_segments, 'q', $self->queue_name);
$uri->query_form(worker_id => $self->worker_id);
$uri;
}
sub _worker_uri {
my $self = shift;
my $uri = $self->base_uri->clone;
$uri->path_segments($uri->path_segments, 'w', $self->queue_name);
$uri;
}
sub rest_register_worker {
my $self = shift;
my $request = HTTP::Request->new(POST => $self->_worker_uri);
$request->content(JSON::encode_json({worker_id => $self->worker_id}));
my $res = $self->ua->request($request);
die "can't register to ".$self->base_uri if (!$res->is_success);
}
sub rest_unregister_worker {
my $self = shift;
my $uri = $self->_worker_uri;
$uri->query_form(worker_id => $self->worker_id);
my $request = HTTP::Request->new(DELETE => $uri);
my $res = $self->ua->request($request);
}
sub rest_fetch_job {
my $self = shift;
my $res = $self->ua->request(HTTP::Request->new(GET => $self->_job_uri));
if ($res->is_success) {
return JSON::decode_json($res->content);
}
else {
$self->logger->log(
level => 'debug',
message => $res->code . ':' . $res->message
);
}
return;
}
sub rest_retry_job {
my ($self, $job) = @_;
my $request = HTTP::Request->new(PUT => $self->_job_uri);
$request->header('Content-Type' => 'application/json');
$request->content(JSON::encode_json($job));
my $res = $self->ua->request($request);
if (!$res->is_success) {
use YAML::Syck; warn Dump $res;
$self->logger->log(
level => 'error',
message => 'failed to update job ('
. $res->code . ':'
. $res->message . ')',
);
}
}
1;
|