Redis限流实践：实现用户消息推送每天最多通知2次的功能

文章目录

• 🚀一、背景
• 🚀二、准备工作
• 🚀三、实现逻辑
• 🚀四、结语

---

🚀一、背景

🚀二、准备工作

首先，我们需要准备好数据库和 Redis 服务。在 MySQL 数据库中创建一个 user_notifications 表, 包含以下字段：

• id：主键自增长ID
• user_id：用户ID
• content：通知内容
• created_at：记录创建时间

此外，还需要安装 Redis 扩展，在 PHP 中可以通过以下命令安装：

$ pecl install redis

常见的推送消息场景如下图。

🚀三、实现逻辑

🔎3.1 查询用户的已发送通知数量

在用户登录或接收新通知时，我们需要查询用户今天已发送的通知数量。我们可以利用 Redis 的 Sorted Set 数据结构来存储每个用户的通知记录。将用户ID作为 Sorted Set 的 key，通知的发送时间戳作为 score 值，这样就可以按照时间顺序存储用户的通知记录。

使用以下代码实现查询用户已发送通知数量的函数：

function getNotificationCount($userId) {
    $redis = new Redis();
    $redis->connect('127.0.0.1', 6379);

    $todayStart = strtotime('today'); // 当天开始的时间戳
    $todayEnd = strtotime('tomorrow') - 1; // 当天结束的时间戳

    $count = $redis->zcount('user_notifications:' . $userId, $todayStart, $todayEnd);

    return $count;
}

🔎3.2 发送通知

在发送通知之前，先检查用户已发送通知数量是否达到限制。如果已发送通知数量大于等于2，则不再发送新通知；否则，保存通知记录到数据库，并将通知记录的发送时间戳添加到 Redis Sorted Set 中。

使用以下代码实现发送通知的函数：

function sendNotification($userId, $content) {
    // 检查用户已发送通知数量
    $count = getNotificationCount($userId);
    if ($count >= 2) {
        return false;
    }

    // 保存通知记录到数据库
    $pdo = new PDO('mysql:host=localhost;dbname=test', 'username', 'password');
    $stmt = $pdo->prepare("INSERT INTO user_notifications (user_id, content, created_at) VALUES (?, ?, NOW())");
    $stmt->execute([$userId, $content]);

    // 将通知记录的发送时间戳添加到 Redis Sorted Set
    $redis = new Redis();
    $redis->connect('127.0.0.1', 6379);
    $redis->zadd('user_notifications:' . $userId, time(), $content);

    return true;
}

🔎3.3 获取用户通知列表

用户可以通过接口或页面查看自己的通知列表。我们可以从数据库中查询用户的通知记录，并按照发送时间倒序排列。

使用以下代码实现获取用户通知列表的函数：

function getNotificationList($userId) {
    $pdo = new PDO('mysql:host=localhost;dbname=test', 'username', 'password');
    $stmt = $pdo->prepare("SELECT * FROM user_notifications WHERE user_id = ? ORDER BY created_at DESC");
    $stmt->execute([$userId]);

    $notifications = $stmt->fetchAll(PDO::FETCH_ASSOC);

    return $notifications;
}

🔎3.4 清理过期通知记录

为了避免 Redis Sorted Set 中存储的用户通知记录过多，我们可以定时清理过期的通知记录。通过设置 Redis 的过期时间来实现自动清理。例如，我们可以设置 Sorted Set 的过期时间为2天，在用户查询通知列表时，先删除过期的通知记录，再返回有效的通知列表。

function cleanExpiredNotifications($userId) {
    $redis = new Redis();
    $redis->connect('127.0.0.1', 6379);

    // 设置 Sorted Set 的过期时间为2天
    $expireTime = strtotime('2 days ago');
    $redis->expireAt('user_notifications:' . $userId, $expireTime);
}

🔎3.5 定时任务

为了每天凌晨清理用户的通知记录，我们可以使用 Linux 的 crontab 来定时执行清理任务。编辑 crontab 文件，添加如下代码：

0 0 * * * php /path/to/clean_expired_notifications.php

并创建 clean_expired_notifications.php 文件，内容如下：

<?php

require_once 'redis.php';

$pdo = new PDO('mysql:host=localhost;dbname=test', 'username', 'password');
$stmt = $pdo->prepare("SELECT DISTINCT user_id FROM user_notifications");
$stmt->execute();

$userIds = $stmt->fetchAll(PDO::FETCH_COLUMN);

foreach ($userIds as $userId) {
    cleanExpiredNotifications($userId);
}

🚀四、结语

通过 PHP 和 Redis 实现用户消息推送每天最多通知2次的功能，并结合定时任务清理过期通知记录，可以有效地避免过多地打扰用户。以上是基本实现逻辑和代码示例，你可以根据自己的实际需求进行修改和扩展，例如根据不同用户设置不同的通知限制次数等。

推荐您阅读本专栏其他内容，PHP进阶实战教程，相信不会让您失望。如果你对上面的功能有疑问，随时欢迎与我交流。