Repository navigation
Expand file tree
/
Copy pathratelimit.py
More file actions
50 lines (40 loc) · 1.59 KB
/
Copy pathratelimit.py
File metadata and controls
50 lines (40 loc) · 1.59 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
"""Rate limiting: token-bucket per gli upstream esterni (Nominatim/Overpass/Photon ~1 req/s)
e limite di richieste per IP client per proteggere il proxy da abusi."""
import threading
import time
from collections import defaultdict, deque
class TokenBucket:
"""Limita il traffico verso un singolo host upstream a ~rate richieste al secondo."""
def __init__(self, rate_per_second=1.0):
self._interval = 1.0 / rate_per_second
self._lock = threading.Lock()
self._next_slot = 0.0
def wait_for_slot(self):
with self._lock:
now = time.monotonic()
start = max(now, self._next_slot)
self._next_slot = start + self._interval
delay = start - now
if delay > 0:
time.sleep(delay)
class ClientRateLimiter:
"""Limita le richieste per IP client (finestra scorrevole)."""
def __init__(self, max_requests=60, window_seconds=60):
self._max = max_requests
self._window = window_seconds
self._hits = defaultdict(deque)
self._lock = threading.Lock()
def allow(self, client_ip):
now = time.monotonic()
with self._lock:
q = self._hits[client_ip]
while q and now - q[0] > self._window:
q.popleft()
if len(q) >= self._max:
return False
q.append(now)
return True
nominatim_bucket = TokenBucket(rate_per_second=1.0)
overpass_bucket = TokenBucket(rate_per_second=1.0)
photon_bucket = TokenBucket(rate_per_second=1.0)
client_limiter = ClientRateLimiter(max_requests=60, window_seconds=60)