全球主机交流论坛

标题: 阿里云alpine系统，检测流量达标则关机脚本 [打印本页]

---

作者: 调查员    时间: 2024-9-18 19:16
标题: 阿里云alpine系统，检测流量达标则关机脚本


此脚本是之前坛友发的，
我稍加修改，
使流量检测更准。
（之前的非常不准，我这个脚本，经过我多天检测，跟阿里云后台的流量变动几乎一致）

添加了五分钟后关机的命令。（怕开机后死循环）


建议定时任务，设为10分钟运行一次。
*/10    *       *       *       *       ash /root/aliyun_traffic.sh >/dev/null
（alpine系统，是用ash代替bash）

1. #!/bin/bash
   
2. 
   
3. # 获取当前脚本的绝对路径
   
4. SCRIPT_PATH=$(realpath "$0")
   
5. 
   
6. # 保存流量数据的目录和文件
   
7. TRAFFIC_DIR="/root/network_traffic"
   
8. TRAFFIC_FILE="$TRAFFIC_DIR/network_traffic.dat"
   
9. LOG_FILE="$TRAFFIC_DIR/network_traffic_monitor.log"
   
10. CURRENT_MONTH=$(date +"%Y-%m")
    
11. SHUTDOWN_THRESHOLD=$((19 * 1024 * 1024 * 1024))  # 19GB 转换为字节的整数表示
    
12. 
    
13. # 要监控的网络接口
    
14. INTERFACE="eth0"
    
15. 
    
16. # 创建保存流量数据的目录
    
17. if [ ! -d "$TRAFFIC_DIR" ]; then
    
18.     mkdir -p "$TRAFFIC_DIR"
    
19.     if [ $? -eq 0 ]; then
    
20.         echo "目录 $TRAFFIC_DIR 创建成功"
    
21.     else
    
22.         echo "无法创建目录 $TRAFFIC_DIR"
    
23.         exit 1
    
24.     fi
    
25. fi
    
26. 
    
27. 
    
28. # 定义日志记录函数
    
29. log_message() {
    
30.     local message="$1"
    
31.     echo "$(date +"%Y-%m-%d %H:%M:%S") - $message" >> $LOG_FILE
    
32. }
    
33. 
    
34. # 初始化日志文件
    
35. if [ ! -f $LOG_FILE ]; then
    
36.     touch $LOG_FILE
    
37.     log_message "日志文件创建成功"
    
38. fi
    
39. 
    
40. # 如果流量文件不存在或者月份不同，则创建并初始化
    
41. if [ ! -f $TRAFFIC_FILE ]; then
    
42.     echo "$CURRENT_MONTH 0 0" > $TRAFFIC_FILE
    
43.     log_message "流量文件创建成功"
    
44. else
    
45.     saved_month=$(awk '{print $1}' $TRAFFIC_FILE)
    
46.     if [ "$saved_month" != "$CURRENT_MONTH" ]; then
    
47.         echo "$CURRENT_MONTH 0 0" > $TRAFFIC_FILE
    
48. # 月初先删除日志，再重建日志文件
    
49.         rm $LOG_FILE
    
50.         touch $LOG_FILE
    
51.         log_message "流量文件月份更新"
    
52.     fi
    
53. fi
    
54. 
    
55. # 读取之前的流量记录
    
56. read saved_month  last_total_out last_check_out < $TRAFFIC_FILE
    
57. 
    
58. # 获取当前接收和发送的字节数
    
59. 
    
60. 
    
61. 
    
62. if ! out_bytes=$(cat /sys/class/net/$INTERFACE/statistics/tx_bytes); then
    
63.     log_message "无法读取发送字节数"
    
64.     exit 1
    
65. fi
    
66. 
    
67. # 如果上次检查值大于当前值，说明流量可能已被重置
    
68. if [ "$out_bytes" -lt "$last_check_out" ]; then
    
69.     log_message "检测到网卡流量重置，恢复累积流量"
    
70.     last_check_out=0
    
71. fi
    
72. 
    
73. # 计算增量流量
    
74. delta_out=$((out_bytes - last_check_out))
    
75. 
    
76. # 计算启动前后的累计流量
    
77. total_out=$((last_total_out + delta_out))
    
78. total_bytes=$((total_out))
    
79. 
    
80. # 检查是否达到阈值
    
81. if [ "$total_bytes" -ge "$SHUTDOWN_THRESHOLD" ]; then
    
82.     log_message "总流量已达到19GB，五分钟后关机"
    
83.     sleep 5m
    
84.     poweroff
    
85. fi
    
86. 
    
87. # 自适应单位输出
    
88. if [ $total_bytes -lt 1024 ]; then
    
89.     total="$total_bytes bytes"
    
90. elif [ $total_bytes -lt $((1024 * 1024)) ]; then
    
91.     total=$(echo "scale=2; $total_bytes / 1024" | bc)
    
92.     total="$total KB"
    
93. elif [ $total_bytes -lt $((1024 * 1024 * 1024)) ]; then
    
94.     total=$(echo "scale=2; $total_bytes / 1024 / 1024" | bc)
    
95.     total="$total MB"
    
96. else
    
97.     total=$(echo "scale=2; $total_bytes / 1024 / 1024 / 1024" | bc)
    
98.     total="$total GB"
    
99. fi
    
100. 
     
101. # 输出结果
     
102. log_message "$total"
     
103. echo "$(date +%R) This Month: $total"
     
104. echo "------------------------------"
     
105. 
     
106. # 将当前流量值保存到文件
     
107. echo "$CURRENT_MONTH $total_out $out_bytes" > $TRAFFIC_FILE
     

复制代码

---

作者: 种植业生产人员    时间: 2024-9-18 19:23
牛逼

---

作者: joshuabu    时间: 2024-9-18 19:33
牛鼻

---

作者: 似毛非毛    时间: 2024-9-18 19:36
mark一下。不知道openwrt能用不

---

作者: jayvzh    时间: 2024-9-20 09:32
感谢楼主，补充debian的，
修改eth0和自己网卡一致，文件编码utf8，chmod +x 执行权限

1. #!/bin/bash
   
2. 
   
3. # 获取当前脚本的绝对路径
   
4. SCRIPT_PATH=$(realpath "$0")
   
5. 
   
6. # 保存流量数据的目录和文件
   
7. TRAFFIC_DIR="/root/network_traffic"
   
8. TRAFFIC_FILE="$TRAFFIC_DIR/network_traffic.dat"
   
9. LOG_FILE="$TRAFFIC_DIR/network_traffic_monitor.log"
   
10. CURRENT_MONTH=$(date +"%Y-%m")
    
11. SHUTDOWN_THRESHOLD=$((19 * 1024 * 1024 * 1024))  # 19GB 转换为字节的整数表示
    
12. 
    
13. # 要监控的网络接口
    
14. INTERFACE="eth0"
    
15. 
    
16. # 创建保存流量数据的目录
    
17. if [ ! -d "$TRAFFIC_DIR" ]; then
    
18.     mkdir -p "$TRAFFIC_DIR"
    
19.     if [ $? -eq 0 ]; then
    
20.         echo "目录 $TRAFFIC_DIR 创建成功"
    
21.     else
    
22.         echo "无法创建目录 $TRAFFIC_DIR"
    
23.         exit 1
    
24.     fi
    
25. fi
    
26. 
    
27. 
    
28. # 定义日志记录函数
    
29. log_message() {
    
30.     local message="$1"
    
31.     echo "$(date +"%Y-%m-%d %H:%M:%S") - $message" >> $LOG_FILE
    
32. }
    
33. 
    
34. # 初始化日志文件
    
35. if [ ! -f $LOG_FILE ]; then
    
36.     touch $LOG_FILE
    
37.     log_message "日志文件创建成功"
    
38. fi
    
39. 
    
40. # 如果流量文件不存在或者月份不同，则创建并初始化
    
41. if [ ! -f $TRAFFIC_FILE ]; then
    
42.     echo "$CURRENT_MONTH 0 0" > $TRAFFIC_FILE
    
43.     log_message "流量文件创建成功"
    
44. else
    
45.     saved_month=$(awk '{print $1}' $TRAFFIC_FILE)
    
46.     if [ "$saved_month" != "$CURRENT_MONTH" ]; then
    
47.         echo "$CURRENT_MONTH 0 0" > $TRAFFIC_FILE
    
48. # 月初先删除日志，再重建日志文件
    
49.         rm $LOG_FILE
    
50.         touch $LOG_FILE
    
51.         log_message "流量文件月份更新"
    
52.     fi
    
53. fi
    
54. 
    
55. # 读取之前的流量记录
    
56. read saved_month  last_total_out last_check_out < $TRAFFIC_FILE
    
57. 
    
58. # 获取当前接收和发送的字节数
    
59. 
    
60. 
    
61. 
    
62. if ! out_bytes=$(cat /sys/class/net/$INTERFACE/statistics/tx_bytes); then
    
63.     log_message "无法读取发送字节数"
    
64.     exit 1
    
65. fi
    
66. 
    
67. # 如果上次检查值大于当前值，说明流量可能已被重置
    
68. if [ "$out_bytes" -lt "$last_check_out" ]; then
    
69.     log_message "检测到网卡流量重置，恢复累积流量"
    
70.     last_check_out=0
    
71. fi
    
72. 
    
73. # 计算增量流量
    
74. delta_out=$((out_bytes - last_check_out))
    
75. 
    
76. # 计算启动前后的累计流量
    
77. total_out=$((last_total_out + delta_out))
    
78. total_bytes=$((total_out))
    
79. 
    
80. # 检查是否达到阈值
    
81. if [ "$total_bytes" -ge "$SHUTDOWN_THRESHOLD" ]; then
    
82.     log_message "总流量已达到19GB，五分钟后关机"
    
83.     sleep 5m
    
84.     /sbin/poweroff  # 使用完整路径
    
85. fi
    
86. 
    
87. # 自适应单位输出
    
88. if [ $total_bytes -lt 1024 ]; then
    
89.     total="$total_bytes bytes"
    
90. elif [ $total_bytes -lt $((1024 * 1024)) ]; then
    
91.     total=$(echo "scale=2; $total_bytes / 1024" | bc)
    
92.     total="$total KB"
    
93. elif [ $total_bytes -lt $((1024 * 1024 * 1024)) ]; then
    
94.     total=$(echo "scale=2; $total_bytes / 1024 / 1024" | bc)
    
95.     total="$total MB"
    
96. else
    
97.     total=$(echo "scale=2; $total_bytes / 1024 / 1024 / 1024" | bc)
    
98.     total="$total GB"
    
99. fi
    
100. 
     
101. # 输出结果
     
102. log_message "$total"
     
103. echo "$(date +%R) This Month: $total"
     
104. echo "------------------------------"
     
105. 
     
106. # 将当前流量值保存到文件
     
107. echo "$CURRENT_MONTH $total_out $out_bytes" > $TRAFFIC_FILE
     

复制代码

脚本用到bc注意安装：
apt install bc cron
crontab -e
*/10 * * * * /bin/bash /root/network_traffic/cdt-shuntdown.sh >/dev/null

---

作者: 随波逐流    时间: 2024-9-20 09:36
好帖子，都支持下

---

作者: xxrrrrr    时间: 2024-9-20 11:20
如何做到流量到限制值后，关闭指定的几个服务，这样就不用关机了，还可以用探针服务查看信息

---

作者: u78    时间: 2024-9-20 11:28

xxrrrrr 发表于 2024-9-20 11:20
如何做到流量到限制值后，关闭指定的几个服务，这样就不用关机了，还可以用探针服务查看信息 ...

把关机的指令改成杀死程序的就行了.....

---

欢迎光临 全球主机交流论坛 (https://443502.xyz/)

Powered by Discuz! X3.4