Skip to main content

sz_orm_core/
lib.rs

1//! # SZ-ORM — 鲜视达 ORM
2//!
3//! Rust 异步 ORM 工作空间(原型阶段),兼容 ThinkORM 风格。
4//!
5//! ## 架构概览
6//!
7//! SZ-ORM 工作空间由 **39 个成员** 组成(37 个 sz-orm-* lib + cli + examples):
8//!
9//! ### 核心引擎 (sz-orm-core)
10//! | 模块 | 功能 |
11//! |------|------|
12//! | `model` | `Model` trait — 定义表名、主键、时间戳、软删除、关联关系 |
13//! | `query` | `QueryBuilder<M>` — 链式 API,支持 SELECT/INSERT/UPDATE/DELETE/聚合/分页/JOIN |
14//! | `dialect` | 多数据库方言 — MySQL (反引号)、PostgreSQL (双引号)、SQLite、Oracle 23ai |
15//! | `pool` | 异步连接池 — 可配置大小、超时、空闲回收、健康检查、最大生命周期 |
16//! | `transaction` | ACID 事务 — 隔离级别、保存点、`TransactionManager` 多事务管理 |
17//! | `migration` | 文件迁移系统 — up/down/rollback/reset/refresh,含 `SchemaBuilder` |
18//! | `cache` | 多级缓存 — `MemoryCache`、`MultiLevelCache`,支持 TTL |
19//! | `value` | 统一值类型 — 20 种变体 (整数/浮点/字符串/字节/UUID/日期/JSON/数组) |
20//! | `db_type` | 数据库类型枚举 — MySQL、PostgreSQL、SQLite、Oracle、Redis、MongoDB 等 11 种 |
21//! | `error` | 错误类型体系 — `DbError`(20 变体)、`PoolError`、`CacheError`、`TxError` |
22//!
23//! ### 数据库适配器
24//! - **sz-orm-sqlx** — sqlx 适配器,连接真实 MySQL/PostgreSQL/SQLite/Oracle
25//! - **sz-orm-sql-validator** — SQL 校验与注入检测
26//!
27//! ### 扩展生态包 (18 个)
28//! | 包名 | 功能 |
29//! |------|------|
30//! | sz-orm-crypto | 加密原语 (AES-256-GCM, PBKDF2, HMAC-SHA256) |
31//! | sz-orm-auth | JWT 鉴权 (HS256) |
32//! | sz-orm-scheduler | Cron 定时任务调度 |
33//! | sz-orm-mqtt | MQTT 客户端 (rumqttc) |
34//! | sz-orm-websocket | WebSocket 服务端 (tokio-tungstenite) |
35//! | sz-orm-queue | 消息队列 (RabbitMQ/lapin, Kafka, NATS, ActiveMQ, RocketMQ, Pulsar) |
36//! | sz-orm-storage | 对象存储 (S3/阿里云/腾讯云/华为云/七牛/又拍云/本地) |
37//! | sz-orm-ai | AI 集成 (Embedding, RAG, Vector) |
38//! | sz-orm-grpc | gRPC 服务/客户端 |
39//! | sz-orm-graphql | GraphQL 查询支持 |
40//! | sz-orm-es | Elasticsearch 集成 |
41//! | sz-orm-tracing | 分布式追踪 |
42//! | sz-orm-logger | 日志系统 |
43//! | sz-orm-swagger | API 文档生成 |
44//! | sz-orm-masking | 数据脱敏 |
45//! | sz-orm-health | 健康检查 |
46//! | sz-orm-audit | 审计日志 |
47//! | sz-orm-batch | 批量操作 |
48//!
49//! ### 高级特性包 (6 个)
50//! | 包名 | 功能 |
51//! |------|------|
52//! | sz-orm-dtx | 分布式事务 |
53//! | sz-orm-rw | 读写分离 |
54//! | sz-orm-sharding | 分库分表 |
55//! | sz-orm-limit | 限流控制 |
56//! | sz-orm-config | 配置管理 |
57//! | sz-orm-mig | 迁移管理增强 |
58//!
59//! ### 平台支持
60//! - **sz-orm-wasm** — WebAssembly 编译目标
61//! - **sz-orm-lc** — 本地/边缘计算
62//! - **sz-orm-back** — 备份与恢复
63//!
64//! ## 快速入门
65//!
66//! ```rust,ignore
67//! use sz_orm_core::*;
68//!
69//! // 1. 定义模型
70//! #[derive(Clone)]
71//! struct User {
72//!     id: i64,
73//!     name: String,
74//!     email: String,
75//! }
76//!
77//! impl Model for User {
78//!     type PrimaryKey = i64;
79//!     fn table_name() -> &'static str { "users" }
80//!     fn pk(&self) -> Self::PrimaryKey { self.id }
81//!     fn set_pk(&mut self, pk: Self::PrimaryKey) { self.id = pk; }
82//! }
83//!
84//! // 2. 构建查询
85//! let dialect = get_dialect(DbType::MySQL).unwrap();
86//! let sql = QueryBuilder::<User>::new(dialect)
87//!     .table("users")
88//!     .select(vec!["id", "name", "email"])
89//!     .where_cond("status = 'active'")
90//!     .order_by("created_at")
91//!     .order_desc("id")
92//!     .limit(10)
93//!     .build_select();
94//!
95//! // 3. 执行前校验
96//! QueryBuilder::<User>::new(get_dialect(DbType::MySQL).unwrap())
97//!     .table("users")
98//!     .select(vec!["id", "name"])
99//!     .validate()?; // 校验 SQL 语法、注入、括号平衡
100//!
101//! // 4. 其他操作
102//! let mut data = std::collections::HashMap::new();
103//! data.insert("name".to_string(), Value::String("Alice".to_string()));
104//! data.insert("age".to_string(), Value::I64(25));
105//!
106//! let insert_sql = QueryBuilder::<User>::new(dialect)
107//!     .table("users")
108//!     .build_insert(&data);
109//!
110//! let update_sql = QueryBuilder::<User>::new(get_dialect(DbType::MySQL).unwrap())
111//!     .table("users")
112//!     .where_cond("id = 1")
113//!     .build_update(&data);
114//!
115//! let delete_sql = QueryBuilder::<User>::new(get_dialect(DbType::MySQL).unwrap())
116//!     .table("users")
117//!     .where_cond("id = 1")
118//!     .build_delete();
119//! ```
120//!
121//! ## 支持的数据库
122//!
123//! | 数据库 | 方言实现 | 真实连接 | 引用方式 |
124//! |--------|---------|---------|---------|
125//! | MySQL | `MySqlDialect` (`` ` `` 反引号) | sz-orm-sqlx | ✅ |
126//! | PostgreSQL | `PostgreSqlDialect` (`"` 双引号) | sz-orm-sqlx | ✅ |
127//! | SQLite 3.35+ | `SqliteDialect` (`"` 双引号) | sz-orm-sqlx | ✅ |
128//! | Oracle 23ai | `OracleDialect` (类型自动映射) | sz-orm-sqlx | ✅ |
129//!
130//! 通过 `get_dialect(DbType::MySQL)` 获取方言实例。每个方言处理:
131//! - 标识符引用风格
132//! - 字符串转义规则
133//! - 分页语法 (LIMIT/OFFSET vs OFFSET/FETCH)
134//! - JSON 提取函数 (JSON_EXTRACT vs #>> vs json_extract vs JSON_VALUE)
135//! - 全文搜索 (MATCH AGAINST vs to_tsvector vs CONTAINS)
136//! - 布尔转整数 (IF/CASE)
137//! - 自增关键字 (AUTO_INCREMENT/GENERATED BY DEFAULT AS IDENTITY)
138//!
139//! ## 核心功能详解
140//!
141//! ### QueryBuilder API
142//!
143//! 所有查询方法返回 `Self`,支持链式调用:
144//!
145//! ```rust,ignore
146//! // 基础查询
147//! QueryBuilder::<M>::new(dialect)
148//!     .table("users")
149//!     .select(vec!["id", "name"])
150//!     .where_cond("status = 'active'")    // AND
151//!     .or_where("role = 'admin'")         // OR
152//!     .where_in("id", vec![Value::I64(1), Value::I64(2)])
153//!     .where_between("age", Value::I64(18), Value::I64(30))
154//!     .where_null("deleted_at")
155//!     .order_by("created_at")
156//!     .order_desc("id")
157//!     .group_by("status")
158//!     .having("COUNT(*) > 5")
159//!     .limit(20)
160//!     .offset(40)
161//!     .page(3, 20)                       // page=3, page_size=20
162//!     .join_inner("posts", "users.id", "posts.user_id")
163//!     .join_left("profiles", "users.id", "profiles.user_id")
164//!     .build_select();
165//!
166//! // 聚合函数
167//! builder.build_count();    // SELECT COUNT(*)
168//! builder.build_exists();   // SELECT EXISTS(...)
169//! builder.build_max("score");
170//! builder.build_min("price");
171//! builder.build_sum("amount");
172//! builder.build_avg("value");
173//! ```
174//!
175//! ### SQL 校验
176//!
177//! ```rust,ignore
178//! // 编译时 + 运行时双重校验
179//! builder.validate()?;              // 校验 SELECT
180//! builder.validate_insert(&data)?;  // 校验 INSERT(含空数据检测)
181//! builder.validate_update(&data)?;  // 校验 UPDATE(含空数据检测)
182//! builder.validate_delete()?;       // 校验 DELETE
183//!
184//! // 校验内容包括:SQL 语法、注入检测、括号平衡、
185//! // 表名/列名合法性、JOIN 列名校验
186//! ```
187//!
188//! ### Model Trait
189//!
190//! ```rust,ignore
191//! pub trait Model: Send + Sync + Sized + 'static {
192//!     type PrimaryKey: Send + Sync + Debug + Display + Clone + Default;
193//!
194//!     fn table_name() -> &'static str;          // 表名(必需)
195//!     fn pk_name() -> &'static str { "id" }     // 主键列名
196//!     fn pk(&self) -> Self::PrimaryKey;         // 获取主键值
197//!     fn set_pk(&mut self, pk: Self::PrimaryKey); // 设置主键值
198//!     fn foreign_key(relation: &str) -> String; // 外键命名 "user_id"
199//!     fn timestamp_fields() -> Option<TimestampFields>; // 自动时间戳
200//!     fn soft_delete_field() -> Option<&'static str>;   // 软删除字段
201//! }
202//!
203//! // ModelExt 扩展
204//! pub trait ModelExt: Model {
205//!     fn columns() -> Vec<&'static str>;     // 所有列
206//!     fn fillable() -> Vec<&'static str>;    // 可填充列
207//!     fn guarded() -> Vec<&'static str>;     // 保护列(默认含主键)
208//!     fn hidden() -> Vec<&'static str>;      // 隐藏列(不序列化)
209//!     fn relations() -> HashMap<&str, Relation>; // 关联关系
210//!     fn fill(&mut self, data: HashMap<String, Value>); // 批量赋值
211//!     fn to_json(&self) -> serde_json::Value; // 序列化
212//! }
213//!
214//! // 四种关联关系
215//! // BelongsTo   — 多对一(Order → User)
216//! // HasMany     — 一对多(User → Orders)
217//! // HasOne      — 一对一(User → Profile)
218//! // BelongsToMany — 多对多(User ↔ Role,通过中间表)
219//! ```
220//!
221//! ### 连接池
222//!
223//! ```rust,ignore
224//! // 通过 Builder 配置
225//! let config = PoolConfigBuilder::new()
226//!     .max_size(100)       // 最大连接数
227//!     .min_idle(10)        // 最小空闲连接
228//!     .acquire_timeout(30) // 获取超时(秒)
229//!     .idle_timeout(600)   // 空闲超时(秒)
230//!     .max_lifetime(1800)  // 最大生命周期(秒)
231//!     .build()?;
232//!
233//! let pool = Pool::new(config, factory)?;
234//! let conn = pool.acquire().await?;  // 获取连接(带超时)
235//! pool.release(conn).await;         // 归还连接
236//! pool.status().await;               // PoolStatus { idle, active, max, min }
237//! pool.reap_idle().await;           // 回收空闲连接
238//! pool.close_all().await;           // 关闭所有连接
239//! ```
240//!
241//! ### 事务
242//!
243//! ```rust,ignore
244//! // 事务选项
245//! let opts = TransactOptions::default()
246//!     .with_isolation(IsolationLevel::Serializable)
247//!     .read_only()
248//!     .with_timeout(Duration::from_secs(30));
249//!
250//! let mut tx = Transaction::new(conn, opts);
251//! tx.execute("INSERT INTO users VALUES (1)").await?;
252//! tx.query("SELECT * FROM users").await?;
253//!
254//! // 保存点(嵌套事务)
255//! let sp = tx.savepoint().await?;         // SAVEPOINT sp_N
256//! tx.rollback_to_savepoint(&sp).await?;   // ROLLBACK TO SAVEPOINT sp_N
257//! tx.release_savepoint(&sp).await?;       // RELEASE SAVEPOINT sp_N
258//!
259//! tx.commit().await?;
260//! // tx.rollback().await?;
261//!
262//! // TransactionManager:管理多个命名事务
263//! let mgr = TransactionManager::new();
264//! mgr.begin("tx1", conn, opts).await?;
265//! mgr.commit("tx1").await?;
266//! mgr.list().await;        // ["tx1"]
267//! mgr.state("tx1").await;  // Some(TransactionState::Committed)
268//! ```
269//!
270//! ### 迁移系统
271//!
272//! ```rust,ignore
273//! // 文件命名:<version>_<name>_up.sql / <version>_<name>_down.sql
274//! // 示例:001_create_users_up.sql, 001_create_users_down.sql
275//!
276//! let resolver = FileMigrationResolver::new(PathBuf::from("./migrations"));
277//! let migrations = resolver.resolve(DbType::MySQL)?;
278//!
279//! let mut migrator = Migrator::new(MigrationContext::default())
280//!     .add_migrations(migrations);
281//!
282//! migrator.migrate().await?;                     // 执行所有待迁移
283//! migrator.up(Some("003")).await?;               // 执行到指定版本
284//! migrator.down(Some("001")).await?;             // 回滚到指定版本
285//! migrator.rollback("002").await?;               // 回滚单个迁移
286//! migrator.reset().await?;                       // 全部回滚 + 重新执行
287//! migrator.refresh().await?;                     // 同 reset
288//! migrator.progress();                            // MigrationProgress { total, applied, pending }
289//!
290//! // SchemaBuilder:程序化建表
291//! let sql = SchemaBuilder::new("users")
292//!     .add_column(ColumnDef::new("id", "INT").not_null().auto_increment())
293//!     .add_column(ColumnDef::new("name", "VARCHAR").length(255).not_null())
294//!     .add_index(IndexDef::new("idx_name", vec!["name"]).unique())
295//!     .add_foreign_key(
296//!         ForeignKeyDef::new("fk_role", "role_id", "roles", "id")
297//!             .on_delete("CASCADE")
298//!     )
299//!     .build(DbType::MySQL);
300//! ```
301//!
302//! ### 值类型 (Value)
303//!
304//! ```rust,ignore
305//! // 20 种变体,覆盖所有数据库类型
306//! Value::Null | Bool(bool) | I8..I64 | U8..U64 | F32 | F64
307//! | String(String) | Bytes(Vec<u8>) | Uuid(String) | Date(String)
308//! | DateTime(String) | Time(String) | Json(String) | Array(Vec<Value>)
309//!
310//! // 类型转换
311//! value.as_str()    // Option<&str>
312//! value.as_i64()    // Option<i64>(支持 F32/F64/Bool/String→i64 转换)
313//! value.as_f64()    // Option<f64>
314//! value.as_bool()   // Option<bool>(支持 "true"/"1"/"yes"/"on" 等)
315//! value.as_bytes()  // Option<&[u8]>
316//! value.to_param()  // Cow<str> — SQL 参数格式
317//!
318//! // From 实现
319//! let v: Value = 42i64.into();
320//! let v: Value = "hello".into();
321//! let v: Value = vec![1u8, 2u8].into();
322//! ```
323//!
324//! ## 错误处理
325//!
326//! 统一错误类型体系,每种错误携带唯一错误码:
327//!
328//! ```rust,ignore
329//! // DbError — 20 种变体,错误码 DB001-DB020
330//! DbError::QueryError("...")
331//! DbError::ConnectionRefused("...")
332//! DbError::ConnectionTimeout("...")
333//! DbError::NotFound("...")
334//! DbError::ConstraintViolation("...")
335//! // ... 等
336//!
337//! // PoolError — 6 种变体,错误码 PL001-PL006
338//! PoolError::Exhausted | Timeout | AlreadyAcquired | InvalidConfig | ...
339//!
340//! // CacheError — 6 种变体,错误码 CH001-CH006
341//! // TxError — 6 种变体(NotStarted, CommitFailed, SavepointError 等)
342//!
343//! // 便捷方法
344//! DbError::query("test failed")        // 创建查询错误
345//! DbError::connection("timeout")       // 创建连接错误
346//! DbError::not_found("user #42")       // 创建未找到错误
347//! err.is_retryable()                   // 是否可重试
348//! err.error_code()                     // "DB001"
349//! ```
350//!
351//! ## 验证方法
352//!
353//! SZ-ORM 通过 **7 线验证体系** 保证质量:
354//!
355//! | 验证方法 | 描述 | 测试文件 |
356//! |---------|------|---------|
357//! | **TDD** | 核心模块 115+ 单元测试 | `core.rs` |
358//! | **集成** | 真实 MySQL/PG/SQLite/Oracle 端到端 | `integration_mysql.rs`, `integration_pg.rs`, `integration_sqlite.rs` |
359//! | **Jepsen** | 29 并发正确性测试 + 10 真实 DB Jepsen | `jepsen.rs`, `real_db_jepsen.rs` |
360//! | **Fuzz** | 11 边界/边缘案例发现 | `fuzz.rs` |
361//! | **Stress** | 77 性能基准测试 | `stress.rs`, `core_bench.rs` |
362//! | **Chaos** | 16 故障鲁棒性测试 | `chaos.rs` |
363//! | **Formal** | 14 形式化验证不变量 | `formal.rs` |
364//!
365//! **总计:2,271 测试,0 失败**(1,635 `#[test]` + 636 `#[tokio::test]`;部分需真实服务)
366//!
367//! ## 类型别名与常量
368//!
369//! ```rust,ignore
370//! // 类型别名
371//! pub type Shared<T> = Arc<T>;
372//! pub type Boxed<T> = Box<T>;
373//! pub type DbResult<T> = Result<T, DbError>;
374//! pub type PoolResult<T> = Result<T, PoolError>;
375//! pub type CacheResult<T> = Result<T, CacheError>;
376//! pub type TxResult<T> = Result<T, TxError>;
377//!
378//! // 默认常量
379//! pub const DEFAULT_BATCH_SIZE: usize = 1000;
380//! pub const DEFAULT_ACQUIRE_TIMEOUT: u64 = 30;   // 秒
381//! pub const DEFAULT_IDLE_TIMEOUT: u64 = 600;      // 秒
382//! pub const DEFAULT_MAX_LIFETIME: u64 = 1800;     // 秒
383//! pub const DEFAULT_MIN_IDLE: u32 = 5;
384//! pub const DEFAULT_MAX_SIZE: u32 = 100;
385//! ```
386//!
387//! ## 导出清单
388//!
389//! `use sz_orm_core::*;` 将导入以下模块的全部公共符号:
390//!
391//! - `async_trait` (重导出)、`bytes::Bytes`、`chrono::{DateTime, Utc}`、`serde::{Deserialize, Serialize}`
392//! - `cache::*` — `Cache`, `MemoryCache`, `MultiLevelCache`, `CacheStats`
393//! - `db_type::*` — `DbType` 枚举 (11 种数据库)
394//! - `dialect::*` — `Dialect`, `MySqlDialect`, `PostgreSqlDialect`, `SqliteDialect`, `OracleDialect`, `get_dialect()`
395//! - `error::*` — `DbError`, `PoolError`, `CacheError`, `TxError`
396//! - `migration::*` — `Migration`, `Migrator`, `SchemaBuilder`, `ColumnDef`, `IndexDef`, `ForeignKeyDef`
397//! - `model::*` — `Model`, `ModelExt`, `Relation`, `BelongsTo`, `HasMany`, `HasOne`, `BelongsToMany`
398//! - `pool::*` — `Pool`, `PoolConfig`, `PoolConfigBuilder`, `Connection`, `ConnectionFactory`, `PoolStatus`
399//! - `query::*` — `QueryBuilder<M>` (链式 SQL 构造器)
400//! - `transaction::*` — `Transaction`, `TransactionManager`, `TransactOptions`, `IsolationLevel`
401//! - `value::*` — `Value` 枚举 (20 种变体)
402
403// 文档完整性:docs.rs 构建文档时启用 missing_docs lint
404// 本地和 CI clippy 不触发,避免 313 个 pub API 缺文档阻塞开发
405// 待文档逐步补齐后再改为全局 #![warn(missing_docs)]
406#![cfg_attr(docsrs, warn(missing_docs))]
407
408use std::sync::Arc;
409
410/// Re-export async traits
411pub use async_trait::async_trait;
412
413/// Re-export common types
414pub use bytes::Bytes;
415pub use chrono::{DateTime, Utc};
416pub use serde::{Deserialize, Serialize};
417
418pub mod accessors;
419pub mod behaviors;
420mod cache;
421pub mod data_permission;
422mod db_type;
423pub mod dialect;
424pub mod dirty_attributes;
425pub mod dynamic_filter;
426pub mod dynamic_sql;
427pub mod entity_graph;
428mod error;
429pub mod find_with_related;
430pub mod guard;
431pub mod hooks;
432pub mod hydration_plugin;
433pub mod i18n;
434pub mod join_dsl;
435pub mod json_query;
436pub mod l2_cache;
437pub mod lambda;
438pub mod migration;
439mod model;
440pub mod observer;
441pub mod optimistic_lock;
442pub mod phinx_migration;
443mod pool;
444mod query;
445pub mod queryable;
446pub mod quick_query;
447pub mod repository;
448pub mod result_map;
449pub mod schema_gen;
450pub mod sql_safety;
451mod transaction;
452pub mod type_handler;
453pub mod typed;
454pub mod typed_ast;
455mod value;
456
457// Re-export proc macros
458pub use sz_orm_macros::sql_string;
459pub use sz_orm_macros::typed_query;
460
461pub use cache::*;
462pub use db_type::*;
463#[allow(ambiguous_glob_reexports)]
464pub use dialect::*;
465pub use error::*;
466#[allow(ambiguous_glob_reexports)]
467pub use migration::*;
468pub use model::*;
469pub use pool::*;
470pub use query::*;
471pub use transaction::*;
472pub use value::*;
473
474/// Alias for `Arc<T>`
475pub type Shared<T> = Arc<T>;
476
477/// Alias for `Box<T>`
478pub type Boxed<T> = Box<T>;
479
480/// Alias for Result<T, DbError>
481pub type DbResult<T> = Result<T, DbError>;
482
483/// Result type for pool operations
484pub type PoolResult<T> = Result<T, PoolError>;
485
486/// Result type for cache operations
487pub type CacheResult<T> = Result<T, CacheError>;
488
489/// Result type for transaction operations
490pub type TxResult<T> = Result<T, TxError>;
491
492/// Default batch size for bulk operations
493pub const DEFAULT_BATCH_SIZE: usize = 1000;
494
495/// Default connection timeout in seconds
496pub const DEFAULT_ACQUIRE_TIMEOUT: u64 = 30;
497
498/// Default idle timeout in seconds
499pub const DEFAULT_IDLE_TIMEOUT: u64 = 600;
500
501/// Default max lifetime in seconds
502pub const DEFAULT_MAX_LIFETIME: u64 = 1800;
503
504/// Default minimum idle connections
505pub const DEFAULT_MIN_IDLE: u32 = 5;
506
507/// Default maximum pool size
508pub const DEFAULT_MAX_SIZE: u32 = 100;
509
510#[cfg(test)]
511mod tests {
512    use super::*;
513    use std::error::Error;
514
515    #[test]
516    fn test_db_type() {
517        assert_eq!(DbType::MySQL.as_str(), "mysql");
518        assert_eq!(DbType::PostgreSQL.as_str(), "postgres");
519        assert_eq!(DbType::Sqlite.as_str(), "sqlite");
520    }
521
522    #[test]
523    fn test_value() {
524        let v = Value::Null;
525        assert!(v.is_null());
526
527        let v = Value::I64(42);
528        assert!(v.is_i64());
529
530        let v = Value::String("hello".to_string());
531        assert!(v.is_string());
532    }
533
534    #[test]
535    fn test_db_error_display() {
536        let err = DbError::QueryError("test query failed".to_string());
537        assert_eq!(format!("{}", err), "Query error: test query failed");
538
539        let err = DbError::ConnectionRefused("localhost".to_string());
540        assert_eq!(format!("{}", err), "Connection refused: localhost");
541    }
542
543    #[test]
544    fn test_db_error_source() {
545        let err = DbError::PoolError(PoolError::Timeout);
546        assert!(err.source().is_some());
547    }
548
549    #[tokio::test]
550    async fn test_async_trait_export() {
551        fn _check_send_sync<T: Send + Sync>() {}
552
553        struct TestImpl;
554        #[async_trait]
555        trait AsyncFoo: Send + Sync {
556            async fn foo(&self);
557        }
558
559        #[async_trait]
560        impl AsyncFoo for TestImpl {
561            async fn foo(&self) {}
562        }
563
564        let impl_ = TestImpl;
565        impl_.foo().await;
566        _check_send_sync::<TestImpl>();
567    }
568
569    // ---- compile-time SQL validation macro tests ----
570
571    /// Valid SQL should compile and be usable as a string
572    #[test]
573    fn test_sql_string_valid_select() {
574        let sql = sql_string!("SELECT * FROM users WHERE id = 1");
575        assert!(sql.contains("SELECT"));
576        assert!(sql.contains("FROM"));
577    }
578
579    #[test]
580    fn test_sql_string_valid_insert() {
581        let sql = sql_string!("INSERT INTO users (name) VALUES ('alice')");
582        assert!(sql.contains("INSERT"));
583    }
584
585    #[test]
586    fn test_sql_string_valid_update() {
587        let sql = sql_string!("UPDATE users SET name = 'bob' WHERE id = 1");
588        assert!(sql.contains("UPDATE"));
589    }
590
591    #[test]
592    fn test_sql_string_valid_delete() {
593        let sql = sql_string!("DELETE FROM users WHERE id = 1");
594        assert!(sql.contains("DELETE"));
595    }
596
597    #[test]
598    fn test_sql_string_valid_create() {
599        let sql = sql_string!("CREATE TABLE test (id INT PRIMARY KEY)");
600        assert!(sql.contains("CREATE"));
601    }
602
603    #[test]
604    fn test_sql_string_with_params() {
605        let sql = sql_string!("SELECT * FROM users WHERE id = ?"; params: 1);
606        assert!(sql.contains("?"));
607    }
608
609    #[test]
610    fn test_sql_string_complex_query() {
611        let sql = sql_string!(
612            "SELECT u.*, o.total FROM users u LEFT JOIN orders o ON u.id = o.user_id WHERE u.status = 'active'"
613        );
614        assert!(sql.contains("LEFT JOIN"));
615    }
616
617    #[test]
618    fn test_sql_string_nested_parens() {
619        let sql = sql_string!("SELECT * FROM (SELECT * FROM users) t");
620        assert!(sql.contains("SELECT"));
621    }
622}