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

在这里插入图片描述

🏆作者简介，黑夜开发者，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 文件，内容如下：

<?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);
}