feat: Async retriever

Signed-off-by: Dmitry Dygalo <dmitry@dygalo.dev>

Author: Stranger6667

.github/workflows/ci.yml

@@ -51,7 +51,7 @@ jobs:
           cache-all-crates: "true"
           key: ${{ matrix.os }}
 
-      - run: cargo test --no-fail-fast
+      - run: cargo test --no-fail-fast --all-features
 
   test-wasm:
     name: Test on WASM

CHANGELOG.md

@@ -2,8 +2,22 @@
 
 ## [Unreleased]
 
+### Breaking Changes
+
+- The `Retrieve` trait's `retrieve` method now accepts URI references as `&Uri<String>` instead of `&Uri<&str>`. 
+  This aligns with the async version and simplifies internal URI handling. The behavior and available methods remain the same, this is purely a type-level change.
+
+  ```rust
+  // Before
+  fn retrieve(&self, uri: &Uri<&str>) -> Result<Value, Box<dyn std::error::Error + Send + Sync>>
+
+  // After
+  fn retrieve(&self, uri: &Uri<String>) -> Result<Value, Box<dyn std::error::Error + Send + Sync>>
+  ```
+
 ### Added
 
+- Support non-blocking retrieval for external resources during schema resolution via the new `resolve-async` feature. [#385](https://github.com/Stranger6667/jsonschema/issues/385)
 - Re-export `referencing::Registry` as `jsonschema::Registry`.
 - `ValidationOptions::with_registry` that allows for providing a predefined `referencing::Registry`. [#682](https://github.com/Stranger6667/jsonschema/issues/682)
 

README.md

@@ -52,7 +52,7 @@ See more usage examples in the [documentation](https://docs.rs/jsonschema).
 
 - 📚 Full support for popular JSON Schema drafts
 - 🔧 Custom keywords and format validators
-- 🌐 Remote reference fetching (network/file)
+- 🌐 Blocking & non-blocking remote reference fetching (network/file)
 - 🎨 `Basic` output style as per JSON Schema spec
 - ✨ Meta-schema validation for schema documents
 - 🔗 Bindings for [Python](https://github.com/Stranger6667/jsonschema/tree/master/crates/jsonschema-py)

crates/jsonschema-py/src/registry.rs

@@ -1,5 +1,3 @@
-use std::sync::Arc;
-
 use jsonschema::Resource;
 use pyo3::{exceptions::PyValueError, prelude::*};
 
@@ -29,7 +27,7 @@ impl Registry {
 
         if let Some(retriever) = retriever {
             let func = into_retriever(retriever)?;
-            options = options.retriever(Arc::new(Retriever { func }));
+            options = options.retriever(Retriever { func });
         }
 
         let pairs = resources.try_iter()?.map(|item| {
@@ -45,7 +43,7 @@ impl Registry {
         let pairs: Result<Vec<_>, PyErr> = pairs.collect();
 
         let registry = options
-            .try_from_resources(pairs?.into_iter())
+            .build(pairs?)
             .map_err(|e| PyValueError::new_err(e.to_string()))?;
 
         Ok(Registry { inner: registry })

crates/jsonschema-py/src/retriever.rs

@@ -10,7 +10,10 @@ pub(crate) struct Retriever<T: Fn(&str) -> PyResult<Value>> {
 }
 
 impl<T: Send + Sync + Fn(&str) -> PyResult<Value>> Retrieve for Retriever<T> {
-    fn retrieve(&self, uri: &Uri<&str>) -> Result<Value, Box<dyn std::error::Error + Send + Sync>> {
+    fn retrieve(
+        &self,
+        uri: &Uri<String>,
+    ) -> Result<Value, Box<dyn std::error::Error + Send + Sync>> {
         Ok((self.func)(uri.as_str())?)
     }
 }

crates/jsonschema-referencing/Cargo.toml

@@ -17,6 +17,13 @@ parking_lot = "0.12.3"
 percent-encoding = "2.3.1"
 serde_json.workspace = true
 
+async-trait = { version = "0.1.86", optional = true }
+futures = { version = "0.3.31", optional = true }
+
+[features]
+default = []
+retrieve-async = ["dep:async-trait", "dep:futures"]
+
 [lints]
 workspace = true
 
@@ -26,6 +33,7 @@ codspeed-criterion-compat = { version = "2.7", default-features = false }
 criterion = { version = "0.5", default-features = false }
 referencing_testsuite = { package = "jsonschema-referencing-testsuite", path = "../jsonschema-referencing-testsuite/" }
 test-case = "3.3.1"
+tokio = { version = "1", features = ["macros", "rt"] }
 
 [[bench]]
 harness = false

crates/jsonschema-referencing/src/anchors/mod.rs

@@ -243,14 +243,11 @@ mod tests {
             },
         }));
 
-        let registry = Registry::try_from_resources(
-            [
-                ("http://example.com".to_string(), root.clone()),
-                ("http://example.com/foo/".to_string(), true_resource),
-                ("http://example.com/foo/bar".to_string(), root.clone()),
-            ]
-            .into_iter(),
-        )
+        let registry = Registry::try_from_resources([
+            ("http://example.com".to_string(), root.clone()),
+            ("http://example.com/foo/".to_string(), true_resource),
+            ("http://example.com/foo/bar".to_string(), root.clone()),
+        ])
         .expect("Invalid resources");
         let resolver = registry
             .try_resolver("http://example.com")
@@ -287,14 +284,11 @@ mod tests {
             },
         }));
 
-        let registry = Registry::try_from_resources(
-            [
-                ("http://example.com".to_string(), two.clone()),
-                ("http://example.com/foo/".to_string(), one),
-                ("http://example.com/foo/bar".to_string(), two.clone()),
-            ]
-            .into_iter(),
-        )
+        let registry = Registry::try_from_resources([
+            ("http://example.com".to_string(), two.clone()),
+            ("http://example.com/foo/".to_string(), one),
+            ("http://example.com/foo/bar".to_string(), two.clone()),
+        ])
         .expect("Invalid resources");
         let resolver = registry
             .try_resolver("http://example.com")
@@ -397,14 +391,11 @@ mod tests {
             },
         }));
 
-        let registry = Registry::try_from_resources(
-            vec![
-                ("http://example.com".to_string(), root.clone()),
-                ("http://example.com/foo/".to_string(), true_resource),
-                ("http://example.com/foo/bar".to_string(), root.clone()),
-            ]
-            .into_iter(),
-        )
+        let registry = Registry::try_from_resources(vec![
+            ("http://example.com".to_string(), root.clone()),
+            ("http://example.com/foo/".to_string(), true_resource),
+            ("http://example.com/foo/bar".to_string(), root.clone()),
+        ])
         .expect("Invalid resources");
 
         let resolver = registry
@@ -442,14 +433,11 @@ mod tests {
         }));
         let three = Draft::Draft201909.create_resource(json!({"$recursiveAnchor": false}));
 
-        let registry = Registry::try_from_resources(
-            vec![
-                ("http://example.com".to_string(), three),
-                ("http://example.com/foo/".to_string(), two.clone()),
-                ("http://example.com/foo/bar".to_string(), one),
-            ]
-            .into_iter(),
-        )
+        let registry = Registry::try_from_resources(vec![
+            ("http://example.com".to_string(), three),
+            ("http://example.com/foo/".to_string(), two.clone()),
+            ("http://example.com/foo/bar".to_string(), one),
+        ])
         .expect("Invalid resources");
 
         let resolver = registry

crates/jsonschema-referencing/src/error.rs

@@ -153,6 +153,7 @@ impl std::error::Error for Error {
     }
 }
 
+/// Errors that can occur during URI handling.
 #[derive(Debug)]
 pub enum UriError {
     Parse {

crates/jsonschema-referencing/src/lib.rs

@@ -27,3 +27,6 @@ pub use retriever::{DefaultRetriever, Retrieve};
 pub(crate) use segments::Segments;
 pub use specification::Draft;
 pub use vocabularies::{Vocabulary, VocabularySet};
+
+#[cfg(feature = "retrieve-async")]
+pub use retriever::AsyncRetrieve;

crates/jsonschema-referencing/src/meta.rs

@@ -1,3 +1,6 @@
+//! Built-in JSON Schema meta-schemas.
+//!
+//! This module provides access to the official JSON Schema meta-schemas for different draft versions.
 use std::sync::Arc;
 
 use once_cell::sync::Lazy;