Multi platform build

Author: max-lt

.dockerignore

@@ -1,6 +1,20 @@
-.vscode
-.git
+# Ignore everything
+*
 
-target
+# Allow bin and src directories
+!/src
+!/bin
 
-.env
+# Allow cargo files
+!Cargo.toml
+!Cargo.lock
+
+# Allow .sqlx directory
+!.sqlx
+
+# Ignore unnecessary files inside allowed directories
+# This should go after the allowed directories
+**/*~
+**/*.log
+**/.DS_Store
+**/Thumbs.db

.github/workflows/docker.yml

@@ -2,7 +2,9 @@ name: Build docker image
 
 on:
   push:
-    branches: [ master ]
+    branches:
+      - master
+      - dev
     tags:
       - 'v*.*.*'
   pull_request:
@@ -22,8 +24,9 @@ jobs:
           images: |
             ghcr.io/${{ github.repository }}
           tags: |
-            type=semver,pattern={{raw}}
             type=raw,value=latest,enable={{is_default_branch}}
+            type=ref,event=branch
+            type=ref,event=tag
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
       - name: Login to GitHub Container Registry
@@ -36,7 +39,9 @@ jobs:
         id: docker_build
         uses: docker/build-push-action@v5
         with:
+          file: Dockerfile.multi
           push: ${{ github.event_name != 'pull_request' }}
           tags: ${{ steps.metadata.outputs.tags }}
           cache-from: type=gha,scope=main
           cache-to: type=gha,mode=max,scope=main
+          platforms: linux/amd64,linux/arm64

Dockerfile

@@ -1,22 +1,22 @@
-FROM rust:1.84 as builder
+FROM rust:1.84 AS builder
 
-RUN mkdir -p /build/openworkers-runner
+RUN mkdir -p /build
 
 ENV RUST_BACKTRACE=1
 ENV RUNTIME_SNAPSHOT_PATH=/build/snapshot.bin
 
-WORKDIR /build/openworkers-runner
+WORKDIR /build
 
-COPY . /build/openworkers-runner
+COPY . /build
 
 RUN touch $RUNTIME_SNAPSHOT_PATH
 
 RUN --mount=type=cache,target=$CARGO_HOME/git \
     --mount=type=cache,target=$CARGO_HOME/registry \
-    --mount=type=cache,target=/build/openworkers-runner/target \
+    --mount=type=cache,target=/build/target \
     cargo run --release --bin snapshot && \
     # Build the runner and copy executable out of the cache so it can be used in the next stage
-    cargo build --release && cp /build/openworkers-runner/target/release/openworkers-runner /build/output
+    cargo build --release && cp /build/target/release/openworkers-runner /build/output
 
 FROM debian:bookworm-slim
 

Dockerfile.multi

@@ -0,0 +1,66 @@
+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.84 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
+
+FROM prepare AS builder
+
+RUN mkdir -p /build
+
+ENV RUST_BACKTRACE=1
+ENV RUNTIME_SNAPSHOT_PATH=/build/snapshot.bin
+
+WORKDIR /build
+
+COPY . /build
+
+RUN touch $RUNTIME_SNAPSHOT_PATH
+
+RUN --mount=type=cache,sharing=locked,target=$CARGO_HOME/git \
+    --mount=type=cache,sharing=locked,target=$CARGO_HOME/registry \
+    --mount=type=cache,sharing=locked,target=/build/openworkers-runner/target \
+    cargo run --release --bin snapshot
+
+FROM builder AS platform
+
+ARG TARGETPLATFORM
+
+RUN echo "Building for $TARGETPLATFORM"
+
+RUN --mount=type=cache,id=apt-$TARGETPLATFORM,sharing=locked,target=$CARGO_HOME/git \
+    --mount=type=cache,id=apt-$TARGETPLATFORM,sharing=locked,target=$CARGO_HOME/registry \
+    --mount=type=cache,id=apt-$TARGETPLATFORM,sharing=locked,target=/build/openworkers-runner/target \
+    case "$TARGETPLATFORM" in \
+        "linux/arm64") cargo build --release --target aarch64-unknown-linux-gnu && \
+                        mv /build/target/aarch64-unknown-linux-gnu/release/openworkers-runner /build/output ;; \
+        "linux/amd64") cargo build --release --target x86_64-unknown-linux-gnu && \
+                        mv /build/target/x86_64-unknown-linux-gnu/release/openworkers-runner  /build/output ;; \
+        *) echo "Unsupported platform: $TARGETPLATFORM" && exit 1 ;; \
+    esac
+
+FROM common AS runner
+
+COPY --from=platform /build/output /usr/local/bin/openworkers-runner
+
+CMD ["/usr/local/bin/openworkers-runner"]
+
+EXPOSE 8080