drizzle 0.2.1

A type-safe SQL query builder 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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

#![cfg(any(feature = "rusqlite", feature = "turso", feature = "libsql"))]
#![cfg(all(feature = "serde", feature = "uuid"))]

use drizzle::core::expr::*;
use drizzle::sqlite::prelude::*;
use serde::{Deserialize, Serialize};
use uuid::Uuid;

#[derive(Serialize, Deserialize, Debug, PartialEq, Clone, Default)]
pub struct Profile {
    age: i64,
    name: String,
    interests: Vec<String>,
}

#[derive(SQLiteFromRow, Serialize, Deserialize, Debug, PartialEq, Clone, Default)]
pub struct UserResult {
    id: Uuid,
    age: Option<i64>,
}

#[SQLiteTable(NAME = "json_users", STRICT)]
struct JsonUser {
    id: Uuid,
    email: String,
    #[column(JSON)]
    profile: Profile,
    raw: serde_json::Value,
}
#[derive(SQLiteSchema)]
struct Schema {
    jsonuser: JsonUser,
}

#[drizzle::test]
fn json_storage(db: &mut TestDb<Schema>) {
    let Schema { jsonuser } = schema;

    let profile = Profile {
        age: 30,
        name: "John".to_string(),
        interests: vec!["Coding".to_string(), "Reading".to_string()],
    };

    let id = Uuid::new_v4();

    let stmt = db.insert(jsonuser).values([InsertJsonUser::new(
        id,
        "john@test.com",
        profile.clone(),
        serde_json::json!({ "enabled": true }),
    )]);

    // let stmt2 =
    //     db.insert(jsonuser)
    //         .values([InsertJsonUser::new(id, "john@test.com", jsonb(profile))]);

    stmt.execute();

    let stmt = db
        .select((
            jsonuser.id,
            cast(
                drizzle::sqlite::expr::json_extract(jsonuser.profile, "age"),
                drizzle::sqlite::types::Integer,
            )
            .alias("age"),
        ))
        .from(jsonuser)
        .r#where(eq(jsonuser.id, id));

    let user: UserResult = stmt.get();

    assert_eq!(user.id, id);
    assert_eq!(user.age, Some(30));

    // The table-generated row owns the custom JSON codec.
    let stmt = db.select(()).from(jsonuser).r#where(eq(jsonuser.id, id));

    let result: SelectJsonUser = stmt.get();

    assert_eq!(result.id, id);
    assert_eq!(result.email, "john@test.com");
    assert_eq!(result.profile, profile);
    assert_eq!(result.raw, serde_json::json!({ "enabled": true }));
}

#[drizzle::test]
fn owned_insert_model_keeps_json_values(db: &mut TestDb<Schema>) {
    use drizzle::core::SQLModel as _;

    let profile = Profile {
        age: 41,
        name: "Ada".to_string(),
        interests: vec!["Engines".to_string()],
    };

    // A JSON column inserts through `json(?)`; the owned model must keep that
    // expression and its value rather than replace them with a bound NULL.
    let id = Uuid::new_v4();
    let row = || {
        InsertJsonUser::new(
            id,
            "ada@test.com",
            profile.clone(),
            serde_json::json!({ "enabled": false }),
        )
    };
    let borrowed = row();
    let owned = row().into_owned();

    let borrowed_values = borrowed.values();
    let owned_values = owned.values();
    assert!(
        borrowed_values.sql().contains("json"),
        "{}",
        borrowed_values.sql()
    );
    assert_eq!(owned_values.sql(), borrowed_values.sql());
    assert!(
        owned_values.params().eq(borrowed_values.params()),
        "owned {:?} != borrowed {:?}",
        owned_values.params().collect::<Vec<_>>(),
        borrowed_values.params().collect::<Vec<_>>()
    );
}