30 lines
825 B
Python
30 lines
825 B
Python
|
import redis
|
||
|
|
|
||
|
|
import setting
|
||
|
|
|
||
|
|
|
||
|
|
class RedisDB:
|
||
|
|
|
||
|
|
DEFAULT_EXPIRE = 300
|
||
|
|
|
||
|
|
def __init__(self, host=setting.REDIS_HOST, port=setting.REDIS_PORT, base=setting.REDIS_BASE):
|
||
|
|
self.host = host
|
||
|
|
self.port = port
|
||
|
|
self.base = base
|
||
|
|
self._client = redis.StrictRedis(host=self.host, port=self.port, db=self.base, retry_on_timeout=True)
|
||
|
|
|
||
|
|
def save_start(self, module, expire=300):
|
||
|
|
lock_key = 'LOCK:{}'.format(module)
|
||
|
|
self._client.ping()
|
||
|
|
if self._client.get(lock_key) is not None:
|
||
|
|
return False
|
||
|
|
else:
|
||
|
|
self._client.set(lock_key, 1)
|
||
|
|
self._client.expire(lock_key, expire)
|
||
|
|
return True
|
||
|
|
|
||
|
|
def save_end(self, module):
|
||
|
|
lock_key = 'LOCK:{}'.format(module)
|
||
|
|
self._client.ping()
|
||
|
|
self._client.delete(lock_key)
|