arel 0.1.0

arel: a sql orm
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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203

use arel::prelude::*;
use chrono::{TimeZone};

#[arel(primary_key="id", locking_column="lock_version")]
struct User {
    #[arel(table_column_name="id")]
    uid: Option<i64>,
    #[arel(table_column_name="desc")]
    desc2: String,
    #[arel(table_column_name="type")]
    r#type: Option<i32>,
    lock_version: Option<i32>,
    done: Option<bool>,
    expired_at: chrono::DateTime<chrono::Utc>,
}

async fn init_db() -> anyhow::Result<()> {
    let db_state = arel::visitors::get_or_init_db_state(|| Box::pin(async {
        sqlx::any::AnyPoolOptions::new().max_connections(5).connect("sqlite::memory:").await
    })).await?;

    sqlx::query("CREATE TABLE IF NOT EXISTS users
            (
                id          INTEGER PRIMARY KEY NOT NULL,
                desc TEXT                NOT NULL,
                done        BOOLEAN             NOT NULL DEFAULT 0,
                lock_version   INT(11) NOT NULL DEFAULT 0,
                expired_at   DATETIME NOT NULL
            );"
    ).execute(db_state.pool()).await?;
    for i in 0..10 {
        sqlx::query(&User::create(json!({
                "desc": format!("test-{}", i),
                "expired_at": "2021-01-01 00:00:00",
            })).to_sql_string()?).execute(db_state.pool()).await?;
    }

    Ok(())
}

async fn with_transaction_and_with_lock_test() -> anyhow::Result<()> {
    // Transaction ok
    let u1 = User::with_transaction(|tx| Box::pin(async {
        let mut u1 = User::query().fetch_one_with_executor(&mut *tx).await?;
        let mut u2 = User::query().fetch_last_with_executor(&mut *tx).await?;
        u1.set_desc2("tx1".to_string());
        u2.set_desc2("tx2".to_string());
        u1.save_with_executor(&mut *tx).await?;
        u2.save_with_executor(&mut *tx).await?;
        Ok(Some(u1))
    })).await?.unwrap();
    assert_eq!(u1.desc2(), Some(&"tx1".to_string()));
    let u1 = User::query().fetch_one().await?;
    assert_eq!(u1.desc2(), Some(&"tx1".to_string()));
    let u2 = User::query().fetch_last().await?;
    assert_eq!(u2.desc2(), Some(&"tx2".to_string()));

    assert!(User::with_transaction(|tx| Box::pin(async {
        let mut u1 = User::query().fetch_one_with_executor(&mut *tx).await?;
        let mut u2 = User::query().fetch_last_with_executor(&mut *tx).await?;
        u1.set_desc2("err_tx1".to_string());
        u2.set_desc2("err_tx2".to_string());
        u1.save_with_executor(&mut *tx).await?;
        u2.save_with_executor(&mut *tx).await?;
        Err(anyhow::anyhow!("test rollback"))
    })).await.is_err());
    let u1 = User::query().fetch_one().await?;
    assert_eq!(u1.desc2(), Some(&"tx1".to_string()));
    let u2 = User::query().fetch_last().await?;
    assert_eq!(u2.desc2(), Some(&"tx2".to_string()));

    // with_lock ok
    let mut u1 = User::query().fetch_one().await?;
    let mut u2 = User::query().fetch_last().await?;
    let u1 = u1.fetch_self().await?.with_lock(|tx| Box::pin(async move {
        u1.set_desc2("with_lock1".to_string()).save_with_executor(&mut *tx).await?;
        u2.set_desc2("with_lock2".to_string()).save_with_executor(&mut *tx).await?;
        Ok(Some(u1))
    })).await?.unwrap();
    assert_eq!(u1.desc2(), Some(&"with_lock1".to_string()));
    let u1 = User::query().fetch_one().await?;
    assert_eq!(u1.desc2(), Some(&"with_lock1".to_string()));
    let u2 = User::query().fetch_last().await?;
    assert_eq!(u2.desc2(), Some(&"with_lock2".to_string()));

    // with_lock rollback
    let mut u1 = User::query().fetch_one().await?;
    let mut u2 = User::query().fetch_last().await?;
    assert!(u1.fetch_self().await?.with_lock(|tx| Box::pin(async move {
        u1.set_desc2("err_with_lock1".to_string());
        u2.set_desc2("err_with_lock2".to_string());
        u1.save_with_executor(&mut *tx).await?;
        u2.save_with_executor(&mut *tx).await?;
        Err(anyhow::anyhow!("test lock rollback"))
    })).await.is_err());
    let u1 = User::query().fetch_one().await?;
    assert_eq!(u1.desc2(), Some(&"with_lock1".to_string()));
    let u2 = User::query().fetch_last().await?;
    assert_eq!(u2.desc2(), Some(&"with_lock2".to_string()));

    Ok(())
}

async fn main_test() -> anyhow::Result<()> {
    init_db().await?;

    with_transaction_and_with_lock_test().await?;

    assert_eq!(User::table_column_names(), vec!["id", "desc", "type", "lock_version", "done", "expired_at"]);
    assert_eq!(User::attr_names(), vec!["uid", "desc2", "r#type", "lock_version", "done", "expired_at"]);
    assert_eq!(User::attr_name_to_table_column_name("uid")?, "id");
    assert_eq!(User::attr_name_to_table_column_name("desc2")?, "desc");
    assert_eq!(User::attr_name_to_table_column_name("r#type")?, "type");
    assert_eq!(User::table_column_name_to_attr_name("id")?, "uid");
    assert_eq!(User::table_column_name_to_attr_name("desc")?, "desc2");
    assert_eq!(User::table_column_name_to_attr_name("type")?, "r#type");
    // assert_eq!(User::uid_table_column_name(), "id");
    // assert_eq!(User::desc_table_column_name(), "desc");
    // assert_eq!(User::done_table_column_name(), "done");
    // assert_eq!(User::expired_at_table_column_name(), "expired_at");

    let count = User::fetch_count().await?;
    assert_eq!(count, 10);

    // query all
    let users = User::query().fetch_all().await?;
    assert_eq!(users.len(), 10);

    // update batch
    User::update_all(json!({"desc": "update_1"})).execute().await?;
    let user = User::query().fetch_one().await?;
    assert_eq!(user.desc2(), Some(&"update_1".to_string()));
    let user = User::fetch_first().await?;
    assert_eq!(user.uid(), Some(&1));
    let user = User::fetch_last().await?;
    eprintln!("------{:?}", user.uid());
    assert_eq!(user.uid(), Some(&10));

    // delete batch
    User::delete_all(json!(["id > ?", 5])).execute().await?;
    let users = User::query().fetch_all().await?;
    assert_eq!(users.len(), 5);

    // query one
    let user = User::query().fetch_one().await?;
    assert_eq!(user.uid(), Some(&1));

    // update
    let mut user = User::query().fetch_one().await?;
    user.set_desc2("custom desc".to_string()).save().await?;
    assert_eq!(user.desc2(), Some(&"custom desc".to_string()));
    let user = User::query().fetch_one().await?;
    assert_eq!(user.desc2(), Some(&"custom desc".to_string()));
    // locking_column
    let mut u1 = User::query().fetch_one().await?;
    let mut u2 = User::query().fetch_one().await?;
    assert!(u1.set_desc2("custom desc".to_string()).save().await.is_ok());
    // u2.set_desc2("custom desc".to_string()).save().await?; // print locking_column error message
    assert!(u2.set_desc2("custom desc".to_string()).save().await.is_err());


    let expired_at = chrono::Utc.ymd(2021, 12, 31).and_hms(23, 59, 59);

    // create
    let mut user = User::new();
    // validates
    assert!(user.save().await.is_err());
    user.set_desc2("create desc".to_string())
        .set_expired_at(expired_at.clone())
        .save().await?;
    let users = User::query().fetch_all().await?;
    assert_eq!(users.len(), 6);

    // // delete
    let mut user = User::query().fetch_one().await?;
    let _result = user.delete().await?;
    let users = User::query().fetch_all().await?;
    assert_eq!(users.len(), 5);

    // query_count
    let count = User::fetch_count().await?;
    assert_eq!(count, 5);
    let count = User::query().where_range("expired_at", ..=expired_at).fetch_count().await?;
    assert_eq!(count, 5);
    let count = User::query().where_range("expired_at", ..expired_at).fetch_count().await?;
    assert_eq!(count, 4);
    let count = User::query().where_range("expired_at", expired_at..).fetch_count().await?;
    assert_eq!(count, 1);

    Ok(())
}

#[test]
fn test_sqlite_sqlx() {
    assert!(
        match tokio_test::block_on(main_test()) {
            Ok(()) => Ok(()),
            Err(e) => {
                eprintln!("err: {:?}", e);
                Err(e)
            }
        }.is_ok());
}