流量监控脚本

### 安装依赖
```bash
 apt install vnstat jq -y
```
启动并自启
```bash
sudo systemctl start vnstat
sudo systemctl enable vnstat
```
### 脚本vim traffic_limiter.sh
```bash
#!/bin/bash

# --- 用户配置 ---
# 你的主网卡名称
INTERFACE="ens3"
# 每月总流量限额 (单位: GiB)
# 例如，设置为 4000 GiB (约等于 4TB)
THRESHOLD_GIB=4000
# 限速值 (tc 使用的单位，例如 mbit)
# 5 MB/s = 40 Mbit/s
LIMIT_RATE="40mbit"
# --- 配置结束 ---

# 脚本状态文件，用于判断当前是否已经限速
STATUS_FILE="/tmp/traffic_limit_status"

# 将 GiB 阈值转换为 KiB，因为 vnstat 的 JSON 输出单位是 KiB
THRESHOLD_KIB=$(echo "$THRESHOLD_GIB * 1024 * 1024" | bc)

# 从 vnstat 获取本月的总流量 (上行 + 下行)，单位是 KiB
# --json m 1 表示获取最近一个月的 JSON 格式数据
# jq 用来解析 JSON，找到对应网卡的总流量
# 注意：如果服务器上有多个网卡，jq 的查询可能需要微调
CURRENT_KIB=$(vnstat --json m 1 | jq ".interfaces[] | select(.name == \"$INTERFACE\") | (.traffic.total.rx + .traffic.total.tx)")

# 检查是否成功获取流量数据
if [ -z "$CURRENT_KIB" ]; then
    echo "无法从 vnstat 获取网卡 $INTERFACE 的流量数据，退出脚本。"
    exit 1
fi

echo "当前流量: $CURRENT_KIB KiB"
echo "流量阈值: $THRESHOLD_KIB KiB"

# 比较当前流量和阈值
if (( $(echo "$CURRENT_KIB > $THRESHOLD_KIB" | bc -l) )); then
    # 流量已超额
    echo "流量已超额，准备施加限速。"
    if [ ! -f "$STATUS_FILE" ]; then
        echo "当前未限速，开始应用限速规则..."
        # 应用 tc 限速规则
        tc qdisc add dev $INTERFACE root handle 1: htb default 11
        tc class add dev $INTERFACE parent 1: classid 1:1 htb rate 1000mbit
        tc class add dev $INTERFACE parent 1:1 classid 1:11 htb rate $LIMIT_RATE
        echo "限速规则已应用。"
        # 创建状态文件标记为“已限速”
        touch "$STATUS_FILE"
    else
        echo "系统已处于限速状态，无需重复操作。"
    fi
else
    # 流量未超额
    echo "流量正常，检查是否需要解除限速。"
    if [ -f "$STATUS_FILE" ]; then
        echo "流量已恢复正常，开始解除限速规则..."
        # 移除 tc 限速规则
        tc qdisc del dev $INTERFACE root
        echo "限速规则已解除。"
        # 删除状态文件
        rm "$STATUS_FILE"
    else
        echo "系统未限速，无需操作。"
    fi
fi

exit 0
```
设置执行权限
```bash
sudo chmod +x traffic_limiter.sh
```
### 设置定时任务 (Cron Job)
```bash
*/5 * * * * /root/traffic_limiter.sh  >> /var/log/traffic_limiter.log 2>&1 # 每五分钟执行一次并记录到日志文件
0 0 1 * * truncate -s 0 /var/log/traffic_limiter.log # 每月清空一次日志文件
```