WEB服务,尤其是分布式的WEB服务,通常会面临需要实现安全高效的全局锁的服务场景。而redis的诸多特性,使得借助其实现的全局锁显得如此简单。本篇文章借助小明应聘魔笛手研发岗位的面试,层层递进的来介绍一下如何通过redis实现全局锁。
面试官老张:看你简历上说,之前的项目中使用了redis,你知道redis为什么是单线程的吗?
小明:多进程或者多线程更多的为了高效利用CPU,但是却对开发人员提出了更高的要求。而redis作为内存型数据库,其性能瓶颈往往不是CPU,而是内存和网络。单线程对于redis来说已经足够快了,也就没有必要再引入复杂的多任务模型。
注:官方文档有提到单线程的redis就可以达到10W的QPS。
面试官老张:不错,我再问一个简单的问题,通过什么命令为一个key指定字符串的值。
小明:set命令。
面试官老张:恩,这个问题有点太简单了哈,那你能介绍一下set命令都可以接收什么参数吗?
小明:这个我有查阅过,redis的set命令除了指定key和value,还有几个可选参数,分别是:
- ex——设置key多少秒之后过期
- px——设置key多少毫秒之后过期
- nx——只有key不存在时,set操作才会生效
- xx——只有key存在时,set操作才会生效
面试官老张:很好,那你可以利用redis实现一把全局锁吗?
小明:那我通过python代码来简单实现一下吧。
from redis import Redis
LOCK_KEY = "global_lock"
AUTO_RELEASE_SECOND = 100
redis = Redis()
def acquire_lock():
if redis.set(LOCK_KEY, "1", nx=True, ex=AUTO_RELEASE_SECOND):
return True
return False
def release_lock():
redis.delete(LOCK_KEY)
利用nx参数,可以保证只有key不存在时才会设置生效,这样就并发操作中只有一个实例可以获取生效状态。
通过ex设置过期时间,可以保证获取锁的实例不会因为崩溃或者忘记释放,导致所有实例都无法获取锁。
注:觉得不过瘾的同学可以看一下redis-py中关于Lock的实现。
面试官老张:非常好。今天的面试就到这吧。