资讯专栏INFORMATION COLUMN

php+redis实现抢购功能

BetaRabbit / 3459人阅读

摘要:实现思路实现分布式锁思路思路很简单,主要用到的函数是,这个应该是实现分布式锁最主要的函数。实现任务队列这里的实现会用到上面的分布式的锁机制,主要是用到了里的有序集合这一数据结构。

实现思路
1.Redis实现分布式锁思路
   思路很简单,主要用到的redis函数是setnx(),这个应该是实现分布式锁最主要的函数。首先是将某一任务标识名(这里用Lock:order作为标识名的例子)作为键存到redis里,并为其设个过期时间,如果是还有Lock:order请求过来,先是通过setnx()看看是否能将Lock:order插入到redis里,可以的话就返回true,不可以就返回false。当然,在我的代码里会比这个思路复杂一些,我会在分析代码时进一步说明。

2.Redis实现任务队列
  这里的实现会用到上面的Redis分布式的锁机制,主要是用到了Redis里的有序集合这一数据结构。例如入队时,通过zset的add()函数进行入队,而出队时,可以用到zset的getScore()函数。另外还可以弹出顶部的几个任务。

  以上就是实现 分布式锁 和 任务队列 的简单思路,如果你看完有点模棱两可,那请看接下来的代码实现。

代码分析
(一)先来分析Redis分布式锁的代码实现

    (1)为避免特殊原因导致锁无法释放,在加锁成功后,锁会被赋予一个生存时间(通过lock方法的参数设置或者使用默认值),超出生存时间锁会被自动释放锁的生存时间默认比较短(秒级),因此,若需要长时间加锁,可以通过expire方法延长锁的生存时间为适当时间,比如在循环内。

    (2)系统级的锁当进程无论何种原因时出现crash时,操作系统会自己回收锁,所以不会出现资源丢失,但分布式锁不用,若一次性设置很长时间,一旦由于各种原因出现进程crash 或者其他异常导致unlock未被调用时,则该锁在剩下的时间就会变成垃圾锁,导致其他进程或者进程重启后无法进入加锁区域。

    先看加锁的实现代码:这里需要主要两个参数,一个是$timeout,这个是循环获取锁的等待时间,在这个时间内会一直尝试获取锁知道超时,如果为0,则表示获取锁失败后直接返回而不再等待;另一个重要参数的$expire,这个参数指当前锁的最大生存时间,以秒为单位的,它必须大于0,如果超过生存时间锁仍未被释放,则系统会自动强制释放。这个参数的最要作用请看上面的(1)里的解释。

    这里先取得当前时间,然后再获取到锁失败时的等待超时的时刻(是个时间戳),再获取到锁的最大生存时刻是多少。这里redis的key用这种格式:”Lock:锁的标识名”,这里就开始进入循环了,先是插入数据到redis里,使用setnx()函数,这函数的意思是,如果该键不存在则插入数据,将最大生存时刻作为值存储,假如插入成功,则对该键进行失效时间的设置,并将该键放在$lockedName数组里,返回true,也就是上锁成功;如果该键存在,则不会插入操作了,这里有一步严谨的操作,那就是取得当前键的剩余时间,假如这个时间小于0,表示key上没有设置生存时间(key是不会不存在的,因为前面setnx会自动创建)如果出现这种状况,那就是进程的某个实例setnx成功后 crash 导致紧跟着的expire没有被调用,这时可以直接设置expire并把锁纳为己用。如果没设置锁失败的等待时间 或者 已超过最大等待时间了,那就退出循环,反之则 隔 $waitIntervalUs 后继续 请求。 这就是加锁的整一个代码分析。
    

</>复制代码

  1. /**
  2. * 加锁
  3. * @param [type] $name 锁的标识名
  4. * @param integer $timeout 循环获取锁的等待超时时间,在此时间内会一直尝试获取锁直到超时,为0表示失败后直接返回不等待
  5. * @param integer $expire 当前锁的最大生存时间(秒),必须大于0,如果超过生存时间锁仍未被释放,则系统会自动强制释放
  6. * @param integer $waitIntervalUs 获取锁失败后挂起再试的时间间隔(微秒)
  7. * @return [type] [description]
  8. */
  9. public function lock($name, $timeout = 0, $expire = 15, $waitIntervalUs = 100000) {
  10. if ($name == null) return false;
  11. //取得当前时间
  12. $now = time();
  13. //获取锁失败时的等待超时时刻
  14. $timeoutAt = $now + $timeout;
  15. //锁的最大生存时刻
  16. $expireAt = $now + $expire;
  17. $redisKey = "Lock:{$name}";
  18. while (true) {
  19. //将rediskey的最大生存时刻存到redis里,过了这个时刻该锁会被自动释放
  20. $result = $this->redisString->setnx($redisKey, $expireAt);
  21. if ($result != false) {
  22. //设置key的失效时间
  23. $this->redisString->expire($redisKey, $expireAt);
  24. //将锁标志放到lockedNames数组里
  25. $this->lockedNames[$name] = $expireAt;
  26. return true;
  27. }
  28. //以秒为单位,返回给定key的剩余生存时间
  29. $ttl = $this->redisString->ttl($redisKey);
  30. //ttl小于0 表示key上没有设置生存时间(key是不会不存在的,因为前面setnx会自动创建)
  31. //如果出现这种状况,那就是进程的某个实例setnx成功后 crash 导致紧跟着的expire没有被调用
  32. //这时可以直接设置expire并把锁纳为己用
  33. if ($ttl < 0) {
  34. $this->redisString->set($redisKey, $expireAt);
  35. $this->lockedNames[$name] = $expireAt;
  36. return true;
  37. }
  38. /*****循环请求锁部分*****/
  39. //如果没设置锁失败的等待时间 或者 已超过最大等待时间了,那就退出
  40. if ($timeout <= 0 || $timeoutAt < microtime(true)) break;
  41. //隔 $waitIntervalUs 后继续 请求
  42. usleep($waitIntervalUs);
  43. }
  44. return false;
  45. }

接着看解锁的代码分析:解锁就简单多了,传入参数就是锁标识,先是判断是否存在该锁,存在的话,就从redis里面通过deleteKey()函数删除掉锁标识即可。

</>复制代码

  1. /**
  2. * 解锁
  3. * @param [type] $name [description]
  4. * @return [type] [description]
  5. */
  6. public function unlock($name) {
  7. //先判断是否存在此锁
  8. if ($this->isLocking($name)) {
  9. //删除锁
  10. if ($this->redisString->deleteKey("Lock:$name")) {
  11. //清掉lockedNames里的锁标志
  12. unset($this->lockedNames[$name]);
  13. return true;
  14. }
  15. }
  16. return false;
  17. }

再贴上删除掉所有锁的方法,其实都一个样,多了个循环遍历而已。

</>复制代码

  1. /**
  2. * 释放当前所有获得的锁
  3. * @return [type] [description]
  4. */
  5. public function unlockAll() {
  6. //此标志是用来标志是否释放所有锁成功
  7. $allSuccess = true;
  8. foreach ($this->lockedNames as $name => $expireAt) {
  9. if (false === $this->unlock($name)) {
  10. $allSuccess = false;
  11. }
  12. }
  13. return $allSuccess;
  14. }

以上就是用Redis实现分布式锁的整一套思路和代码实现的总结和分享,这里我附上正一个实现类的代码,代码里我基本上对每一行进行了注释,方便大家快速看懂并且能模拟应用。想要深入了解的请看整个类的代码:

</>复制代码

  1. /**
  2. *在redis上实现分布式锁
  3. */
  4. class RedisLock {
  5. private $redisString;
  6. private $lockedNames = [];
  7. public function __construct($param = NULL) {
  8. $this->redisString = RedisFactory::get($param)->string;
  9. }
  10. /**
  11. * 加锁
  12. * @param [type] $name 锁的标识名
  13. * @param integer $timeout 循环获取锁的等待超时时间,在此时间内会一直尝试获取锁直到超时,为0表示失败后直接返回不等待
  14. * @param integer $expire 当前锁的最大生存时间(秒),必须大于0,如果超过生存时间锁仍未被释放,则系统会自动强制释放
  15. * @param integer $waitIntervalUs 获取锁失败后挂起再试的时间间隔(微秒)
  16. * @return [type] [description]
  17. */
  18. public function lock($name, $timeout = 0, $expire = 15, $waitIntervalUs = 100000) {
  19. if ($name == null) return false;
  20. //取得当前时间
  21. $now = time();
  22. //获取锁失败时的等待超时时刻
  23. $timeoutAt = $now + $timeout;
  24. //锁的最大生存时刻
  25. $expireAt = $now + $expire;
  26. $redisKey = "Lock:{$name}";
  27. while (true) {
  28. //将rediskey的最大生存时刻存到redis里,过了这个时刻该锁会被自动释放
  29. $result = $this->redisString->setnx($redisKey, $expireAt);
  30. if ($result != false) {
  31. //设置key的失效时间
  32. $this->redisString->expire($redisKey, $expireAt);
  33. //将锁标志放到lockedNames数组里
  34. $this->lockedNames[$name] = $expireAt;
  35. return true;
  36. }
  37. //以秒为单位,返回给定key的剩余生存时间
  38. $ttl = $this->redisString->ttl($redisKey);
  39. //ttl小于0 表示key上没有设置生存时间(key是不会不存在的,因为前面setnx会自动创建)
  40. //如果出现这种状况,那就是进程的某个实例setnx成功后 crash 导致紧跟着的expire没有被调用
  41. //这时可以直接设置expire并把锁纳为己用
  42. if ($ttl < 0) {
  43. $this->redisString->set($redisKey, $expireAt);
  44. $this->lockedNames[$name] = $expireAt;
  45. return true;
  46. }
  47. /*****循环请求锁部分*****/
  48. //如果没设置锁失败的等待时间 或者 已超过最大等待时间了,那就退出
  49. if ($timeout <= 0 || $timeoutAt < microtime(true)) break;
  50. //隔 $waitIntervalUs 后继续 请求
  51. usleep($waitIntervalUs);
  52. }
  53. return false;
  54. }
  55. /**
  56. * 解锁
  57. * @param [type] $name [description]
  58. * @return [type] [description]
  59. */
  60. public function unlock($name) {
  61. //先判断是否存在此锁
  62. if ($this->isLocking($name)) {
  63. //删除锁
  64. if ($this->redisString->deleteKey("Lock:$name")) {
  65. //清掉lockedNames里的锁标志
  66. unset($this->lockedNames[$name]);
  67. return true;
  68. }
  69. }
  70. return false;
  71. }
  72. /**
  73. * 释放当前所有获得的锁
  74. * @return [type] [description]
  75. */
  76. public function unlockAll() {
  77. //此标志是用来标志是否释放所有锁成功
  78. $allSuccess = true;
  79. foreach ($this->lockedNames as $name => $expireAt) {
  80. if (false === $this->unlock($name)) {
  81. $allSuccess = false;
  82. }
  83. }
  84. return $allSuccess;
  85. }
  86. /**
  87. * 给当前所增加指定生存时间,必须大于0
  88. * @param [type] $name [description]
  89. * @return [type] [description]
  90. */
  91. public function expire($name, $expire) {
  92. //先判断是否存在该锁
  93. if ($this->isLocking($name)) {
  94. //所指定的生存时间必须大于0
  95. $expire = max($expire, 1);
  96. //增加锁生存时间
  97. if ($this->redisString->expire("Lock:$name", $expire)) {
  98. return true;
  99. }
  100. }
  101. return false;
  102. }
  103. /**
  104. * 判断当前是否拥有指定名字的所
  105. * @param [type] $name [description]
  106. * @return boolean [description]
  107. */
  108. public function isLocking($name) {
  109. //先看lonkedName[$name]是否存在该锁标志名
  110. if (isset($this->lockedNames[$name])) {
  111. //从redis返回该锁的生存时间
  112. return (string)$this->lockedNames[$name] = (string)$this->redisString->get("Lock:$name");
  113. }
  114. return false;
  115. }
  116. }

文章版权归作者所有,未经允许请勿转载,若此文章存在违规行为,您可以联系管理员删除。

转载请注明本文地址:https://www.ucloud.cn/yun/22413.html

相关文章

  • SegmentFault 技术周刊 Vol.37 - 分布式缓存利器:Redis

    摘要:持久化到中反向代理的负载均衡基于的集群搭建如何实现从中订阅消息转发到客户端的扩展是阻塞式,使用订阅发布模式时,会导致整个进程进入阻塞。缓存是用于解决高并发场景下系统的性能及稳定性问题的银弹。 showImg(https://segmentfault.com/img/bVYE6k?w=900&h=385); Redis 是由意大利程序员 Salvatore Sanfilippo(昵称:a...

    binaryTree 评论0 收藏0
  • 秒杀系统优化方案之缓存、队列、锁设计思路

    摘要:一为什么难秒杀系统难做的原因库存只有一份,所有人会在集中的时间读和写这些数据。又例如抢票,亦与秒杀类似,瞬时流量更甚。 一、为什么难     秒杀系统难做的原因:库存只有一份,所有人会在集中的时间读和写这些数据。例如小米手机每周二的秒杀,可能手机只有1万部,但瞬时进入的流量可能是几百几千万。又例如12306抢票,亦与秒杀类似,瞬时流量更甚。 主要需要解决的问题有两个: 高并发对数据库...

    dinfer 评论0 收藏0

发表评论

0条评论

最新活动
阅读需要支付1元查看
<