rullst-orm 1.1.13

An Active Record ORM for Rust, inspired by Laravel's Eloquent.
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

use rullst_orm::{Eloquent, sqlx::FromRow};
use rullst_orm::schema::{Schema, Blueprint};

#[derive(Debug, Clone, FromRow, rullst_orm::Eloquent)]
#[eloquent(table = "accounts")]
pub struct Account {
    pub id: i32,
    pub owner: String,
    pub balance: i32,
}

#[tokio::main]
async fn main() -> Result<(), rullst_orm::sqlx::Error> {
    // 1. Initialize DB
    let _ = std::fs::remove_file("transactions_test.db");
    std::fs::File::create("transactions_test.db").unwrap();
    Eloquent::init("sqlite://transactions_test.db").await?;

    // 2. Schema
    Schema::create("accounts", |table: &mut Blueprint| {
        table.id();
        table.string("owner").not_null();
        table.integer("balance").not_null();
    }).await?;

    // 3. Create initial accounts
    let mut a = Account { id: 0, owner: "Alice".to_string(), balance: 100 };
    a.save().await?;
    let mut b = Account { id: 0, owner: "Bob".to_string(), balance: 100 };
    b.save().await?;

    println!("--- Initial Balances ---");
    println!("Alice: ${}", Account::find(1).await?.unwrap().balance);
    println!("Bob: ${}", Account::find(2).await?.unwrap().balance);

    // ---------------------------------------------------------
    // Scenario 1: A Successful Transaction (Transfer $50 from Alice to Bob)
    // ---------------------------------------------------------
    println!("\n--- Attempting Successful Transfer ($50) ---");
    let mut tx1 = Eloquent::begin_transaction().await?;
    
    // Using `get_with_tx` and `save_with_tx`
    let mut alice = Account::query().where_eq("owner", "Alice").first_with_tx(&mut tx1).await?.unwrap();
    let mut bob = Account::query().where_eq("owner", "Bob").first_with_tx(&mut tx1).await?.unwrap();
    
    alice.balance -= 50;
    bob.balance += 50;
    
    alice.save_with_tx(&mut tx1).await?;
    bob.save_with_tx(&mut tx1).await?;
    
    tx1.commit().await?; // COMMITTED!
    println!("Transfer committed successfully.");

    // ---------------------------------------------------------
    // Scenario 2: A Failed Transaction (Transfer $200 from Alice to Bob, but Alice doesn't have enough)
    // ---------------------------------------------------------
    println!("\n--- Attempting Failed Transfer ($200) ---");
    let mut tx2 = Eloquent::begin_transaction().await?;
    
    let mut alice2 = Account::query().where_eq("owner", "Alice").first_with_tx(&mut tx2).await?.unwrap();
    let mut bob2 = Account::query().where_eq("owner", "Bob").first_with_tx(&mut tx2).await?.unwrap();
    
    alice2.balance -= 200; // Oops, this is invalid logically!
    bob2.balance += 200;
    
    // We save the corrupted state to the transaction
    alice2.save_with_tx(&mut tx2).await?;
    bob2.save_with_tx(&mut tx2).await?;
    
    // Wait, let's simulate an error causing a rollback!
    println!("Simulating an error... Rolling back!");
    tx2.rollback().await?; // ROLLED BACK!

    // ---------------------------------------------------------
    // Final Verification
    // ---------------------------------------------------------
    println!("\n--- Final Balances (Expected: Alice $50, Bob $150) ---");
    println!("Alice: ${}", Account::find(1).await?.unwrap().balance);
    println!("Bob: ${}", Account::find(2).await?.unwrap().balance);

    Ok(())
}