handle black and white list, add support to interval

1 files changed, 19 insertions, 0 deletions

diff --git a/lib/Plack/Middleware/Throttle/Interval.pm b/lib/Plack/Middleware/Throttle/Interval.pm
index cbe7d59..3e61220 100644
--- a/lib/Plack/Middleware/Throttle/Interval.pm
+++ b/lib/Plack/Middleware/Throttle/Interval.pm
@@ -3,10 +3,29 @@ package Plack::Middleware::Throttle::Interval;
 use Moose;
 extends 'Plack::Middleware::Throttle';
 
+has min => (is => 'rw', isa => 'Int', default => 0, lazy => 1);
+
 sub allowed {
+    my ($self, $key) = @_;
+
+    my $t1 = time();
+    my $t0 = $self->backend->get($key);
+    $self->backend->set($key, $t1);
+
+    if (!$t0 || ($t1 - $t0) > $self->min) {
+        return 1;
+    }else{
+        return 0;
+    }
 }
 
 sub cache_key {
+    my ( $self, $env ) = @_;
+    $self->client_identifier($env);
+}
+
+sub reset_time {
+    time + 1;
 }
 
 1;