mirror of
https://github.com/xcat2/confluent.git
synced 2024-12-25 12:41:39 +00:00
86 lines
2.3 KiB
Perl
86 lines
2.3 KiB
Perl
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2014 IBM Corporation
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
use strict;
|
|
use warnings;
|
|
|
|
package Confluent::TLV;
|
|
|
|
use JSON;
|
|
|
|
sub new {
|
|
my $proto = shift;
|
|
my $class = ref($proto) || $proto;
|
|
my $self = {};
|
|
$self->{handle} = shift;
|
|
$self->{json} = JSON->new;
|
|
bless($self, $class);
|
|
return $self;
|
|
}
|
|
|
|
sub recv {
|
|
my $self = shift;
|
|
my $data;
|
|
my $tl = '';
|
|
do {
|
|
sysread($self->{handle}, $data, 4 - length($tl));
|
|
$tl .= $data;
|
|
} while (length($tl) < 4);
|
|
$tl = unpack("N", $tl);
|
|
if ($tl & (1 << 31)) {
|
|
die "Reserved bit used, protocol violated";
|
|
}
|
|
my $length = $tl & 16777215; # lower 24 bits only
|
|
my $datatype = ($tl >> 24);
|
|
my $bytedata = '';
|
|
while (length($bytedata) < $length) {
|
|
sysread($self->{handle}, $data, $length - length($bytedata));
|
|
$bytedata .= $data;
|
|
}
|
|
if ($datatype == 0) {
|
|
return $bytedata;
|
|
} elsif ($datatype == 1) {
|
|
return $self->{json}->utf8->decode($bytedata);
|
|
} else {
|
|
die "Unexpected data type $datatype";
|
|
}
|
|
}
|
|
|
|
sub send {
|
|
my $self = shift;
|
|
my $data = shift;
|
|
if (ref $data eq 'HASH') { # Need to do JSON
|
|
my $json = $self->{json}->utf8->encode($data);
|
|
my $typelength = length($json);
|
|
if ($typelength > 16777215) {
|
|
die "Data too large";
|
|
}
|
|
$typelength |= 16777216;
|
|
my $handle = $self->{handle};
|
|
print $handle pack("N", $typelength) . $json;
|
|
$self->{handle}->flush();
|
|
} elsif (not ref $data) { # text data
|
|
my $typelength = length($data);
|
|
if ($typelength > 16777215) {
|
|
die "Data too large";
|
|
}
|
|
my $handle = $self->{handle};
|
|
print $handle pack("N", $typelength) . $data;
|
|
$self->{handle}->flush();
|
|
}
|
|
}
|
|
|
|
1;
|