From c8c02f6f6e5034563d70d49be5a24a7424e82fb4 Mon Sep 17 00:00:00 2001
From: Billy Chan <ccw.billy.123@gmail.com>
Date: Thu, 28 Mar 2024 16:25:31 +0800
Subject: [PATCH] Upstream Changes - 5 (#2176)

* Upstream Changes - 5

* clippy

* more
---
 sea-orm-migration/src/migrator.rs |  4 ++--
 src/executor/delete.rs            | 12 ++++++------
 src/executor/insert.rs            | 20 +++++++++++---------
 src/executor/paginator.rs         | 26 +++++++++++++-------------
 4 files changed, 32 insertions(+), 30 deletions(-)

diff --git a/sea-orm-migration/src/migrator.rs b/sea-orm-migration/src/migrator.rs
index d5880288..05329e5f 100644
--- a/sea-orm-migration/src/migrator.rs
+++ b/sea-orm-migration/src/migrator.rs
@@ -313,7 +313,7 @@ where
     }
 
     // Drop all tables
-    let stmt = query_tables(db);
+    let stmt = query_tables(db).await;
     let rows = db.query_all(db_backend.build(&stmt)).await?;
     for row in rows.into_iter() {
         let table_name: String = row.try_get("", "table_name")?;
@@ -437,7 +437,7 @@ where
     Ok(())
 }
 
-fn query_tables<C>(db: &C) -> SelectStatement
+async fn query_tables<C>(db: &C) -> SelectStatement
 where
     C: ConnectionTrait,
 {
diff --git a/src/executor/delete.rs b/src/executor/delete.rs
index 08d1d8f1..e1663a6c 100644
--- a/src/executor/delete.rs
+++ b/src/executor/delete.rs
@@ -1,6 +1,4 @@
-use crate::{
-    error::*, ActiveModelTrait, ConnectionTrait, DeleteMany, DeleteOne, EntityTrait, Statement,
-};
+use crate::{error::*, ActiveModelTrait, ConnectionTrait, DeleteMany, DeleteOne, EntityTrait};
 use sea_query::DeleteStatement;
 use std::future::Future;
 
@@ -56,8 +54,7 @@ impl Deleter {
     where
         C: ConnectionTrait,
     {
-        let builder = db.get_database_backend();
-        exec_delete(builder.build(&self.query), db)
+        exec_delete(self.query, db)
     }
 }
 
@@ -68,10 +65,13 @@ where
     Deleter::new(query).exec(db).await
 }
 
-async fn exec_delete<C>(statement: Statement, db: &C) -> Result<DeleteResult, DbErr>
+async fn exec_delete<C>(query: DeleteStatement, db: &C) -> Result<DeleteResult, DbErr>
 where
     C: ConnectionTrait,
 {
+    let builder = db.get_database_backend();
+    let statement = builder.build(&query);
+
     let result = db.execute(statement).await?;
     Ok(DeleteResult {
         rows_affected: result.rows_affected(),
diff --git a/src/executor/insert.rs b/src/executor/insert.rs
index 1bd074d6..34b80337 100644
--- a/src/executor/insert.rs
+++ b/src/executor/insert.rs
@@ -1,7 +1,6 @@
 use crate::{
     error::*, ActiveModelTrait, ColumnTrait, ConnectionTrait, EntityTrait, Insert, IntoActiveModel,
-    Iterable, PrimaryKeyToColumn, PrimaryKeyTrait, SelectModel, SelectorRaw, Statement, TryFromU64,
-    TryInsert,
+    Iterable, PrimaryKeyToColumn, PrimaryKeyTrait, SelectModel, SelectorRaw, TryFromU64, TryInsert,
 };
 use sea_query::{FromValueTuple, Iden, InsertStatement, Query, ValueTuple};
 use std::{future::Future, marker::PhantomData};
@@ -176,8 +175,7 @@ where
         C: ConnectionTrait,
         A: 'a,
     {
-        let builder = db.get_database_backend();
-        exec_insert(self.primary_key, builder.build(&self.query), db)
+        exec_insert(self.primary_key, self.query, db)
     }
 
     /// Execute an insert operation
@@ -208,13 +206,16 @@ where
 
 async fn exec_insert<A, C>(
     primary_key: Option<ValueTuple>,
-    statement: Statement,
+    statement: InsertStatement,
     db: &C,
 ) -> Result<InsertResult<A>, DbErr>
 where
     C: ConnectionTrait,
     A: ActiveModelTrait,
 {
+    let db_backend = db.get_database_backend();
+    let statement = db_backend.build(&statement);
+
     type PrimaryKey<A> = <<A as ActiveModelTrait>::Entity as EntityTrait>::PrimaryKey;
     type ValueTypeOf<A> = <PrimaryKey<A> as PrimaryKeyTrait>::ValueType;
 
@@ -259,7 +260,8 @@ where
     C: ConnectionTrait,
 {
     let db_backend = db.get_database_backend();
-    let exec_result = db.execute(db_backend.build(&insert_statement)).await?;
+    let insert_statement = db_backend.build(&insert_statement);
+    let exec_result = db.execute(insert_statement).await?;
     Ok(exec_result.rows_affected())
 }
 
@@ -281,15 +283,15 @@ where
                     .map(|c| c.select_as(c.into_returning_expr(db_backend))),
             );
             insert_statement.returning(returning);
+            let insert_statement = db_backend.build(&insert_statement);
             SelectorRaw::<SelectModel<<A::Entity as EntityTrait>::Model>>::from_statement(
-                db_backend.build(&insert_statement),
+                insert_statement,
             )
             .one(db)
             .await?
         }
         false => {
-            let insert_res =
-                exec_insert::<A, _>(primary_key, db_backend.build(&insert_statement), db).await?;
+            let insert_res = exec_insert::<A, _>(primary_key, insert_statement, db).await?;
             <A::Entity as EntityTrait>::find_by_id(insert_res.last_insert_id)
                 .one(db)
                 .await?
diff --git a/src/executor/paginator.rs b/src/executor/paginator.rs
index 71b55bc4..55512585 100644
--- a/src/executor/paginator.rs
+++ b/src/executor/paginator.rs
@@ -67,19 +67,19 @@ where
     /// Get the total number of items
     pub async fn num_items(&self) -> Result<u64, DbErr> {
         let builder = self.db.get_database_backend();
-        let stmt = builder.build(
-            SelectStatement::new()
-                .expr(Expr::cust("COUNT(*) AS num_items"))
-                .from_subquery(
-                    self.query
-                        .clone()
-                        .reset_limit()
-                        .reset_offset()
-                        .clear_order_by()
-                        .to_owned(),
-                    Alias::new("sub_query"),
-                ),
-        );
+        let stmt = SelectStatement::new()
+            .expr(Expr::cust("COUNT(*) AS num_items"))
+            .from_subquery(
+                self.query
+                    .clone()
+                    .reset_limit()
+                    .reset_offset()
+                    .clear_order_by()
+                    .to_owned(),
+                Alias::new("sub_query"),
+            )
+            .to_owned();
+        let stmt = builder.build(&stmt);
         let result = match self.db.query_one(stmt).await? {
             Some(res) => res,
             None => return Ok(0),