docker

.env

@@ -2,4 +2,4 @@ POSTGRES_USER=testuser
 POSTGRES_PASSWORD=testuserpwd
 POSTGRES_DB=locust
 POSTGRES_HOST=postgres-locust
-POSTGRES_PORT=5436
+POSTGRES_PORT=5432

Dockerfile

@@ -0,0 +1,20 @@
+FROM lukemathwalker/cargo-chef:latest-rust-1 AS chef
+WORKDIR /app
+
+FROM chef AS planner
+COPY . .
+RUN cargo chef prepare --recipe-path recipe.json
+
+FROM chef AS builder 
+COPY --from=planner /app/recipe.json recipe.json
+# Build dependencies - this is the caching Docker layer!
+RUN cargo chef cook --release --recipe-path recipe.json
+# Build application
+COPY . .
+RUN cargo build --release --bin locust
+
+# We do not need the Rust toolchain to run the binary!
+FROM debian:bookworm-slim AS runtime
+WORKDIR /app
+COPY --from=builder /app/target/release/locust /usr/local/bin
+ENTRYPOINT ["/usr/local/bin/locust"]

docker-compose.yml

@@ -1,5 +1,19 @@
 services:
+  locust:
+    networks:
+      - locust-net
+    image: locust:latest
+    container_name: locust
+    build:
+      context: .
+    env_file: .env
+    depends_on:
+      - postgres-locust
+      - flyway-locust
+
   postgres-locust:
+    networks:
+      - locust-net
     container_name: postgres-locust
     image: postgres:15-alpine
     environment:
@@ -17,6 +31,8 @@ services:
       - postgres-locust:/var/lib/postgresql/data/
 
   flyway-locust:
+    networks:
+      - locust-net
     container_name: flyway-locust
     environment:
       FLYWAY_USER: ${POSTGRES_USER}
@@ -30,5 +46,9 @@ services:
     depends_on:
       - postgres-locust
 
+networks:
+  locust-net:
+    external: false
+
 volumes:
   postgres-locust:

locust-core/src/lib.rs

@@ -9,13 +9,13 @@ pub async fn new_pool() -> Result<PgPool, Error> {
     let user = env::var("POSTGRES_USER").unwrap();
     let pwd = env::var("POSTGRES_PASSWORD").unwrap_or("password".into());
     let db = env::var("POSTGRES_DB").unwrap_or("postgres".into());
-    // let host = env::var("POSTGRES_HOST").unwrap_or("localhost".into());
-    let host = "localhost";
+    let host = env::var("POSTGRES_HOST").unwrap_or("localhost".into());
     let port = env::var("POSTGRES_PORT")
         .unwrap_or("5432".into())
         .parse::<usize>()
         .expect("Invalid psql port");
     let conn_string = format!("postgresql://{}:{}@{}:{}/{}", user, pwd, host, port, db);
+    println!("{}", conn_string);
 
     let pool = PgPoolOptions::new().connect(&conn_string).await?;
 

src/main.rs

@@ -57,8 +57,8 @@ impl ServiceWrapper {
 async fn main() {
     tracing_subscriber::fmt::init();
 
-    let mut private_key_bytes: &[u8] = include_bytes!("ca/hudsucker.key");
-    let mut ca_cert_bytes: &[u8] = include_bytes!("ca/hudsucker.cer");
+    let mut private_key_bytes: &[u8] = include_bytes!("ca/locust.key");
+    let mut ca_cert_bytes: &[u8] = include_bytes!("ca/locust.cer");
     let private_key = tokio_rustls::rustls::PrivateKey(
         pemfile::pkcs8_private_keys(&mut private_key_bytes)
             .next()
@@ -84,6 +84,7 @@ async fn main() {
         db: Arc::new(db_pool),
     };
 
+    println!("Starting up proxy server!");
     if let Err(e) = wrapper.start(shutdown_signal()).await {
         error!("{}", e);
     }