add http port env var

Author: resoluteCoder

Cargo.toml

@@ -12,3 +12,7 @@ tokio = { version = "1.41.1", features = ["macros", "rt-multi-thread"] }
 tower-http = { version = "0.6.2", features = ["trace"] }
 tracing = "0.1.41"
 tracing-subscriber = "0.3.19"
+
+[[bin]]
+name = "local-detectors"
+path = "src/main.rs"

Dockerfile

@@ -0,0 +1,22 @@
+# Use the official Rust image as a base
+FROM rust:latest
+
+# Set the working directory to /app
+WORKDIR /app
+
+# Copy the Cargo.toml file into the working directory
+COPY Cargo.toml .
+
+COPY src/ /app/src/
+
+# Build the Rust application
+RUN cargo build --release
+
+# Copy the built application into the working directory
+COPY target/release/* .
+
+# Expose the port that the application will listen on
+EXPOSE 8080
+
+# Run the command to start the application when the container is launched
+CMD ["/app/local-detectors"]

src/detectors.rs

@@ -5,6 +5,7 @@ use regex::Regex;
 use axum::{http::StatusCode, response::IntoResponse, Json};
 use serde::{Deserialize, Serialize};
 use serde_json::json;
+use tracing::info;
 
 #[derive(Serialize, Deserialize, Debug)]
 #[cfg_attr(test, derive(PartialEq))]
@@ -92,9 +93,14 @@ fn regex_match(
     }
 }
 
+// pub async fn handle_text_contents(Json(payload): Json<DetectionRequest>) {
+//     info!("incoming payload: {:#?}", payload);
+// }
 pub async fn handle_text_contents(
     Json(payload): Json<DetectionRequest>,
 ) -> Result<impl IntoResponse, (StatusCode, String)> {
+    println!("hi");
+    info!("incoming payload: {:?}", payload);
     if payload.detector_params.regex.is_empty() {
         return Err((StatusCode::BAD_REQUEST, format!("empty regex")));
     }

src/main.rs

@@ -1,4 +1,4 @@
-use std::net::SocketAddr;
+use std::{env, net::SocketAddr};
 
 use axum::{
     routing::{get, post},
@@ -18,6 +18,14 @@ async fn main() {
         .compact()
         .init();
 
+    let mut http_port = 8080;
+    if let Ok(port) = env::var("HTTP_PORT") {
+        match port.parse::<u16>() {
+            Ok(port) => http_port = port,
+            Err(err) => println!("{}", err),
+        }
+    }
+
     let app = Router::new()
         .route("/health", get(|| async { "Hello, World!" }))
         .route("/api/v1/text/contents", post(handle_text_contents))
@@ -27,7 +35,7 @@ async fn main() {
                 .on_response(trace::DefaultOnResponse::new().level(Level::INFO)),
         );
 
-    let addr = SocketAddr::from(([127, 0, 0, 1], 8080));
+    let addr = SocketAddr::from(([127, 0, 0, 1], http_port));
     tracing::info!("listening on {}", addr);
 
     let listener = tokio::net::TcpListener::bind(addr).await.unwrap();