From 66fc20ee54f3040f22ee3b4a5f48d11e84c21056 Mon Sep 17 00:00:00 2001
From: "peng.xu" <peng.xu@zilliz.com>
Date: Mon, 14 Oct 2019 15:35:29 +0800
Subject: [PATCH] update build.sh

update build.sh
---
 build.sh | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/build.sh b/build.sh
index 2b3c89bbf9..8e142d0115 100755
--- a/build.sh
+++ b/build.sh
@@ -21,12 +21,17 @@ function build_image() {
 case "$1" in
 
 all)
+    [[ -z $MISHARDS_REGISTRY ]] && {
+        echo -e "${YELLOW}Error: Please set docker registry first:${ENDC}\n\t${BOLD}export MISHARDS_REGISTRY=xxxx\n${ENDC}"
+        exit 1
+    }
+
     version=""
     [[ ! -z $2 ]] && version=":${2}"
-    build_image "Dockerfile" "registry.zilliz.com/milvus/mishards${version}" "registry.zilliz.com/milvus/mishards"
+    build_image "Dockerfile" "${MISHARDS_REGISTRY}${version}" "${MISHARDS_REGISTRY}"
     ;;
 *)
     echo "Usage: [option...] {base | apps}"
-    echo "all,      Usage: build.sh all [tagname|] => registry.zilliz.com/milvus/mishards:\${tagname}"
+    echo "all,      Usage: build.sh all [tagname|] => {docker_registry}:\${tagname}"
     ;;
 esac