Add cargo-chef and multi-platform native builds

Author: max-lt

.github/workflows/docker.yml

@@ -38,7 +38,7 @@ jobs:
         id: docker_build
         uses: docker/build-push-action@v5
         with:
-          file: Dockerfile
+          file: Dockerfile.multi
           push: ${{ github.event_name != 'pull_request' }}
           tags: ${{ steps.metadata.outputs.tags }}
           cache-from: type=gha,scope=main

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "openworkers-logs"
-version = "0.1.0"
+version = "0.1.1"
 edition = "2021"
 
 [dependencies]

Cargo.toml

@@ -1,14 +1,24 @@
-FROM rust:1.91 AS builder
+FROM rust:1.91 AS chef
+RUN cargo install cargo-chef
+WORKDIR /build
 
-RUN mkdir -p /build
+FROM chef AS planner
+COPY . .
+RUN cargo chef prepare --recipe-path recipe.json
 
+FROM chef AS builder
 ENV RUST_BACKTRACE=1
 ENV SQLX_OFFLINE=true
 
-WORKDIR /build
-
-COPY . /build
+COPY --from=planner /build/recipe.json recipe.json
+# Build dependencies - this is the caching Docker layer!
+RUN --mount=type=cache,target=$CARGO_HOME/git \
+    --mount=type=cache,target=$CARGO_HOME/registry \
+    --mount=type=cache,target=/build/target \
+    cargo chef cook --release --recipe-path recipe.json
 
+# Build application
+COPY . .
 RUN --mount=type=cache,target=$CARGO_HOME/git \
     --mount=type=cache,target=$CARGO_HOME/registry \
     --mount=type=cache,target=/build/target \

Dockerfile

@@ -0,0 +1,74 @@
+FROM debian:bookworm-slim AS common
+
+RUN apt-get update \
+    # Install ca-certificates and wget (used for healthcheck)
+    && apt-get install -y ca-certificates wget \
+    && rm -rf /var/lib/apt/lists/*
+
+FROM --platform=$BUILDPLATFORM rust:1.91 AS prepare
+
+RUN apt-get update && apt-get install -y \
+    gcc-aarch64-linux-gnu \
+    gcc-x86-64-linux-gnu \
+    libc6-dev-amd64-cross \
+    libc6-dev-arm64-cross
+
+ENV CC_x86_64_unknown_linux_gnu=x86_64-linux-gnu-gcc
+ENV CC_aarch64_unknown_linux_gnu=aarch64-linux-gnu-gcc
+
+ENV CARGO_TARGET_X86_64_UNKNOWN_LINUX_GNU_LINKER=x86_64-linux-gnu-gcc
+ENV CARGO_TARGET_AARCH64_UNKNOWN_LINUX_GNU_LINKER=aarch64-linux-gnu-gcc
+
+RUN rustup target add x86_64-unknown-linux-gnu \
+                      aarch64-unknown-linux-gnu
+
+RUN cargo install cargo-chef
+
+FROM prepare AS planner
+
+WORKDIR /build
+COPY . .
+RUN cargo chef prepare --recipe-path recipe.json
+
+FROM prepare AS builder
+
+WORKDIR /build
+
+ENV RUST_BACKTRACE=1
+ENV SQLX_OFFLINE=true
+
+COPY --from=planner /build/recipe.json recipe.json
+
+ARG TARGETPLATFORM
+
+# Build dependencies for the target platform
+RUN --mount=type=cache,id=cargo-$TARGETPLATFORM,sharing=locked,target=$CARGO_HOME/git \
+    --mount=type=cache,id=cargo-$TARGETPLATFORM,sharing=locked,target=$CARGO_HOME/registry \
+    --mount=type=cache,id=cargo-$TARGETPLATFORM,sharing=locked,target=/build/target \
+    case "$TARGETPLATFORM" in \
+        "linux/arm64") cargo chef cook --release --target aarch64-unknown-linux-gnu --recipe-path recipe.json ;; \
+        "linux/amd64") cargo chef cook --release --target x86_64-unknown-linux-gnu --recipe-path recipe.json ;; \
+        *) echo "Unsupported platform: $TARGETPLATFORM" && exit 1 ;; \
+    esac
+
+# Build application
+COPY . .
+
+RUN --mount=type=cache,id=cargo-$TARGETPLATFORM,sharing=locked,target=$CARGO_HOME/git \
+    --mount=type=cache,id=cargo-$TARGETPLATFORM,sharing=locked,target=$CARGO_HOME/registry \
+    --mount=type=cache,id=cargo-$TARGETPLATFORM,sharing=locked,target=/build/target \
+    case "$TARGETPLATFORM" in \
+        "linux/arm64") cargo build --release --target aarch64-unknown-linux-gnu && \
+                        cp /build/target/aarch64-unknown-linux-gnu/release/openworkers-logs /build/output ;; \
+        "linux/amd64") cargo build --release --target x86_64-unknown-linux-gnu && \
+                        cp /build/target/x86_64-unknown-linux-gnu/release/openworkers-logs /build/output ;; \
+        *) echo "Unsupported platform: $TARGETPLATFORM" && exit 1 ;; \
+    esac
+
+FROM common AS runner
+
+COPY --from=builder /build/output /usr/local/bin/openworkers-logs
+
+CMD ["/usr/local/bin/openworkers-logs"]
+
+EXPOSE 8080