PVE如何优雅地实现硬盘柜随需开机，用完下线 - Chiphell - 分享与交流用户体验

#!/bin/bash
# 虚拟机编号
VM_ID=102
# 轮询间隔时间（秒）
POLL_INTERVAL=60
# 最大轮询次数
MAX_RETRIES=10
# 检查虚拟机是否已关机
check_vm_status() {
local vm_id=$1
local status=$(qm status "$vm_id" | awk '{print $2}')
if [[ "$status" == "stopped" ]]; then
return 0
else
return 1
fi
}
# 动态筛选 USB 类型的硬盘设备
get_usb_devices() {
lsblk -o NAME,TRAN | grep -i "usb" | awk '{print $1}'
}
# 安全断开 USB 设备
safely_remove_usb() {
local device=$1
# 检查设备是否存在
if [[ ! -e /dev/$device ]]; then
echo "Device /dev/$device not found. Skipping."
return
fi
# 检查是否安装了 udisksctl
if command -v udisksctl >/dev/null 2>&1; then
echo "Using udisksctl to safely remove /dev/$device..."
# 卸载设备的所有分区
for part in $(lsblk -ln -o NAME /dev/$device); do
if mount | grep -q "/dev/$part"; then
udisksctl unmount -b /dev/$part 2>/dev/null
echo "Unmounted /dev/$part"
fi
done
# 关闭设备电源
udisksctl power-off -b /dev/$device
echo "Device /dev/$device powered off."
else
echo "udisksctl not found. Falling back to /sys method for /dev/$device..."
# 卸载设备的所有分区
for part in $(lsblk -ln -o NAME /dev/$device); do
if mount | grep -q "/dev/$part"; then
umount /dev/$part 2>/dev/null
echo "Unmounted /dev/$part"
fi
done
# 从系统中移除设备
echo 1 > /sys/block/$device/device/delete
echo "Device /dev/$device removed from system."
fi
}
# 主逻辑
main() {
local retries=0
while true; do
if (( retries >= MAX_RETRIES )); then
echo "Maximum retries reached. Exiting."
exit 1
fi
# 检查虚拟机状态
if check_vm_status "$VM_ID"; then
echo "VM $VM_ID is offline. Proceeding with USB device removal."
# 动态获取所有 USB 类型硬盘设备
local usb_devices=$(get_usb_devices)
if [[ -z "$usb_devices" ]]; then
echo "No USB devices found. Exiting."
break
fi
# 安全断开所有筛选出的 USB 设备
for device in $usb_devices; do
safely_remove_usb "$device"
done
echo "All USB devices have been safely removed."
break
else
echo "VM $VM_ID is still running. Retrying in $POLL_INTERVAL seconds..."
fi
retries=$((retries + 1))
sleep "$POLL_INTERVAL"
done
}
main

复制代码