mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 05:18:52 +08:00
aa84124d24
Signed-off-by: Filip Haltmayer <filip.haltmayer@zilliz.com>
531 lines
14 KiB
Bash
Executable File
531 lines
14 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
namespace="default"
|
|
root_path="by-dev"
|
|
operation="migrate"
|
|
image_tag="milvusdb/milvus:v2.2.0"
|
|
meta_migration_pod_tag="milvusdb/meta-migration:v2.2.0-bugfix-20230112"
|
|
remove_migrate_pod_after_migrate="false"
|
|
storage_class=""
|
|
external_etcd_svc=""
|
|
etcd_svc=""
|
|
|
|
#-n namespace: The namespace that Milvus is installed in.
|
|
#-i milvus_instance: The name of milvus instance.
|
|
#-s source_version: The milvus source version.
|
|
#-t target_version: The milvus target version.
|
|
#-r root_path: The milvus meta root path.
|
|
#-w image_tag: The new milvus image tag.
|
|
#-o operation: The operation: migrate/rollback.
|
|
#-m meta_migration_pod_tag: The image for meta migration pod.
|
|
#-d remove_migrate_pod_after_migrate: Remove migration pod after successful migration.
|
|
#-c storage_class: The storage class for meta migration pvc.
|
|
#-e external_etcd_svc: The endpoints for etcd which used by milvus.
|
|
while getopts "n:i:s:t:r:w:o:m:c:e:d:" opt_name
|
|
do
|
|
case $opt_name in
|
|
n) namespace=$OPTARG;;
|
|
i) instance_name=$OPTARG;;
|
|
s) source_version=$OPTARG;;
|
|
t) target_version=$OPTARG;;
|
|
r) root_path=$OPTARG;;
|
|
w) image_tag=$OPTARG;;
|
|
o) operation=$OPTARG;;
|
|
m) meta_migration_pod_tag=$OPTARG;;
|
|
d) remove_migrate_pod_after_migrate="true";;
|
|
c) storage_class=$OPTARG;;
|
|
e) external_etcd_svc=$OPTARG;;
|
|
*) echo "Unkonwen parameters";;
|
|
esac
|
|
done
|
|
|
|
if [ ! $instance_name ]; then
|
|
echo "Missing argument instance_name, please add it. For example:'./migrate.sh -i milvus-instance-name'"
|
|
exit 1
|
|
fi
|
|
|
|
if [ ! $source_version ]; then
|
|
echo "Missing argument source_version, please add it. For example:'./migrate.sh -s 2.1.4'"
|
|
exit 1
|
|
fi
|
|
|
|
if [ ! $target_version ]; then
|
|
echo "Missing argument target_version, please add it. For example:'./migrate.sh -t 2.2.0'"
|
|
exit 1
|
|
fi
|
|
|
|
if [ ! $image_tag ]; then
|
|
echo "Missing argument image_tag, please add it. For example:'./migrate.sh -w milvusdb/milvus:v2.2.0'"
|
|
exit 1
|
|
fi
|
|
|
|
deployments=$(kubectl get deploy -n $namespace -l app.kubernetes.io/instance=$instance_name,app.kubernetes.io/name=milvus --output=jsonpath={.items..metadata.name})
|
|
deploys=()
|
|
replicas=()
|
|
for d in $deployments; do
|
|
component=$(kubectl get deploy $d -n $namespace --output=jsonpath={.metadata.labels.component})
|
|
replica=$(kubectl get deploy $d -n $namespace --output=jsonpath={.spec.replicas})
|
|
if [ "$component" = "attu" ]; then
|
|
continue
|
|
fi
|
|
deploys+=("$d")
|
|
done
|
|
|
|
if [ ${#deploys[@]} -eq 0 ]; then
|
|
echo "There is no Milvus instance $instance_name in the namespace $namespace"
|
|
exit 1
|
|
fi
|
|
|
|
if [ ! $external_etcd_svc ]; then
|
|
svc=$(kubectl get svc -n $namespace -l app.kubernetes.io/instance=$instance_name,app.kubernetes.io/name=etcd -o name | grep -v headless | cut -d '/' -f 2)
|
|
|
|
if [ ! $svc ]; then
|
|
echo "Missing etcd service, please add it. For example:'./migrate.sh -e <etcd-svc-ip>:<etcd-svc-port>'"
|
|
exit 1
|
|
else
|
|
port=$(kubectl get svc -n $namespace $svc --output=jsonpath='{.spec.ports[?(@.name == "client")].port}')
|
|
if [ ! $port ]; then
|
|
echo "Missing etcd service port..."
|
|
exit 1
|
|
fi
|
|
etcd_svc="$svc:$port"
|
|
fi
|
|
else
|
|
etcd_svc="$external_etcd_svc"
|
|
fi
|
|
|
|
scs=$(kubectl get storageclass --output=jsonpath='{.items..metadata.name}')
|
|
if [ ! $storage_class ]; then
|
|
# check whether there a default storageclass
|
|
exists=0
|
|
for sc in $scs; do
|
|
default=$(kubectl get storageclass $sc --output=jsonpath='{.metadata.annotations.storageclass\.kubernetes\.io/is-default-class}')
|
|
if [ "$default" = "true" ] ; then
|
|
exists=1
|
|
break
|
|
fi
|
|
done
|
|
|
|
if [ $exists -eq 0 ] ; then
|
|
echo "No default storageclass, please specify a storageclass via -c <storage-class-name>"
|
|
exit 1
|
|
fi
|
|
else
|
|
exists=0
|
|
for sc in $scs; do
|
|
if [ "$sc" = "$storage_class" ] ; then
|
|
exists=1
|
|
break
|
|
fi
|
|
done
|
|
|
|
if [ $exists -eq 0 ]; then
|
|
echo "Nonexistent storageclass, please make sure specified storageclass exists..."
|
|
exit 1
|
|
fi
|
|
fi
|
|
|
|
echo "Migration milvus meta will take four steps:"
|
|
echo "1. Stop the milvus components"
|
|
echo "2. Backup the milvus meta"
|
|
echo "3. Migrate the milvus meta"
|
|
echo "4. Startup milvus components in new image version"
|
|
echo
|
|
|
|
|
|
function store_deploy_replicas(){
|
|
# store deploy replicas in a configmap
|
|
# first check whether config exists
|
|
data=$(kubectl get configmap "milvus-deploy-replicas-$instance_name" -n $namespace --output=jsonpath={.data.replicas} 2>/dev/null)
|
|
if [ ! "$data" ]; then
|
|
for d in ${deploys[@]}; do
|
|
replica=$(kubectl get deploy $d -n $namespace --output=jsonpath={.spec.replicas})
|
|
replicas+=("$d:$replica")
|
|
done
|
|
cat <<EOF | kubectl apply -n $namespace -f -
|
|
apiVersion: v1
|
|
kind: ConfigMap
|
|
metadata:
|
|
name: milvus-deploy-replicas-${instance_name}
|
|
data:
|
|
replicas: "${replicas[@]}"
|
|
EOF
|
|
else
|
|
for d in $data; do
|
|
replicas+=("$d")
|
|
done
|
|
fi
|
|
}
|
|
|
|
function stop_milvus_deploy(){
|
|
echo "Stop milvus deployments: ${deploys[@]}"
|
|
kubectl scale deploy -n $namespace "${deploys[@]}" --replicas=0 > /dev/null
|
|
wait_for_milvus_stopped "${deploys[@]}"
|
|
echo "Stopped..."
|
|
}
|
|
|
|
function wait_for_milvus_stopped(){
|
|
while true
|
|
do
|
|
total=0
|
|
for deploy in $1;
|
|
do
|
|
count=$(kubectl get deploy $deploy -n $namespace --output=jsonpath={.status.replicas})
|
|
count=${count:-0}
|
|
total=`expr $total + $count`
|
|
done
|
|
if [ $total -eq 0 ]; then
|
|
break
|
|
fi
|
|
sleep 5
|
|
done
|
|
# wait for the session key expire
|
|
sleep 75
|
|
}
|
|
|
|
function wait_for_backup_done(){
|
|
backup_pod_name="milvus-meta-migration-backup-${instance_name}"
|
|
configmap_name="milvus-meta-migration-config-${instance_name}"
|
|
while true
|
|
do
|
|
status=$(kubectl get pod -n $namespace $backup_pod_name --output=jsonpath={.status.phase})
|
|
case $status in
|
|
Succeeded)
|
|
echo "Meta backup is done..."
|
|
kubectl annotate configmap $configmap_name -n $namespace backup=succeed
|
|
break
|
|
;;
|
|
Failed)
|
|
echo "Meta backup is failed..."
|
|
echo "Here is the log:"
|
|
kubectl logs $backup_pod_name -n $namespace
|
|
echo
|
|
exit 2
|
|
;;
|
|
*)
|
|
sleep 10
|
|
continue
|
|
;;
|
|
esac
|
|
done
|
|
}
|
|
|
|
function wait_for_migrate_done(){
|
|
migrate_pod_name="milvus-meta-migration-${instance_name}"
|
|
while true
|
|
do
|
|
status=$(kubectl get pod -n $namespace $migrate_pod_name --output=jsonpath={.status.phase})
|
|
case $status in
|
|
Succeeded)
|
|
echo "Migration is done..."
|
|
echo
|
|
break
|
|
;;
|
|
Failed)
|
|
echo "Migration is failed..."
|
|
echo "Here is the log:"
|
|
kubectl logs $migrate_pod_name -n $namespace
|
|
echo
|
|
exit 3
|
|
;;
|
|
*)
|
|
sleep 10
|
|
continue
|
|
;;
|
|
esac
|
|
done
|
|
}
|
|
|
|
function wait_for_rollback_done(){
|
|
rollback_pod_name="milvus-meta-migration-rollback-${instance_name}"
|
|
while true
|
|
do
|
|
status=$(kubectl get pod -n $namespace $rollback_pod_name --output=jsonpath={.status.phase})
|
|
case $status in
|
|
Succeeded)
|
|
echo "Rollback is done..."
|
|
echo
|
|
break
|
|
;;
|
|
Failed)
|
|
echo "Rollback is failed..."
|
|
echo "Here is the log:"
|
|
kubectl logs $rollback_pod_name -n $namespace
|
|
echo
|
|
exit 5
|
|
;;
|
|
*)
|
|
sleep 10
|
|
continue
|
|
;;
|
|
esac
|
|
done
|
|
}
|
|
|
|
function generate_migrate_meta_cm_and_pvc(){
|
|
# check whether pvc exists
|
|
exists=$(kubectl get pvc "milvus-meta-migration-backup-${instance_name}" -n $namespace --output=jsonpath={.metadata.name} 2>/dev/null)
|
|
if [ ! $exists ]; then
|
|
cat <<EOF | kubectl apply -n $namespace -f -
|
|
apiVersion: v1
|
|
kind: PersistentVolumeClaim
|
|
metadata:
|
|
name: milvus-meta-migration-backup-${instance_name}
|
|
spec:
|
|
accessModes:
|
|
- ReadWriteOnce
|
|
storageClassName: $storage_class
|
|
resources:
|
|
requests:
|
|
storage: 10Gi
|
|
EOF
|
|
fi
|
|
cat <<EOF | kubectl apply -n $namespace -f -
|
|
apiVersion: v1
|
|
kind: ConfigMap
|
|
metadata:
|
|
name: milvus-meta-migration-config-${instance_name}
|
|
data:
|
|
backup.yaml: |+
|
|
cmd:
|
|
type: backup
|
|
|
|
config:
|
|
sourceVersion: $source_version
|
|
targetVersion: $target_version
|
|
backupFilePath: /milvus/data/migration.bak
|
|
|
|
metastore:
|
|
type: etcd
|
|
|
|
etcd:
|
|
endpoints:
|
|
- $etcd_svc
|
|
rootPath: $root_path
|
|
metaSubPath: meta
|
|
kvSubPath: kv
|
|
migration.yaml: |+
|
|
cmd:
|
|
type: run
|
|
|
|
config:
|
|
sourceVersion: $source_version
|
|
targetVersion: $target_version
|
|
backupFilePath: /milvus/data/migration.bak
|
|
|
|
metastore:
|
|
type: etcd
|
|
|
|
etcd:
|
|
endpoints:
|
|
- $etcd_svc
|
|
rootPath: $root_path
|
|
metaSubPath: meta
|
|
kvSubPath: kv
|
|
rollback.yaml: |+
|
|
cmd:
|
|
type: rollback
|
|
|
|
config:
|
|
sourceVersion: $source_version
|
|
targetVersion: $target_version
|
|
backupFilePath: /milvus/data/migration.bak
|
|
|
|
metastore:
|
|
type: etcd
|
|
|
|
etcd:
|
|
endpoints:
|
|
- $etcd_svc
|
|
rootPath: $root_path
|
|
metaSubPath: meta
|
|
kvSubPath: kv
|
|
EOF
|
|
}
|
|
|
|
function backup_meta(){
|
|
echo "Backuping meta..."
|
|
echo "Checking whether backup exists..."
|
|
configmap_name="milvus-meta-migration-config-${instance_name}"
|
|
backup_exists=$(kubectl get configmap $configmap_name -n $namespace --output=jsonpath={.metadata.annotations.backup})
|
|
if [ "$backup_exists" = "succeed" ]; then
|
|
echo "Found previous backups, skip meta backup..."
|
|
else
|
|
cat <<EOF | kubectl apply -n $namespace -f -
|
|
apiVersion: v1
|
|
kind: Pod
|
|
metadata:
|
|
name: milvus-meta-migration-backup-${instance_name}
|
|
spec:
|
|
restartPolicy: Never
|
|
containers:
|
|
- name: meta-migration
|
|
image: $meta_migration_pod_tag
|
|
command: ["/bin/sh"]
|
|
args:
|
|
- -c
|
|
- /milvus/bin/meta-migration -config=/milvus/configs/meta/backup.yaml
|
|
volumeMounts:
|
|
- name: backup
|
|
mountPath: /milvus/data
|
|
- name: config
|
|
mountPath: /milvus/configs/meta
|
|
volumes:
|
|
- name: backup
|
|
persistentVolumeClaim:
|
|
claimName: milvus-meta-migration-backup-${instance_name}
|
|
- name: config
|
|
configMap:
|
|
name: milvus-meta-migration-config-${instance_name}
|
|
EOF
|
|
wait_for_backup_done
|
|
fi
|
|
}
|
|
|
|
function rollback_meta(){
|
|
generate_migrate_meta_cm_and_pvc
|
|
echo "Checking whether backup exists..."
|
|
backup_exists=$(kubectl get configmap milvus-meta-migration-config -n $namespace --output=jsonpath={.metadata.annotations.backup})
|
|
if [ "$backup_exists" = "succeed" ]; then
|
|
echo "Found previous backups, start meta rollback..."
|
|
cat <<EOF | kubectl apply -n $namespace -f -
|
|
apiVersion: v1
|
|
kind: Pod
|
|
metadata:
|
|
name: milvus-meta-migration-rollback-${instance_name}
|
|
spec:
|
|
restartPolicy: Never
|
|
containers:
|
|
- name: meta-migration
|
|
image: $meta_migration_pod_tag
|
|
command: ["/bin/sh"]
|
|
args:
|
|
- -c
|
|
- /milvus/bin/meta-migration -config=/milvus/configs/meta/rollback.yaml
|
|
volumeMounts:
|
|
- name: backup
|
|
mountPath: /milvus/data
|
|
- name: config
|
|
mountPath: /milvus/configs/meta
|
|
volumes:
|
|
- name: backup
|
|
persistentVolumeClaim:
|
|
claimName: milvus-meta-migration-backup-${instance_name}
|
|
- name: config
|
|
configMap:
|
|
name: milvus-meta-migration-config-${instance_name}
|
|
EOF
|
|
wait_for_rollback_done
|
|
else
|
|
echo "No backup exists, abort..."
|
|
exit 4
|
|
fi
|
|
}
|
|
|
|
function migrate_meta(){
|
|
generate_migrate_meta_cm_and_pvc
|
|
|
|
backup_meta
|
|
echo "Migrating meta..."
|
|
cat <<EOF | kubectl apply -n $namespace -f -
|
|
apiVersion: v1
|
|
kind: Pod
|
|
metadata:
|
|
name: milvus-meta-migration-${instance_name}
|
|
spec:
|
|
restartPolicy: Never
|
|
containers:
|
|
- name: meta-migration
|
|
image: $meta_migration_pod_tag
|
|
command: ["/bin/sh"]
|
|
args:
|
|
- -c
|
|
- /milvus/bin/meta-migration -config=/milvus/configs/meta/migration.yaml
|
|
volumeMounts:
|
|
- name: backup
|
|
mountPath: /milvus/data
|
|
- name: config
|
|
mountPath: /milvus/configs/meta
|
|
volumes:
|
|
- name: backup
|
|
persistentVolumeClaim:
|
|
claimName: milvus-meta-migration-backup-${instance_name}
|
|
- name: config
|
|
configMap:
|
|
name: milvus-meta-migration-config-${instance_name}
|
|
EOF
|
|
wait_for_migrate_done
|
|
}
|
|
|
|
function start_milvus_deploy(){
|
|
echo "Starting milvus components..."
|
|
for d in "${replicas[@]}"
|
|
do
|
|
IFS=':'
|
|
read -a item <<< "$d"
|
|
deploy_name="${item[0]}"
|
|
replica_count="${item[1]}"
|
|
echo "Starting $deploy_name..."
|
|
component=$(kubectl get deploy -n $namespace $deploy_name --output=jsonpath={.metadata.labels.component})
|
|
kubectl patch deployment $deploy_name -n $namespace -p "{\"spec\":{\"template\": {\"spec\": {\"containers\":[{\"name\": \"$component\", \"image\": \"$image_tag\"}]}}}}"
|
|
kubectl scale deployment $deploy_name -n $namespace --replicas="$replica_count"
|
|
done
|
|
}
|
|
|
|
function wait_for_milvus_ready(){
|
|
total_component_count=${#replicas[@]}
|
|
while true
|
|
do
|
|
ready_count=0
|
|
for deploy in "${replicas[@]}"
|
|
do
|
|
IFS=':'
|
|
read -a item <<< "$deploy"
|
|
deploy_name="${item[0]}"
|
|
replica_count="${item[1]}"
|
|
count=$(kubectl get deploy $deploy_name -n $namespace --output=jsonpath={.status.readyReplicas})
|
|
if [ "$count" = "$replica_count" ] ; then
|
|
ready_count=`expr $ready_count + 1`
|
|
fi
|
|
done
|
|
if [ "$total_component_count" = "$ready_count" ]; then
|
|
break
|
|
fi
|
|
sleep 5
|
|
done
|
|
}
|
|
|
|
store_deploy_replicas
|
|
stop_milvus_deploy
|
|
echo
|
|
case $operation in
|
|
migrate)
|
|
echo "Starting to migrate milvus meta..."
|
|
migrate_meta
|
|
start_milvus_deploy
|
|
echo
|
|
echo "Upgrading is done. Waiting for milvus components to be ready again..."
|
|
wait_for_milvus_ready
|
|
if [ "$remove_migrate_pod_after_migrate" = "true" ]; then
|
|
kubectl delete pods -n $namespace "milvus-meta-migration-backup-${instance_name}" "milvus-meta-migration-${instance_name}"
|
|
kubectl delete pvc -n $namespace "milvus-meta-migration-backup-${instance_name}"
|
|
kubectl delete configmap -n $namespace "milvus-meta-migration-config-${instance_name}" "milvus-deploy-replicas-${instance_name}"
|
|
echo
|
|
fi
|
|
echo "All milvus components are running. Enjoy your vector search..."
|
|
;;
|
|
rollback)
|
|
echo "Starting to rollback milvus meta..."
|
|
rollback_meta
|
|
start_milvus_deploy
|
|
echo
|
|
echo "Rollbacking is done. Waiting for milvus components to be ready again..."
|
|
wait_for_milvus_ready
|
|
echo "All milvus components are running. Enjoy your vector search..."
|
|
;;
|
|
*)
|
|
echo "Invalid operation..."
|
|
exit 6
|
|
;;
|
|
esac
|