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

04-11 1296阅读

🏆作者简介，黑夜开发者，CSDN领军人物，全栈领域优质创作者✌，CSDN博客专家，阿里云社区专家博主，2023年6月CSDN上海赛道top4。

🏆数年电商行业从业经验，历任核心研发工程师，项目技术负责人。

🏆本文已收录于PHP专栏：PHP进阶实战教程。

🏆另有专栏PHP入门基础教程，希望各位大佬多多支持❤️。

🎉欢迎 👍点赞✍评论⭐收藏

文章目录

🚀一、背景
🚀二、准备工作
🚀三、实现逻辑
- 🔎3.1 查询用户的已发送通知数量
- 🔎3.2 发送通知
- 🔎3.3 获取用户通知列表
- 🔎3.4 清理过期通知记录
- 🔎3.5 定时任务
- 🚀四、结语
  🚀一、背景
  
  在开发应用程序中，经常需要向用户推送消息通知，但是为了避免过多的打扰用户，我们希望限制每天最多通知2次。本篇博文将介绍如何使用 PHP 和 Redis 实现这一功能。
  
  🚀二、准备工作
  
  首先，我们需要准备好数据库和 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 文件，内容如下：