orm_demo/
orm_demo.rs

1use rustio_core::{Db, Error, Model, Row, Value};
2
3#[derive(Debug)]
4struct User {
5    id: i64,
6    name: String,
7    is_admin: bool,
8}
9
10impl Model for User {
11    const TABLE: &'static str = "users";
12    const COLUMNS: &'static [&'static str] = &["id", "name", "is_admin"];
13    const INSERT_COLUMNS: &'static [&'static str] = &["name", "is_admin"];
14
15    fn id(&self) -> i64 {
16        self.id
17    }
18
19    fn from_row(row: Row<'_>) -> Result<Self, Error> {
20        Ok(Self {
21            id: row.get_i64("id")?,
22            name: row.get_string("name")?,
23            is_admin: row.get_bool("is_admin")?,
24        })
25    }
26
27    fn insert_values(&self) -> Vec<Value> {
28        vec![self.name.clone().into(), self.is_admin.into()]
29    }
30}
31
32#[tokio::main]
33async fn main() -> Result<(), Error> {
34    let db = Db::memory().await?;
35    db.execute(
36        "CREATE TABLE users (
37            id INTEGER PRIMARY KEY AUTOINCREMENT,
38            name TEXT NOT NULL,
39            is_admin INTEGER NOT NULL
40        )",
41    )
42    .await?;
43
44    let alice_id = User { id: 0, name: "Alice".into(), is_admin: false }
45        .create(&db)
46        .await?;
47    let bob_id = User { id: 0, name: "Bob".into(), is_admin: true }
48        .create(&db)
49        .await?;
50    println!("created ids: alice={alice_id} bob={bob_id}");
51
52    let alice = User::find(&db, alice_id).await?.expect("alice");
53    println!("find alice: {alice:?}");
54
55    let all = User::all(&db).await?;
56    println!("all: {all:?}");
57
58    let renamed = User { id: alice_id, name: "Alicia".into(), is_admin: false };
59    renamed.update(&db).await?;
60    let after_update = User::find(&db, alice_id).await?.unwrap();
61    println!("after update: {after_update:?}");
62
63    User::delete(&db, bob_id).await?;
64    println!("remaining after delete bob: {:?}", User::all(&db).await?);
65
66    Ok(())
67}
orm_demo/orm_demo.rs

orm_demo/
orm_demo.rs
