ormer 0.1.8

An ORM framework with a usage style similar to Linq, supporting Sqlite, PostgresQL, MySQL
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

#![cfg(any(feature = "sqlite", feature = "postgresql", feature = "mysql"))]

use ormer::query::builder::Select;

mod _test_common;

// 使用宏定义测试专用模型（唯一表名）
define_test_user_for_join!(User, "test_range_users_1");
define_test_role!(Role, "test_range_roles_1");

async fn test_basic_range_impl(config: &_test_common::DbConfig) {
    // 测试 range(0..10)
    let select = Select::<User>::new().range(0..10);
    let (sql, _) = select.to_sql_with_params(config.0);

    assert!(sql.contains("LIMIT 10"));
    assert!(sql.contains("OFFSET 0"));
    println!("SQL with range(0..10): {}", sql);
}

async fn test_range_with_offset_impl(config: &_test_common::DbConfig) {
    // 测试 range(10..20)
    let select = Select::<User>::new().range(10..20);
    let (sql, _) = select.to_sql_with_params(config.0);

    assert!(sql.contains("LIMIT 10"));
    assert!(sql.contains("OFFSET 10"));
    println!("SQL with range(10..20): {}", sql);
}

async fn test_range_with_filter_impl(config: &_test_common::DbConfig) {
    // 测试 range 与 filter 组合
    let select = Select::<User>::new().filter(|p| p.age.ge(18)).range(5..15);
    let (sql, _) = select.to_sql_with_params(config.0);

    assert!(sql.contains("WHERE"));
    assert!(sql.contains("LIMIT 10"));
    assert!(sql.contains("OFFSET 5"));
    println!("SQL with filter and range: {}", sql);
}

async fn test_range_with_order_by_impl(config: &_test_common::DbConfig) {
    // 测试 range 与 order_by 组合
    let select = Select::<User>::new().order_by(|p| p.age.desc()).range(0..5);
    let (sql, _) = select.to_sql_with_params(config.0);

    assert!(sql.contains("ORDER BY"));
    assert!(sql.contains("LIMIT 5"));
    assert!(sql.contains("OFFSET 0"));
    println!("SQL with order_by and range: {}", sql);
}

async fn test_range_single_record_impl(config: &_test_common::DbConfig) {
    // 测试单条记录 range(0..1)
    let select = Select::<User>::new().range(0..1);
    let (sql, _) = select.to_sql_with_params(config.0);

    assert!(sql.contains("LIMIT 1"));
    assert!(sql.contains("OFFSET 0"));
    println!("SQL with range(0..1): {}", sql);
}

async fn test_no_range_impl(config: &_test_common::DbConfig) {
    // 测试不使用 range 时不生成 LIMIT/OFFSET
    let select = Select::<User>::new();
    let (sql, _) = select.to_sql_with_params(config.0);

    assert!(!sql.contains("LIMIT"));
    assert!(!sql.contains("OFFSET"));
    println!("SQL without range: {}", sql);
}

async fn test_related_select_range_impl(config: &_test_common::DbConfig) {
    // 测试 RelatedSelect 的 range 功能
    let select = Select::<User>::new()
        .from::<User, Role>()
        .filter(|p, q| p.id.eq(q.uid))
        .range(0..10);
    let (sql, _) = select.to_sql_with_params(config.0);

    assert!(sql.contains("LIMIT 10"));
    assert!(sql.contains("OFFSET 0"));
    println!("RelatedSelect SQL with range: {}", sql);
}

async fn test_left_join_range_impl(config: &_test_common::DbConfig) {
    // 测试 LEFT JOIN 的 range 功能
    let select = Select::<User>::new()
        .left_join::<Role>(|p, q| p.id.eq(q.uid))
        .range(0..10);
    let (sql, _) = select.to_sql_with_params(config.0);

    assert!(sql.contains("LEFT JOIN"));
    assert!(sql.contains("LIMIT 10"));
    assert!(sql.contains("OFFSET 0"));
    println!("LeftJoin SQL with range: {}", sql);
}

async fn test_range_calculation_impl(config: &_test_common::DbConfig) {
    // 验证 LIMIT 和 OFFSET 的计算是否正确
    let test_cases = vec![
        (0..10, 10, 0),
        (5..15, 10, 5),
        (10..20, 10, 10),
        (0..1, 1, 0),
        (100..200, 100, 100),
    ];

    for (range, expected_limit, expected_offset) in test_cases {
        let select = Select::<User>::new().range(range);
        let (sql, _) = select.to_sql_with_params(config.0);

        assert!(
            sql.contains(&format!("LIMIT {}", expected_limit)),
            "Expected LIMIT {} in SQL: {}",
            expected_limit,
            sql
        );
        assert!(
            sql.contains(&format!("OFFSET {}", expected_offset)),
            "Expected OFFSET {} in SQL: {}",
            expected_offset,
            sql
        );
    }
    println!("All range calculation tests passed!");
}

test_on_all_dbs!(test_basic_range_impl);
test_on_all_dbs!(test_range_with_offset_impl);
test_on_all_dbs!(test_range_with_filter_impl);
test_on_all_dbs!(test_range_with_order_by_impl);
test_on_all_dbs!(test_range_single_record_impl);
test_on_all_dbs!(test_no_range_impl);
test_on_all_dbs!(test_related_select_range_impl);
test_on_all_dbs!(test_left_join_range_impl);
test_on_all_dbs!(test_range_calculation_impl);