123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238 |
- <?php
- namespace Illuminate\Cache;
- use Closure;
- use Illuminate\Contracts\Cache\Repository as Cache;
- use Illuminate\Support\InteractsWithTime;
- class RateLimiter
- {
- use InteractsWithTime;
- /**
- * The cache store implementation.
- *
- * @var \Illuminate\Contracts\Cache\Repository
- */
- protected $cache;
- /**
- * The configured limit object resolvers.
- *
- * @var array
- */
- protected $limiters = [];
- /**
- * Create a new rate limiter instance.
- *
- * @param \Illuminate\Contracts\Cache\Repository $cache
- * @return void
- */
- public function __construct(Cache $cache)
- {
- $this->cache = $cache;
- }
- /**
- * Register a named limiter configuration.
- *
- * @param string $name
- * @param \Closure $callback
- * @return $this
- */
- public function for(string $name, Closure $callback)
- {
- $this->limiters[$name] = $callback;
- return $this;
- }
- /**
- * Get the given named rate limiter.
- *
- * @param string $name
- * @return \Closure|null
- */
- public function limiter(string $name)
- {
- return $this->limiters[$name] ?? null;
- }
- /**
- * Attempts to execute a callback if it's not limited.
- *
- * @param string $key
- * @param int $maxAttempts
- * @param \Closure $callback
- * @param int $decaySeconds
- * @return mixed
- */
- public function attempt($key, $maxAttempts, Closure $callback, $decaySeconds = 60)
- {
- if ($this->tooManyAttempts($key, $maxAttempts)) {
- return false;
- }
- if (is_null($result = $callback())) {
- $result = true;
- }
- return tap($result, function () use ($key, $decaySeconds) {
- $this->hit($key, $decaySeconds);
- });
- }
- /**
- * Determine if the given key has been "accessed" too many times.
- *
- * @param string $key
- * @param int $maxAttempts
- * @return bool
- */
- public function tooManyAttempts($key, $maxAttempts)
- {
- if ($this->attempts($key) >= $maxAttempts) {
- if ($this->cache->has($this->cleanRateLimiterKey($key).':timer')) {
- return true;
- }
- $this->resetAttempts($key);
- }
- return false;
- }
- /**
- * Increment (by 1) the counter for a given key for a given decay time.
- *
- * @param string $key
- * @param int $decaySeconds
- * @return int
- */
- public function hit($key, $decaySeconds = 60)
- {
- return $this->increment($key, $decaySeconds);
- }
- /**
- * Increment the counter for a given key for a given decay time by a given amount.
- *
- * @param string $key
- * @param int $decaySeconds
- * @param int $amount
- * @return int
- */
- public function increment($key, $decaySeconds = 60, $amount = 1)
- {
- $key = $this->cleanRateLimiterKey($key);
- $this->cache->add(
- $key.':timer', $this->availableAt($decaySeconds), $decaySeconds
- );
- $added = $this->cache->add($key, 0, $decaySeconds);
- $hits = (int) $this->cache->increment($key, $amount);
- if (! $added && $hits == 1) {
- $this->cache->put($key, 1, $decaySeconds);
- }
- return $hits;
- }
- /**
- * Get the number of attempts for the given key.
- *
- * @param string $key
- * @return mixed
- */
- public function attempts($key)
- {
- $key = $this->cleanRateLimiterKey($key);
- return $this->cache->get($key, 0);
- }
- /**
- * Reset the number of attempts for the given key.
- *
- * @param string $key
- * @return mixed
- */
- public function resetAttempts($key)
- {
- $key = $this->cleanRateLimiterKey($key);
- return $this->cache->forget($key);
- }
- /**
- * Get the number of retries left for the given key.
- *
- * @param string $key
- * @param int $maxAttempts
- * @return int
- */
- public function remaining($key, $maxAttempts)
- {
- $key = $this->cleanRateLimiterKey($key);
- $attempts = $this->attempts($key);
- return $maxAttempts - $attempts;
- }
- /**
- * Get the number of retries left for the given key.
- *
- * @param string $key
- * @param int $maxAttempts
- * @return int
- */
- public function retriesLeft($key, $maxAttempts)
- {
- return $this->remaining($key, $maxAttempts);
- }
- /**
- * Clear the hits and lockout timer for the given key.
- *
- * @param string $key
- * @return void
- */
- public function clear($key)
- {
- $key = $this->cleanRateLimiterKey($key);
- $this->resetAttempts($key);
- $this->cache->forget($key.':timer');
- }
- /**
- * Get the number of seconds until the "key" is accessible again.
- *
- * @param string $key
- * @return int
- */
- public function availableIn($key)
- {
- $key = $this->cleanRateLimiterKey($key);
- return max(0, $this->cache->get($key.':timer') - $this->currentTime());
- }
- /**
- * Clean the rate limiter key from unicode characters.
- *
- * @param string $key
- * @return string
- */
- public function cleanRateLimiterKey($key)
- {
- return preg_replace('/&([a-z])[a-z]+;/i', '$1', htmlentities($key));
- }
- }
|