ormer 0.2.12

/// 公共模块 - 包含共享辅助函数、宏定义、连接池和统一接口
pub mod connection_pool;

pub mod common_helpers;

/// SQLite/DuckDB 共用的 DDL 内省与约束校验
pub mod ddl_introspection;

use crate::abstract_layer::DbType;
use crate::model::{Value, VersionSnapshotUpdate};

/// 宏定义模块 - 用于减少重复代码
#[macro_use]
pub mod macros;

/// 流式查询连接管理模块
pub mod stream_connection;
pub use stream_connection::StreamConnection;

/// 统一使用 unified 模块提供接口，当启用任一数据库 feature 时可用
mod unified;
pub use crate::hooks::WithoutHooksExecutor;
#[cfg(feature = "mysql")]
pub(crate) use unified::isolation_level_sql;
#[cfg(any(feature = "clickhouse", feature = "influxdb"))]
pub(crate) use unified::compute_pending_migrations;
pub use unified::{
    AggregateFuture, BatchFuture, BatchManyFuture, BatchQueries, BatchQuery, BatchQueryFuture,
    BlockDeleteExecutor, BlockDeleteResult, CollectFuture, CreateTableExecutor, Database,
    DatabaseScope, DeleteExecutor, DerivedTableCollectFuture, DerivedTableSelectExecutor,
    DoubleIncludedCollectFuture, DoubleIncludedSelectExecutor, DropTableExecutor, FirstFuture,
    FourTableCountFuture, IncludedCollectFuture, IncludedSelectExecutor, InnerJoinedSelectExecutor,
    InsertExecutor, InsertGraphExecutor, InsertOrIgnoreExecutor, InsertOrUpdateExecutor,
    InsertPartialExecutor, IsolationLevel, LeftJoinCollectFuture, LeftJoinedSelectExecutor,
    ModelCollectWithFuture, MultiTableCountFuture, NestedInclude, ProjectionCollectFuture,
    ProjectionSelectExecutor, RawCollectFuture, RawSelectExecutor, RelatedCollectFuture,
    RelatedCountFuture, RelatedSelectExecutor, RelationNestedLoader, ReplicatedDatabase,
    ReplicatedDatabaseBuilder, RightJoinedSelectExecutor, SaveExecutor, ScopedDeleteExecutor,
    ScopedUpdateExecutor, SelectExecutor, SelectStream, SelectStreamIterator, Transaction,
    TransactionFuture, TransactionOptions, TruncateTableExecutor, UnionSelectExecutor,
    UpdateExecutor, UpdateGraphExecutor,
};

// 旧类型名过渡别名（已合并，保留 re-export 以兼容现有导入路径）：
// - Mapped/Grouped* → Projection*（R2）
// - Transaction*Insert* / TransactionRaw* / TransactionSave* → 合并入对应普通执行器（R3）
// - PooledRawSelectExecutor → unified::RawSelectExecutor（R3）
#[allow(deprecated)]
pub use unified::{
    GroupedCollectFuture, GroupedSelectExecutor, MappedCollectFuture, MappedSelectExecutor,
    PooledRawSelectExecutor, TransactionInsertExecutor, TransactionInsertOrIgnoreExecutor,
    TransactionInsertOrUpdateExecutor, TransactionRawCollectFuture, TransactionRawSelectExecutor,
    TransactionSaveExecutor,
};

// 连接池类型 - 根据启用的 feature 导出
pub use connection_pool::{
    ConnectionPool, PooledConnection, PooledDatabaseScope, ReplicatedConnectionPool,
    ReplicatedPoolBuilder,
};

#[derive(Debug, Clone)]
pub struct SingleSqlStatement {
    pub sql: String,
    pub params: Vec<Value>,
    pub param_rust_types: Option<Vec<&'static str>>,
    pub versioned: bool,
    pub version_update: Option<VersionSnapshotUpdate>,
}

impl SingleSqlStatement {
    pub fn new(sql: impl Into<String>, params: Vec<Value>) -> Self {
        Self {
            sql: sql.into(),
            params,
            param_rust_types: None,
            versioned: false,
            version_update: None,
        }
    }

    pub fn with_param_rust_types(mut self, param_rust_types: Vec<&'static str>) -> Self {
        self.param_rust_types = Some(param_rust_types);
        self
    }

    pub fn with_optimistic_lock(
        mut self,
        versioned: bool,
        version_update: Option<VersionSnapshotUpdate>,
    ) -> Self {
        self.versioned = versioned;
        self.version_update = version_update;
        self
    }
}

#[derive(Debug, Clone)]
pub struct SqlStatement {
    pub db_type: DbType,
    pub statements: Vec<SingleSqlStatement>,
}

impl SqlStatement {
    pub fn single(db_type: DbType, sql: impl Into<String>, params: Vec<Value>) -> Self {
        Self {
            db_type,
            statements: vec![SingleSqlStatement::new(sql, params)],
        }
    }

    pub fn batch(db_type: DbType, statements: Vec<SingleSqlStatement>) -> Self {
        Self {
            db_type,
            statements,
        }
    }
}

pub trait SqlExecutor: Sized {
    type Output;

    fn to_sql(&self) -> crate::Result<SqlStatement>;

    fn execute_with_sql(
        self,
        sql: SqlStatement,
    ) -> impl std::future::Future<Output = crate::Result<Self::Output>>;

    fn execute(self) -> impl std::future::Future<Output = crate::Result<Self::Output>> {
        async move {
            let sql = self.to_sql()?;
            self.execute_with_sql(sql).await
        }
    }
}

/// 统一的数据库执行入口。
///
/// 这个 trait 只覆盖 repository/service 最常用的读写方法，返回现有执行器类型，
/// 避免再包一层 Box/Rc 之类的间接层。
pub trait DbExecutor {
    fn select<T: crate::model::Model>(&self) -> SelectExecutor<'_, T>;

    fn select_column<T: crate::model::Model, V>(&self) -> ProjectionSelectExecutor<'_, T, V>;

    fn batch<'a, B>(&'a self, batch: B) -> BatchFuture<'a, B>
    where
        B: BatchQueries<'a>,
    {
        BatchFuture::new(batch)
    }

    fn batch_many<'a, I, Q>(&'a self, queries: I) -> BatchManyFuture<'a, Q>
    where
        I: IntoIterator<Item = Q>,
        Q: BatchQuery<'a>,
    {
        BatchManyFuture::new(queries)
    }
}