File Coverage

blib/lib/Katsubushi/Client.pm

Criterion	Covered	Total	%
statement	17	47	36.1
branch	0	10	0.0
condition			n/a
subroutine	6	9	66.6
pod	0	3	0.0
total	23	69	33.3

package Katsubushi::Client;

use 5.008001;
use strict;
use warnings;

our $VERSION = "0.1";

use Cache::Memcached::Fast;

use Carp qw(croak);
use Class::Tiny +{
    servers => sub { ["127.0.0.1:11212"] },
    _models => sub {
        my $self = shift;
        return [ map {
            Cache::Memcached::Fast->new({ servers => [ $_ ] });
        } @{$self->servers} ];
    },
};

sub BUILD {
    my $self = shift;

    if (scalar @{$self->servers} < 1) {
        die 'there are no servers.';
    }
}

sub fetch {
    my $self = shift;

    my $id = 0;

    # retry at once for only main (the first of list) server
    for my $model ($self->_models->[0], @{$self->_models}) {
        $id = $model->get('id');
        last if $id;
    }

    unless ($id) {
        croak 'Failed to fetch new id';
    }

    return $id;
}

sub fetch_multi {
    my $self = shift;
    my $n    = shift;
    my @keys = ( 1 .. $n );
    my $ids;
    # retry at once for only main (the first of list) server
    for my $model ($self->_models->[0], @{$self->_models}) {
        $ids = $model->get_multi(@keys);
        last if scalar(values %$ids) == $n;
    }

    if (scalar values %$ids != $n) {
        croak "Failed to fetch new $n ids";
    }

    return map { $ids->{$_} } @keys;
}

1;
__END__