pool生成関数を切り出す

テストのときとかに使いまわしやすくなった

src/app.rs

@@ -1,9 +1,8 @@
-use anyhow::{Context, Result};
+use anyhow::Result;
 use axum::{
     routing::{get, post},
     Router,
 };
-use sqlx::postgres::PgPool;
 use std::net::SocketAddr;
 use std::sync::Arc;
 
@@ -13,11 +12,9 @@ pub mod equipment;
 /// サーバーの実体
 /// データベースを起動してエントリポイントに応じて関数を呼び出す
 pub async fn app(bind: SocketAddr) -> Result<()> {
-    let database_url =
-        std::env::var("DATABASE_URL").context("Environment variable not set: DATABASE_URL")?;
+    let conn = Arc::new(crate::database::create_pool().await?);
 
     // migrateファイルを適用
-    let conn = Arc::new(PgPool::connect(&database_url).await?);
     crate::database::migrate(&mut conn.acquire().await?).await?;
 
     // pathと関数の実体の紐づけ

src/app/equipment.rs

@@ -18,16 +18,13 @@ pub async fn insert_equipment(
 mod tests {
     use axum::{extract::Json, http::StatusCode};
     use serde_json::json;
-    use sqlx::postgres::PgPool;
     use std::sync::Arc;
 
     use crate::app::equipment::insert_equipment;
 
     #[tokio::test]
     async fn test_insert_equipment() {
-        let database_url = std::env::var("DATABASE_URL").unwrap();
-
-        let conn = Arc::new(PgPool::connect(&database_url).await.unwrap());
+        let conn = Arc::new(crate::database::create_pool().await.unwrap());
 
         let status_code = insert_equipment(
             Json(

src/database.rs

@@ -11,6 +11,7 @@
 //!
 
 use anyhow::{Context, Result};
+use sqlx::{pool::Pool, postgres::PgPool, Postgres};
 
 /// 物品登録を行う関数を提供する
 pub mod insert_equipment;
@@ -26,3 +27,12 @@ where
         .await
         .context("Failed to run migrations")
 }
+
+/// poolを生成する
+pub async fn create_pool() -> Result<Pool<Postgres>> {
+    let database_url =
+        std::env::var("DATABASE_URL").context("Environment variable not set: DATABASE_URL")?;
+    let conn = PgPool::connect(&database_url).await?;
+
+    Ok(conn)
+}