在Dockerfile中动态设置LD_PRELOAD环境变量（多架构场景）

问题描述 #

我使用GitHub Actions的docker/build-push-action@v6构建linux/amd64、linux/arm64多架构Docker镜像，需要在Dockerfile中动态设置LD_PRELOAD环境变量，其值为/usr/lib/${ARCH}-linux-gnu/libjemalloc.so.2——其中ARCH对应x86_64（适配amd64架构）或aarch64（适配arm64架构）。由于RUN块中设置的环境变量会在块结束后丢失，特此寻求实现动态设置的方法。

现有配置 #

GitHub Actions 配置

- name: Build and push
        uses: docker/build-push-action@v6
        with:
          context: my-dir
          push: true
          tags: myrepo.io/my-registry/my-app:${{ github.event_name == 'repository_dispatch' && github.event.client_payload.latest_ver || github.event.inputs.certified_version }}-multiarch
          file: Dockerfile_multiarch
          platforms: linux/amd64,linux/arm64
          provenance: false

Dockerfile 内容

FROM ubuntu:20.04

ENV DEBIAN_FRONTEND noninteractive

# Install required packages and dependencies for RPM
RUN apt-get update && \
    apt-get install -y wget curl vim less git python3 linux-tools-common sysstat procps libjemalloc-dev gnupg && \
    rm -rf /var/lib/apt/lists/*

ENV JAVA_HOME="/usr/lib/jvm/java-21-amazon-corretto"
ENV PATH=$JAVA_HOME/bin:$PATH

//todo: How can I do this?
ENV LD_PRELOAD=/usr/lib/${ARCH}-linux-gnu/libjemalloc.so.2

解决方案 #

方法一：利用Docker内置构建参数（推荐） #

Docker在构建多架构镜像时，会自动为每个目标平台传递内置构建参数TARGETARCH：

• 构建linux/amd64时，TARGETARCH值为x86_64
• 构建linux/arm64时，TARGETARCH值为aarch64

直接修改Dockerfile，替换todo部分：

# 利用Docker内置的TARGETARCH参数动态设置LD_PRELOAD
ARG TARGETARCH
ENV LD_PRELOAD=/usr/lib/${TARGETARCH}-linux-gnu/libjemalloc.so.2

无需修改GitHub Actions配置，docker/build-push-action会自动处理多平台构建时的参数传递，每个架构的镜像会生成对应的LD_PRELOAD值。

方法二：手动传递架构构建参数（自定义场景） #

如果需要自定义架构映射逻辑，可以在GitHub Actions中手动传递build-arg，配合多平台构建：

修改GitHub Actions配置，添加build-args字段：

- name: Build and push
        uses: docker/build-push-action@v6
        with:
          context: my-dir
          push: true
          tags: myrepo.io/my-registry/my-app:${{ github.event_name == 'repository_dispatch' && github.event.client_payload.latest_ver || github.event.inputs.certified_version }}-multiarch
          file: Dockerfile_multiarch
          platforms: linux/amd64,linux/arm64
          provenance: false
          build-args: |
            ARCH=${{ TARGETARCH }}

然后修改Dockerfile：

ARG ARCH
ENV LD_PRELOAD=/usr/lib/${ARCH}-linux-gnu/libjemalloc.so.2

方法三：运行时动态设置（灵活适配场景） #

如果需要在容器启动时根据实际运行的架构动态设置变量，可以编写entrypoint脚本：

1. 修改Dockerfile，添加entrypoint脚本并设置为入口：

# 创建entrypoint脚本，运行时判断架构并设置LD_PRELOAD
RUN echo '#!/bin/bash
ARCH=$(uname -m)
if [ "$ARCH" = "x86_64" ]; then
    export LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libjemalloc.so.2
elif [ "$ARCH" = "aarch64" ]; then
    export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2
fi
exec "$@"' > /entrypoint.sh && chmod +x /entrypoint.sh

# 设置容器入口
ENTRYPOINT ["/entrypoint.sh"]
# 替换为你的应用启动命令
CMD ["java", "-jar", "your-app.jar"]

这种方式会在容器启动时自动检测当前运行的架构，设置对应的LD_PRELOAD值，适合需要在单一镜像中兼容多架构的场景（不过多架构镜像本身是分平台构建的，方法一的构建时设置更高效）。

内容的提问来源于stack exchange，提问作者Prashant Pandey