Redis实战记录之限制操作频率

<?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 根据用户操作动作时间范围，进行频率的控制和失效释放
}

<?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