src/Biz/PostFilter/Event/TokenBucketEventSubscriber.php

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

<?php

namespace Biz\PostFilter\Event;

use Codeages\Biz\Framework\Event\Event;
use Codeages\PluginBundle\Event\EventSubscriber;
use Symfony\Component\EventDispatcher\EventSubscriberInterface;
use Topxia\Service\Common\ServiceKernel;

class TokenBucketEventSubscriber extends EventSubscriber implements EventSubscriberInterface
{
    public static function getSubscribedEvents()
    {
        return array(
            'thread.before_create' => 'before',
            'thread.create' => 'incrToken',

            'thread.post.before_create' => 'before',
            'thread.post.create' => 'incrToken',

            'course.thread.before_create' => 'before',
            'course.thread.create' => 'incrToken',

            'course.thread.post.before_create' => 'before',
            'course.thread.post.create' => 'incrToken',

            'group.thread.before_create' => 'before',
            'group.thread.create' => 'incrToken',

            'group.thread.post.before_create' => 'before',
            'group.thread.post.create' => 'incrToken',
        );
    }

    public function before(Event $event)
    {
        $currentUser = ServiceKernel::instance()->getCurrentUser();
        if ($currentUser->isAdmin() || $currentUser->isSuperAdmin() || $currentUser->isTeacher()) {
            return;
        }

        $currentIp = $currentUser->currentIp;

        if (!($this->getTokenBucketService()->hasToken($currentIp, 'thread')
            && $this->getTokenBucketService()->hasToken($currentUser['id'], 'threadLoginedUser'))) {
            $event->stopPropagation();
        }
    }

    public function incrToken(Event $event)
    {
        $currentUser = ServiceKernel::instance()->getCurrentUser();
        if ($currentUser->isAdmin() || $currentUser->isSuperAdmin() || $currentUser->isTeacher()) {
            return;
        }

        $currentIp = $currentUser->currentIp;

        $this->getTokenBucketService()->incrToken($currentIp, 'thread');
        $this->getTokenBucketService()->incrToken($currentUser['id'], 'threadLoginedUser');