miniorm 0.4.1

a *very* simple ORM built on top of sqlx
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

use iso_currency::Currency;
use miniorm::prelude::*;
use rust_decimal::Decimal;
use rust_decimal_macros::dec;
use serde::{Deserialize, Serialize};
use sqlx::{types::chrono::NaiveDate, FromRow};

#[derive(Clone, Debug, Serialize, Deserialize, Eq, PartialEq)]
pub struct Ticker(pub String);

#[derive(Clone, Debug, Serialize, Deserialize, Eq, PartialEq)]
pub struct Stock {
    pub ticker: Ticker,
    pub currency: Currency,
}

#[derive(Clone, Debug, Serialize, Deserialize, Eq, PartialEq)]
pub enum Instrument {
    Cash(Currency),
    Stock(Stock),
}

#[derive(Copy, Clone, Debug, Eq, PartialEq, Serialize, Deserialize)]
pub enum Operation {
    Buy,
    Sell,
    Dividend,
    Deposit,
    Withdrawal,
}

#[derive(Clone, Debug, Eq, PartialEq, FromRow, Entity)]
pub struct Transaction {
    #[postgres(DATE NOT NULL)]
    pub date: NaiveDate,

    #[sqlx(json)]
    #[postgres(JSONB NOT NULL)]
    pub operation: Operation,

    #[sqlx(json)]
    #[postgres(JSONB NOT NULL)]
    pub instrument: Instrument,

    #[postgres(DECIMAL NOT NULL)]
    pub quantity: Decimal,

    #[postgres(DECIMAL NOT NULL)]
    pub unit_price: Decimal,

    #[postgres(DECIMAL NOT NULL)]
    pub taxes: Decimal,

    #[postgres(DECIMAL NOT NULL)]
    pub fees: Decimal,

    #[sqlx(json)]
    #[postgres(JSONB NOT NULL)]
    pub currency: Currency,

    #[postgres(DECIMAL NOT NULL)]
    pub exchange_rate: Decimal,
}

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    dotenv::dotenv().ok();

    println!("== POSTGRES ==");
    let url = std::env::var("POSTGRES_URL").expect("POSTGRES_URL env variable not set");
    let db = sqlx::PgPool::connect(&url).await?;
    let store = miniorm::Store::new(db);

    println!("Recreating table...");
    store.recreate_table().await?;

    let aapl = Stock {
        ticker: Ticker("AAPL".into()),
        currency: Currency::USD,
    };

    let tx = Transaction {
        date: NaiveDate::from_ymd_opt(2024, 3, 15).unwrap(),
        operation: Operation::Buy,
        instrument: Instrument::Stock(aapl),
        quantity: dec!(10),
        unit_price: dec!(170.0),
        taxes: dec!(10.2),
        fees: dec!(5.5),
        currency: Currency::USD,
        exchange_rate: dec!(0.9),
    };

    println!("Inserting...");
    let tx = store.create(tx).await?;

    println!("Retrieveing by id...");
    let mut fetched = store.read(tx.id()).await?;
    assert_eq!(tx, fetched);

    println!("Updating by id");
    fetched.operation = Operation::Sell;
    let after_update = store.update(fetched).await?;

    println!("Listing all...");
    let all = store.list().await?;
    assert_eq!(all.len(), 1);
    assert_eq!(&after_update, &all[0]);

    println!("Deleting by id...");
    store.delete(tx.id()).await?;

    println!("Checking delete successful");
    assert!(matches!(
        store.read(tx.id()).await,
        Err(sqlx::Error::RowNotFound)
    ));

    Ok(())
}