support arm64 compilation based on docker-build

Signed-off-by: yejin-fxn <[email protected]>

Author: yejin-fxn

docker-build/build_cluster_docker.sh

@@ -6,6 +6,8 @@
 #!/usr/bin/env bash
 set -e
 
+: "${Platform:=amd64}"
+
 BASEDIR=$(dirname "$0")
 cd $BASEDIR
 WORKINGDIR=`pwd`
@@ -32,16 +34,17 @@ echo "  Tag: ${TAG}"
 echo "  BASEDIR: ${BASEDIR}"
 echo "  WORKINGDIR: ${WORKINGDIR}"
 echo "  source_code_dir: ${source_code_dir}"
+echo "  Platform: ${Platform}"
 
 package() {
-  docker run --rm -u $(id -u):$(id -g) -v ${source_code_dir}:/data/projects/fate/FATE-Serving --entrypoint="" maven:3.6-jdk-8 /bin/bash -c "cd /data/projects/fate/FATE-Serving && mvn clean package -DskipTests"
+  docker run --rm --platform linux/amd64 -v ${source_code_dir}:/data/projects/fate/FATE-Serving --entrypoint="" maven:3.6-jdk-8 /bin/bash -c "cd /data/projects/fate/FATE-Serving && mvn clean package -DskipTests"
 }
 
 buildModule() {
   for module in "serving-proxy" "serving-server" "serving-admin"
   do
       echo "### START BUILDING ${module} ###"
-      docker build --build-arg version=${version} -t ${PREFIX}/${module}:${TAG} -f ${source_code_dir}/docker-build/${module}/Dockerfile ${source_code_dir}
+      docker build --platform linux/${Platform} --build-arg version=${version} -t ${PREFIX}/${module}:${TAG} -f ${source_code_dir}/docker-build/${module}/Dockerfile.${Platform} ${source_code_dir}
       echo "### FINISH BUILDING ${module} ###"
       echo ""
   done;

docker-build/serving-admin/Dockerfile renamed to docker-build/serving-admin/Dockerfile.amd64

@@ -0,0 +1,25 @@
+FROM openjdk:8u342-jre as builder
+
+
+WORKDIR /data/projects/fate-serving/serving-admin/
+USER root
+
+ARG version
+
+COPY fate-serving-admin/target/fate-serving-admin-${version}-release.zip ./
+
+RUN unzip fate-serving-admin-${version}-release.zip && \
+    rm fate-serving-admin-${version}-release.zip && \
+    ln -s fate-serving-admin-*.jar fate-serving-admin.jar
+
+FROM openjdk:8u342-jre
+
+WORKDIR /data/projects/fate-serving/serving-admin/
+
+COPY --from=builder /data/projects/fate-serving/serving-admin  /data/projects/fate-serving/serving-admin
+
+USER root
+
+EXPOSE 8350
+
+CMD java -Dspring.config.location=conf/application.properties -cp "conf/:lib/*:fate-serving-admin.jar" com.webank.ai.fate.serving.admin.Bootstrap

docker-build/serving-admin/Dockerfile.arm64

@@ -0,0 +1,25 @@
+FROM openjdk:8u342-jre as builder
+
+WORKDIR /data/projects/fate-serving/serving-proxy/
+
+USER root
+
+ARG version
+
+COPY fate-serving-proxy/target/fate-serving-proxy-${version}-release.zip ./
+
+RUN unzip fate-serving-proxy-${version}-release.zip && \
+    rm fate-serving-proxy-${version}-release.zip && \
+    ln -s fate-serving-proxy-*.jar fate-serving-proxy.jar
+
+FROM openjdk:8u342-jre
+
+WORKDIR /data/projects/fate-serving/serving-proxy/
+
+COPY --from=builder /data/projects/fate-serving/serving-proxy  /data/projects/fate-serving/serving-proxy
+
+USER root
+
+EXPOSE 8059 8869 8879
+
+CMD java -Dspring.config.location=conf/application.properties -cp conf/:lib/*:fate-serving-proxy.jar com.webank.ai.fate.serving.proxy.bootstrap.Bootstrap -c conf/application.properties

docker-build/serving-proxy/Dockerfile renamed to docker-build/serving-proxy/Dockerfile.amd64

@@ -0,0 +1,25 @@
+FROM openjdk:8u342-jre as builder
+
+WORKDIR /data/projects/fate-serving/serving-server/
+
+USER root
+
+ARG version
+
+COPY fate-serving-server/target/fate-serving-server-${version}-release.zip ./
+
+RUN unzip fate-serving-server-${version}-release.zip && \
+    rm fate-serving-server-${version}-release.zip && \
+    ln -s fate-serving-server-*.jar fate-serving-server.jar
+
+FROM openjdk:8u342-jre
+
+WORKDIR /data/projects/fate-serving/serving-server/
+
+COPY --from=builder /data/projects/fate-serving/serving-server  /data/projects/fate-serving/serving-server
+
+USER root
+
+EXPOSE 8000
+
+CMD java -cp conf/:lib/*:extension/*:fate-serving-server.jar com.webank.ai.fate.serving.Bootstrap -c conf/serving-server.properties

docker-build/serving-proxy/Dockerfile.arm64

@@ -30,8 +30,8 @@
         <module>fate-serving-register</module>
         <module>fate-serving-common</module>
         <module>fate-serving-proxy</module>
-<!--        <module>fate-serving-admin</module>-->
-<!--        <module>fate-serving-admin-ui</module>-->
+        <module>fate-serving-admin</module>
+        <module>fate-serving-admin-ui</module>
         <module>fate-serving-extension</module>
         <module>fate-serving-sdk</module>
     </modules>