lib/Plack/Middleware/APIRateLimit/Backend/Redis.pm


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

package Plack::Middleware::APIRateLimit::Backend::Redis;

use Moose;
extends 'Plack::Middleware::APIRateLimit::Backend';

use AnyEvent::Redis;

has store => (
    is      => 'rw',
    isa     => 'AnyEvent::Redis',
    lazy    => 1,
    default => sub {
        return AnyEvent::Redis->new(
            host => '127.0.0.1',
            port => 6378,
        );
    }
);

sub BUILD {
    my $self = shift;
    if (scalar @_) {
        $self->store(AnyEvent::Redis->new(@_));
    }
    return $self;
}

sub get {
    my ( $self, $key ) = @_;
    my $val = $self->store->get($key)->recv;
    if ( !$val ) {
        $self->store->set( $key => 1 )->recv;
        $val = 1;
    }
    return $val;
}

sub incr {
    my ($self, $key) = @_;
    return $self->store->incr($key)->recv;
}

1;