rust-ef 1.7.0

Rust Entity Framework - An EFCore-inspired ORM 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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

//! Batch SELECT benchmark — measures `to_list` query throughput for tables
//! pre-seeded with 100 / 500 / 1000 rows, both unfiltered and filtered.
//!
//! Run with: `cargo bench -p rust-ef --bench bench_query`

use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
use rust_ef::db_context::{DbContext, DbContextOptionsBuilder};
use rust_ef::linq;
use rust_ef::prelude::*;
use rust_ef_sqlite::DbContextOptionsBuilderExt as _;
use std::sync::Arc;
use tokio::runtime::Runtime;
use tokio::sync::Mutex;

/// Simple single-PK entity used for query throughput measurement.
#[derive(Debug, Clone, EntityType)]
#[table("bench_widgets")]
struct BenchWidget {
    #[primary_key]
    #[auto_increment]
    id: i32,
    #[required]
    #[max_length(100)]
    name: String,
    value: f64,
    tenant_id: i32,
}

/// Seeds a fresh in-memory SQLite context with `n` rows.
async fn seeded_ctx(n: usize) -> DbContext {
    let mut builder = DbContextOptionsBuilder::new();
    builder.use_sqlite_in_memory();
    let options = builder.build();
    let mut ctx = DbContext::from_options(&options).expect("DbContext");
    ctx.set::<BenchWidget>();
    ctx.ensure_created().await.expect("ensure_created");
    for i in 0..n {
        ctx.set::<BenchWidget>().add(BenchWidget {
            id: 0,
            name: format!("widget-{i}"),
            value: i as f64 * 1.5,
            tenant_id: (i / 10) as i32,
        });
    }
    let result = ctx.save_changes().await.expect("save_changes");
    assert_eq!(result.added, n, "seed should insert all rows");
    ctx
}

/// Unfiltered `to_list` — loads every row.
async fn select_all(ctx: &Mutex<DbContext>) {
    let mut guard = ctx.lock().await;
    let rows = guard
        .set::<BenchWidget>()
        .query()
        .to_list()
        .await
        .expect("to_list");
    assert!(!rows.is_empty(), "should load rows");
}

/// Filtered `to_list` via `linq!` — loads a subset matching a tenant.
async fn select_filtered(ctx: &Mutex<DbContext>) {
    let mut guard = ctx.lock().await;
    let rows = linq!(guard.set::<BenchWidget>(), |w: BenchWidget| w.tenant_id

        == 1)
    .to_list()
    .await
    .expect("to_list");
    assert!(!rows.is_empty(), "should load matching rows");
}

fn bench_batch_select(c: &mut Criterion) {
    let rt = Runtime::new().expect("tokio runtime");

    let mut group = c.benchmark_group("batch_select");

    // Unfiltered full-table scan. Seed once per size; the context is reused
    // across all samples of that size (queries are read-only).
    for &n in &[100usize, 500, 1000] {
        let ctx = Arc::new(Mutex::new(rt.block_on(seeded_ctx(n))));
        group.bench_with_input(BenchmarkId::new("to_list_all", n), &n, |b, _| {
            let ctx = ctx.clone();
            b.to_async(&rt).iter(move || {
                let ctx = ctx.clone();
                async move { select_all(&ctx).await }
            });
        });
    }

    // Filtered query (WHERE tenant_id = 1).
    for &n in &[100usize, 500, 1000] {
        let ctx = Arc::new(Mutex::new(rt.block_on(seeded_ctx(n))));
        group.bench_with_input(BenchmarkId::new("to_list_filtered", n), &n, |b, _| {
            let ctx = ctx.clone();
            b.to_async(&rt).iter(move || {
                let ctx = ctx.clone();
                async move { select_filtered(&ctx).await }
            });
        });
    }

    group.finish();
}

criterion_group!(benches, bench_batch_select);
criterion_main!(benches);