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.
103 lines
3.0 KiB
103 lines
3.0 KiB
6 years ago
|
# coding=utf-8
|
||
|
|
|
||
5 years ago
|
from __future__ import absolute_import
|
||
|
6 years ago
|
import datetime
|
||
|
|
import threading
|
||
|
|
|
||
|
|
lock = threading.Lock()
|
||
|
|
|
||
|
|
|
||
|
|
class TempIntent(object):
|
||
|
|
timeout = 1000 # milliseconds
|
||
|
|
store = None
|
||
|
|
|
||
|
|
def __init__(self, timeout=1000, store=None):
|
||
|
|
self.timeout = timeout
|
||
|
|
if store is None:
|
||
|
|
raise NotImplementedError
|
||
|
|
|
||
|
|
self.store = store
|
||
|
|
|
||
|
|
def get(self, kind, *keys):
|
||
|
|
with lock:
|
||
|
|
# iter all requested keys
|
||
|
|
for key in keys:
|
||
|
|
hit = False
|
||
|
|
|
||
|
|
# skip key if invalid
|
||
|
|
if not key:
|
||
|
|
continue
|
||
|
|
|
||
|
|
# valid kind?
|
||
|
|
if kind in self.store:
|
||
|
|
now = datetime.datetime.now()
|
||
|
|
key = str(key)
|
||
|
|
|
||
|
|
# iter all known kinds (previously created)
|
||
|
|
for known_key in self.store[kind].keys():
|
||
|
|
# may need locking, for now just play it safe
|
||
|
|
data = self.store[kind].get(known_key, {})
|
||
|
|
ends = data.get("timeout")
|
||
|
|
if not ends:
|
||
|
|
continue
|
||
|
|
|
||
|
|
timed_out = False
|
||
|
|
if now > ends:
|
||
|
|
timed_out = True
|
||
|
|
|
||
|
|
# key and kind in storage, and not timed out = hit
|
||
|
|
if known_key == key and not timed_out:
|
||
|
|
hit = True
|
||
|
|
|
||
|
|
if timed_out:
|
||
|
|
try:
|
||
|
|
del self.store[kind][key]
|
||
|
|
except:
|
||
|
|
continue
|
||
|
|
|
||
|
|
if hit:
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
def resolve(self, kind, key):
|
||
|
|
with lock:
|
||
|
|
if kind in self.store and key in self.store[kind]:
|
||
|
|
del self.store[kind][key]
|
||
|
|
return True
|
||
|
|
return False
|
||
|
|
|
||
|
|
def set(self, kind, key, data=None, timeout=None):
|
||
|
|
with lock:
|
||
|
|
if kind not in self.store:
|
||
|
|
self.store[kind] = {}
|
||
|
|
|
||
|
|
key = str(key)
|
||
|
|
self.store[kind][key] = {
|
||
|
|
"data": data,
|
||
|
|
"timeout": datetime.datetime.now() + datetime.timedelta(milliseconds=timeout or self.timeout)
|
||
|
|
}
|
||
|
|
|
||
|
|
def has(self, kind, key):
|
||
|
|
with lock:
|
||
|
|
if kind not in self.store:
|
||
|
|
return False
|
||
|
|
return key in self.store[kind]
|
||
|
|
|
||
|
|
def cleanup(self):
|
||
|
|
now = datetime.datetime.now()
|
||
|
|
clear_all = False
|
||
|
|
for kind, info in self.store.items():
|
||
|
|
for key, intent_data in info.items():
|
||
|
|
# legacy intent data, clear everything
|
||
|
|
if not isinstance(intent_data, dict):
|
||
|
|
clear_all = True
|
||
|
|
continue
|
||
|
|
|
||
|
|
if now > intent_data["timeout"]:
|
||
|
|
del self.store[kind][key]
|
||
|
|
if clear_all:
|
||
|
|
self.store.clear()
|
||
|
|
|
||
|
|
self.store.save()
|
||
|
|
|