批量创建OpenStack实例-CSDN博客

本文链接：https://blog.csdn.net/u013469753/article/details/147353610
在Linux终端实现批量创建OpenStack实例，支持支持统计、并发创建、安全确认、重试机制、日志。
#!/bin/bash
# ======================================================================
# 增强版OpenStack实例创建脚本（修复日志功能）
# 功能：支持统计、并发创建、安全确认、重试机制
# 更新日期：2025年4月19日
# ======================================================================

#######################################
# 配置区（用户可自定义参数）
#######################################

# --------------------
# 实例配置
# --------------------
FLAVOR="BJ1-2U4GB64GB"                # 实例规格（必填）
SECURITY_GROUP="default"               # 安全组名称（默认安全组）
NETWORK_NAME="BJself1"                 # 网络名称（必须存在）

# --------------------
# 资源限制配置
# --------------------
MAX_INSTANCES_per_Image_per_Zone=2     # 单个镜像在每个可用区中最大实例数
MAX_TOTAL_INSTANCES_ALL_ZONE=10        # 实时查询所有可用区内的实例总数上限
AVAILABILITY_ZONES=(                   # 可用区列表（格式：nova:zone_name）
    "nova:bjcontroller1"
    "nova:bjcompute1"
)

# --------------------
# 运行控制配置
# --------------------
SLEEP_TIME=5                           # 资源检查间隔时间（秒）
LOG_FILE="./log_launch_Instance_$(date +%Y%m%d).log"   # 日志文件路径

# --------------------
# 颜色定义（终端输出美化）
# --------------------
RED='\033[0;31m'    # 红色 - 错误信息
GREEN='\033[0;32m'  # 绿色 - 成功信息
YELLOW='\033[0;33m' # 黄色 - 警告信息
NC='\033[0m'        # 无颜色 - 重置终端颜色


#######################################
# 初始化区（加载环境/资源检查）
#######################################

# --------------------
# 加载OpenStack认证信息
# --------------------
source /openstack/admin-openrc.sh || { 
    echo -e "${RED}错误：加载环境变量失败${NC}" | tee -a "$LOG_FILE" >&2
    exit 1
}

# --------------------
# 获取网络ID（不使用 awk）
# --------------------
fixnet=$(openstack network list -f value -c Name -c ID | grep -w "$NETWORK_NAME" | cut -d' ' -f2)
if [ -z "$fixnet" ]; then
    echo -e "${RED}错误：未找到网络 '$NETWORK_NAME'${NC}" | tee -a "$LOG_FILE" >&2
    exit 1
fi

# --------------------
# 获取可用镜像列表
# --------------------
imagelist=$(glance image-list | grep -o '[a-zA-Z0-9]\{8\}-[a-zA-Z0-9]\{4\}-[a-zA-Z0-9]\{4\}-[a-zA-Z0-9]\{4\}-[a-zA-Z0-9]\{12\}')
if [ -z "$imagelist" ]; then
    echo -e "${RED}错误：未找到可用镜像${NC}" | tee -a "$LOG_FILE" >&2
    exit 1
fi


#######################################
# 统计变量声明
#######################################
declare -A ZONE_IMAGE_COUNT            # 每个可用区已使用的镜像数量
declare -A ZONE_INSTANCE_COUNT         # 每个可用区已创建的实例数量
TOTAL_INSTANCES=0                      # 全局累计实例总数

# 初始化统计变量
for zone in "${AVAILABILITY_ZONES[@]}"; do
    ZONE_IMAGE_COUNT[$zone]=""
    ZONE_INSTANCE_COUNT[$zone]=0
done


#######################################
# 功能函数区
#######################################

# --------------------
# 函数：get_instance_count
# 用途：获取指定镜像在指定可用区的实例数量
# --------------------
get_instance_count() {
    local image=$1
    local zone=$2
    openstack server list --long -f value -c "Image" -c "Availability Zone" | 
        grep -w "$image" | grep -w "$zone" | wc -l
}

# --------------------
# 函数：get_total_instances_all_zones
# 用途：获取所有可用区的实例总数
# --------------------
get_total_instances_all_zones() {
    openstack server list --all-projects -c ID -f value | wc -l
}

# --------------------
# 函数：create_instance
# 用途：创建单个实例
# --------------------
create_instance() {
    local image=$1
    local zone=$2
    echo -e "${GREEN}$(date '+%Y-%m-%d %H:%M:%S') [创建实例]- (镜像: $image| 可用区节点: $zone)${NC}" | tee -a "$LOG_FILE"
    
    # 调用OpenStack API创建实例
    if ! openstack server create \
        --flavor "$FLAVOR" \
        --image "$image" \
        --nic "net-id=$fixnet" \
        --security-group "$SECURITY_GROUP" \
        --availability-zone "$zone" \
        "instance-$(date +%s)" >> "$LOG_FILE" 2>&1; then
        echo -e "${RED}错误：创建实例失败 [镜像: $image | 可用区: $zone]${NC}" | tee -a "$LOG_FILE" >&2
        return 1
    fi
    
    # 更新统计信息
    ((ZONE_INSTANCE_COUNT[$zone]++))
    ((TOTAL_INSTANCES++))
    
    # 更新已使用镜像数
    if ! echo "${ZONE_IMAGE_COUNT[$zone]}" | grep -q "$image"; then
        ZONE_IMAGE_COUNT[$zone]+=" $image"
    fi
}


#######################################
# 流程控制区
#######################################

# --------------------
# 安全确认机制
# --------------------
if [[ -t 0 ]]; then
    echo -e "${YELLOW}警告：即将开始创建实例！${NC}" | tee -a "$LOG_FILE"
    read -p "是否继续？(y/N): " -n 1 confirm
    echo  # 换行
    [[ "${confirm,,}" != "y" ]] && {
        echo -e "${YELLOW}操作已取消${NC}" | tee -a "$LOG_FILE"
        exit 0
    }
fi

# --------------------
# 中断信号处理
# --------------------
trap 'echo -e "\n${YELLOW}捕获到中断信号，输出当前统计信息...${NC}" | tee -a "$LOG_FILE";
      for zone in "${AVAILABILITY_ZONES[@]}"; do
          echo -e "\n可用区 $zone 中：" | tee -a "$LOG_FILE"
          # 统计已使用镜像数（基于 ZONE_IMAGE_COUNT[$zone]）
          used_images=$(echo "${ZONE_IMAGE_COUNT[$zone]}" | wc -w)
          echo -e " - 中断信号：已使用镜像数: $used_images" | tee -a "$LOG_FILE"
          echo -e " - 中断信号：已创建实例数: ${ZONE_INSTANCE_COUNT[$zone]}" | tee -a "$LOG_FILE"
      done;
      echo -e "\n${GREEN}中断信号：全局统计：总计创建实例数 = $TOTAL_INSTANCES${NC}" | tee -a "$LOG_FILE";
      exit 1' SIGINT


#######################################
# 主逻辑区
#######################################

echo -e "\n${GREEN}====== 开始创建实例 [时间: $(date +%T)]======${NC}" | tee -a "$LOG_FILE"
image_index=0

# 遍历所有镜像
for image in $imagelist; do
    ((image_index++))
    echo -e "\n${GREEN}[镜像进度] 第 $image_index 个镜像 ($image)${NC}" | tee -a "$LOG_FILE"
    
    # 遍历所有可用区与节点
    for zone in "${AVAILABILITY_ZONES[@]}"; do
        echo -e "${GREEN}检查可用区节点: $zone${NC}" | tee -a "$LOG_FILE"
        
        # 检查所有可用区实例总数限制
        while [ $(get_total_instances_all_zones) -ge $MAX_TOTAL_INSTANCES_ALL_ZONE ]; do
            echo -e "${YELLOW}等待：此时刻所有节点的实例总数已达上限 $MAX_TOTAL_INSTANCES_ALL_ZONE ，等待资源释放...${NC}" | tee -a "$LOG_FILE"
            sleep $SLEEP_TIME
        done
        
        # 检查镜像实例数量限制
        while [ $(get_instance_count "$image" "$zone") -ge $MAX_INSTANCES_per_Image_per_Zone ]; do
            echo -e "${YELLOW}等待：镜像 $image 在 $zone 的实例数已达上限 ($MAX_INSTANCES_per_Image_per_Zone)${NC}" | tee -a "$LOG_FILE"
            sleep $SLEEP_TIME
        done
        
        # 执行实例创建
        create_instance "$image" "$zone"
    done
done


#######################################
# 最终统计输出
#######################################

echo -e "\n${GREEN}====== 正常结束执行结果统计 ======${NC}" | tee -a "$LOG_FILE"
for zone in "${AVAILABILITY_ZONES[@]}"; do
    echo -e "\n${GREEN}可用区 [$zone] 统计：${NC}" | tee -a "$LOG_FILE"
    used_images=$(echo "${ZONE_IMAGE_COUNT[$zone]}" | wc -w)
    echo -e " - 使用镜像数: $used_images" | tee -a "$LOG_FILE"
    echo -e " - 创建实例数: ${ZONE_INSTANCE_COUNT[$zone]}" | tee -a "$LOG_FILE"
done
echo -e "\n${GREEN}最后统计：累计成功创建 $TOTAL_INSTANCES 个实例${NC}" | tee -a "$LOG_FILE"