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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
package Net::Riak::Transport::PBC::Message;
use Moose;
use MooseX::Types::Moose qw/Str HashRef Int/;
use Net::Riak::Types 'Socket';
use Net::Riak::Transport::PBC::Code qw/
REQ_CODE EXPECTED_RESP RESP_CLASS RESP_DECODER/;
use Net::Riak::Transport::PBC::Transport;
has socket => (
is => 'rw',
isa => Socket,
predicate => 'has_socket',
);
has request => (
isa => 'Str',
is => 'ro',
lazy_build => 1,
);
has request_code => (
required => 1,
isa => Int,
is => 'ro',
lazy_build => 1,
);
has message_type => (
required => 1,
isa => Str,
is => 'ro',
trigger => sub {
$_[0]->{message_type} = 'Rpb'.$_[1];
}
);
has params => (
is => 'ro',
isa => HashRef,
);
sub _build_request_code {
my $self = shift;
return REQ_CODE($self->message_type);
}
sub _build_request {
my $self = shift;
$self->_pack_request( $self->request_code, $self->encode );
}
sub _pack_request {
my ($self, $code, $req) = @_;
my $h = pack('c', $code) . $req;
use bytes;
my $len = length $h;
return pack('N',$len).$h;
}
sub encode {
my $self = shift;
return $self->message_type->can('encode')
? $self->message_type->encode( $self->params )
: '';
}
sub decode {
my ($self, $type, $raw_content) = @_;
return 'Rpb'.$type->decode($raw_content);
}
sub send {
my ($self, $cb) = @_;
die "No socket? did you forget to ->connect?" unless $self->has_socket;
$self->socket->print($self->request);
my $resp = $self->handle_response;
return $resp unless $cb;
$cb->($resp);
while (!$resp->done) {
$resp = $self->handle_response;
# use YAML::Syck; warn Dump $resp;
$cb->($resp);
}
return 1;
}
sub handle_response {
my $self = shift;
my ($code, $resp) = $self->_unpack_response;
my $expected_code = EXPECTED_RESP($self->request_code);
if ($expected_code != $code) {
# TODO throw object
die "Expecting response type "
. RESP_CLASS($expected_code)
. " got " . RESP_CLASS($code);
}
return 1 unless RESP_DECODER($code);
return RESP_DECODER($code)->decode($resp);
}
sub _unpack_response {
my $self = shift;
my ( $len, $code, $msg );
$self->socket->read( $len, 4 );
$len = unpack( 'N', $len );
$self->socket->read( $code, 1 );
$code = unpack( 'c', $code );
$self->socket->read( $msg, $len - 1 );
return ( $code, $msg );
}
1;
|
