invr 0.2.3

Declarative data validation engine using invariants executed on Polars DataFrames.
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

use crate::engine::Engine;
use crate::error::{ApplicationError, ApplicationResult};
use crate::infrastructure::polars::{checks, kind::PolarsKind};
use crate::report::Report;
use crate::spec::Spec;
use polars::prelude::*;

/// Polars-based execution engine for evaluating invariants on a `DataFrame`.
///
/// This engine implements the generic `Engine` trait for `PolarsKind` invariants.
///
/// Execution flow:
/// 1. Receives a `Spec` containing a collection of invariants.
/// 2. Delegates execution to `checks::run_all`.
/// 3. Collects all produced `Violation`s.
/// 4. Builds and returns a `Report`.
///
/// This struct is intentionally stateless and lightweight.
///
/// # Example
/// ```ignore
/// let engine = EnginePolarsDataFrame;
/// let report = engine.execute(&df, &spec)?;
/// ```
pub struct EnginePolarsDataFrame;

impl Engine for EnginePolarsDataFrame {
    type Dataset = DataFrame;
    type Kind = PolarsKind;

    /// Executes all invariants from the provided `Spec` against the given `DataFrame`.
    ///
    /// Returns a `Report` containing zero or more violations.
    ///
    /// Errors occurring during Polars expression evaluation are
    /// converted into `ApplicationError::engine_failure`.
    fn execute(
        &self,
        dataset: &Self::Dataset,
        spec: &Spec<Self::Kind>,
    ) -> ApplicationResult<Report> {
        let violations = checks::run_all(dataset, spec.invariants())
            .map_err(|e| ApplicationError::engine_failure(e.to_string()))?;

        let mut report = Report::new();
        report.extend(violations);
        Ok(report)
    }
}