You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
1.7 KiB
67 lines
1.7 KiB
6 years ago
|
from .core import Interface
|
||
|
|
from threading import Lock
|
||
|
|
|
||
|
|
|
||
|
|
class Dict(Interface):
|
||
|
|
def __init__(self):
|
||
|
|
self.lock = Lock()
|
||
|
|
self.data = dict()
|
||
|
|
Interface.__init__(self)
|
||
|
|
|
||
|
|
def __getstate__(self):
|
||
|
|
return {'data': self.data}
|
||
|
|
|
||
|
|
def __setstate__(self, state):
|
||
|
|
Interface.__setstate__(self, state)
|
||
|
|
Dict.__init__(self)
|
||
|
|
self.data = state['data']
|
||
|
|
|
||
|
|
def append(self, data, lock=True, **kwargs):
|
||
|
|
if lock: self.lock.acquire()
|
||
|
|
try:
|
||
|
|
for k, v in data.items():
|
||
|
|
if k not in self.data:
|
||
|
|
self.data[k] = []
|
||
|
|
self.data[k].append(v)
|
||
|
|
finally:
|
||
|
|
if lock: self.lock.release()
|
||
|
|
|
||
|
|
def _get(self, keys, lock=True, **kwargs):
|
||
|
|
assert isinstance(keys, (list, tuple, set))
|
||
|
|
if lock:
|
||
|
|
self.lock.acquire()
|
||
|
|
try:
|
||
|
|
result = [b''.join(self.data.get(key, [])) for key in keys]
|
||
|
|
finally:
|
||
|
|
if lock:
|
||
|
|
self.lock.release()
|
||
|
|
return result
|
||
|
|
|
||
|
|
def _iset(self, key, value, lock=True):
|
||
|
|
""" Idempotent set """
|
||
|
|
if lock:
|
||
|
|
self.lock.acquire()
|
||
|
|
try:
|
||
|
|
self.data[key] = [value]
|
||
|
|
finally:
|
||
|
|
if lock:
|
||
|
|
self.lock.release()
|
||
|
|
|
||
|
|
def _delete(self, keys, lock=True):
|
||
|
|
if lock:
|
||
|
|
self.lock.acquire()
|
||
|
|
try:
|
||
|
|
for key in keys:
|
||
|
|
if key in self.data:
|
||
|
|
del self.data[key]
|
||
|
|
finally:
|
||
|
|
if lock:
|
||
|
|
self.lock.release()
|
||
|
|
|
||
|
|
def drop(self):
|
||
|
|
self._iset_seen.clear()
|
||
|
|
self.data.clear()
|
||
|
|
|
||
|
|
def __exit__(self, *args):
|
||
|
|
self.drop()
|