1
#!/bin/bash
2
#================================
3
# 作者:     久棹
4
# 用途:     在 Linux 上自动安装/卸载 Prometheus Server(二进制方式)
5
# 适用系统: CentOS 7/8/9, Rocky, Ubuntu 18.04+, Debian 10+ 等 systemd 系统
6
#================================
7

8
#================================
9
# 第一部分: 定义变量
10
# 把常用的路径、版本号、下载地址定义为变量，方便后续修改和维护
11
#================================
12

13
# --- Prometheus 相关配置 ---
14
# Prometheus 的版本号（要与 download 目录下的 tar.gz 文件名匹配）
15
# 注意: 变量名用 PROM_VERSION，避免与 /etc/os-release 里的 VERSION 冲突
16
PROM_VERSION=3.13.2
17
# 系统架构
18
ARCH=amd64
19
# Prometheus 压缩包的文件名
20
SOFTWARE=prometheus-${PROM_VERSION}.linux-${ARCH}.tar.gz
21
# Prometheus 在 GitHub 上的下载地址
22
URL=https://github.com/prometheus/prometheus/releases/download/v${PROM_VERSION}/${SOFTWARE}
23

24
# --- 本地目录 ---
25
# 预下载文件的存放目录（脚本所在目录下的 download/）
26
DOWNLOAD=./download
27
# 解压后的临时目录名（解压后先改名 prometheus）
28
# 复用 ARCH 变量，避免硬编码 amd64
29
TMPDIR=prometheus-${PROM_VERSION}.linux-${ARCH}
30
# systemd 服务文件的存放路径
31
SYSTEMD_DIR=/etc/systemd/system
32
# systemd 服务名（与笔记一致）
33
SERVICE=prometheus
34

35
# --- 安装路径配置（mv 方案，无软链接） ---
36
# 二进制文件 mv 到 /usr/local/bin/（全局命令）
37
BIN_DIR=/usr/local/bin
38
# 配置文件目录
39
CONFIG_DIR=/etc/prometheus
40
# 数据目录（systemd 里 --storage.tsdb.path 指定）
41
DATA_DIR=/var/lib/prometheus
42
# 日志目录（systemd 里 &>> 重定向目标）
43
LOG_DIR=/var/log/prometheus
44

45
# --- 服务监听配置 ---
46
HOSTIP=0.0.0.0
47
PORT=9090
48
# 主机名（安装完成提示用）
49
HOSTNAME=`hostname`
50

51
# 加载操作系统的 ID 变量（如 centos、ubuntu），用于判断系统类型
52
. /etc/os-release
53
OS_VERSION=$ID
54

55
#================================
56
# 第二部分: 工具函数
57
#================================
58

59
#----------------------------------------------------
60
# 颜色变量
61
#----------------------------------------------------
62
RED='\e[31m'
63
GREEN='\e[32m'
64
YELLOW='\e[33m'
65
BLUE='\e[34m'
66
RESET='\e[0m'
67

68
#----------------------------------------------------
69
# 函数名: color
70
# 作用:   按编号输出彩色文本，调用简洁（color 1 "文本"）
71
# 参数:
72
#   $1 - 颜色编号（1红 2绿 3黄 4蓝）
73
#   $* - 要输出的文本
74
#----------------------------------------------------
75
function color(){
76
    local c=$RESET
77
    case $1 in
78
        1) c=$RED ;;
79
        2) c=$GREEN ;;
80
        3) c=$YELLOW ;;
81
        4) c=$BLUE ;;
82
    esac
83
    shift
84
    echo -e "${c}$@${RESET}"
85
}
86

87
#----------------------------------------------------
88
# 函数名: check_root
89
# 作用:   检查是否以 root 用户运行，非 root 直接退出
90
#----------------------------------------------------
91
function check_root(){
92
    if [ "$(id -u)" -ne 0 ]; then
93
        color 1 "错误: 请使用 root 用户或 sudo 运行此脚本！"
94
        color 1 "用法: sudo bash $0 install"
95
        exit 1
96
    fi
97
}
98

99
#----------------------------------------------------
100
# 函数名: prepare
101
# 作用:   下载 Prometheus 软件包（如果本地已经有就跳过下载，节省时间）
102
#----------------------------------------------------
103
function prepare(){
104
    # download 目录不存在则创建
105
    [ -d ${DOWNLOAD} ] || mkdir -p ${DOWNLOAD}
106

107
    # 判断系统是否安装 wget
108
    if [ "$OS_VERSION" == "centos" ] || [ "$OS_VERSION" == "rhel" ] || [ "$OS_VERSION" == "rocky" ]; then
109
        rpm -qa | grep -q wget || yum -y install wget
110
    elif [ "$OS_VERSION" == "ubuntu" ] || [ "$OS_VERSION" == "debian" ]; then
111
        dpkg -l | grep -q wget || apt -y install wget
112
    fi
113

114
    # 判断文件是否存在且非空，不存在才下载
115
    if [ -s ${DOWNLOAD}/${SOFTWARE} ]; then
116
        echo "软件包已存在，跳过下载: ${DOWNLOAD}/${SOFTWARE}"
117
    else
118
        echo "正在下载 Prometheus v${PROM_VERSION} ..."
119
        wget -T 20 -t 3 ${URL} -O ${DOWNLOAD}/${SOFTWARE}
120
        if [ $? -ne 0 ]; then
121
            rm -f ${DOWNLOAD}/${SOFTWARE}
122
            color 1 "错误: 无法下载 ${SOFTWARE} ，请检查网络连接！"
123
            color 1 "下载地址: ${URL}"
124
            color 1 "提示: 若网络不佳，可先手动下载放到 ${DOWNLOAD}/ 目录"
125
            exit 100
126
        fi
127
        echo "下载完成: ${SOFTWARE}"
128
    fi
129
}
130

131
#================================
132
# 第三部分: 安装函数
133
#================================
134

135
function InstallPrometheus(){
136
    echo "========================================="
137
    echo "  开始安装 Prometheus v${PROM_VERSION}"
138
    echo "========================================="
139

140
    # --- 第 1 步: 检查 root 权限 ---
141
    check_root
142

143
    # --- 第 2 步: 准备环境 + 下载软件包 ---
144
    prepare
145

146
    # --- 第 3 步: 解压到 /usr/local/ 并改名 ---
147
    echo "正在解压 Prometheus ..."
148
    tar xf ${DOWNLOAD}/${SOFTWARE} -C /usr/local/
149
    mv /usr/local/${TMPDIR} /usr/local/prometheus
150
    echo "解压完成，目录已改名: /usr/local/prometheus"
151

152
    # --- 第 4 步: 二进制 mv 到 /usr/local/bin/（全局命令） ---
153
    echo "正在移动二进制到 /usr/local/bin/ ..."
154
    mv /usr/local/prometheus/prometheus ${BIN_DIR}/
155
    mv /usr/local/prometheus/promtool ${BIN_DIR}/
156
    echo "二进制已安装: ${BIN_DIR}/prometheus ${BIN_DIR}/promtool"
157

158
    # --- 第 5 步: 创建配置目录并移动配置文件 ---
159
    echo "正在配置目录与配置文件 ..."
160
    mkdir -p ${CONFIG_DIR}
161
    mv /usr/local/prometheus/prometheus.yml ${CONFIG_DIR}/
162
    echo "配置文件已安装: ${CONFIG_DIR}/prometheus.yml"
163

164
    # --- 第 6 步: 删除源目录（二进制和配置都已 mv 走，目录没用了） ---
165
    echo "正在删除源目录 ..."
166
    rm -rf /usr/local/prometheus
167

168
    # --- 第 7 步: 创建数据与日志目录 ---
169
    # 数据目录: systemd 里 --storage.tsdb.path 指定
170
    # 日志目录: systemd 里 &>> 重定向目标，不存在会导致启动失败
171
    echo "正在创建数据与日志目录 ..."
172
    mkdir -p ${DATA_DIR} ${LOG_DIR}
173
    echo "数据目录: ${DATA_DIR}"
174
    echo "日志目录: ${LOG_DIR}"
175

176
    # --- 第 8 步: 生成 systemd 服务文件 ---
177
    echo "正在生成 systemd 服务文件 ..."
178
    cat > ${SYSTEMD_DIR}/${SERVICE}.service <<EOF
179
[Unit]
180
# 服务描述
181
Description=Prometheus Server
182
# 官方文档地址
183
Documentation=https://prometheus.io/docs/introduction/overview/
184
# 等待网络就绪后再启动
185
After=network-online.target
186
# 主动声明依赖网络在线目标
187
Wants=network-online.target
188

189
[Service]
190
# 前台常驻进程：ExecStart 拉起即算成功（不 fork 不后台化）
191
Type=simple
192
# 异常退出时自动重启
193
Restart=on-failure
194
# 重启间隔 3 秒
195
RestartSec=3
196
# 文件描述符上限，防止高并发连接数超出
197
LimitNOFILE=65535
198
# 启动命令（bash -c 包裹，日志重定向到文件）
199
# ⚠️ 必须包 /bin/bash -c：systemd 直接 exec 二进制不经 shell，&>> 会被当参数报错
200
ExecStart=/bin/bash -c "${BIN_DIR}/prometheus \
201
    --config.file=${CONFIG_DIR}/prometheus.yml \
202
    --web.enable-lifecycle \
203
    --storage.tsdb.path=${DATA_DIR} \
204
    --storage.tsdb.retention.time=60d \
205
    --storage.tsdb.retention.size=512MB \
206
    --web.listen-address=${HOSTIP}:${PORT} \
207
    --web.max-connections=65535 \
208
    --web.read-timeout=5m \
209
    --query.timeout=10s \
210
    --query.max-concurrency=20 \
211
    --log.level=info \
212
    --log.format=json \
213
    &>> ${LOG_DIR}/prometheus.log"
214
# ⚠️ 不用 ExecReload！ExecStart 是 bash -c，SIGHUP 会杀掉 bash 导致进程树全灭
215
#    热加载只能走 HTTP API: curl -X POST http://localhost:9090/-/reload
216

217
[Install]
218
# 开机自启
219
WantedBy=multi-user.target
220
EOF
221
    echo "systemd 服务文件已生成: ${SYSTEMD_DIR}/${SERVICE}.service"
222

223
    # --- 第 9 步: 启动服务并设置开机自启 ---
224
    systemctl daemon-reload
225
    systemctl enable --now ${SERVICE} &> /dev/null
226
    echo "服务已启动并设置为开机自启"
227

228
    # --- 第 10 步: 验证安装结果 ---
229
    sleep 2
230
    if [ "$(systemctl is-active ${SERVICE})" == "active" ]; then
231
        color 2 ""
232
        color 2 "========================================="
233
        color 2 "  安装成功！Prometheus v${PROM_VERSION} 已运行"
234
        color 2 "========================================="
235
        color 2 ""
236
        echo "服务状态: $(systemctl is-active ${SERVICE})"
237
        echo "监听端口:"
238
        ss -lnt | grep ${PORT} || netstat -lnt | grep ${PORT}
239
        echo ""
240
        echo "WebUI 地址: http://${HOSTIP}:${PORT}/"
241
        echo "日志文件: ${LOG_DIR}/prometheus.log"
242
        echo ""
243
        echo "服务命令:"
244
        echo "  查看状态: systemctl status ${SERVICE}"
245
        echo "  重启服务: systemctl restart ${SERVICE}"
246
        echo "  热加载配置: curl -X POST http://localhost:${PORT}/-/reload"
247
    else
248
        color 1 "错误: 服务启动失败，请查看日志: ${LOG_DIR}/prometheus.log"
249
        exit 1
250
    fi
251
}
252

253
#==================
254
# 第四部分: 卸载函数
255
#==================
256

257
function RemovePrometheus(){
258
    echo "========================================="
259
    echo "  开始卸载 Prometheus"
260
    echo "========================================="
261

262
    # --- 第 1 步: 检查 root 权限 ---
263
    check_root
264

265
    # --- 第 2 步: 停止服务并禁用开机自启 ---
266
    echo "正在停止服务并取消开机自启 ..."
267
    systemctl disable --now ${SERVICE} &> /dev/null
268

269
    # --- 第 3 步: 删除 systemd 服务文件 ---
270
    echo "正在删除 systemd 服务文件 ..."
271
    rm -f ${SYSTEMD_DIR}/${SERVICE}.service
272

273
    # --- 第 4 步: 删除二进制文件 ---
274
    echo "正在删除二进制文件 ..."
275
    rm -f ${BIN_DIR}/prometheus ${BIN_DIR}/promtool
276

277
    # --- 第 5 步: 删除配置/数据/日志目录 ---
278
    echo "正在删除配置、数据、日志目录 ..."
279
    rm -rf ${CONFIG_DIR} ${DATA_DIR} ${LOG_DIR}
280

281
    # --- 第 6 步: 重新加载 systemd 配置 ---
282
    systemctl daemon-reload
283

284
    # --- 第 7 步: 验证卸载结果 ---
285
    echo ""
286
    color 4 "卸载完成！以下内容已清除:"
287
    echo "  - ${SYSTEMD_DIR}/${SERVICE}.service"
288
    echo "  - ${BIN_DIR}/prometheus ${BIN_DIR}/promtool"
289
    echo "  - ${CONFIG_DIR} ${DATA_DIR} ${LOG_DIR}"
290
    echo "软件包保留: ${DOWNLOAD}/${SOFTWARE}"
291
    echo ""
292
}
293

294
#=============================
295
# 第五部分: 主函数 —— 脚本的入口
296
#=============================
297

298
function main(){
299
    case $1 in
300
        install|i)
301
            InstallPrometheus
302
            ;;
303
        remove|r)
304
            RemovePrometheus
305
            ;;
306
        *)
307
            echo "用法: $0 {install|i|remove|r}"
308
            echo ""
309
            echo "  示例:"
310
            echo "    安装: $0 install"
311
            echo "    卸载: $0 remove"
312
            echo "    简写: $0 i        (安装)"
313
            echo "    简写: $0 r        (卸载)"
314
            ;;
315
    esac
316
}
317

318
# 调用 main 函数，把命令行的第一个参数传进去
319
main $1