File Coverage

blib/lib/POE/Component/Client/Stomp/Utils.pm

Criterion	Covered	Total	%
statement	12	72	16.6
branch	0	8	0.0
condition	0	3	0.0
subroutine	4	17	23.5
pod	13	13	100.0
total	29	113	25.6

package POE::Component::Client::Stomp::Utils;

use 5.008;
use strict;
use warnings;

use Net::Stomp::Frame;

our $VERSION = '0.02';

# -------------------------------------------------------------------

sub new {
    my $proto = shift;

    my $self = {};
    my $class = ref($proto) || $proto;

    $self->{transaction_id} = 0;
    $self->{session_id} = 0;
    $self->{message_id} = 0;

    bless($self, $class);

    return $self;

}

# -------------------------------------------------------------------
# Mutators
# -----------------------------------------------------------

sub transaction_id {
    my ($self, $a) = @_;

    $self->{transaction_id} = $a if (defined $a);
    return $self->{transaction_id};

}

sub session_id {
    my ($self, $a) = @_;

    $self->{session_id} = $a if (defined $a);
    return $self->{session_id};

}

sub message_id {
    my ($self, $a) = @_;

    $self->{message_id} = $a if (defined $a);
    return $self->{message_id};

}

# -----------------------------------------------------------
# Methods
# -----------------------------------------------------------

sub connect {
    my ($self, $params) = @_;

    my $body = '';
    my $command = 'CONNECT';

    return(Net::Stomp::Frame->new({command => $command,
                                   headers => $params,
                                   body => $body}));

}

sub subscribe {
    my ($self, $params) = @_;

    my $body = '';
    my $command = 'SUBSCRIBE';

    return(Net::Stomp::Frame->new({command => $command,
                                   headers => $params,
                                   body => $body}));

}

sub unsubscribe {
    my ($self, $params) = @_;

    my $body = '';
    my $command = 'UNSUBSCRIBE';

    return(Net::Stomp::Frame->new({command => $command,
                                   headers => $params,
                                   body => $body}));

}

sub begin {
    my ($self, $params) = @_;

    my $body = '';
    my $command = 'BEGIN';

    $self->transaction_id($params->{transaction});

    return(Net::Stomp::Frame->new({command => $command,
                                   headers => $params,
                                   body => $body}));

}

sub commit {
    my ($self, $params) = @_;

    my $body = '';
    my $command = 'COMMIT';

    $params->{transaction} = $self->transaction_id;

    return(Net::Stomp::Frame->new({command => $command,
                                   headers => $params,
                                   body => $body}));

}

sub ack {
    my ($self, $params) = @_;

    my $body = '';
    my $command = 'ACK';

    $params->{transaction} = $self->transaction_id
        if ($self->{transaction_id} > 0);

    return(Net::Stomp::Frame->new({command => $command,
                                   headers => $params,
                                   body => $body}));

}

sub abort {
    my ($self, $params) = @_;

    my $body = '';
    my $command = 'ABORT';

    $params->{transaction} = $self->transaction_id;
    $self->{transaction_id} = 0;

    return(Net::Stomp::Frame->new({command => $command,
                                   headers => $params,
                                   body => $body}));

}

sub disconnect {
    my ($self, $params) = @_;

    my $body = '';
    my $command = 'DISCONNECT';

    return(Net::Stomp::Frame->new({command => $command,
                                   headers => $params,
                                   body => $body}));

}

sub send {
    my ($self, $params) = @_;

    my $body = $params->{data};
    my $command = 'SEND';

    delete $params->{data};
    $params->{'content-length'} = length($body);

    return(Net::Stomp::Frame->new({command => $command,
                                   headers => $params,
                                   body => $body}));

}

1;

__END__