123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169 |
- #!/usr/bin/env bash
- # Licensed to the LF AI & Data foundation under one
- # or more contributor license agreements. See the NOTICE file
- # distributed with this work for additional information
- # regarding copyright ownership. The ASF licenses this file
- # to you under the Apache License, Version 2.0 (the
- # "License"); you may not use this file except in compliance
- # with the License. You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- run_embed() {
- cat << EOF > embedEtcd.yaml
- 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
- cat << EOF > user.yaml
- # Extra config to override default milvus.yaml
- EOF
- 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 $(pwd)/volumes/milvus:/var/lib/milvus \
- -v $(pwd)/embedEtcd.yaml:/milvus/configs/embedEtcd.yaml \
- -v $(pwd)/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.13-hotfix \
- 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_container() {
- 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 milvus container failed."
- exit 1
- fi
- echo "Delete milvus container successfully."
- }
- delete() {
- delete_container
- sudo rm -rf $(pwd)/volumes
- sudo rm -rf $(pwd)/embedEtcd.yaml
- sudo rm -rf $(pwd)/user.yaml
- echo "Delete successfully."
- }
- upgrade() {
- read -p "Please confirm if you'd like to proceed with the upgrade. The default will be to the latest version. Confirm with 'y' for yes or 'n' for no. > " check
- if [ "$check" == "y" ] ||[ "$check" == "Y" ];then
- res=`sudo docker ps -a|grep milvus-standalone|wc -l`
- if [ $res -eq 1 ]
- then
- stop
- delete_container
- fi
- curl -sfL https://raw.githubusercontent.com/milvus-io/milvus/master/scripts/standalone_embed.sh -o standalone_embed_latest.sh && \
- bash standalone_embed_latest.sh start 1> /dev/null && \
- echo "Upgrade successfully."
- else
- echo "Exit upgrade"
- exit 0
- fi
- }
- case $1 in
- restart)
- stop
- start
- ;;
- start)
- start
- ;;
- stop)
- stop
- ;;
- upgrade)
- upgrade
- ;;
- delete)
- delete
- ;;
- *)
- echo "please use bash standalone_embed.sh restart|start|stop|upgrade|delete"
- ;;
- esac
|