NV\RequestLimitBundle\Utils\RequestRestrictor

Complexity

Total Complexity

5

Size/Duplication

Total Lines	46
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	restrictRequestByUserId()	0	3	1
A	__construct()	0	3	1
A	restrictRequestByIp()	0	3	1
A	restrictByKey()	0	8	2

<?php

namespace NV\RequestLimitBundle\Utils;

use NV\RequestLimitBundle\Exception\RequestLimitReachedException;
use NV\RequestLimitBundle\Storage\StorageManager;

class RequestRestrictor
{
    /**
     * @var StorageManager
     */
    private $storageManager;

    /**
     * @param StorageManager $storageManager
     */
    public function __construct(StorageManager $storageManager)
    {
        $this->storageManager = $storageManager;
    }

    /**
     * @param $userIp
     * @return mixed
     */
    public function restrictRequestByIp($userIp)
    {
        return $this->restrictByKey($userIp);
    }

    /**
     * @param $userId
     * @return mixed
     */
    public function restrictRequestByUserId($userId)
    {
       return $this->restrictByKey($userId);
    }

    /**
     * @param $key
     * @return mixed
     * @throws RequestLimitReachedException
     */
    private function restrictByKey($key) {
        if ($this->storageManager->hasItem($key)) {
            throw new RequestLimitReachedException();
        }

        $this->storageManager->setItem($key);

        return $this->storageManager->getItem($key);
    }
}