redis实现锁 · php笔记

## redis 实现锁 ```php // 其他服务不在注册 register_shutdown_function 而是添加事件 $shutdownEvent = []; // 注册一个会在php中止时执行的函数 register_shutdown_function('shutdown_function'); function shutdown_function() { global $shutdownEvent; // 执行注册的事件 foreach ($shutdownEvent as $key => $f) { if (is_callable($f)) { $f(); } } $e = error_get_last(); if ($e) { $errorInfo = friendlyErrorType($e['type']); // print_r($e); switch ($e['type']) { case E_PARSE: case E_ERROR: case E_CORE_ERROR: case E_COMPILE_ERROR: case E_USER_ERROR: $errorStr = "appErrorHandler: type: {$e['type']} ({$errorInfo[0]} - {$errorInfo[1]}); message: {$e['message']}; file: {$e['file']}; line: {$e['line']}"; trace($errorStr, 'run_error'); trace('[shutdown_function:trace]', '', true); error(1, $errorStr); break; } } // 写入日志 trace('[shutdown_function:trace]', '', true); } // 使用 redis实现的简单无阻塞的锁（没有考虑复杂的情况） // https://www.cnblogs.com/fengff/p/10913492.html function lock($key, $ok, $gone) { global $redis, $shutdownEvent; $roomid = 'room_' . time() . rand(1, 9999); // 初始化一个空函数防止下面报错，任何环节，代码节点都有可能发生异常甚至不确定性故障 $delLock = function () { }; // trace('lock: ' . $key . PHP_EOL, '>debug-lock'); // exit; try { $isLockOk = $redis->setNX($key, $roomid); // var_dump($isLockOk);exit; if ($isLockOk) { // 获得锁 // echo PHP_EOL . 'lock ok!' . PHP_EOL . PHP_EOL; // 防止A进程获得锁后，由于A执行时间太长，锁被回收了 // 此时B获得锁，等A执行完毕，删除时删除了B持有锁，它以为删除的是自己的，因为它并不是到自己的锁被回收了 // 设置过期是因为防止任务挂了锁没被回收，这个问题不会有了，所以我们不设置过期 // 不能因为执行时间长的设置过期，不然还会有两个进行并发了 // 但是为了规范严谨，我们删除时还是严格检查 $delLock = function () use ($key, $redis, $roomid) { if ($redis->get($key) == $roomid) { $redis->del($key); trace('delLock: ' . $key . ' ' . $roomid . PHP_EOL, '>debug-lock'); // echo 'delLock: ' . $key . ' ' . $roomid . PHP_EOL; } }; // trace('lock: ok!', '>debug-lock'); // 能保存上下文 // 像js的闭包一样，赞！ $shutdownEvent['unLock'] = function () use ($key, $redis, $roomid, $delLock) { // 只有发生错误时，才使用事件删除key // ctrl + c 退出就没办法了（不算异常），除非能获取脚本推出code if (!error_get_last()) { return; } // 获取了锁的脚本，脚本退出时删也没事 // ctrl + c 退出 register_shutdown_function 不会执行 // 这种情况那就只有使用信号了 trace('shutdownEvent unLock: ' . $key . ' ' . $roomid . PHP_EOL, '>debug-lock'); // echo 'shutdownEvent unLock: ' . $key . ' ' . $roomid . PHP_EOL; // $redis 连接也可能失效，删除有很多种失败的情况 // 只能尽量将最容易失败的地方处理好 $delLock(); }; // 注意: 注册事件这段代码必须放在这里: 获得锁之后 - 执行代码之前 , 不可预料的出错主要是在 $ok() 这里 // trace('lock: ok -> shutdownEvent', '>debug-lock'); $res = $ok(); // 如果任务发生错误崩了，这下面不会执行，但是没关系 shutdownEvent 事件还会执行 $delLock(); // unset($shutdownEvent['unLock']); // 如果没异常，delLock() 顺利执行了，就不用继续执行事件了 } else { // 当前已在运行，直接退出 // echo PHP_EOL . 'Be gone' . PHP_EOL . PHP_EOL; $res = $gone(); } } catch (\Exception $e) { $delLock(); throw $e; } // appErrorHandler: type: 1 (E_ERROR - 致命错误); message: Class 'sModel' not found; file: D:\wamp64\www\patent_management_system\include\rockFun.php; line: 564 // todo: 如果出现致命错误，会导致 key 没删掉 // https://www.php.net/manual/zh/function.register-shutdown-function.php // 是可以多次注册的，是追加不是覆盖，没有问题 // 只注册一次 // 这个无法实现只注册一次 // register_shutdown_function(); // 改用注册 shutdown 事件，框架调用 // 即使这样肯定也不能万无一失 return $res; } ``` ---- ### 用法： ```php // 使任务保持“单例”执行 lock('lock_' . $fun, function () use ($fun) { $fun(); }, function () use ($fun) { echo "crontab_run: {$fun} task running ..." . PHP_EOL; // 如果当前任务已经在执行了，就什么都不做 }); ``` ---- ### 待实现 - 可重入 - 阻塞等特性 ...