perf: Cache reference lookups for subschemas

Author: Stranger6667

CHANGELOG.rst

@@ -3,6 +3,7 @@ v4.3.0
 
 * Fix undesired fallback to brute force container uniqueness check on
   certain input types (#893)
+* Cache reference lookups for subschemas (#893)
 * Implement a PEP544 Protocol for validator classes (#890)
 
 v4.2.1

jsonschema/validators.py

@@ -767,22 +767,32 @@ def _finditem(self, schema, key):
                 yield each
             values.extendleft(each.values())
 
-    def resolve(self, ref):
-        """
-        Resolve the given reference.
-        """
-        url = self._urljoin_cache(self.resolution_scope, ref).rstrip("/")
+    @lru_cache()
+    def _find_subschemas(self):
+        return list(self._finditem(self.referrer, "$id"))
 
+    @lru_cache()
+    def _find_in_subschemas(self, url):
         uri, fragment = urldefrag(url)
-
-        for subschema in self._finditem(self.referrer, "$id"):
+        for subschema in self._find_subschemas():
             target_uri = self._urljoin_cache(
                 self.resolution_scope, subschema["$id"],
             )
             if target_uri.rstrip("/") == uri.rstrip("/"):
                 if fragment:
                     subschema = self.resolve_fragment(subschema, fragment)
                 return url, subschema
+        return None
+
+    def resolve(self, ref):
+        """
+        Resolve the given reference.
+        """
+        url = self._urljoin_cache(self.resolution_scope, ref).rstrip("/")
+
+        match = self._find_in_subschemas(url)
+        if match is not None:
+            return match
 
         return url, self._remote_cache(url)