vps交流
当前位置：嘟嘟社区 > bbs > 正文

阿里云alpine系统，检测流量达标则关机脚本

2024-09-18 分类：bbs

本帖最后由 调查员 于 2024-9-18 19:18 编辑

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

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

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

  1. #!/bin/bash
  3. # 获取当前脚本的绝对路径
  4. SCRIPT_PATH=$(realpath "$0")
  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 转换为字节的整数表示
  13. # 要监控的网络接口
  14. INTERFACE="eth0"
  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
  28. # 定义日志记录函数
  29. log_message() {
  30.     local message="$1"
  31.     echo "$(date +"%Y-%m-%d %H:%M:%S") – $message" >> $LOG_FILE
  32. }
  34. # 初始化日志文件
  35. if [ ! -f $LOG_FILE ]; then
  36.     touch $LOG_FILE
  37.     log_message "日志文件创建成功"
  38. fi
  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
  55. # 读取之前的流量记录
  56. read saved_month  last_total_out last_check_out < $TRAFFIC_FILE
  58. # 获取当前接收和发送的字节数
  62. if ! out_bytes=$(cat /sys/class/net/$INTERFACE/statistics/tx_bytes); then
  63.     log_message "无法读取发送字节数"
  64.     exit 1
  65. fi
  67. # 如果上次检查值大于当前值，说明流量可能已被重置
  68. if [ "$out_bytes" -lt "$last_check_out" ]; then
  69.     log_message "检测到网卡流量重置，恢复累积流量"
  70.     last_check_out=0
  71. fi
  73. # 计算增量流量
  74. delta_out=$((out_bytes – last_check_out))
  76. # 计算启动前后的累计流量
  77. total_out=$((last_total_out + delta_out))
  78. total_bytes=$((total_out))
  80. # 检查是否达到阈值
  81. if [ "$total_bytes" -ge "$SHUTDOWN_THRESHOLD" ]; then
  82.     log_message "总流量已达到19GB，五分钟后关机"
  83.     sleep 5m
  84.     poweroff
  85. fi
  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
  101. # 输出结果
  102. log_message "$total"
  103. echo "$(date +%R) This Month: $total"
  104. echo "——————————"
  106. # 将当前流量值保存到文件
  107. echo "$CURRENT_MONTH $total_out $out_bytes" > $TRAFFIC_FILE

复制代码
牛逼
牛鼻
mark一下。不知道openwrt能用不
标签：