feat: support multi-arch (amd64+arm64) Docker image builds (#3)

Author: yunyu950908

.dockerignore

@@ -0,0 +1,3 @@
+*
+!server/
+server/bin/

.github/workflows/build-swr.yml

@@ -14,17 +14,14 @@ jobs:
         with:
           fetch-depth: 1
 
-      - name: Set up Go
-        uses: actions/setup-go@v5
-        with:
-          go-version-file: server/go.mod
-          cache-dependency-path: server/go.sum
-
       - name: Set image variables
         run: |
           echo "SHORT_SHA=${GITHUB_SHA::7}" >> "$GITHUB_ENV"
           echo "SWR_IMG=${{ secrets.SWR_REGISTRY }}/${{ secrets.SWR_ORGANIZATION }}/mnemo-server" >> "$GITHUB_ENV"
 
+      - name: Set up QEMU
+        uses: docker/setup-qemu-action@v3
+
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
 
@@ -35,13 +32,15 @@ jobs:
           username: ${{ secrets.SWR_USERNAME }}
           password: ${{ secrets.SWR_PASSWORD }}
 
-      - name: Build image
-        run: REGISTRY=${{ secrets.SWR_REGISTRY }}/${{ secrets.SWR_ORGANIZATION }} COMMIT=${{ env.SHORT_SHA }} make docker
-
-      - name: Tag latest
-        run: docker tag "$SWR_IMG:$SHORT_SHA" "$SWR_IMG:latest"
-
-      - name: Push image tags
-        run: |
-          docker push "$SWR_IMG:$SHORT_SHA"
-          docker push "$SWR_IMG:latest"
+      - name: Build and push multi-arch image
+        uses: docker/build-push-action@v6
+        with:
+          context: .
+          file: ./server/Dockerfile
+          platforms: linux/amd64,linux/arm64
+          push: true
+          tags: |
+            ${{ env.SWR_IMG }}:${{ env.SHORT_SHA }}
+            ${{ env.SWR_IMG }}:latest
+          cache-from: type=gha
+          cache-to: type=gha,mode=max

AGENTS.md

@@ -59,7 +59,7 @@ cd server && MNEMO_DSN="user:pass@tcp(host:4000)/db?parseTime=true" go run ./cmd
 - `INSERT ... ON DUPLICATE KEY UPDATE` is the expected upsert pattern.
 - Atomic version bump happens in SQL: `SET version = version + 1`.
 - `X-Mnemo-Agent-Id` is the per-agent identity header for memory requests.
-- Always use `make` targets for building and Docker image operations — never construct raw `go build` or `docker build` commands from scratch. Use `make build-linux` for the server binary and `REGISTRY=<ecr> COMMIT=<tag> make docker` for images.
+- Always use `make` targets for building and Docker image operations — never construct raw `go build` or `docker build` commands from scratch. Use `make build-linux` for the server binary, `REGISTRY=<ecr> COMMIT=<tag> make docker` for local single-arch images, and `make docker-push` for multi-arch (amd64+arm64) build & push.
 
 ## Go style
 

Makefile

@@ -1,7 +1,7 @@
 MAKEFILE_DIR:=$(shell dirname $(realpath $(firstword $(MAKEFILE_LIST))))
 IMG ?= $(REGISTRY)/mnemo-server:$(COMMIT)
 
-.PHONY: build vet clean run test test-integration docker
+.PHONY: build vet clean run test test-integration docker docker-push
 
 build:
 	mkdir -p $(MAKEFILE_DIR)/server/bin
@@ -26,6 +26,9 @@ clean:
 run: build
 	cd server && MNEMO_DSN="$(MNEMO_DSN)" ./bin/mnemo-server
 
-docker: build-linux
-	docker build --platform=linux/amd64 -q -f ./server/Dockerfile -t $(IMG) .
+docker:
+	docker buildx build --platform=linux/amd64 --load -f ./server/Dockerfile -t $(IMG) .
+
+docker-push:
+	docker buildx build --platform=linux/amd64,linux/arm64 --push -f ./server/Dockerfile -t $(IMG) .
 

server/Dockerfile

@@ -1,12 +1,18 @@
-# FROM golang:1.22-alpine AS builder
-# WORKDIR /app
-# COPY go.mod go.sum ./
-# RUN go mod download
-# COPY . .
-# RUN CGO_ENABLED=0 go build -o /mnemo-server ./cmd/mnemo-server
+# --- Build stage: always runs on host arch for speed ---
+FROM --platform=$BUILDPLATFORM golang:1.24-alpine AS builder
 
+ARG TARGETOS
+ARG TARGETARCH
+
+WORKDIR /app
+COPY server/go.mod server/go.sum ./
+RUN go mod download
+COPY server/ .
+RUN CGO_ENABLED=0 GOOS=${TARGETOS} GOARCH=${TARGETARCH} go build -o /mnemo-server ./cmd/mnemo-server
+
+# --- Runtime stage: target arch ---
 FROM alpine:3.19
 RUN apk add --no-cache ca-certificates
-COPY ./server/bin/mnemo-server /mnemo-server
+COPY --from=builder /mnemo-server /mnemo-server
 EXPOSE 8080
 ENTRYPOINT ["/mnemo-server"]