src/ApiBundle/EventListener/BizRateLimitListener.php line 21

Open in your IDE?
  1. <?php
  3. namespace ApiBundle\EventListener;
  5. use ApiBundle\Event\ResourceEvent;
  6. use Codeages\Biz\Framework\Context\Biz;
  8. class BizRateLimitListener
  9. {
  10.     private $biz;
  12.     public function __construct(Biz $biz)
  13.     {
  14.         $this->biz = $biz;
  15.     }
  17.     private $ruleMap = [
  18.         ['SmsCenter', 'post', 'register_sms_rate_limiter'],
  19.     ];
  21.     public function handle(ResourceEvent $event)
  22.     {
  23.         $request = $event->getRequest();
  24.         foreach ($this->ruleMap as $rule) {
  25.             if ($this->isRateLimitApi($rule, $event)) {
  26.                 $rateLimiter = $this->getRateLimiter($rule[2]);
  27.                 $rateLimiter->handle($request);
  28.             }
  29.         }
  30.     }
  32.     private function isRateLimitApi($rule, ResourceEvent $event)
  33.     {
  34.         $request = $event->getRequest();
  35.         $resourceProxy = $event->getResourceProxy();
  36.         $class = get_class($resourceProxy->getResource());
  37.         $className = $this->getClassName($class);
  39.         return 0 === strcasecmp($rule[0], $className)
  40.             && 0 === strcasecmp($rule[1], $request->getMethod());
  41.     }
  43.     private function getClassName($class)
  44.     {
  45.         $path = explode('\\', $class);
  47.         return array_pop($path);
  48.     }
  50.     /**
  51.      * @return \AppBundle\Component\RateLimit\RateLimiterInterface
  52.      */
  53.     private function getRateLimiter($name)
  54.     {
  55.         return $this->biz[$name];
  56.     }
  57. }