drizzle 0.1.14

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

#![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,
}
#[derive(SQLiteSchema)]
struct Schema {
    jsonuser: JsonUser,
}

#[derive(SQLiteFromRow, Debug, PartialEq, Default)]
pub struct JsonReadResult {
    id: Uuid,
    email: String,
    #[json]
    profile: Profile,
}

#[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())]);

    // 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));

    // Test reading full JSON column back via #[json] on FromRow struct
    let stmt = db
        .select((jsonuser.id, jsonuser.email, jsonuser.profile))
        .from(jsonuser)
        .r#where(eq(jsonuser.id, id));

    let result: JsonReadResult = stmt.get();

    assert_eq!(result.id, id);
    assert_eq!(result.email, "john@test.com");
    assert_eq!(result.profile, profile);
}