File Coverage

blib/lib/WWW/RabbitMQ/Broker/Shovel.pm

Criterion	Covered	Total	%
statement	6	39	15.3
branch	0	10	0.0
condition	0	5	0.0
subroutine	2	8	25.0
pod	6	6	100.0
total	14	68	20.5

package WWW::RabbitMQ::Broker::Shovel;

use strict;
use warnings;

our $VERSION = '0.03';

sub new
{
	my $class  = shift;
	my $broker = shift;

	my $self = {
		_config => ref($_[0]) ? $_[0] : {@_},
	};

	unless ($broker && ref($broker) eq 'WWW::RabbitMQ::Broker') {
		die "ERROR: Did not receive a proper broker object.\n";
	}

	$self->{_config}{vhost} ||= '/';
	$self->{_broker} = $broker;
	return bless($self, $class);
}

sub delete
{
	my $self = shift;
	$self->{_broker}->httpMethod('DELETE')->apiCall("parameters/shovel/$self->{_config}{vhost}/$self->{_config}{name}");
	return {deleted => 1};
}

sub get
{
	my $self = shift;
	my $res  = $self->{_broker}->httpMethod('GET')->apiCall("parameters/shovel/$self->{_config}{vhost}/$self->{_config}{name}");
	return $self->{_config} = $res;
}

sub getConfig
{
	my $self = shift;
	my $get  = shift;
	$self->get;
	if ($get eq 'all') {
		return $self->{_config};
	}
	else {
		return $self->{_config}{$get};
	}
}

sub getOrPut
{
	my $self = shift;

	my $res;
	eval {
		$res = $self->{_broker}->httpMethod('GET')->apiCall("parameters/shovel/$self->{_config}{vhost}/$self->{_config}{name}");
	};

	if ($!) {
		unless ($! =~ /Invalid/) {
			die "ERROR[404]: Invalid Method\n";
		}
		my $res = $self->{_broker}->httpMethod('PUT')->apiCall(
			"parameters/shovel/$self->{_config}{vhost}/$self->{_config}{name}",
			$self->{_config},
		);
		return $self->{_config} = $res;
	}
	else {
		return $self->{_config} = $res;
	}
}

sub put
{
	my $self = shift;
	my $res = $self->{_broker}->httpMethod('PUT')->apiCall(
		"parameters/shovel/$self->{_config}{vhost}/$self->{_config}{name}",
		$self->{_config},
	);
	return $self->{_config} = $res;
}

1;

__END__