rs2-stream 0.3.3

A high-performance, production-ready async streaming library for Rust.
Documentation 
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
51
52
53
54
55

//! Schema validation system for RS2 streams

use async_trait::async_trait;
use jsonschema::{validator_for, Validator};
use serde_json::Value;
use std::sync::Arc;

#[derive(Debug, thiserror::Error)]
pub enum SchemaError {
    #[error("Validation failed: {0}")]
    ValidationFailed(String),
    #[error("Missing schema: {0}")]
    MissingSchema(String),
    #[error("Parse error: {0}")]
    ParseError(String),
}

#[async_trait]
pub trait SchemaValidator: Send + Sync {
    async fn validate(&self, data: &[u8]) -> Result<(), SchemaError>;
    fn get_schema_id(&self) -> String;
}

/// Production-ready JSON Schema validator for RS2 streams.
#[derive(Clone)]
pub struct JsonSchemaValidator {
    schema_id: String,
    compiled: Arc<Validator>,
}

impl JsonSchemaValidator {
    /// Create a new validator from a JSON schema value.
    pub fn new(schema_id: &str, schema: Value) -> Self {
        let compiled = validator_for(&schema).expect("Invalid JSON schema");
        Self {
            schema_id: schema_id.to_string(),
            compiled: Arc::new(compiled),
        }
    }
}

#[async_trait]
impl SchemaValidator for JsonSchemaValidator {
    async fn validate(&self, data: &[u8]) -> Result<(), SchemaError> {
        let value: Value =
            serde_json::from_slice(data).map_err(|e| SchemaError::ParseError(e.to_string()))?;
        if let Err(error) = self.compiled.validate(&value) {
            return Err(SchemaError::ValidationFailed(error.to_string()));
        }
        Ok(())
    }
    fn get_schema_id(&self) -> String {
        self.schema_id.clone()
    }
}