Add test entry point to see container lifecycle in stage

Dockerfile

@@ -1,4 +1,4 @@
 FROM --platform=linux/amd64 ubuntu:22.04 as build-image
 
 WORKDIR /src
 RUN apt-get update && apt-get install -y \
@@ -16,21 +16,21 @@
     wget
 
 RUN curl https://sh.rustup.rs -sSf | sh -s -- -y
 ENV PATH "/root/.cargo/bin:${PATH}"
 ENV RUSTUP_HOME "/root/.rustup"
 ENV CARGO_HOME "/root/.cargo"
 RUN rustup toolchain install nightly-2024-07-10
 RUN rustup default nightly-2024-07-10
 RUN rustup component add cargo
 RUN cargo install cargo-build-deps \
     && cargo install cargo-edit
 
 FROM --platform=linux/amd64 build-image as build-app
 WORKDIR /src/gpu-iris-mpc
 COPY . .
 RUN cargo build --release --target x86_64-unknown-linux-gnu --bin nccl --bin server --bin client --bin key-manager --bin upgrade-server --bin upgrade-client --bin upgrade-checker --bin reshare-server --bin reshare-client
 
 FROM --platform=linux/amd64 ghcr.io/worldcoin/iris-mpc-base:cuda12_2-nccl2_22_3_1
 ENV DEBIAN_FRONTEND=noninteractive
 
 # Include client, server and key-manager, upgrade-client and upgrade-server binaries
@@ -44,5 +44,10 @@
 COPY --from=build-app /src/gpu-iris-mpc/target/x86_64-unknown-linux-gnu/release/reshare-server /bin/reshare-server
 COPY --from=build-app /src/gpu-iris-mpc/target/x86_64-unknown-linux-gnu/release/reshare-client /bin/reshare-client
 
+
+# Copy your wrapper entrypoint script into the final container
+COPY test_entrypoint.sh /test_entrypoint.sh
+RUN chmod +x /test_entrypoint.sh
+
 USER 65534
-ENTRYPOINT ["/bin/server"]
+ENTRYPOINT ["/test_entrypoint.sh", "/bin/server"]

deploy/stage/smpcv2-0-stage/values-iris-mpc.yaml

@@ -84,7 +84,7 @@ env:
     value: "true"
 
   - name: SMPC__INIT_DB_SIZE
-    value: "800000"
+    value: "8000000"
 
   - name: SMPC__MAX_DB_SIZE
     value: "1000000"

deploy/stage/smpcv2-1-stage/values-iris-mpc.yaml

@@ -84,7 +84,7 @@ env:
     value: "true"
 
   - name: SMPC__INIT_DB_SIZE
-    value: "800000"
+    value: "8000000"
 
   - name: SMPC__MAX_DB_SIZE
     value: "1000000"

deploy/stage/smpcv2-2-stage/values-iris-mpc.yaml

@@ -84,7 +84,7 @@ env:
     value: "true"
 
   - name: SMPC__INIT_DB_SIZE
-    value: "800000"
+    value: "8000000"
 
   - name: SMPC__MAX_DB_SIZE
     value: "1000000"

test_entrypoint.sh

@@ -0,0 +1,16 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+# Function that prints the date on exit
+on_exit() {
+  echo "[Container End] $(date)  :: Container is stopping..."
+}
+
+# Register the function above to run when the script exits
+trap on_exit EXIT
+
+# Print a timestamp when container starts
+echo "[Container Start] $(date)  :: Container is starting..."
+
+# Execute the main process
+exec "$@"