RateLimit\InMemoryRateLimiter

Complexity

Total Complexity

7

Size/Duplication

Total Lines	49
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Test Coverage

Coverage

100%

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	limit()	0	10	2
A	limitSilently()	0	13	1
A	key()	0	4	1
A	hit()	0	13	3

<?php

declare(strict_types=1);

namespace RateLimit;

use RateLimit\Exception\LimitExceeded;

final class InMemoryRateLimiter implements RateLimiter, SilentRateLimiter
{
    /** @var array */
    private $store = [];

    public function limit(string $identifier, Rate $rate): void
    {
        $key = $this->key($identifier, $rate->getInterval());

        $current = $this->hit($key, $rate);

        if ($current > $rate->getOperations()) {
            throw LimitExceeded::for($identifier, $rate);
        }
    }

    public function limitSilently(string $identifier, Rate $rate): Status
    {
        $key = $this->key($identifier, $rate->getInterval());

        $current = $this->hit($key, $rate);

        return Status::from(
            $identifier,
            $current,
            $rate->getOperations(),
            $this->store[$key]['reset_time']
        );
    }

    private function key(string $identifier, int $interval): string
    {
        return "$identifier:$interval:" . floor(time() / $interval);
    }

    private function hit(string $key, Rate $rate): int
    {
        if (!isset($this->store[$key])) {
            $this->store[$key] = [
                'current' => 1,
                'reset_time' => time() + $rate->getInterval(),
            ];
        } elseif ($this->store[$key]['current'] <= $rate->getOperations()) {
            $this->store[$key]['current']++;
        }

        return $this->store[$key]['current'];
    }
}