jandieman/searxng
2
0
Fork 0
mirror of https://github.com/searxng/searxng.git synced 2025-12-23 12:10:00 +00:00
Code Issues Projects Releases Wiki Activity

add rate limiting per engine

Browse Source
This commit is contained in:
Marc Abonce Seguin
2022-04-24 23:52:13 -05:00
parent 666cd1f635
commit 85eca4fb22
8 changed files with 90 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
searx/shared/shared_simple.py
View File

@@ -16,14 +16,28 @@ class SimpleSharedDict(shared_abstract.SharedDict):
def get_int(self, key: str) -> Optional[int]:
return self.d.get(key, None)
def set_int(self, key: str, value: int):
def set_int(self, key: str, value: int, expire: Optional[int] = None):
self.d[key] = value
if expire:
self._expire(key, expire)
def get_str(self, key: str) -> Optional[str]:
return self.d.get(key, None)
def set_str(self, key: str, value: str):
def set_str(self, key: str, value: str, expire: Optional[int] = None):
self.d[key] = value
if expire:
self._expire(key, expire)
def _expire(self, key: str, expire: int):
t = threading.Timer(expire, lambda k, d: d.pop(k), args=[key, self.d])
t.daemon = True
t.start()
def run_locked(func, *args):
# SimpleSharedDict is not actually shared, so no locking needed
return func(*args)
def schedule(delay, func, *args):