894 lines
33 KiB
Bash
894 lines
33 KiB
Bash
#!/bin/bash
|
||
|
||
# 功能：自动安装或卸载 Prometheus、Node Exporter、Grafana、Blackbox Exporter 和 Alertmanager，支持用户选择安装或卸载组件及指定版本或默认安装最新版本，适配系统架构
|
||
# 作者：Grok3
|
||
# 使用方法：chmod +x install_prometheus.sh && ./install_prometheus.sh
|
||
|
||
# 设置颜色输出
|
||
RED='\033[0;31m'
|
||
GREEN='\033[0;32m'
|
||
YELLOW='\033[1;33m'
|
||
NC='\033[0m' # No Color
|
||
|
||
# 检查是否为 root 用户或有 sudo 权限
|
||
if [ "$EUID" -ne 0 ]; then
|
||
echo -e "${RED}错误：请以 root 用户或使用 sudo 权限运行此脚本！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
# 定义常量
|
||
INSTALL_DIR="/opt/prometheus"
|
||
DATA_DIR="/opt/prometheus/data"
|
||
SERVICE_FILE="/etc/systemd/system/prometheus.service"
|
||
NODE_EXPORTER_INSTALL_DIR="/opt/node_exporter"
|
||
NODE_EXPORTER_SERVICE_FILE="/etc/systemd/system/node-exporter.service"
|
||
GRAFANA_CONFIG_DIR="/etc/grafana"
|
||
GRAFANA_SERVICE_FILE="/etc/systemd/system/grafana-server.service"
|
||
BLACKBOX_INSTALL_DIR="/opt/blackbox_exporter"
|
||
BLACKBOX_DATA_DIR="/opt/blackbox_exporter/data"
|
||
BLACKBOX_SERVICE_FILE="/etc/systemd/system/blackbox-exporter.service"
|
||
ALERTMANAGER_INSTALL_DIR="/opt/alertmanager"
|
||
ALERTMANAGER_DATA_DIR="/opt/alertmanager/data"
|
||
ALERTMANAGER_SERVICE_FILE="/etc/systemd/system/alertmanager.service"
|
||
|
||
# 卸载函数
|
||
uninstall_components() {
|
||
echo -e "${YELLOW}请选择要卸载的组件：${NC}"
|
||
echo -e "${YELLOW}1. Grafana${NC}"
|
||
echo -e "${YELLOW}2. Prometheus${NC}"
|
||
echo -e "${YELLOW}3. Node Exporter${NC}"
|
||
echo -e "${YELLOW}4. Blackbox Exporter${NC}"
|
||
echo -e "${YELLOW}5. Alertmanager${NC}"
|
||
echo -e "${YELLOW}请输入要卸载的组件编号（用空格分隔多个选项，例如：1 3）：${NC}"
|
||
read -r UNINSTALL_SELECTION
|
||
|
||
# 初始化卸载标志
|
||
UNINSTALL_GRAFANA="n"
|
||
UNINSTALL_PROMETHEUS="n"
|
||
UNINSTALL_NODE_EXPORTER="n"
|
||
UNINSTALL_BLACKBOX="n"
|
||
UNINSTALL_ALERTMANAGER="n"
|
||
|
||
# 解析用户输入的选择
|
||
for SELECTED in $UNINSTALL_SELECTION; do
|
||
case $SELECTED in
|
||
1)
|
||
UNINSTALL_GRAFANA="y"
|
||
echo -e "${GREEN}将卸载 Grafana。${NC}"
|
||
;;
|
||
2)
|
||
UNINSTALL_PROMETHEUS="y"
|
||
echo -e "${GREEN}将卸载 Prometheus。${NC}"
|
||
;;
|
||
3)
|
||
UNINSTALL_NODE_EXPORTER="y"
|
||
echo -e "${GREEN}将卸载 Node Exporter。${NC}"
|
||
;;
|
||
4)
|
||
UNINSTALL_BLACKBOX="y"
|
||
echo -e "${GREEN}将卸载 Blackbox Exporter。${NC}"
|
||
;;
|
||
5)
|
||
UNINSTALL_ALERTMANAGER="y"
|
||
echo -e "${GREEN}将卸载 Alertmanager。${NC}"
|
||
;;
|
||
*)
|
||
echo -e "${RED}警告：无效选项 $SELECTED，已忽略。${NC}"
|
||
;;
|
||
esac
|
||
done
|
||
|
||
# 如果没有选择卸载任何组件，则退出
|
||
if [ "$UNINSTALL_PROMETHEUS" != "y" ] && [ "$UNINSTALL_NODE_EXPORTER" != "y" ] && [ "$UNINSTALL_GRAFANA" != "y" ] && [ "$UNINSTALL_BLACKBOX" != "y" ] && [ "$UNINSTALL_ALERTMANAGER" != "y" ]; then
|
||
echo -e "${RED}错误：未选择卸载任何组件，脚本退出！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "${YELLOW}正在卸载已选择的组件...${NC}"
|
||
|
||
# 卸载 Prometheus
|
||
if [ "$UNINSTALL_PROMETHEUS" == "y" ]; then
|
||
# 检查服务是否存在并执行操作
|
||
if systemctl list-units --full -all | grep -q "prometheus.service"; then
|
||
if systemctl is-active --quiet prometheus; then
|
||
echo -e "${YELLOW}正在停止 Prometheus 服务...${NC}"
|
||
systemctl stop prometheus
|
||
fi
|
||
if systemctl is-enabled --quiet prometheus; then
|
||
echo -e "${YELLOW}正在禁用 Prometheus 服务...${NC}"
|
||
systemctl disable prometheus
|
||
fi
|
||
fi
|
||
if [ -f "$SERVICE_FILE" ]; then
|
||
echo -e "${YELLOW}正在删除 Prometheus systemd 服务文件...${NC}"
|
||
rm -f "$SERVICE_FILE"
|
||
systemctl daemon-reload
|
||
systemctl reset-failed
|
||
fi
|
||
|
||
# 删除安装目录和数据目录
|
||
if [ -d "$INSTALL_DIR" ]; then
|
||
echo -e "${YELLOW}正在删除 Prometheus 安装目录 ($INSTALL_DIR)...${NC}"
|
||
rm -rf "$INSTALL_DIR"
|
||
fi
|
||
if [ -d "$DATA_DIR" ]; then
|
||
echo -e "${YELLOW}正在删除 Prometheus 数据目录 ($DATA_DIR)...${NC}"
|
||
rm -rf "$DATA_DIR"
|
||
fi
|
||
|
||
# 删除用户和组
|
||
if id prometheus &> /dev/null; then
|
||
echo -e "${YELLOW}正在删除 Prometheus 用户和组...${NC}"
|
||
userdel prometheus || true
|
||
fi
|
||
if getent group prometheus &> /dev/null; then
|
||
groupdel prometheus || true
|
||
fi
|
||
echo -e "${GREEN}Prometheus 已成功卸载！${NC}"
|
||
fi
|
||
|
||
# 卸载 Node Exporter
|
||
if [ "$UNINSTALL_NODE_EXPORTER" == "y" ]; then
|
||
# 检查服务是否存在并执行操作
|
||
if systemctl list-units --full -all | grep -q "node-exporter.service"; then
|
||
if systemctl is-active --quiet node-exporter; then
|
||
echo -e "${YELLOW}正在停止 Node Exporter 服务...${NC}"
|
||
systemctl stop node-exporter
|
||
fi
|
||
if systemctl is-enabled --quiet node-exporter; then
|
||
echo -e "${YELLOW}正在禁用 Node Exporter 服务...${NC}"
|
||
systemctl disable node-exporter
|
||
fi
|
||
fi
|
||
if [ -f "$NODE_EXPORTER_SERVICE_FILE" ]; then
|
||
echo -e "${YELLOW}正在删除 Node Exporter systemd 服务文件...${NC}"
|
||
rm -f "$NODE_EXPORTER_SERVICE_FILE"
|
||
systemctl daemon-reload
|
||
systemctl reset-failed
|
||
fi
|
||
|
||
# 删除安装目录
|
||
if [ -d "$NODE_EXPORTER_INSTALL_DIR" ]; then
|
||
echo -e "${YELLOW}正在删除 Node Exporter 安装目录 ($NODE_EXPORTER_INSTALL_DIR)...${NC}"
|
||
rm -rf "$NODE_EXPORTER_INSTALL_DIR"
|
||
fi
|
||
|
||
# 删除用户和组
|
||
if id node_exporter &> /dev/null; then
|
||
echo -e "${YELLOW}正在删除 Node Exporter 用户和组...${NC}"
|
||
userdel node_exporter || true
|
||
fi
|
||
if getent group node_exporter &> /dev/null; then
|
||
groupdel node_exporter || true
|
||
fi
|
||
echo -e "${GREEN}Node Exporter 已成功卸载！${NC}"
|
||
fi
|
||
|
||
# 卸载 Grafana
|
||
if [ "$UNINSTALL_GRAFANA" == "y" ]; then
|
||
# 检查服务是否存在并执行操作
|
||
if systemctl list-units --full -all | grep -q "grafana-server.service"; then
|
||
if systemctl is-active --quiet grafana-server; then
|
||
echo -e "${YELLOW}正在停止 Grafana 服务...${NC}"
|
||
systemctl stop grafana-server
|
||
fi
|
||
if systemctl is-enabled --quiet grafana-server; then
|
||
echo -e "${YELLOW}正在禁用 Grafana 服务...${NC}"
|
||
systemctl disable grafana-server
|
||
fi
|
||
fi
|
||
if [ -f "$GRAFANA_SERVICE_FILE" ] || command -v grafana-server &> /dev/null; then
|
||
echo -e "${YELLOW}正在删除 Grafana systemd 服务文件...${NC}"
|
||
rm -f "$GRAFANA_SERVICE_FILE"
|
||
systemctl daemon-reload
|
||
systemctl reset-failed
|
||
fi
|
||
|
||
# 删除配置目录
|
||
if [ -d "$GRAFANA_CONFIG_DIR" ]; then
|
||
echo -e "${YELLOW}正在删除 Grafana 配置目录 ($GRAFANA_CONFIG_DIR)...${NC}"
|
||
rm -rf "$GRAFANA_CONFIG_DIR"
|
||
fi
|
||
|
||
# 卸载 Grafana 包（如果通过包管理器安装）
|
||
if command -v apt &> /dev/null; then
|
||
apt remove -y grafana || true
|
||
elif command -v yum &> /dev/null; then
|
||
yum remove -y grafana || true
|
||
fi
|
||
echo -e "${GREEN}Grafana 已成功卸载！${NC}"
|
||
fi
|
||
|
||
# 卸载 Blackbox Exporter
|
||
if [ "$UNINSTALL_BLACKBOX" == "y" ]; then
|
||
# 检查服务是否存在并执行操作
|
||
if systemctl list-units --full -all | grep -q "blackbox-exporter.service"; then
|
||
if systemctl is-active --quiet blackbox-exporter; then
|
||
echo -e "${YELLOW}正在停止 Blackbox Exporter 服务...${NC}"
|
||
systemctl stop blackbox-exporter
|
||
fi
|
||
if systemctl is-enabled --quiet blackbox-exporter; then
|
||
echo -e "${YELLOW}正在禁用 Blackbox Exporter 服务...${NC}"
|
||
systemctl disable blackbox-exporter
|
||
fi
|
||
fi
|
||
if [ -f "$BLACKBOX_SERVICE_FILE" ]; then
|
||
echo -e "${YELLOW}正在删除 Blackbox Exporter systemd 服务文件...${NC}"
|
||
rm -f "$BLACKBOX_SERVICE_FILE"
|
||
systemctl daemon-reload
|
||
systemctl reset-failed
|
||
fi
|
||
|
||
# 检查并删除配置文件
|
||
if [ -f "$BLACKBOX_INSTALL_DIR/config.yml" ]; then
|
||
echo -e "${YELLOW}正在删除 Blackbox Exporter 配置文件...${NC}"
|
||
rm -f "$BLACKBOX_INSTALL_DIR/config.yml"
|
||
fi
|
||
|
||
# 删除安装目录和数据目录
|
||
if [ -d "$BLACKBOX_INSTALL_DIR" ]; then
|
||
echo -e "${YELLOW}正在删除 Blackbox Exporter 安装目录 ($BLACKBOX_INSTALL_DIR)...${NC}"
|
||
rm -rf "$BLACKBOX_INSTALL_DIR"
|
||
fi
|
||
if [ -d "$BLACKBOX_DATA_DIR" ]; then
|
||
echo -e "${YELLOW}正在删除 Blackbox Exporter 数据目录 ($BLACKBOX_DATA_DIR)...${NC}"
|
||
rm -rf "$BLACKBOX_DATA_DIR"
|
||
fi
|
||
|
||
# 删除用户和组
|
||
if id blackbox &> /dev/null; then
|
||
echo -e "${YELLOW}正在删除 Blackbox Exporter 用户和组...${NC}"
|
||
userdel blackbox || true
|
||
fi
|
||
if getent group blackbox &> /dev/null; then
|
||
groupdel blackbox || true
|
||
fi
|
||
echo -e "${GREEN}Blackbox Exporter 已成功卸载！${NC}"
|
||
fi
|
||
|
||
# 卸载 Alertmanager
|
||
if [ "$UNINSTALL_ALERTMANAGER" == "y" ]; then
|
||
# 检查服务是否存在并执行操作
|
||
if systemctl list-units --full -all | grep -q "alertmanager.service"; then
|
||
if systemctl is-active --quiet alertmanager; then
|
||
echo -e "${YELLOW}正在停止 Alertmanager 服务...${NC}"
|
||
systemctl stop alertmanager
|
||
fi
|
||
if systemctl is-enabled --quiet alertmanager; then
|
||
echo -e "${YELLOW}正在禁用 Alertmanager 服务...${NC}"
|
||
systemctl disable alertmanager
|
||
fi
|
||
fi
|
||
if [ -f "$ALERTMANAGER_SERVICE_FILE" ]; then
|
||
echo -e "${YELLOW}正在删除 Alertmanager systemd 服务文件...${NC}"
|
||
rm -f "$ALERTMANAGER_SERVICE_FILE"
|
||
systemctl daemon-reload
|
||
systemctl reset-failed
|
||
fi
|
||
|
||
# 删除安装目录和数据目录
|
||
if [ -d "$ALERTMANAGER_INSTALL_DIR" ]; then
|
||
echo -e "${YELLOW}正在删除 Alertmanager 安装目录 ($ALERTMANAGER_INSTALL_DIR)...${NC}"
|
||
rm -rf "$ALERTMANAGER_INSTALL_DIR"
|
||
fi
|
||
if [ -d "$ALERTMANAGER_DATA_DIR" ]; then
|
||
echo -e "${YELLOW}正在删除 Alertmanager 数据目录 ($ALERTMANAGER_DATA_DIR)...${NC}"
|
||
rm -rf "$ALERTMANAGER_DATA_DIR"
|
||
fi
|
||
|
||
# 删除用户和组
|
||
if id alertmanager &> /dev/null; then
|
||
echo -e "${YELLOW}正在删除 Alertmanager 用户和组...${NC}"
|
||
userdel alertmanager || true
|
||
fi
|
||
if getent group alertmanager &> /dev/null; then
|
||
groupdel alertmanager || true
|
||
fi
|
||
echo -e "${GREEN}Alertmanager 已成功卸载！${NC}"
|
||
fi
|
||
|
||
echo -e "${GREEN}所选组件卸载完成！${NC}"
|
||
exit 0
|
||
}
|
||
|
||
# 主菜单：选择安装或卸载
|
||
echo -e "${YELLOW}请选择操作：${NC}"
|
||
echo -e "${YELLOW}1. 安装组件${NC}"
|
||
echo -e "${YELLOW}2. 卸载组件${NC}"
|
||
echo -e "${YELLOW}请输入操作编号（1 或 2）：${NC}"
|
||
read -r OPERATION
|
||
|
||
case $OPERATION in
|
||
1)
|
||
echo -e "${GREEN}选择了安装操作。${NC}"
|
||
;;
|
||
2)
|
||
echo -e "${GREEN}选择了卸载操作。${NC}"
|
||
uninstall_components
|
||
;;
|
||
*)
|
||
echo -e "${RED}错误：无效选项 $OPERATION，脚本退出！${NC}"
|
||
exit 1
|
||
;;
|
||
esac
|
||
|
||
# 获取系统架构
|
||
ARCH=$(uname -m)
|
||
case $ARCH in
|
||
x86_64)
|
||
PROM_ARCH="linux-amd64"
|
||
;;
|
||
aarch64|arm64)
|
||
PROM_ARCH="linux-arm64"
|
||
;;
|
||
*)
|
||
echo -e "${RED}错误：不支持的系统架构 ($ARCH)！${NC}"
|
||
exit 1
|
||
;;
|
||
esac
|
||
echo -e "${GREEN}检测到系统架构：$ARCH，使用 Prometheus 架构：$PROM_ARCH${NC}"
|
||
|
||
# 检查 wget 是否安装
|
||
if ! command -v wget &> /dev/null; then
|
||
echo -e "${YELLOW}wget 未安装，正在安装...${NC}"
|
||
if command -v apt &> /dev/null; then
|
||
apt update && apt install -y wget
|
||
elif command -v yum &> /dev/null; then
|
||
yum install -y wget
|
||
else
|
||
echo -e "${RED}错误：无法确定包管理器，无法安装 wget！${NC}"
|
||
exit 1
|
||
fi
|
||
fi
|
||
|
||
# 交互式选择安装的组件
|
||
echo -e "${YELLOW}请选择要安装的组件：${NC}"
|
||
echo -e "${YELLOW}1. Grafana${NC}"
|
||
echo -e "${YELLOW}2. Prometheus${NC}"
|
||
echo -e "${YELLOW}3. Node Exporter${NC}"
|
||
echo -e "${YELLOW}4. Blackbox Exporter${NC}"
|
||
echo -e "${YELLOW}5. Alertmanager${NC}"
|
||
echo -e "${YELLOW}请输入要安装的组件编号（用空格分隔多个选项，例如：1 3）：${NC}"
|
||
read -r SELECTION
|
||
|
||
# 初始化安装标志
|
||
INSTALL_GRAFANA="n"
|
||
INSTALL_PROMETHEUS="n"
|
||
INSTALL_NODE_EXPORTER="n"
|
||
INSTALL_BLACKBOX="n"
|
||
INSTALL_ALERTMANAGER="n"
|
||
|
||
# 解析用户输入的选择
|
||
for SELECTED in $SELECTION; do
|
||
case $SELECTED in
|
||
1)
|
||
INSTALL_GRAFANA="y"
|
||
echo -e "${GREEN}将安装 Grafana。${NC}"
|
||
;;
|
||
2)
|
||
INSTALL_PROMETHEUS="y"
|
||
echo -e "${GREEN}将安装 Prometheus。${NC}"
|
||
;;
|
||
3)
|
||
INSTALL_NODE_EXPORTER="y"
|
||
echo -e "${GREEN}将安装 Node Exporter。${NC}"
|
||
;;
|
||
4)
|
||
INSTALL_BLACKBOX="y"
|
||
echo -e "${GREEN}将安装 Blackbox Exporter。${NC}"
|
||
;;
|
||
5)
|
||
INSTALL_ALERTMANAGER="y"
|
||
echo -e "${GREEN}将安装 Alertmanager。${NC}"
|
||
;;
|
||
*)
|
||
echo -e "${RED}警告：无效选项 $SELECTED，已忽略。${NC}"
|
||
;;
|
||
esac
|
||
done
|
||
|
||
# 如果没有选择安装任何组件，则退出
|
||
if [ "$INSTALL_PROMETHEUS" != "y" ] && [ "$INSTALL_NODE_EXPORTER" != "y" ] && [ "$INSTALL_GRAFANA" != "y" ] && [ "$INSTALL_BLACKBOX" != "y" ] && [ "$INSTALL_ALERTMANAGER" != "y" ]; then
|
||
echo -e "${RED}错误：未选择安装任何组件，脚本退出！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
# 安装 Prometheus 的逻辑
|
||
if [ "$INSTALL_PROMETHEUS" == "y" ]; then
|
||
echo -e "${YELLOW}正在获取 Prometheus 最新版本...${NC}"
|
||
VERSION=$(curl -s https://api.github.com/repos/prometheus/prometheus/releases/latest | grep '"tag_name":' | sed -E 's/.*"v([^"]+)".*/\1/')
|
||
if [ -z "$VERSION" ]; then
|
||
echo -e "${RED}错误：无法获取 Prometheus 最新版本，请检查网络！${NC}"
|
||
exit 1
|
||
fi
|
||
echo -e "${GREEN}Prometheus 最新版本为：$VERSION${NC}"
|
||
DOWNLOAD_URL="https://github.com/prometheus/prometheus/releases/download/v${VERSION}/prometheus-${VERSION}.${PROM_ARCH}.tar.gz"
|
||
|
||
# 检查 Prometheus 下载链接是否有效
|
||
echo -e "${YELLOW}正在验证 Prometheus 下载链接...${NC}"
|
||
HTTP_STATUS=$(curl -s -I "$DOWNLOAD_URL" | grep -i "HTTP/" | awk '{print $2}' | head -1)
|
||
if [[ "$HTTP_STATUS" != "200" && "$HTTP_STATUS" != "302" ]]; then
|
||
echo -e "${RED}错误：无效的 Prometheus 下载链接 ($DOWNLOAD_URL)！${NC}"
|
||
echo -e "${YELLOW}HTTP 状态码：$HTTP_STATUS${NC}"
|
||
exit 1
|
||
fi
|
||
echo -e "${GREEN}Prometheus 下载链接有效，HTTP 状态码：$HTTP_STATUS${NC}"
|
||
|
||
echo -e "${YELLOW}正在创建 Prometheus 安装目录和数据目录...${NC}"
|
||
mkdir -p "$INSTALL_DIR" "$DATA_DIR"
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：创建 Prometheus 目录失败！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "${YELLOW}正在下载 Prometheus v${VERSION}...${NC}"
|
||
wget -O /tmp/prometheus.tar.gz "$DOWNLOAD_URL"
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：下载 Prometheus 失败，请检查网络！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "${YELLOW}正在解压并安装 Prometheus...${NC}"
|
||
tar -xzf /tmp/prometheus.tar.gz -C /tmp
|
||
mv /tmp/prometheus-${VERSION}.${PROM_ARCH}/* "$INSTALL_DIR"
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：解压或移动 Prometheus 文件失败！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
# 清理临时文件
|
||
rm -rf /tmp/prometheus.tar.gz /tmp/prometheus-${VERSION}.${PROM_ARCH}
|
||
|
||
# 创建 Prometheus 用户和组
|
||
echo -e "${YELLOW}正在创建 Prometheus 用户和组...${NC}"
|
||
useradd -M -s /bin/false prometheus || true
|
||
chown -R prometheus:prometheus "$INSTALL_DIR" "$DATA_DIR"
|
||
|
||
# 创建 Prometheus systemd 服务文件
|
||
echo -e "${YELLOW}正在创建 Prometheus systemd 服务文件...${NC}"
|
||
cat > "$SERVICE_FILE" <<EOF
|
||
[Unit]
|
||
Description=Prometheus
|
||
Wants=network-online.target
|
||
After=network-online.target
|
||
|
||
[Service]
|
||
User=prometheus
|
||
Group=prometheus
|
||
Type=simple
|
||
ExecStart=$INSTALL_DIR/prometheus --config.file $INSTALL_DIR/prometheus.yml --storage.tsdb.path $DATA_DIR
|
||
|
||
[Install]
|
||
WantedBy=multi-user.target
|
||
EOF
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：创建 Prometheus 服务文件失败！${NC}"
|
||
exit 1
|
||
fi
|
||
echo -e "${GREEN}Prometheus 安装完成！${NC}"
|
||
fi
|
||
|
||
# 安装 Node Exporter 的逻辑
|
||
if [ "$INSTALL_NODE_EXPORTER" == "y" ]; then
|
||
echo -e "${YELLOW}正在获取 Node Exporter 最新版本...${NC}"
|
||
NODE_EXPORTER_VERSION=$(curl -s https://api.github.com/repos/prometheus/node_exporter/releases/latest | grep '"tag_name":' | sed -E 's/.*"v([^"]+)".*/\1/')
|
||
if [ -z "$NODE_EXPORTER_VERSION" ]; then
|
||
echo -e "${RED}错误：无法获取 Node Exporter 最新版本，请检查网络！${NC}"
|
||
exit 1
|
||
fi
|
||
echo -e "${GREEN}Node Exporter 最新版本为：$NODE_EXPORTER_VERSION${NC}"
|
||
NODE_EXPORTER_DOWNLOAD_URL="https://github.com/prometheus/node_exporter/releases/download/v${NODE_EXPORTER_VERSION}/node_exporter-${NODE_EXPORTER_VERSION}.${PROM_ARCH}.tar.gz"
|
||
|
||
# 检查 Node Exporter 下载链接是否有效
|
||
echo -e "${YELLOW}正在验证 Node Exporter 下载链接...${NC}"
|
||
HTTP_STATUS=$(curl -s -I "$NODE_EXPORTER_DOWNLOAD_URL" | grep -i "HTTP/" | awk '{print $2}' | head -1)
|
||
if [[ "$HTTP_STATUS" != "200" && "$HTTP_STATUS" != "302" ]]; then
|
||
echo -e "${RED}错误：无效的 Node Exporter 下载链接 ($NODE_EXPORTER_DOWNLOAD_URL)！${NC}"
|
||
echo -e "${YELLOW}HTTP 状态码：$HTTP_STATUS${NC}"
|
||
exit 1
|
||
fi
|
||
echo -e "${GREEN}Node Exporter 下载链接有效，HTTP 状态码：$HTTP_STATUS${NC}"
|
||
|
||
echo -e "${YELLOW}正在创建 Node Exporter 安装目录...${NC}"
|
||
mkdir -p "$NODE_EXPORTER_INSTALL_DIR"
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：创建 Node Exporter 目录失败！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "${YELLOW}正在下载 Node Exporter v${NODE_EXPORTER_VERSION}...${NC}"
|
||
wget -O /tmp/node_exporter.tar.gz "$NODE_EXPORTER_DOWNLOAD_URL"
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：下载 Node Exporter 失败，请检查网络！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "${YELLOW}正在解压并安装 Node Exporter...${NC}"
|
||
tar -xzf /tmp/node_exporter.tar.gz -C /tmp
|
||
mv /tmp/node_exporter-${NODE_EXPORTER_VERSION}.${PROM_ARCH}/* "$NODE_EXPORTER_INSTALL_DIR"
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：解压或移动 Node Exporter 文件失败！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
# 清理临时文件
|
||
rm -rf /tmp/node_exporter.tar.gz /tmp/node_exporter-${NODE_EXPORTER_VERSION}.${PROM_ARCH}
|
||
|
||
# 创建 Node Exporter 用户和组
|
||
echo -e "${YELLOW}正在创建 Node Exporter 用户和组...${NC}"
|
||
useradd -M -s /bin/false node_exporter || true
|
||
chown -R node_exporter:node_exporter "$NODE_EXPORTER_INSTALL_DIR"
|
||
|
||
# 创建 Node Exporter systemd 服务文件
|
||
echo -e "${YELLOW}正在创建 Node Exporter systemd 服务文件...${NC}"
|
||
cat > "$NODE_EXPORTER_SERVICE_FILE" <<EOF
|
||
[Unit]
|
||
Description=Node Exporter
|
||
Wants=network-online.target
|
||
After=network-online.target
|
||
|
||
[Service]
|
||
User=node_exporter
|
||
Group=node_exporter
|
||
Type=simple
|
||
ExecStart=$NODE_EXPORTER_INSTALL_DIR/node_exporter
|
||
|
||
[Install]
|
||
WantedBy=multi-user.target
|
||
EOF
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：创建 Node Exporter 服务文件失败！${NC}"
|
||
exit 1
|
||
fi
|
||
echo -e "${GREEN}Node Exporter 安装完成！${NC}"
|
||
fi
|
||
|
||
# 安装 Grafana 的逻辑
|
||
if [ "$INSTALL_GRAFANA" == "y" ]; then
|
||
echo -e "${YELLOW}正在安装 Grafana...${NC}"
|
||
if command -v apt &> /dev/null; then
|
||
apt update
|
||
apt install -y software-properties-common
|
||
add-apt-repository -y "deb https://packages.grafana.com/oss/deb stable main"
|
||
curl -s https://packages.grafana.com/gpg.key | apt-key add -
|
||
apt update
|
||
apt install -y grafana
|
||
elif command -v yum &> /dev/null; then
|
||
cat > /etc/yum.repos.d/grafana.repo <<EOF
|
||
[grafana]
|
||
name=grafana
|
||
baseurl=https://packages.grafana.com/oss/rpm
|
||
repo_gpgcheck=1
|
||
enabled=1
|
||
gpgcheck=1
|
||
gpgkey=https://packages.grafana.com/gpg.key
|
||
sslverify=1
|
||
sslcacert=/etc/pki/tls/certs/ca-bundle.crt
|
||
EOF
|
||
yum install -y grafana
|
||
else
|
||
echo -e "${RED}错误：无法确定包管理器，无法安装 Grafana！${NC}"
|
||
exit 1
|
||
fi
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：安装 Grafana 失败！${NC}"
|
||
exit 1
|
||
fi
|
||
echo -e "${GREEN}Grafana 安装完成！${NC}"
|
||
fi
|
||
|
||
# 安装 Blackbox Exporter 的逻辑
|
||
if [ "$INSTALL_BLACKBOX" == "y" ]; then
|
||
echo -e "${YELLOW}正在获取 Blackbox Exporter 最新版本...${NC}"
|
||
BLACKBOX_VERSION=$(curl -s https://api.github.com/repos/prometheus/blackbox_exporter/releases/latest | grep '"tag_name":' | sed -E 's/.*"v([^"]+)".*/\1/')
|
||
if [ -z "$BLACKBOX_VERSION" ]; then
|
||
echo -e "${RED}错误：无法获取 Blackbox Exporter 最新版本，请检查网络！${NC}"
|
||
exit 1
|
||
fi
|
||
echo -e "${GREEN}Blackbox Exporter 最新版本为：$BLACKBOX_VERSION${NC}"
|
||
BLACKBOX_DOWNLOAD_URL="https://github.com/prometheus/blackbox_exporter/releases/download/v${BLACKBOX_VERSION}/blackbox_exporter-${BLACKBOX_VERSION}.${PROM_ARCH}.tar.gz"
|
||
|
||
# 检查 Blackbox Exporter 下载链接是否有效
|
||
echo -e "${YELLOW}正在验证 Blackbox Exporter 下载链接...${NC}"
|
||
HTTP_STATUS=$(curl -s -I "$BLACKBOX_DOWNLOAD_URL" | grep -i "HTTP/" | awk '{print $2}' | head -1)
|
||
if [[ "$HTTP_STATUS" != "200" && "$HTTP_STATUS" != "302" ]]; then
|
||
echo -e "${RED}错误：无效的 Blackbox Exporter 下载链接 ($BLACKBOX_DOWNLOAD_URL)！${NC}"
|
||
echo -e "${YELLOW}HTTP 状态码：$HTTP_STATUS${NC}"
|
||
exit 1
|
||
fi
|
||
echo -e "${GREEN}Blackbox Exporter 下载链接有效，HTTP 状态码：$HTTP_STATUS${NC}"
|
||
|
||
echo -e "${YELLOW}正在创建 Blackbox Exporter 安装目录和数据目录...${NC}"
|
||
mkdir -p "$BLACKBOX_INSTALL_DIR" "$BLACKBOX_DATA_DIR"
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：创建 Blackbox Exporter 目录失败！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "${YELLOW}正在下载 Blackbox Exporter v${BLACKBOX_VERSION}...${NC}"
|
||
wget -O /tmp/blackbox_exporter.tar.gz "$BLACKBOX_DOWNLOAD_URL"
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：下载 Blackbox Exporter 失败，请检查网络！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "${YELLOW}正在解压并安装 Blackbox Exporter...${NC}"
|
||
tar -xzf /tmp/blackbox_exporter.tar.gz -C /tmp
|
||
mv /tmp/blackbox_exporter-${BLACKBOX_VERSION}.${PROM_ARCH}/* "$BLACKBOX_INSTALL_DIR"
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：解压或移动 Blackbox Exporter 文件失败！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
# 清理临时文件
|
||
rm -rf /tmp/blackbox_exporter.tar.gz /tmp/blackbox_exporter-${BLACKBOX_VERSION}.${PROM_ARCH}
|
||
|
||
# 创建 Blackbox Exporter 用户和组
|
||
echo -e "${YELLOW}正在创建 Blackbox Exporter 用户和组...${NC}"
|
||
useradd -M -s /bin/false blackbox || true
|
||
chown -R blackbox:blackbox "$BLACKBOX_INSTALL_DIR" "$BLACKBOX_DATA_DIR"
|
||
|
||
# 创建 Blackbox Exporter 配置文件
|
||
echo -e "${YELLOW}正在创建 Blackbox Exporter 配置文件...${NC}"
|
||
cat > "$BLACKBOX_INSTALL_DIR/config.yml" <<EOF
|
||
modules:
|
||
http_2xx:
|
||
prober: http
|
||
http:
|
||
preferred_ip_protocol: "ip4"
|
||
http_post_2xx:
|
||
prober: http
|
||
http:
|
||
method: POST
|
||
tcp_connect:
|
||
prober: tcp
|
||
pop3s_banner:
|
||
prober: tcp
|
||
tcp:
|
||
query_response:
|
||
- expect: "^+OK"
|
||
tls: true
|
||
tls_config:
|
||
insecure_skip_verify: false
|
||
grpc:
|
||
prober: grpc
|
||
grpc:
|
||
tls: true
|
||
preferred_ip_protocol: "ip4"
|
||
grpc_plain:
|
||
prober: grpc
|
||
grpc:
|
||
tls: false
|
||
service: "service1"
|
||
ssh_banner:
|
||
prober: tcp
|
||
tcp:
|
||
query_response:
|
||
- expect: "^SSH-2.0-"
|
||
- send: "SSH-2.0-blackbox-ssh-check"
|
||
ssh_banner_extract:
|
||
prober: tcp
|
||
timeout: 5s
|
||
tcp:
|
||
query_response:
|
||
- expect: "^SSH-2.0-([^ -]+)(?: (.*))?$"
|
||
labels:
|
||
- name: ssh_version
|
||
value: "${1}"
|
||
- name: ssh_comments
|
||
value: "${2}"
|
||
irc_banner:
|
||
prober: tcp
|
||
tcp:
|
||
query_response:
|
||
- send: "NICK prober"
|
||
- send: "USER prober prober prober :prober"
|
||
- expect: "PING :([^ ]+)"
|
||
send: "PONG ${1}"
|
||
- expect: "^:[^ ]+ 001"
|
||
icmp:
|
||
prober: icmp
|
||
icmp_ttl5:
|
||
prober: icmp
|
||
timeout: 5s
|
||
icmp:
|
||
ttl: 5
|
||
EOF
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：创建 Blackbox Exporter 配置文件失败！${NC}"
|
||
exit 1
|
||
fi
|
||
chown blackbox:blackbox "$BLACKBOX_INSTALL_DIR/config.yml"
|
||
|
||
# 创建 Blackbox Exporter systemd 服务文件
|
||
echo -e "${YELLOW}正在创建 Blackbox Exporter systemd 服务文件...${NC}"
|
||
cat > "$BLACKBOX_SERVICE_FILE" <<EOF
|
||
[Unit]
|
||
Description=Blackbox Exporter
|
||
Wants=network-online.target
|
||
After=network-online.target
|
||
|
||
[Service]
|
||
User=blackbox
|
||
Group=blackbox
|
||
Type=simple
|
||
ExecStart=$BLACKBOX_INSTALL_DIR/blackbox_exporter --config.file=$BLACKBOX_INSTALL_DIR/config.yml
|
||
|
||
[Install]
|
||
WantedBy=multi-user.target
|
||
EOF
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：创建 Blackbox Exporter 服务文件失败！${NC}"
|
||
exit 1
|
||
fi
|
||
echo -e "${GREEN}Blackbox Exporter 安装完成！${NC}"
|
||
fi
|
||
|
||
# 安装 Alertmanager 的逻辑
|
||
if [ "$INSTALL_ALERTMANAGER" == "y" ]; then
|
||
echo -e "${YELLOW}正在获取 Alertmanager 最新版本...${NC}"
|
||
ALERTMANAGER_VERSION=$(curl -s https://api.github.com/repos/prometheus/alertmanager/releases/latest | grep '"tag_name":' | sed -E 's/.*"v([^"]+)".*/\1/')
|
||
if [ -z "$ALERTMANAGER_VERSION" ]; then
|
||
echo -e "${RED}错误：无法获取 Alertmanager 最新版本，请检查网络！${NC}"
|
||
exit 1
|
||
fi
|
||
echo -e "${GREEN}Alertmanager 最新版本为：$ALERTMANAGER_VERSION${NC}"
|
||
ALERTMANAGER_DOWNLOAD_URL="https://github.com/prometheus/alertmanager/releases/download/v${ALERTMANAGER_VERSION}/alertmanager-${ALERTMANAGER_VERSION}.${PROM_ARCH}.tar.gz"
|
||
|
||
# 检查 Alertmanager 下载链接是否有效
|
||
echo -e "${YELLOW}正在验证 Alertmanager 下载链接...${NC}"
|
||
HTTP_STATUS=$(curl -s -I "$ALERTMANAGER_DOWNLOAD_URL" | grep -i "HTTP/" | awk '{print $2}' | head -1)
|
||
if [[ "$HTTP_STATUS" != "200" && "$HTTP_STATUS" != "302" ]]; then
|
||
echo -e "${RED}错误：无效的 Alertmanager 下载链接 ($ALERTMANAGER_DOWNLOAD_URL)！${NC}"
|
||
echo -e "${YELLOW}HTTP 状态码：$HTTP_STATUS${NC}"
|
||
exit 1
|
||
fi
|
||
echo -e "${GREEN}Alertmanager 下载链接有效，HTTP 状态码：$HTTP_STATUS${NC}"
|
||
|
||
echo -e "${YELLOW}正在创建 Alertmanager 安装目录和数据目录...${NC}"
|
||
mkdir -p "$ALERTMANAGER_INSTALL_DIR" "$ALERTMANAGER_DATA_DIR"
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：创建 Alertmanager 目录失败！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "${YELLOW}正在下载 Alertmanager v${ALERTMANAGER_VERSION}...${NC}"
|
||
wget -O /tmp/alertmanager.tar.gz "$ALERTMANAGER_DOWNLOAD_URL"
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：下载 Alertmanager 失败，请检查网络！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
echo -e "${YELLOW}正在解压并安装 Alertmanager...${NC}"
|
||
tar -xzf /tmp/alertmanager.tar.gz -C /tmp
|
||
mv /tmp/alertmanager-${ALERTMANAGER_VERSION}.${PROM_ARCH}/* "$ALERTMANAGER_INSTALL_DIR"
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：解压或移动 Alertmanager 文件失败！${NC}"
|
||
exit 1
|
||
fi
|
||
|
||
# 清理临时文件
|
||
rm -rf /tmp/alertmanager.tar.gz /tmp/alertmanager-${ALERTMANAGER_VERSION}.${PROM_ARCH}
|
||
|
||
# 创建 Alertmanager 用户和组
|
||
echo -e "${YELLOW}正在创建 Alertmanager 用户和组...${NC}"
|
||
useradd -M -s /bin/false alertmanager || true
|
||
chown -R alertmanager:alertmanager "$ALERTMANAGER_INSTALL_DIR" "$ALERTMANAGER_DATA_DIR"
|
||
|
||
# 创建 Alertmanager systemd 服务文件
|
||
echo -e "${YELLOW}正在创建 Alertmanager systemd 服务文件...${NC}"
|
||
cat > "$ALERTMANAGER_SERVICE_FILE" <<EOF
|
||
[Unit]
|
||
Description=Alertmanager
|
||
Wants=network-online.target
|
||
After=network-online.target
|
||
|
||
[Service]
|
||
User=alertmanager
|
||
Group=alertmanager
|
||
Type=simple
|
||
ExecStart=$ALERTMANAGER_INSTALL_DIR/alertmanager --config.file=$ALERTMANAGER_INSTALL_DIR/alertmanager.yml --storage.path=$ALERTMANAGER_DATA_DIR
|
||
|
||
[Install]
|
||
WantedBy=multi-user.target
|
||
EOF
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "${RED}错误：创建 Alertmanager 服务文件失败！${NC}"
|
||
exit 1
|
||
fi
|
||
echo -e "${GREEN}Alertmanager 安装完成！${NC}"
|
||
fi
|
||
|
||
# 重新加载 systemd 并启动服务
|
||
echo -e "${YELLOW}正在启动已安装的服务...${NC}"
|
||
systemctl daemon-reload
|
||
systemctl reset-failed
|
||
|
||
if [ "$INSTALL_PROMETHEUS" == "y" ]; then
|
||
systemctl start prometheus
|
||
systemctl enable prometheus
|
||
if systemctl is-active --quiet prometheus; then
|
||
echo -e "${GREEN}Prometheus 安装成功并已启动！${NC}"
|
||
echo -e "${GREEN}访问 http://你的服务器IP:9090 查看 Prometheus Web UI。${NC}"
|
||
else
|
||
echo -e "${RED}错误：Prometheus 服务启动失败！${NC}"
|
||
echo -e "${YELLOW}请检查日志：journalctl -u prometheus -f${NC}"
|
||
exit 1
|
||
fi
|
||
fi
|
||
|
||
if [ "$INSTALL_NODE_EXPORTER" == "y" ]; then
|
||
if [ -f "$NODE_EXPORTER_SERVICE_FILE" ]; then
|
||
systemctl start node-exporter
|
||
systemctl enable node-exporter
|
||
if systemctl is-active --quiet node-exporter; then
|
||
echo -e "${GREEN}Node Exporter 安装成功并已启动！${NC}"
|
||
echo -e "${GREEN}访问 http://你的服务器IP:9100/metrics 查看 Node Exporter 指标。${NC}"
|
||
else
|
||
echo -e "${RED}错误：Node Exporter 服务启动失败！${NC}"
|
||
echo -e "${YELLOW}请检查日志：journalctl -u node-exporter -f${NC}"
|
||
exit 1
|
||
fi
|
||
else
|
||
echo -e "${RED}错误：Node Exporter 服务文件未找到 ($NODE_EXPORTER_SERVICE_FILE)！${NC}"
|
||
exit 1
|
||
fi
|
||
fi
|
||
|
||
if [ "$INSTALL_GRAFANA" == "y" ]; then
|
||
systemctl start grafana-server
|
||
systemctl enable grafana-server
|
||
if systemctl is-active --quiet grafana-server; then
|
||
echo -e "${GREEN}Grafana 安装成功并已启动！${NC}"
|
||
echo -e "${GREEN}访问 http://你的服务器IP:3000 查看 Grafana Web UI（默认用户/密码：admin/admin）。${NC}"
|
||
else
|
||
echo -e "${RED}错误：Grafana 服务启动失败！${NC}"
|
||
echo -e "${YELLOW}请检查日志：journalctl -u grafana-server -f${NC}"
|
||
exit 1
|
||
fi
|
||
fi
|
||
|
||
if [ "$INSTALL_BLACKBOX" == "y" ]; then
|
||
systemctl start blackbox-exporter
|
||
systemctl enable blackbox-exporter
|
||
if systemctl is-active --quiet blackbox-exporter; then
|
||
echo -e "${GREEN}Blackbox Exporter 安装成功并已启动！${NC}"
|
||
echo -e "${GREEN}访问 http://你的服务器IP:9115/metrics 查看 Blackbox Exporter 指标。${NC}"
|
||
else
|
||
echo -e "${RED}错误：Blackbox Exporter 服务启动失败！${NC}"
|
||
echo -e "${YELLOW}请检查日志：journalctl -u blackbox-exporter -f${NC}"
|
||
exit 1
|
||
fi
|
||
fi
|
||
|
||
if [ "$INSTALL_ALERTMANAGER" == "y" ]; then
|
||
systemctl start alertmanager
|
||
systemctl enable alertmanager
|
||
if systemctl is-active --quiet alertmanager; then
|
||
echo -e "${GREEN}Alertmanager 安装成功并已启动！${NC}"
|
||
echo -e "${GREEN}访问 http://你的服务器IP:9093 查看 Alertmanager Web UI。${NC}"
|
||
else
|
||
echo -e "${RED}错误：Alertmanager 服务启动失败！${NC}"
|
||
echo -e "${YELLOW}请检查日志：journalctl -u alertmanager -f${NC}"
|
||
exit 1
|
||
fi
|
||
fi
|
||
|
||
# 总结安装结果
|
||
echo -e "${GREEN}安装完成！${NC}"
|
||
if [ "$INSTALL_PROMETHEUS" == "y" ]; then
|
||
echo -e "${GREEN}Prometheus 版本：v${VERSION}${NC}"
|
||
fi
|
||
if [ "$INSTALL_NODE_EXPORTER" == "y" ]; then
|
||
echo -e "${GREEN}Node Exporter 版本：v${NODE_EXPORTER_VERSION}${NC}"
|
||
fi
|
||
if [ "$INSTALL_GRAFANA" == "y" ]; then
|
||
echo -e "${GREEN}Grafana 已安装（版本由包管理器确定）${NC}"
|
||
fi
|
||
if [ "$INSTALL_BLACKBOX" == "y" ]; then
|
||
echo -e "${GREEN}Blackbox Exporter 版本：v${BLACKBOX_VERSION}${NC}"
|
||
fi
|
||
if [ "$INSTALL_ALERTMANAGER" == "y" ]; then
|
||
echo -e "${GREEN}Alertmanager 版本：v${ALERTMANAGER_VERSION}${NC}"
|
||
fi
|