redis 锁 详解redis中的锁以及使用场景

# Multi 命令用于标记一个事务块的开始事务块内的多条命令会按照先后顺序被放进一个队列当中，最后由 EXEC 命令原子性( atomic )地执行
> multi（开始一个redis事物）
incr books
incr books
> exec （执行事物）
> discard （丢弃事物）

> setnx lock:codehole true
.... do something critical ....
> del lock:codehole

> setnx lock:codehole true
> expire lock:codehole 5
.... do something critical ....
> del lock:codehole

> set lock:codehole true ex 5 nx
''' do something '''
> del lock:codehole

#! /usr/bin/env python
# -*- coding: utf-8 -*-
import redis
def sale(rs):
  while True:
    with rs.pipeline() as p:
      try:
        p.watch('apple')          # 监听key值为apple的数据数量改变
        count = int(rs.get('apple'))
        print('拿取到了苹果的数量: %d' % count)
        p.multi()             # 事务开始
        if count> 0 :           # 如果此时还有库存
          p.set('apple', count - 1)
          p.execute()          # 执行事务
        p.unwatch()
        break               # 当库存成功减一或没有库存时跳出执行循环
      except Exception as e:         # 当出现watch监听值出现修改时，WatchError异常抛出
        print('[Error]: %s' % e)
        continue              # 继续尝试执行

rs = redis.Redis(host='127.0.0.1', port=6379)   # 连接redis
rs.set('apple',1000)                # # 首先在redis中设置某商品apple 对应数量value值为1000
sale(rs)

import redis
import uuid
import time

# 1.初始化连接函数
def get_conn(host="127.0.0.1",port=6379):
  rs = redis.Redis(host=host, port=port)
  return rs

# 2. 构建redis锁
def acquire_lock(rs, lock_name, expire_time=10):
  '''
  rs: 连接对象
  lock_name: 锁标识
  acquire_time: 过期超时时间
  return -> False 获锁失败 or True 获锁成功
  '''
  identifier = str(uuid.uuid4())
  end = time.time() + expire_time
  while time.time() < end:
    # 当获取锁的行为超过有效时间，则退出循环，本次取锁失败，返回False
    if rs.setnx(lock_name, identifier): # 尝试取得锁
      return identifier
    # time.sleep(.001)
    return False

# 3. 释放锁
def release_lock(rs, lockname, identifier):
  '''
  rs: 连接对象
  lockname: 锁标识
  identifier: 锁的value值，用来校验
  '''


  if rs.get(lockname).decode() == identifier: # 防止其他进程同名锁被误删
    rs.delete(lockname)
    return True      # 删除锁
  else:
    return False      # 删除失败

#有过期时间的锁
def acquire_expire_lock(rs, lock_name, expire_time=10, locked_time=10):
  '''
  rs: 连接对象
  lock_name: 锁标识
  acquire_time: 过期超时时间
  locked_time: 锁的有效时间
  return -> False 获锁失败 or True 获锁成功
  '''
  identifier = str(uuid.uuid4())
  end = time.time() + expire_time
  while time.time() < end:
    # 当获取锁的行为超过有效时间，则退出循环，本次取锁失败，返回False
    if rs.setnx(lock_name, identifier): # 尝试取得锁
      # print('锁已设置: %s' % identifier)
      rs.expire(lock_name, locked_time)
      return identifier
    time.sleep(.001)
  return False


'''在业务函数中使用上面的锁'''
def sale(rs):
  start = time.time()      # 程序启动时间
  with rs.pipeline() as p:
    '''
    通过管道方式进行连接
    多条命令执行结束，一次性获取结果
    '''

    while 1:
      lock = acquire_lock(rs, 'lock')
      if not lock: # 持锁失败
        continue

      #开始监测"lock"
      p.watch("lock")
      try:
        #开启事务
        p.multi()
        count = int(rs.get('apple')) # 取量
        p.set('apple', count-1)   # 减量
        # time.sleep(5)
        
        #提交事务
        p.execute()
        print('当前库存量: %s' % count)
        #成功则跳出循环
        break
      except:
        #事务失败对应处理
        print("修改数据失败")

      #无论成功与否最终都需要释放锁
      finally:

        res = release_lock(rs, 'lock', lock)
        #释放锁成功，
        if res:
          print("删除锁成功")
        #释放锁失败，强制删除
        else:
          print("删除锁失败,强制删除锁")
          res = rs.delete('lock')
          print(res)

    print('[time]: %.2f' % (time.time() - start))

rs = redis.Redis(host='127.0.0.1', port=6379)   # 连接redis
# rs.set('apple',1000)               # # 首先在redis中设置某商品apple 对应数量value值为1000
sale(rs)

def acquire_expire_lock(rs, lock_name, expire_time=10, locked_time=10):
  '''
  rs: 连接对象
  lock_name: 锁标识
  acquire_time: 过期超时时间
  locked_time: 锁的有效时间
  return -> False 获锁失败 or True 获锁成功
  '''
  identifier = str(uuid.uuid4())
  end = time.time() + expire_time
  while time.time() < end:
    # 当获取锁的行为超过有效时间，则退出循环，本次取锁失败，返回False
    if rs.setnx(lock_name, identifier): # 尝试取得锁
      # print('锁已设置: %s' % identifier)
      rs.expire(lock_name, locked_time)
      return identifier
    time.sleep(.001)
  return False