Redis实战之限制操做频率

最近沉迷于业务开发没法自拔 🤣，有一段时间没有更新博文了，后续博文内容计划把一些业务场景下的实战方案，或者比较好的设计思路进行分享，就不像以前围绕着一个主题，消耗不少的时间去整理相关内容(憋大招)，后续可能一篇的内容量就没那么丰富，可是尽量针对一个点进行更细化，或者更深刻的分析，经过不断分享和自我复盘，进行经验的沉淀，同时提升博文分享的频率 🤙

场景

场景1

留言功能限制，30秒 内只能评论 10次，超出次数不让能再评论，并提示：过于频繁

场景2

点赞功能限制，10秒 内只能点赞 10次，超出次数后不能再点赞，并禁止操做 1个小时，提示：过于频繁，被禁止操做1小时

场景3

上传记录功能，限制一天只能上传 100次，超出次数不让能再上传，并提示：超出今日上线

抽离本质

在业务开发的过程当中，咱们不断的参与各类业务场景的方案设计，每每很容易碰到很相似的场景，只不过当前所属的业务模块不同，其实这些需求的本质是解决同一个问题，当遇到这种场景的时候，咱们须要根据本身经验分析抽离出需求的本质问题，实现一个通用的解决方案，让本身的解决方案更有价值，这可能就是区别于你是有灵魂的工程师仍是cp（copy paste）最强王者吧。

分析上面3个业务场景，能够从中发现其中有类似的逻辑，称它为同类的问题，如今咱们就是要抽离这个问题，设计一个通用的解决方案，勾画相同逻辑流程图：

经过分析上面的需求场景，抽离出他们都须要的那些条件：

• 限制对象：用户
• 限制操做（评论，点赞，记录， ...）
• 时间范围X秒内
• 限制操做数Y次
• 超出后禁止操做时间Z（秒/具体时间）
• 超出后不让再操做，并提示

（最小时间单位用秒：天/小时/分钟均可换算成秒，用秒能够解决更多的场景）

若是把功能抽离成一个通用函数是否是大概是这样：

<?php
/**
 * 频率限制
 * @param string $action 操做动做
 * @param int $userId 发起操做的用户ID
 * @param int $time 时间范围X秒内
 * @param int $number 限制操做数Y次
 * @param array $expire 超出封印时间Z ['type'=>1,'ttl'=>过时时间/秒] ['type'=>2,'ttl'=>具体过时时间戳] 二选一
 * @return bool
 * @throws \Exception
 */
public static function frequencyLimit(string $action, int $userId, int $time, int $number, $expire = [])
{
    // todo 根据用户操做动做时间范围，进行频率的控制和失效释放
}

解决方案落地

功能中须要对用户发起的操做和时间，以及累计次数进行存储，而且须要失效过时的清理，若是这个时候咱们依赖mysql作存储，想一想都觉的挺痛苦，这里主角：redis 终于登场了，基于redis特性，incr的原子操做和key 支持过时机制，内存存储的效率优点，能够相对简单灵活而且又高效的完成目的。

这里简单实现个通用功能的代码：

<?php
/**
 * 频率限制
 * @param string $action 操做动做
 * @param int $userId 发起操做的用户ID
 * @param int $time 时间范围X秒内
 * @param int $number 限制操做数Y次
 * @param array $expire  超出封印时间Z ['type'=>1,'ttl'=>过时时间/秒] ['type'=>2,'ttl'=>具体过时时间戳] 二选一
 * @return bool
 * @throws \Exception
 */
public function frequencyLimit(string $action, int $userId, int $time, int $number, $expire = [])
{
    if (empty($action) || $userId <= 0 || $time <= 0 || $number <= 0) {
        throw new \Exception('非法参数');
    }
    $key = 'act:limit:' . $action . ':' . $userId;
    $r = RedisClient::connect();
    //获取当前累计次数
    $current = intval($r->get($key));
    if ($current >= $number) return false;
    //累计并返回最新值
    $current = $r->incr($key);
    //第一次累加，设置控制操做频率的有效时间
    if ($current === 1) $r->expire($key, $time);
    //未超出限制次数先放过
    if ($current < $number) return true;
    //超出后根据须要从新设置过时失效时间 $current === $number 判断保证只从新设置一次
    $type = empty($expire['type']) ? 0 : intval($expire['type']);
    $ttl = empty($expire['ttl']) ? 0 : intval($expire['ttl']);
    if ($current === $number && $ttl > 0 && in_array($type, [1, 2])) {
        if ($type === 1) $r->expire($key, $ttl);
        if ($type === 2) $r->expireAt($key, $ttl);
    }
    return false;
}
//场景1

/**
 * 评论限制
 * @param int $userId
 * @return bool|string
 */
public function doComment(int $userId)
{
    try {
        $pass = FrequencyLimit::doHandle('comment', $userId, 30, 10);
        if (!$pass) return '过于频繁';
        // todo 评论逻辑
        return true;
    } catch (\Exception $e) {
        return $e->getMessage();
    }
}

//场景2
/**
 * 点赞限制
 * @param int $userId
 * @return bool|string
 */
public function doLike(int $userId)
{
    try {
        $pass = FrequencyLimit::doHandle('like', $userId, 10, 10, ['type' => 1, 'ttl' => 1 * 60 * 60]);
        if (!$pass) return '过于频繁，被禁止操做1小时';
        // todo 点赞逻辑
        return true;
    } catch (\Exception $e) {
        return $e->getMessage();
    }
}

//场景3

/**
 * 上传限制
 * @param int $userId
 * @return bool|string
 */
public function doUpload(int $userId)
{
    try {
        $expire = strtotime(date('Y-m-d', strtotime(+1 . 'days')));
        $pass = FrequencyLimit::doHandle('upload', $userId, 1 * 24 * 60 * 60, 100, ['type' => 2, 'ttl' => $expire]);
        if (!$pass) return '超出今日上线';
        // todo 上传逻辑
        return true;
    } catch (\Exception $e) {
        return $e->getMessage();
    }
}

//场景N

编码上能够根据你设计这个通用方案的复杂度进行进一步抽象，如抽象成频率限制的功能类 等

总结

• 对类似的业务场景进行分析，发现本质问题并设计通用的解决方案
• 让解决方案更有价值，作一个有灵魂的开发者
• 熟练掌握redis，充分利用它的特性和优点

---

首发于Github🌈大话WEB开发，欢迎Star 🥰