Update Dockerfile

Dockerfile

@@ -1,154 +1,31 @@
-# Use an appropriate base image
-FROM ubuntu:20.04
+# Use the official Milvus image as the base image
+FROM milvusdb/milvus:v2.4.5
 
-# Install necessary packages
-RUN apt-get update && apt-get install -y \
-    curl \
-    docker.io \
-    && rm -rf /var/lib/apt/lists/*
-
-# Create the necessary directories
+# Create necessary directories
 RUN mkdir -p /var/lib/milvus /milvus/configs
 
-# Create the configuration files
-RUN cat << EOF > /milvus/configs/embedEtcd.yaml
-listen-client-urls: http://0.0.0.0:2379
+# Create the embedEtcd.yaml file
+RUN echo "listen-client-urls: http://0.0.0.0:2379
 advertise-client-urls: http://0.0.0.0:2379
 quota-backend-bytes: 4294967296
 auto-compaction-mode: revision
-auto-compaction-retention: '1000'
-EOF
-
-RUN cat << EOF > /milvus/configs/user.yaml
-# Extra config to override default milvus.yaml
-EOF
-
-# Create the Bash script directly in the Dockerfile
-RUN cat << 'EOF' > /usr/local/bin/start_milvus.sh
-#!/bin/bash
-
-run_embed() {
-    sudo docker run -d \
-        --name milvus-standalone \
-        --security-opt seccomp:unconfined \
-        -e ETCD_USE_EMBED=true \
-        -e ETCD_DATA_DIR=/var/lib/milvus/etcd \
-        -e ETCD_CONFIG_PATH=/milvus/configs/embedEtcd.yaml \
-        -e COMMON_STORAGETYPE=local \
-        -v /var/lib/milvus:/var/lib/milvus \
-        -v /milvus/configs/embedEtcd.yaml:/milvus/configs/embedEtcd.yaml \
-        -v /milvus/configs/user.yaml:/milvus/configs/user.yaml \
-        -p 19530:19530 \
-        -p 9091:9091 \
-        -p 2379:2379 \
-        --health-cmd="curl -f http://localhost:9091/healthz" \
-        --health-interval=30s \
-        --health-start-period=90s \
-        --health-timeout=20s \
-        --health-retries=3 \
-        milvusdb/milvus:v2.4.5 \
-        milvus run standalone  1> /dev/null
-}
-
-wait_for_milvus_running() {
-    echo "Wait for Milvus Starting..."
-    while true
-    do
-        res=$(sudo docker ps|grep milvus-standalone|grep healthy|wc -l)
-        if [ $res -eq 1 ]
-        then
-            echo "Start successfully."
-            echo "To change the default Milvus configuration, add your settings to the user.yaml file and then restart the service."
-            break
-        fi
-        sleep 1
-    done
-}
-
-start() {
-    res=$(sudo docker ps|grep milvus-standalone|grep healthy|wc -l)
-    if [ $res -eq 1 ]
-    then
-        echo "Milvus is running."
-        exit 0
-    fi
-
-    res=$(sudo docker ps -a|grep milvus-standalone|wc -l)
-    if [ $res -eq 1 ]
-    then
-        sudo docker start milvus-standalone 1> /dev/null
-    else
-        run_embed
-    fi
-
-    if [ $? -ne 0 ]
-    then
-        echo "Start failed."
-        exit 1
-    fi
-
-    wait_for_milvus_running
-}
-
-stop() {
-    sudo docker stop milvus-standalone 1> /dev/null
-
-    if [ $? -ne 0 ]
-    then
-        echo "Stop failed."
-        exit 1
-    fi
-    echo "Stop successfully."
-
-}
-
-delete() {
-    res=$(sudo docker ps|grep milvus-standalone|wc -l)
-    if [ $res -eq 1 ]
-    then
-        echo "Please stop Milvus service before delete."
-        exit 1
-    fi
-    sudo docker rm milvus-standalone 1> /dev/null
-    if [ $? -ne 0 ]
-    then
-        echo "Delete failed."
-        exit 1
-    fi
-    sudo rm -rf /var/lib/milvus
-    sudo rm -rf /milvus/configs/embedEtcd.yaml
-    sudo rm -rf /milvus/configs/user.yaml
-    echo "Delete successfully."
-}
+auto-compaction-retention: '1000'" > /milvus/configs/embedEtcd.yaml
 
-case $1 in
-    restart)
-        stop
-        start
-        ;;
-    start)
-        start
-        ;;
-    stop)
-        stop
-        ;;
-    delete)
-        delete
-        ;;
-    *)
-        echo "please use bash start_milvus.sh restart|start|stop|delete"
-        ;;
-esac
-EOF
+# Create the user.yaml file
+RUN echo "# Extra config to override default milvus.yaml" > /milvus/configs/user.yaml
 
-# Make the script executable
-RUN chmod +x /usr/local/bin/start_milvus.sh
+# Set environment variables
+ENV ETCD_USE_EMBED=true
+ENV ETCD_DATA_DIR=/var/lib/milvus/etcd
+ENV ETCD_CONFIG_PATH=/milvus/configs/embedEtcd.yaml
+ENV COMMON_STORAGETYPE=local
 
-# Expose the necessary ports
+# Expose necessary ports
 EXPOSE 19530 9091 2379
 
-# Set the entrypoint to the script
-ENTRYPOINT ["/usr/local/bin/start_milvus.sh"]
+# Health check configuration
+HEALTHCHECK --interval=30s --timeout=20s --start-period=90s --retries=3 \
+    CMD curl -f http://localhost:9091/healthz || exit 1
 
-# Default command to start the service
-CMD ["start"]
+# Command to run Milvus standalone
+CMD ["milvus", "run", "standalone"]