Lamoda\Tactician\RateLimit\RateLimitMiddleware

Complexity

Total Complexity

4

Size/Duplication

Total Lines	39
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	8	2
A	execute()	0	14	2

<?php

declare(strict_types=1);

namespace Lamoda\Tactician\RateLimit;

use Lamoda\Tactician\RateLimit\RateLimiter\RateLimitProviderInterface;
use Lamoda\Tactician\RateLimit\RateLimiter\RateLimiterInterface;
use League\Tactician\Middleware;
use Psr\Log\LoggerInterface;
use Psr\Log\NullLogger;

final class RateLimitMiddleware implements Middleware
{
    /**
     * @var RateLimitProviderInterface
     */
    private $rateLimitProvider;
    /**
     * @var RateLimiterInterface
     */
    private $rateLimiter;
    /**
     * @var LoggerInterface
     */
    private $logger;

    public function __construct(
        RateLimitProviderInterface $rateLimitProvider,
        RateLimiterInterface $rateLimiter,
        LoggerInterface $logger = null
    ) {
        $this->rateLimitProvider = $rateLimitProvider;
        $this->rateLimiter = $rateLimiter;
        $this->logger = $logger ?: new NullLogger();
    }

    public function execute($command, callable $next)
    {
        $rateLimit = $this->rateLimitProvider->provide($command);

        if (null !== $rateLimit) {
            $this->logger->info('Throttling command with key {key} for {limit} on {time} ms', [
                'key' => $rateLimit->getKey(),
                'limit' => $rateLimit->getLimit(),
                'time' => $rateLimit->getMilliseconds(),
            ]);
            $this->rateLimiter->throttle($rateLimit);
        }

        return $next($command);
    }
}