shell脚本---检测机器HDD盘的写速率

你的名字 2022-09-03 01:20 243阅读 0赞

#!/bin/bash
    #需求
    #10.12.111.X 网段的机器, 检查磁盘类型, 如果是hdd 的话, 先检查 磁盘剩余空间, run 一个dd 命令, 生成一个 100G 的文件,然后判断它的平均写入时间, 跟我们的基准值做比较, < 基准值 20% exit 1
    
    function Inspect_Disk_Type(){
        
        #获取磁盘名字 并根据每个磁盘的名字进行检查磁盘类型(如果磁盘有多个 怎么排除？)
        echo  -e "\033[35m ---------------------开始获取磁盘信息---------------------------- \033[0m"
        disk_name=$(lsblk -d -o name|grep -v nvme0n1|grep -v NAME)
        #检查磁盘类型
        disk_type=$(lsblk -d -o name,rota|grep -v NAME|grep $disk_name|awk '{print $2}')
        echo -e  "diskname: $disk_name\ndisktype:$disk_type\n"
        echo -e  "\033[32m --------------------磁盘信息获取完毕----------------------------- \033[0m"
    }
    
    function If_Disk_HDD(){
        
        #判断是否是HDD类型磁盘
        echo -e "\033[35m --------------开始检测磁盘是否为HDD类型磁盘-------------------- \033[0m"
        if [[ $disk_type == 1 ]]
        then
            echo -e "\033[32m -------------------$disk_name 是HDD类型的磁盘-------------------- \033[0m" && $1
        else
            echo -e "\033[31m --------------- $disk_name 不是HDD类型的磁盘，不符合检测要求并开始退出---------------- \033[0m"
            exit 1
        fi
    }
    
    function Check_Dick_Size(){
        
        #检查磁盘剩余空间
        echo -e "\033[35m ----------------开始检查磁盘剩余容量获取----------------------- \033[0m"
        disk_capacity=$(df -BG|grep $disk_name |awk '{print $4}'|awk '{sub(/.$/,"")}1')
        echo -e "\033[32m ----------获取到 $disk_name 磁盘剩余空间为:$disk_capacity G \033[0m"
    }
    
    function If_Disk_Size(){
        
        #进行判断容量是否满足大小
        test_size=100
        contrast_size=$(awk "BEGIN{print $test_size * 1.2}")
        echo -e "\033[35m ------------------开始进行磁盘容量检测------------------------ \033[0m "
        if [ $disk_capacity -ge $contrast_size ]
        then
            echo -e "\033[35m --------------------磁盘 $disk_name 容量满足要求 并 开始执行磁盘检测程序-------------------- \033[0m "
        else
            echo -e "\033[31m ---------------------------磁盘容量不满足要求 程序开始退出----------------------- \033[0m "
            exit 1
         fi
    }
    
    function Run_DD(){
        
        #判断目录是否存在 不存在则创建
        if [  -d ./.IO_information ]
        then
            echo "-------./.IO_information 目录已经存在 -----------"
        else
            mkdir ./.IO_information
        fi
        #如果磁盘空间够的话 运行dd命令成一个test_size大小的文件
        dd if=/dev/zero bs=10240 count=10000000 of=./$test_size.G.txt >./.IO_information/IO_information.txt 2>&1
        new_io_rate=$(grep copied ./.IO_information/IO_information.txt|awk '{print $(NF-1),$(NF)}')
        echo -e "\033[32m --------------磁盘的读写速率为：$new_io_rate --------------------- \033[0m"
        rm -rf ./$test_size.G.txt
        new_time=$(grep copied ./.IO_information/IO_information.txt|awk '{print $(NF-3)}')
        echo -e "\033[32m --------------耗时为：$new_time --------------------- \033[0m"
    }
    
    function Update_Benchmark(){
        
        #基准检测
        old_io_rate=$(grep copied ./.IO_information/IO_information.txt|awk '{print $(NF-1),$(NF)}')
        old_time=$(grep copied ./.IO_information/IO_information.txt|awk '{print $(NF-3)}')
        echo -e "\033[32m ---------------开始基准检测-------------------- \033[0m"
        comparison_results1=$(expr $new_time \> $old_time)
        if [ $comparison_results1 -eq 0 ]
        then
            echo $old_time >./.IO_information/time.txt
            echo -e "\033[35m --------------------基准仍为 $old_io_rate为准------------------ \033[0m "
        else
            echo -e "\033[31m ---------------------------基准有变化 新的基准为 $new_io_rate ----------------------- \033[0m "
            echo $new_io_rate >./.IO_information/IO_Benchmark.txt
            echo $new_time >./.IO_information/time.txt
        fi
    }
    
    function Judge_Disk(){
        
        #根据基准进行判断磁盘写速度
        #耗时比较少的基准时间
        time_benchmark=$(cat ./.IO_information/time.txt)
        #120%的基准时间 用于和新获取的时间进行对比
        eighty_percent=$(awk "BEGIN{print $time_benchmark *1.2}")
        #新获取的时间$new_time
        comparison_results2=$(expr $new_time \> $eighty_percent)
        if [ $comparison_results2 -eq 0 ]
        then
            echo -e  " \033[31m ------------ 磁盘的写入速度低于基准的80% 请及时检查磁盘------------ \033[0m "
        else
            echo -e "  \033[32m ----------------磁盘的写入速度并无异常------------- \033[0m "
        fi
    }
    
    function main(){
        
        Inspect_Disk_Type #获取磁盘信息
        If_Disk_HDD Check_Dick_Size #判断是否为HDD类型的磁盘
        If_Disk_Size #判断磁盘容量是否满足大小
        Run_DD
        filetime=$(stat -c %Y ./.IO_information/IO_information.txt)
        new_filetime=$(date +%s)
        if [ $[ $new_filetime - $filetime ] -gt 18000 ]
        then
            Update_Benchmark #进行基准检测
            Judge_Disk #判断磁盘的情况
        else
            echo -e "第一次运行并没有旧基准"
        fi
    }
    main

输出效果如下：  
![脚本执行的效果图][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjUwNjU5OQ_size_16_color_FFFFFF_t_70]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjUwNjU5OQ_size_16_color_FFFFFF_t_70]: /images/20220829/9daa8783e6a3468a9224d4967d08511a.png