Redis 分布式锁

<?php


class RedisDistributedLock
{
    private $redisConfig; //Redis配置
    private $lockKey; //锁的键名
    private $acquireTimeout; //超时时间
    private $redis; //$reids实例
    private $isNegtive = true; //是否是悲观锁
    private $identify = ""; //锁的唯一标识,防止锁被别的进程误删

    public function __construct($redisConfig, $lockKey, $acquireTimeout = 5, $isNegtive = true)
    {
        if (!$redisConfig || !is_array($redisConfig)) {
            throw new \Exception("redis配置错误");
        }
        if (!$lockKey) {
            throw new \Exception("lockKey不能为空");
        }

        try {
            $this->redisConfig = $redisConfig;
            $this->lockKey = $lockKey;
            $this->isNegtive = $isNegtive;
            $this->acquireTimeout = $acquireTimeout;
            $this->redis = new \Redis();
            $this->redis->connect($redisConfig["host"], $redisConfig["port"]);
        } catch (\RedisException $e) {
            throw $e;
        }
    }

    /**
     * 获取锁
     */
    public function lock()
    {
        $v = uniqid();
        if ($this->isNegtive) { //悲观锁
            $endtime = microtime(true) * 1000 + $this->acquireTimeout * 1000;
            while (microtime(true) * 1000 < $endtime) { //每隔一段时间尝试获取一次锁
                $acquired = $this->redis->set($this->lockKey, $v, ["NX", "EX" => $this->acquireTimeout]);
                if ($acquired) { //获取锁成功,返回true
                    $this->identify = $v;
                    return true;
                }
                usleep(100);
            }
            return false; //获取锁超时,返回false
        } else { //乐观锁
            /**
             * 乐观锁只尝试一次,成功返回true,失败返回false
             */
            $acquired = $this->redis->set($this->lockKey, $v, ["NX", "EX" => $this->acquireTimeout]);
            if ($acquired) {
                $this->identify = $v;
                return true;
            }
            return false;
        }
    }

    /**
     * 释放锁
     */
    public function unlock()
    {
        if (!$this->identify) { //没有成功获得锁,直接返回false
            return false;
        }
        /**
         * 由于判断是否相等和删除是两步操作,因此使用 lua 脚本来保证原子性
         */
        $script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
        $result = $this->redis->eval($script, [$this->lockKey, $this->identify], 1);
        if ($result) {
            return true;
        }
        return false;
    }
}
php
本作品采用《CC 协议》,转载必须注明作者和本文链接
本帖由系统于 5个月前 自动加精
讨论数量: 4

写的很棒。我之前用redis分布式锁没有封装成对象使用。但是锁的唯一标识,防止锁被别的进程误删这一块对应的处理是$this->redis->eval($script, [$this->lockKey, $this->identify], 1)吗?这句话我没太理解,还请赐教!

5个月前 评论
pwake (楼主) 5个月前
easyer (作者) 5个月前
pwake (楼主) 5个月前

你好,请问怎么用,表示对 redis 不太熟悉

5个月前 评论
pwake (楼主) 5个月前

这个锁可以用来防止表单重复提交吗

5个月前 评论
Mr_Guo 4个月前
Laravel00

你好,可以把分布式锁的原理,如何用,怎么用,用到哪个地方一并写上,详细的做一下分析,这样比直接贴代码会好很多。因为也有很多人看不懂代码

2天前 评论

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