PHP+Redis解决实际问题一:订单限流

1、本系列文章每期都将解决一个Redis实际问题
2、每期问题将在每期的评论中选取
3、问题限Redis相关,其它问题如果本人感兴趣也不排除开辟新系列
4、本人常用PHP所以解决方案以PHP为主
5、评论里没有合适的提问时我会自己给自己出题

问题描述:

本期为第一期,所以只能自己出题了

如何用Redis给订单限流,如每M秒允许N个访问

解决方案:

<?php

    /**
     * 是否允许放行
     * @param string $key       redis键前缀
     * @param int $timeInterval 时间间隔(秒)
     * @param int $max          时间间隔内最大放行数
     * @return bool             是否放行
     * @throws Exception
     * @example
     * <pre>
     * //每秒放行一个
     * isAllow('my_allow');
     *
     * //每秒放行3个
     * isAllow('my_allow',1,3);
     *
     * //每3秒放行2个
     * isAllow('my_allow',3,2);
     */
    function isAllow(string $key, int $timeInterval=1, int $max=1):bool{
        if($timeInterval<1){
            throw new Exception('时间间隔必须大于0');
        }
        if($max<1){
            throw new Exception('最大放行数必须大于0');
        }

        $redis=new Redis();
        $redis->connect('192.168.31.187');
        if(!$redis->isConnected()){
            throw new Exception('Redis服务连接失败');
        }

        //对时间戳取模,使得每$timeInterval秒取得同一个时间戳
        $time=time();
        $key.=':'.($time-($time%$timeInterval));

        //自增并返回自增后的结果
        $index=$redis->incr($key);

        //如果是第一个访问,设置键的过期时间
        if($index===1){
            $redis->expire($key,$timeInterval+1);
        }

        return $index<$max+1;
    }

代码解读:

  1. 对时间取模,使得键名每$timeInterval秒更新一次
  2. incr()方法自增键的值,如果键不存在则先创建一个值为0的键再进行自增
  3. 根据自增原理,同键名下第N次自增返回的值就是N
  4. 键名每$timeInterval秒更新一次,所以在创建键的$timeInterval+1秒后键就不再有价值

以上,欢迎大家提问纠错补充优化

本作品采用《CC 协议》,转载必须注明作者和本文链接
讨论数量: 1

不错,测试了一下理解啦

4周前 评论

讨论应以学习和精进为目的。请勿发布不友善或者负能量的内容,与人为善,比聪明更重要!