TrueNAS SCALE - 硬盘休眠

#!/bin/bash

LOG_DIR="/root/disk_monitor"
mkdir -p $LOG_DIR

STATS_FILE="$LOG_DIR/spindown_stats.txt"
DEBUG_FILE="$LOG_DIR/spindown_debug.log"

DISKS=("/dev/sda" "/dev/sdb")
declare -A last_state
declare -A standby_count
declare -A active_count

# 初始化状态为 unknown
for disk in "${DISKS[@]}"; do
    last_state[$disk]="unknown"
    standby_count[$disk]=0
    active_count[$disk]=0
done

# 添加调试函数
debug_log() {
    echo "$(date '+%Y-%m-%d %H:%M:%S'): $1" >> $DEBUG_FILE
}

# 更新统计信息的函数
update_stats() {
    local current_time="$1"
    local today=$(date '+%Y-%m-%d')
    
    echo "=== 统计信息 ($current_time) ===" > $STATS_FILE
    echo "今日日期: $today" >> $STATS_FILE
    echo "监控开始时间: $(head -n 1 $DEBUG_FILE 2>/dev/null | cut -d':' -f1 || echo $current_time)" >> $STATS_FILE
    echo "" >> $STATS_FILE
    
    for disk in "${DISKS[@]}"; do
        disk_name=$(basename $disk)
        echo "硬盘 $disk_name:" >> $STATS_FILE
        echo "- 进入休眠次数: ${standby_count[$disk]}" >> $STATS_FILE
        echo "- 唤醒次数: ${active_count[$disk]}" >> $STATS_FILE
        echo "" >> $STATS_FILE
    done
}

while true; do
    current_time=$(date '+%Y-%m-%d %H:%M:%S')
    
    for disk in "${DISKS[@]}"; do
        disk_name=$(basename $disk)
        current_state=$(hdparm -C $disk 2>/dev/null | grep "drive state" | awk '{print $NF}')
        
        # 添加调试信息
        debug_log "硬盘: $disk_name"
        debug_log "当前状态: $current_state"
        debug_log "上次状态: ${last_state[$disk]}"
        
        if [ "${last_state[$disk]}" != "$current_state" ]; then
            debug_log "状态发生变化"
            if [ "$current_state" = "standby" ]; then
                ((standby_count[$disk]++))
                debug_log "$disk_name 进入休眠，计数: ${standby_count[$disk]}"
            elif [ "$current_state" = "active/idle" ]; then
                ((active_count[$disk]++))
                debug_log "$disk_name 被唤醒，计数: ${active_count[$disk]}"
            fi
            
            # 更新状态前记录
            debug_log "更新状态: ${last_state[$disk]} -> $current_state"
            last_state[$disk]=$current_state
            # 更新状态后验证
            debug_log "更新后的状态: ${last_state[$disk]}"
            
            update_stats "$current_time"
        else
            debug_log "状态未变化"
        fi
        
        debug_log "-------------------"
    done
    
    # 日志轮转（保持最近7天的日志）
    find $LOG_DIR -name "spindown_debug.log.*" -mtime +7 -delete 2>/dev/null
    if [ $(stat -c%s "$DEBUG_FILE" 2>/dev/null) -gt 1048576 ]; then  # 1MB
        mv $DEBUG_FILE "$DEBUG_FILE.$(date '+%Y%m%d')"
    fi
    
    sleep 300  # 每5分钟检查一次
done