From 2ad41872dab4f61b8fefc2e4d9d858cc45864153 Mon Sep 17 00:00:00 2001
From: WAR10CK <war10ck.efz@gmail.com>
Date: Tue, 26 Aug 2025 11:39:14 +0800
Subject: [PATCH] enhance: support enable asan for docker image builder
 (#44047)

#44045

Signed-off-by: YangYanbin <warlock.yyb@alibaba-inc.com>
---
 build/build_image.sh | 13 +++++++++++--
 docker-compose.yml   |  1 +
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/build/build_image.sh b/build/build_image.sh
index 9480dcb368..b65e6bac6c 100755
--- a/build/build_image.sh
+++ b/build/build_image.sh
@@ -44,8 +44,17 @@ fi
 
 echo ${IMAGE_ARCH}
 
-
-BUILD_ARGS="${BUILD_ARGS:---build-arg TARGETARCH=${IMAGE_ARCH}}"
+if [[ "$USE_ASAN" == "ON" ]]; then
+    if [[ "$OS_NAME" == "ubuntu20.04" ]]; then
+        BUILD_ARGS="${BUILD_ARGS:---build-arg TARGETARCH=${IMAGE_ARCH} --build-arg MILVUS_ASAN_LIB=/milvus/lib/libasan.so.6}"
+    elif [[ "$OS_NAME" == "ubuntu22.04" ]]; then
+        BUILD_ARGS="${BUILD_ARGS:---build-arg TARGETARCH=${IMAGE_ARCH} --build-arg MILVUS_ASAN_LIB=/milvus/lib/libasan.so.8}"
+    else
+        BUILD_ARGS="${BUILD_ARGS:---build-arg TARGETARCH=${IMAGE_ARCH}}"
+    fi
+else
+    BUILD_ARGS="${BUILD_ARGS:---build-arg TARGETARCH=${IMAGE_ARCH}}"
+fi
 
 pushd "${toplevel}"
 
diff --git a/docker-compose.yml b/docker-compose.yml
index 6e58a06e16..21a7739b06 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -32,6 +32,7 @@ services:
       CONAN_USER_HOME: /home/milvus
       AZURE_STORAGE_CONNECTION_STRING: ${AZURITE_CONNECTION_STRING}
       ENABLE_GCP_NATIVE: ${ENABLE_GCP_NATIVE}
+      USE_ASAN: ${USE_ASAN}
     volumes: &builder-volumes
       - .:/go/src/github.com/milvus-io/milvus:delegated
       - ${DOCKER_VOLUME_DIRECTORY:-.docker}/${IMAGE_ARCH}-${OS_NAME}-ccache:/ccache:delegated