sqlx pgsql 测试完成

Author: fankaiLiu

templates/classic/_base/src/main.rs.liquid

@@ -8,7 +8,13 @@ use scheduled_thread_pool::ScheduledThreadPool;
 {%- endif %}
 {%- endif %}
 {%- if db_lib == "sqlx" %}
+{%- if db_type == "sqlite" %}
 use sqlx::SqlitePool;
+{%- elsif db_type == "postgres" %}
+use sqlx::PgPool;
+{%- elsif db_type == "mysql" %}
+use sqlx::mysql::MySqlPool;
+{%- endif %}
 {%- endif %}
 use dotenvy::dotenv;
 use salvo::catcher::Catcher;
@@ -87,12 +93,18 @@ async fn main() {
     {%- endif %}
 
     {%- elsif db_lib == "sqlx" %}
+    {%- if db_type == "sqlite" %}
     let sqlx_pool = SqlitePool::connect(&config::get().db.url)
+    {%- elsif db_type == "postgres" %}
+    let sqlx_pool = PgPool::connect(&config::get().db.url)
+    {%- elsif db_type == "mysql" %}
+    let sqlx_pool = MySqlPool::connect(&config::get().db.url)
+    {%- endif %}
         .await
         .expect("Database connection failed.");
     crate::db::SQLX_POOL
         .set(sqlx_pool)
-        .expect("diesel pool should be set");
+        .expect("sqlx pool should be set");
     {%- endif %}
 
     let _guard = config.log.guard();

templates/classic/_base/src/routers/mod.rs

@@ -30,7 +30,7 @@ pub fn root() -> Router {
                         .get(user::list_users)
                         .post(user::create_user)
                         .push(
-                            Router::with_path("{id}")
+                            Router::with_path("{user_id}")
                                 .put(user::update_user)
                                 .delete(user::delete_user),
                         ),

templates/classic/sqlx/src/config/db_config.rs

@@ -53,4 +53,4 @@ fn default_connection_timeout() -> u64 {
 }
 fn default_statement_timeout() -> u64 {
     30000
-}
+}

templates/classic/sqlx/src/db/mod.rs

@@ -0,0 +1,21 @@
+use std::sync::OnceLock;
+
+{%- if db_type == "sqlite" %}
+use sqlx::SqlitePool;
+pub static SQLX_POOL: OnceLock<SqlitePool> = OnceLock::new();
+pub fn pool() -> &'static SqlitePool {
+    SQLX_POOL.get().expect("sqlx pool should be set")
+}
+{%- elsif db_type == "postgres" %}
+use sqlx::postgres::PgPool;
+pub static SQLX_POOL: OnceLock<PgPool> = OnceLock::new();
+pub fn pool() -> &'static PgPool {
+    SQLX_POOL.get().expect("sqlx pool should be set")
+}
+{%- elsif db_type == "mysql" %}
+use sqlx::mysql::MySqlPool;
+pub static SQLX_POOL: OnceLock<MySqlPool> = OnceLock::new();
+pub fn pool() -> &'static MySqlPool {
+    SQLX_POOL.get().expect("sqlx pool should be set")
+}
+{%- endif %}