87 строки
4.0 KiB
Bash
Executable File
87 строки
4.0 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
# Copyright (c) Microsoft Corporation.
|
|
# Licensed under the MIT license.
|
|
|
|
# parameters
|
|
ENV_CODE=${1:-"aoi"}
|
|
|
|
AI_MODEL_INFRA_RESOURCE_NAME=${2:-$AI_MODEL_INFRA_RESOURCE_NAME}
|
|
AI_MODEL_INFRA_TYPE=${3:-${AI_MODEL_INFRA_TYPE:-"batch-account"}} # Currently supported values are aks and batch-account
|
|
|
|
NO_DELETE_DATA_RESOURCE_GROUP=${NO_DELETE_DATA_RESOURCE_GROUP:-"false"}
|
|
DATA_RESOURCE_GROUP="${ENV_CODE}-data-rg"
|
|
|
|
NO_DELETE_MONITORING_RESOURCE_GROUP=${NO_DELETE_MONITORING_RESOURCE_GROUP:-"false"}
|
|
MONITORING_RESOURCE_GROUP="${ENV_CODE}-monitor-rg"
|
|
|
|
NO_DELETE_NETWORKING_RESOURCE_GROUP=${NO_DELETE_NETWORKING_RESOURCE_GROUP:-"false"}
|
|
NETWORKING_RESOURCE_GROUP="${ENV_CODE}-network-rg"
|
|
|
|
NO_DELETE_ORCHESTRATION_RESOURCE_GROUP=${NO_DELETE_ORCHESTRATION_RESOURCE_GROUP:-"false"}
|
|
ORCHESTRATION_RESOURCE_GROUP="${ENV_CODE}-orc-rg"
|
|
|
|
NO_DELETE_PIPELINE_RESOURCE_GROUP=${NO_DELETE_PIPELINE_RESOURCE_GROUP:-"false"}
|
|
PIPELINE_RESOURCE_GROUP="${ENV_CODE}-pipeline-rg"
|
|
|
|
if [[ ${NO_DELETE_DATA_RESOURCE_GROUP} != "true" ]] && [[ ${NO_DELETE_DATA_RESOURCE_GROUP} == "false" ]]; then
|
|
set -x
|
|
az group delete --name ${DATA_RESOURCE_GROUP} --no-wait --yes
|
|
az deployment sub delete --name ${DATA_RESOURCE_GROUP} --no-wait
|
|
set +x
|
|
fi
|
|
|
|
if [ "${NO_DELETE_MONITORING_RESOURCE_GROUP}" != "true" ] && [ "${NO_DELETE_MONITORING_RESOURCE_GROUP}" == "false" ]; then
|
|
set -x
|
|
az group delete --name ${MONITORING_RESOURCE_GROUP} --no-wait --yes
|
|
az deployment sub delete --name ${MONITORING_RESOURCE_GROUP} --no-wait
|
|
set +x
|
|
fi
|
|
|
|
if [ "${NO_DELETE_NETWORKING_RESOURCE_GROUP}" != "true" ] && [ "${NO_DELETE_NETWORKING_RESOURCE_GROUP}" == "false" ]; then
|
|
set -x
|
|
az group delete --name ${NETWORKING_RESOURCE_GROUP} --no-wait --yes
|
|
az deployment sub delete --name ${NETWORKING_RESOURCE_GROUP} --no-wait
|
|
set +x
|
|
fi
|
|
|
|
if [ "${NO_DELETE_ORCHESTRATION_RESOURCE_GROUP}" != "true" ] && [ "${NO_DELETE_ORCHESTRATION_RESOURCE_GROUP}" == "false" ]; then
|
|
set -x
|
|
az group delete --name ${ORCHESTRATION_RESOURCE_GROUP} --no-wait --yes
|
|
az deployment sub delete --name ${ORCHESTRATION_RESOURCE_GROUP} --no-wait
|
|
set +x
|
|
fi
|
|
|
|
if [ "${NO_DELETE_PIPELINE_RESOURCE_GROUP}" != "true" ] && [ "${NO_DELETE_PIPELINE_RESOURCE_GROUP}" == "false" ]; then
|
|
set -x
|
|
az group delete --name ${PIPELINE_RESOURCE_GROUP} --no-wait --yes
|
|
az deployment sub delete --name ${PIPELINE_RESOURCE_GROUP} --no-wait
|
|
set +x
|
|
fi
|
|
|
|
if [[ "$AI_MODEL_INFRA_TYPE" == "batch-account" ]]; then
|
|
BATCH_ACCOUNT_POOL_NAME=${BATCH_ACCOUNT_POOL_NAME:-"${ENV_CODE}-data-cpu-pool"}
|
|
NO_BATCH_ACCOUNT_POOL_DELETE=${NO_BATCH_ACCOUNT_POOL_DELETE:-"false"}
|
|
if [[ ! -z "$AI_MODEL_INFRA_RESOURCE_NAME" ]] && [ "${NO_BATCH_ACCOUNT_POOL_DELETE}" != "true" ] && [ "${NO_BATCH_ACCOUNT_POOL_DELETE}" == "false" ]; then
|
|
echo "Getting the batch account details"
|
|
BATCH_ACCOUNT_ID=$(az batch account list --query "[?name == '${AI_MODEL_INFRA_RESOURCE_NAME}'].id" -o tsv)
|
|
BATCH_ACCOUNT_RG_NAME=$(az resource show --ids ${BATCH_ACCOUNT_ID} --query resourceGroup -o tsv)
|
|
AZURE_BATCH_ACCESS_KEY=$(az batch account keys list --name ${AI_MODEL_INFRA_RESOURCE_NAME} --resource-group ${BATCH_ACCOUNT_RG_NAME} | jq ".primary")
|
|
AZURE_BATCH_ENDPOINT=$(az batch account show --name ${AI_MODEL_INFRA_RESOURCE_NAME} --resource-group ${BATCH_ACCOUNT_RG_NAME} --query accountEndpoint -o tsv)
|
|
|
|
echo "Cleaning the batch job: ${BATCH_ACCOUNT_POOL_NAME}"
|
|
az batch job delete --yes \
|
|
--job-id ${BATCH_ACCOUNT_POOL_NAME} \
|
|
--account-endpoint ${AZURE_BATCH_ENDPOINT} \
|
|
--account-key ${AZURE_BATCH_ACCESS_KEY} \
|
|
--account-name ${AI_MODEL_INFRA_RESOURCE_NAME}
|
|
if [[ $? == 0 ]]; then
|
|
echo "Cleanup the batch pool: ${BATCH_ACCOUNT_POOL_NAME}"
|
|
az batch pool delete --yes \
|
|
--pool-id ${BATCH_ACCOUNT_POOL_NAME} \
|
|
--account-endpoint ${AZURE_BATCH_ENDPOINT} \
|
|
--account-key ${AZURE_BATCH_ACCESS_KEY} \
|
|
--account-name ${AI_MODEL_INFRA_RESOURCE_NAME}
|
|
fi
|
|
fi
|
|
fi |