基于etcd的分布式锁,简单说就是利用etcd.test_and_set函数来判断lock key是否被占用,存在那就说明有人占用。在创建key的时候加入了ttl,防止因为进程异常退出而没有释放锁。
更新: 支持指定renew时间
renew(ttl=3)
待修复的问题:
针对etcd key 加入watch机制,解决客户端意外退出没有释放锁的问题.
曾经写过关于分布式互斥锁的文章:
zookeeper: http://xiaorui.cc/2015/04/09/python-zookeeper解决redis做分布式锁带来的坑/
redis: http://xiaorui.cc/2014/12/19/python使用redis实现协同控制的分布式锁/
git clone https://github.com/rfyiamcool/pyetcdlock.git
cd pyetcdlock
python setup.py
pypi的安装方式 (话说,pypi有些问题,你在pypi search搜东西的时候,不显示详细信息)
pip instlal pyetcdlock
模块说明:
ttl : expire过期时间,这样可以指定锁定的时间
renew : 如果你的ttl时间将要过期了,那么可以使用renew(),再加点锁定的ttl时间
force_acquire : 强制获取锁
import etcd
from pyetcdnetlock import Lock
client = client = etcd.Client(host='127.0.0.1')
lock = Lock(client, 'path/to/my/key', ttl=30, renewSecondsPrior=5)
if lock.acquire(timeout=20):
# some work
lock.renew()
# some other work
lock.release()
else
# failed to acquire the lock in 20 seconds
pass
OR
with lock as l:
#如果时间不够,可以用renew()参数,加点时间
l.renew()
# some other work