# 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.

function get_milvus_process() {
    echo $(ps -e | grep milvus | grep -v grep |  grep run-with-subprocess | awk '{print $1}')
}

echo "Stopping milvus..."
BASEDIR=$(dirname "$0")
timeout=120
if [ ! -z "$1" ]; then
  timeout=$1
fi

if [ -z $(get_milvus_process) ]; then
  echo "No milvus process"
  exit 0
fi
kill -15 $(get_milvus_process)

start=$(date +%s)
while :
do
  sleep 1
  if [ -z $(get_milvus_process) ]; then
    echo "Milvus stopped"
    break
  fi
  if [ $(( $(date +%s) - $start )) -gt $timeout ]; then
    echo "Milvus timeout stopped"
    kill -15 $(get_milvus_process)
    break
  fi
done