add API_HOST env var

osuAkatsuki · Nov 5, 2023 · 363250a · 363250a
1 parent 2f61733
commit 363250a
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/.env.example b/.env.example
@@ -1,4 +1,5 @@
 APP_ENV=dev
+API_HOST=127.0.0.1
 API_PORT=8665
 DATABASE_HOST=localhost
 DATABASE_PORT=3306
@@ -18,4 +19,4 @@ REDIS_PASSWORD=
 REDIS_DATABASE=0
 BEATMAPS_PATH=
 SERVICE_READINESS_TIMEOUT=60
-RUST_LOG=performance_service=info
+RUST_LOG=performance_service=info
diff --git a/src/api/mod.rs b/src/api/mod.rs
@@ -21,6 +21,7 @@ fn api_router() -> Router {
 
 pub async fn serve(ctx: Context) -> anyhow::Result<()> {
     let server_port = ctx.config.api_port.unwrap();
+    let server_host = ctx.config.api_host.clone().unwrap();
 
     let app = api_router().layer(
         ServiceBuilder::new()
@@ -29,7 +30,7 @@ pub async fn serve(ctx: Context) -> anyhow::Result<()> {
     );
 
     log::info!("serving on {}", server_port);
-    axum::Server::bind(&format!("127.0.0.1:{}", server_port).parse()?)
+    axum::Server::bind(&format!("{}:{}", server_host, server_port).parse()?)
         .serve(app.into_make_service())
         .await?;
 

diff --git a/src/config.rs b/src/config.rs
@@ -3,6 +3,9 @@ pub struct Config {
     #[clap(long, env)]
     pub app_component: String,
 
+    #[clap(long, env)]
+    pub api_host: Option<String>,
+
     #[clap(long, env)]
     pub api_port: Option<u16>,