Merge pull request #4 from akquinet/caching

Caching

powerdns_api_proxy/config.py

@@ -35,6 +35,7 @@ def load_config(path: Optional[Path] = None) -> ProxyConfig:
     return config
 
 
+@lru_cache(maxsize=1000)
 def token_defined(config: ProxyConfig, token: str) -> bool:
     sha512 = hashlib.sha512()
     sha512.update(token.encode())
@@ -57,6 +58,7 @@ def dependency_check_token_defined(
     check_token_defined(load_config(), X_API_Key)
 
 
+@lru_cache(maxsize=1000)
 def get_environment_for_token(
     config: ProxyConfig, token: str
 ) -> ProxyConfigEnvironment:

powerdns_api_proxy/models.py

@@ -1,3 +1,4 @@
+from functools import lru_cache
 from typing import TypedDict
 
 from fastapi import HTTPException
@@ -79,6 +80,16 @@ def __init__(self, **data):
                 # populate zones lookup
                 self._zones_lookup[zone.name] = zone
 
+    def __hash__(self):
+        return hash(
+            self.name
+            + self.token_sha512
+            + str(self.global_read_only)
+            + str(self.global_search)
+            + str(self.zones)
+        )
+
+    @lru_cache(maxsize=10000)
     def get_zone_if_allowed(self, zone: str) -> ProxyConfigZone:
         '''
         Returns the zone config for the given zone name
@@ -122,6 +133,14 @@ def api_token_defined(cls, v):
             raise ValueError('pdns_api_token must a non-empty string')
         return v
 
+    def __hash__(self):
+        return hash(
+            self.pdns_api_url
+            + self.pdns_api_token
+            + str(self.pdns_api_verify_ssl)
+            + str(self.environments)
+        )
+
 
 class ResponseAllowed(BaseModel):
     zones: list[ProxyConfigZone]