ormer 0.1.21

A minimalist ORM framework that supports SQLite, PostgreSQL, MySQL, and SqlServer
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

// 钩子自动触发示例
// 展示如何实现钩子并在数据库操作中自动触发

use ormer::{AfterInsert, AfterUpdate, BeforeInsert, BeforeUpdate, Model};
#[cfg(feature = "sqlite")]
use ormer::{Database, DbType};
#[cfg(feature = "sqlite")]
use std::sync::atomic::{AtomicUsize, Ordering};

// 全局计数器
#[cfg(feature = "sqlite")]
static INSERT_COUNT: AtomicUsize = AtomicUsize::new(0);
#[cfg(feature = "sqlite")]
static UPDATE_COUNT: AtomicUsize = AtomicUsize::new(0);

#[cfg(feature = "sqlite")]
#[derive(Debug, Model)]
#[table = "hook_demo_users"]
struct HookDemoUser {
    #[primary(auto)]
    id: i32,
    name: String,
    email: String,
}

// 实现插入前钩子
#[cfg(feature = "sqlite")]
#[async_trait::async_trait]
impl BeforeInsert for HookDemoUser {
    async fn before_insert(&mut self) {
        INSERT_COUNT.fetch_add(1, Ordering::SeqCst);
        println!("BeforeInsert: 准备插入用户 {}", self.name);
    }
}

// 实现插入后钩子
#[cfg(feature = "sqlite")]
#[async_trait::async_trait]
impl AfterInsert for HookDemoUser {
    async fn after_insert(&self) {
        println!(
            "AfterInsert: 用户 {} 已成功插入 (ID: {})",
            self.name, self.id
        );
    }
}

// 实现更新前钩子
#[cfg(feature = "sqlite")]
#[async_trait::async_trait]
impl BeforeUpdate for HookDemoUser {
    async fn before_update(&mut self) {
        UPDATE_COUNT.fetch_add(1, Ordering::SeqCst);
        println!("BeforeUpdate: 准备更新用户 {}", self.name);
    }
}

// 实现更新后钩子
#[cfg(feature = "sqlite")]
#[async_trait::async_trait]
impl AfterUpdate for HookDemoUser {
    async fn after_update(&self) {
        println!("AfterUpdate: 用户 {} 已成功更新", self.name);
    }
}

#[cfg(feature = "sqlite")]
#[tokio::test]
async fn test_hooks_demo() {
    #[cfg(feature = "sqlite")]
    {
        let db = Database::connect(DbType::Sqlite, ":memory:").await.unwrap();

        // 创建表
        db.create_table::<HookDemoUser>().execute().await.unwrap();

        // 插入数据 - 钩子会自动触发
        let user = HookDemoUser {
            id: 0,
            name: "Alice".to_string(),
            email: "alice@example.com".to_string(),
        };

        // 注意：当前版本需要手动调用钩子
        // 未来版本将支持自动触发
        let mut user = user;
        user.before_insert().await;
        let _ = db.insert(&user).execute().await.unwrap();
        user.after_insert().await;

        assert_eq!(INSERT_COUNT.load(Ordering::SeqCst), 1);

        // 更新数据
        user.name = "Alice Updated".to_string();
        user.before_update().await;
        // 这里应该调用 update 方法
        user.after_update().await;

        assert_eq!(UPDATE_COUNT.load(Ordering::SeqCst), 1);
    }
}