diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/Net/Riak/Transport/PBC/Message.pm | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/lib/Net/Riak/Transport/PBC/Message.pm b/lib/Net/Riak/Transport/PBC/Message.pm index 75170de..6eebc6f 100644 --- a/lib/Net/Riak/Transport/PBC/Message.pm +++ b/lib/Net/Riak/Transport/PBC/Message.pm @@ -110,12 +110,17 @@ sub handle_response { sub _unpack_response { my $self = shift; my ( $len, $code, $msg ); - $self->socket->read( $len, 4 ); + _check($self->socket->read( $len, 4 )); $len = unpack( 'N', $len ); - $self->socket->read( $code, 1 ); + _check($self->socket->read( $code, 1 )); $code = unpack( 'c', $code ); - $self->socket->read( $msg, $len - 1 ); + _check($self->socket->read( $msg, $len - 1 )); return ( $code, $msg ); } +sub _check { + defined $_[0] + or die "failure in reading from the socket. Error were : $!"; +} + 1; |