blob: 4a8cd7f39afd41f15fe154e0340e64ecb1452c66 (
plain) (
blame)
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
package Plack::Middleware::Throttle::Limiter;
use Moose;
extends 'Plack::Middleware::Throttle';
has _request_done => (
is => 'rw',
isa => 'Int',
default => 0,
clearer => '_clear_request_done'
);
sub allowed {
my ( $self, $key ) = @_;
$self->backend->incr($key);
$self->request_done($key);
( $self->_request_done > $self->max ) ? return 0 : return 1;
}
sub request_done {
my ( $self, $key ) = @_;
$self->_request_done( $self->backend->get($key) || 0 );
}
sub add_headers {
my ( $self, $res ) = @_;
my $headers = $res->[1];
Plack::Util::header_set( $headers, 'X-RateLimit-Limit', $self->max );
Plack::Util::header_set( $headers, 'X-RateLimit-Remaining',
( $self->max - $self->_request_done ) );
Plack::Util::header_set( $headers, 'X-RateLimit-Reset',
$self->reset_time );
$self->_clear_request_done;
return $res;
}
1;
__END__
=head1 NAME
Plack::Middleware::Throttle - A Plack Middleware for rate-limiting incoming HTTP requests.
=head1 SYNOPSIS
=head1 DESCRIPTION
=head1 AUTHOR
franck cuny E<lt>franck@lumberjaph.netE<gt>
=head1 SEE ALSO
=head1 LICENSE
This library is free software; you can redistribute it and/or modify
it under the same terms as Perl itself.
=cut
|
