Redis實(shí)戰(zhàn)記錄之限制操作頻率

?php
/**
 * 頻率限制
 * @param string $action 操作動(dòng)作
 * @param int $userId 發(fā)起操作的用戶ID
 * @param int $time 時(shí)間范圍X秒內(nèi)
 * @param int $number 限制操作數(shù)Y次
 * @param array $expire 超出封印時(shí)間Z ['type'=>1,'ttl'=>過期時(shí)間/秒] ['type'=>2,'ttl'=>具體過期時(shí)間戳] 二選一
 * @return bool
 * @throws \Exception
 */
public static function frequencyLimit(string $action, int $userId, int $time, int $number, $expire = [])
{
  // todo 根據(jù)用戶操作動(dòng)作時(shí)間范圍，進(jìn)行頻率的控制和失效釋放
}

?php
/**
 * 頻率限制
 * @param string $action 操作動(dòng)作
 * @param int $userId 發(fā)起操作的用戶ID
 * @param int $time 時(shí)間范圍X秒內(nèi)
 * @param int $number 限制操作數(shù)Y次
 * @param array $expire 超出封印時(shí)間Z ['type'=>1,'ttl'=>過期時(shí)間/秒] ['type'=>2,'ttl'=>具體過期時(shí)間戳] 二選一
 * @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('非法參數(shù)');
  }
  $key = 'act:limit:' . $action . ':' . $userId;
  $r = RedisClient::connect();
  //獲取當(dāng)前累計(jì)次數(shù)
  $current = intval($r->get($key));
  if ($current >= $number) return false;
  //累計(jì)并返回最新值
  $current = $r->incr($key);
  //第一次累加，設(shè)置控制操作頻率的有效時(shí)間
  if ($current === 1) $r->expire($key, $time);
  //未超出限制次數(shù)先放過
  if ($current  $number) return true;
  //超出后根據(jù)需要重新設(shè)置過期失效時(shí)間 $current === $number 判斷保證只重新設(shè)置一次
  $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

/**
 * 評(píng)論限制
 * @param int $userId
 * @return bool|string
 */
public function doComment(int $userId)
{
  try {
    $pass = FrequencyLimit::doHandle('comment', $userId, 30, 10);
    if (!$pass) return '過于頻繁';
    // todo 評(píng)論邏輯
    return true;
  } catch (\Exception $e) {
    return $e->getMessage();
  }
}

//場景2
/**
 * 點(diǎn)贊限制
 * @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小時(shí)';
    // todo 點(diǎn)贊邏輯
    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