Update Nlbench.sh

This commit is contained in:
Jensfrank 2024-08-19 15:48:51 +08:00 committed by GitHub
parent c3686314ee
commit b11b633b23
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -100,78 +100,77 @@ check_root() {
fi
}
# 检测操作系统类型和版本
# 检测操作系统
detect_os() {
if [ -f /etc/os-release ]; then
. /etc/os-release
os_type=$ID
os_version=$VERSION_ID
elif type lsb_release >/dev/null 2>&1; then
os_type=$(lsb_release -si)
os_version=$(lsb_release -sr)
elif [ -f /etc/lsb-release ]; then
. /etc/lsb-release
os_type=$DISTRIB_ID
os_version=$DISTRIB_RELEASE
elif [ -f /etc/redhat-release ]; then
os_type=$(cat /etc/redhat-release | cut -d ' ' -f 1)
os_version=$(cat /etc/redhat-release | sed 's/.*release \([0-9\.]*\).*/\1/')
elif [ -f /etc/gentoo-release ]; then
os_type="gentoo"
os_version=$(cat /etc/gentoo-release | cut -d ' ' -f 5)
elif [ -f /etc/debian_version ]; then
os_type="debian"
elif [ -f /etc/fedora-release ]; then
os_type="fedora"
elif [ -f /etc/centos-release ]; then
os_type="centos"
else
os_type=$(uname -s)
os_version=$(uname -r)
fi
os_type=$(echo "$os_type" | tr '[:upper:]' '[:lower:]')
echo -e "${YELLOW}检测到的系统: $os_type $os_version${NC}"
os_type=$(echo $os_type | tr '[:upper:]' '[:lower:]')
echo "检测到的操作系统: $os_type"
}
# 更新系统
update_system() {
detect_os || return 1
detect_os
if [ $? -ne 0 ]; then
echo -e "${RED}无法检测操作系统。${NC}"
return 1
fi
case "${os_type,,}" in
ubuntu|debian|linuxmint|elementary|pop)
update_cmd="apt-get update"
upgrade_cmd="apt-get upgrade -y"
install_cmd="apt-get install -y"
clean_cmd="apt-get autoremove -y"
;;
centos|rhel|fedora|rocky|almalinux|openeuler)
if command -v dnf &>/dev/null; then
update_cmd="dnf check-update"
upgrade_cmd="dnf upgrade -y"
install_cmd="dnf install -y"
clean_cmd="dnf autoremove -y"
else
update_cmd="yum check-update"
upgrade_cmd="yum upgrade -y"
install_cmd="yum install -y"
clean_cmd="yum autoremove -y"
fi
;;
opensuse*|sles)
update_cmd="zypper refresh"
upgrade_cmd="zypper update -y"
install_cmd="zypper install -y"
upgrade_cmd="zypper dup -y"
clean_cmd="zypper clean -a"
;;
arch|manjaro)
update_cmd="pacman -Sy"
upgrade_cmd="pacman -Su --noconfirm"
install_cmd="pacman -S --noconfirm"
upgrade_cmd="pacman -Syu --noconfirm"
clean_cmd="pacman -Sc --noconfirm"
;;
alpine)
update_cmd="apk update"
upgrade_cmd="apk upgrade"
install_cmd="apk add"
clean_cmd="apk cache clean"
;;
gentoo)
update_cmd="emerge --sync"
upgrade_cmd="emerge -uDN @world"
install_cmd="emerge"
clean_cmd="emerge --depclean"
;;
cloudlinux)
update_cmd="yum check-update"
upgrade_cmd="yum upgrade -y"
install_cmd="yum install -y"
clean_cmd="yum clean all"
;;
*)
echo -e "${RED}不支持的 Linux 发行版: $os_type${NC}"
@ -180,37 +179,69 @@ update_system() {
esac
echo -e "${YELLOW}正在更新系统...${NC}"
sudo $update_cmd && sudo $upgrade_cmd
if [ $? -ne 0 ]; then
echo -e "${RED}系统更新失败。${NC}"
return 1
sudo $update_cmd
if [ $? -eq 0 ]; then
sudo $upgrade_cmd
if [ $? -eq 0 ]; then
echo -e "${GREEN}系统更新完成。${NC}"
echo -e "${YELLOW}正在清理系统...${NC}"
sudo $clean_cmd
echo -e "${GREEN}系统清理完成。${NC}"
# 检查是否需要重启
if [ -f /var/run/reboot-required ]; then
echo -e "${YELLOW}系统更新需要重启才能完成。请在方便时重启系统。${NC}"
fi
return 0
fi
fi
echo -e "${GREEN}系统更新完成。${NC}"
# 检查是否需要重启
if [ -f /var/run/reboot-required ]; then
echo -e "${YELLOW}系统更新需要重启才能完成。请在方便时重启系统。${NC}"
fi
return 0
echo -e "${RED}系统更新失败。${NC}"
return 1
}
# 更新系统并安装依赖
# 定义支持的操作系统类型
SUPPORTED_OS=("ubuntu" "debian" "linuxmint" "elementary" "pop" "centos" "rhel" "fedora" "rocky" "almalinux" "openeuler" "opensuse" "sles" "arch" "manjaro" "alpine" "gentoo" "cloudlinux")
# 安装依赖
install_dependencies() {
echo -e "${YELLOW}正在检查并安装必要的依赖项...${NC}"
# 确保 os_type 已定义
if [ -z "$os_type" ]; then
detect_os
fi
# 更新系统
update_system || echo -e "${RED}系统更新失败。继续安装依赖项。${NC}"
# 安装依赖
local dependencies=("curl" "wget" "iperf3" "bc")
# 检查是否为支持的操作系统
if [[ ! " ${SUPPORTED_OS[@]} " =~ " ${os_type} " ]]; then
echo -e "${RED}不支持的操作系统: $os_type${NC}"
return 1
fi
case "${os_type,,}" in
gentoo)
install_cmd="emerge"
for dep in "${dependencies[@]}"; do
if ! emerge -p $dep &>/dev/null; then
echo -e "${YELLOW}正在安装 $dep...${NC}"
if ! sudo emerge $dep; then
if ! sudo $install_cmd $dep; then
echo -e "${RED}无法安装 $dep。请手动安装此依赖项。${NC}"
fi
else
echo -e "${GREEN}$dep 已安装。${NC}"
fi
done
;;
alpine)
install_cmd="apk add"
for dep in "${dependencies[@]}"; do
if ! command -v "$dep" &> /dev/null; then
echo -e "${YELLOW}正在安装 $dep...${NC}"
if ! sudo $install_cmd "$dep"; then
echo -e "${RED}无法安装 $dep。请手动安装此依赖项。${NC}"
fi
else