launch-instances-cluster.sh

#!/bin/bash

## 样例
## bash launch-instances-cluster.sh -s redis-cluster -t t4g.small -o al2

set -e

## 命令行参数
while getopts 's:t:o:' OPT; do
    case $OPT in
        s) SUT_NAME="$OPTARG";;
        t) INSTANCE_TYPE="$OPTARG";;
        o) OS_TYPE="$OPTARG";;  ## al2, al2023, ubuntu2004, ubuntu2204
    esac
done

if [[ -z ${SUT_NAME} ]]
then
    echo "$0: You MUST specify SUT name with option -s ."
    echo "$0: It can be: redis, mysql, mongo, etc ."
    exit
fi

if [[ -z ${INSTANCE_TYPE} ]]
then
    echo "$0: You MUST specify Instance Type with option -t ."
    exit
fi

if [[ -z ${OS_TYPE} ]]
then
    echo "$0: You do not specify OS Type with option -o, So we will use OS(Amazon Linux 2) by default. "
    OS_TYPE=al2
fi

REGION_NAME="us-west-2"
echo "" > /tmp/temp-setting
echo "export REGION_NAME=${REGION_NAME}" >> /tmp/temp-setting
echo "export INSTANCE_TYPE=${INSTANCE_TYPE}" >> /tmp/temp-setting
echo "export OS_TYPE=${OS_TYPE}" >> /tmp/temp-setting
echo "export SUT_NAME=${SUT_NAME}" >> /tmp/temp-setting

## 根据实例类型、OS 类型查找最新的 AMI。
bash search_latest_ami.sh

source /tmp/temp-setting

## 创建对应的 terraform 配置文件目录
cd ${SUT_NAME}
rm -rf tf_cfg_${SUT_NAME}
cp -rf tf_cfg_template tf_cfg_${SUT_NAME}
cd tf_cfg_${SUT_NAME}

## 修改 variables.tf 内容 
sed -i "s/INSTANCE_NAME_XXX/SUT_${SUT_NAME}/g" variables.tf
sed -i "s/INSTANCE_TYPE_XXX/${INSTANCE_TYPE}/g" variables.tf
sed -i "s/AMI_ID_XXX/${AMI_ID}/g" variables.tf
sed -i "s/USERDATA_FILE_XXX/userdata.sh/g" variables.tf

## 修改 userdata-xx.sh 的 SUT_NAME 
sed -i "s/SUT_XXX/${SUT_NAME}/g" userdata.sh

## 使用 terraform 启动实例
terraform init
terraform plan
terraform apply --auto-approve

echo "$0: Terraform completed."

if [[ ${SUT_NAME} == "redis-cluster" ]] || \
   [[ ${SUT_NAME} == "cassandra-cluster" ]] ; then
        INSTANCE_IP_MASTER=$(terraform output -raw instance_private_ip_0)
        INSTANCE_IP_MASTER1=$(terraform output -raw instance_private_ip_01)
        INSTANCE_IP_MASTER2=$(terraform output -raw instance_private_ip_02)
        INSTANCE_IP_SLAVE=$(terraform output -raw instance_private_ip_1)
        INSTANCE_IP_SLAVE1=$(terraform output -raw instance_private_ip_11)
        INSTANCE_IP_SLAVE2=$(terraform output -raw instance_private_ip_12)
        echo "export INSTANCE_IP_MASTER=${INSTANCE_IP_MASTER}" >> /tmp/temp-setting
        echo "export INSTANCE_IP_MASTER1=${INSTANCE_IP_MASTER1}" >> /tmp/temp-setting
        echo "export INSTANCE_IP_MASTER2=${INSTANCE_IP_MASTER2}" >> /tmp/temp-setting
        echo "export INSTANCE_IP_SLAVE=${INSTANCE_IP_SLAVE}" >> /tmp/temp-setting
        echo "export INSTANCE_IP_SLAVE1=${INSTANCE_IP_SLAVE1}" >> /tmp/temp-setting
        echo "export INSTANCE_IP_SLAVE2=${INSTANCE_IP_SLAVE2}" >> /tmp/temp-setting
else 
        echo "${SUT_NAME} not supported. "
        exit 1
fi

cd ..
mv tf_cfg_${SUT_NAME}  tf_cfg_${SUT_NAME}_${INSTANCE_TYPE}_${OS_TYPE}_${INSTANCE_IP_MASTER}
cp /tmp/temp-setting tf_cfg_${SUT_NAME}_${INSTANCE_TYPE}_${OS_TYPE}_${INSTANCE_IP_MASTER}/temp-setting

echo "$0: Start waiting 180 seconds for UserData completed."
sleep 180

cd ..
bash ./setup-cluster.sh