Added healthcheck and launch/shutdown message.

oracle · Aug 7, 2023 · e6b6efe · e6b6efe
1 parent 65fe6ac
commit e6b6efe
Show file tree

Hide file tree

Showing 3 changed files with 443 additions and 0 deletions.
diff --git a/container-image/Dockerfile b/container-image/Dockerfile
@@ -30,6 +30,7 @@ RUN chmod a+x ${MLFLOW_DIR}/launch_mlflow.sh
 ENV MLFLOW_DIR=${MLFLOW_DIR}
 
 EXPOSE 5000
+HEALTHCHECK --interval=30s CMD curl -f -sI http://localhost:5000 || exit 1
 
 RUN if [ -f ${MLFLOW_DIR}/oci_mlflow*.whl ]; then \
         local_whl=$(find ${MLFLOW_DIR} -name "*.whl" -exec basename {} \; | head -n 1 ); \

diff --git a/container-image/run/launch_mlflow.sh b/container-image/run/launch_mlflow.sh
@@ -7,6 +7,10 @@ set -m -e -o pipefail
 
 conda activate oci-mlflow
 
+echo "========== MLflow server is launchng... "==========
+
 mlflow server $@
 
+echo ""========== MLflow server is shutting down... "=========="
+
 exit $LastExitCode