Shell脚本:向磁盘中批量写入数据

一、关于本文

工作要做的监控系统需要监控磁盘空间的使用率并报警。在测试这个功能的时候需要模拟两个场景：一是磁盘空间不断增长超过设定的阈值时，需要触发报警机制；二是磁盘空间降落到低于报警阈值的时候，不再进行报警。为了测试这两个场景，我写了下面三个脚本：

1）initializer.sh：创建目录TestDir，并创建一个大文件template

2）duplicator.sh：不断复制文件template，直到磁盘空间使用率超过输入的参数为止

3）cleaner.sh：清除前面两个脚本留下的痕迹，即删除目录TestDir

二、initializer.sh

最开始创建一个大文件的方式是通过Shell向文件中写入字符并复制的方式，代码如下（initializer.sh.old）：

#!/bin/sh

#本脚本用于初始化工作，创建文件夹TestDir并写入一个大小为100M的目录

#创建文件TestDir
if [ -x "./TestDir" ]; then
    rm -rf TestDir
fi
mkdir TestDir
cd TestDir
mkdir Template
cd Template

touch template.txt

#制作大小为100K的文件template.txt
string=""
repetend="012345678901234|"
for((i=1;i<6400;i++))
do
    string=$string$repetend
done
echo $string >> template.txt

#复制1000个该文件
i=0
while [ true ]; do
    if [ "$i" -gt 1020 ]; then
        break
    fi
    cp template.txt $i
    ((i++))
done

echo "文件制造完毕，空间占用信息如下"
pwd .
du -sh .

cd ../..

exit 0

这种方式效率比较低，并且代码比较长，后来改用dd命令实现功能（initializer.sh）：

#!/bin/sh

#本脚本用于初始化工作，创建文件夹TestDir并写入一个大小为100M的目录

#创建文件TestDir
if [ -x "./TestDir" ]; then
    rm -rf TestDir
fi
mkdir TestDir
cd TestDir

dd if=/dev/zero of=template bs=1M count=1024

pwd .
du -sh .

cd ..

exit 0

这个脚本创建了TestDir目录，并在里面写入了一个1.1GB的文件template

三、duplicator.sh

脚本duplicator.sh接受一个5-95的数字，作为阈值。这个脚本不断复制initializer.sh创建的template文件，直到里面指定的磁盘空间使用率超过输入的阈值时，才停止运行。

#!/bin/sh

#运行本脚本前请先运行脚本 initializer.sh
#本脚本用于不断复制文件，直到给出的参数阈值超过当前磁盘空间利用率
#输入参数：磁盘空间使用率阈值

#函数：打印脚本使用说明
function usage()
{
    echo "Usage: ./duplicator [threshold]"
    echo "threshold is an integer in the range of [1,99]"
    echo "*Run initializer.sh before run this script"
    exit 0
}

#脚本有且只有一个输入
if [ "$#" -ne 1 ]; then
    echo "脚本应有且只有一个输入"
    usage
fi

#脚本的输入必须为5-95之间的正整数
threshold=`echo $1 | bc`
if [ "$threshold" -lt 5 -o "$threshold" -gt 95 ]; then
    echo "脚本的输入必须为5-95之间的正整数"
    usage
fi

#目录TestDir必须存在
if [ ! -d ./TestDir ]; then
    echo "缺少目录 TestDir"
    usage
fi

#文件TestDir/template必须存在
if [ ! -f ./TestDir/template ]; then
    echo "缺少文件 TestDir/template"
    usage
fi

cd TestDir

#复制文件，超过输入的阈值为止
i=0
while [ true ]; do
    cur=`df -h | grep /dev/sda3 | awk '{printf substr($5,1,length($5)-1)}'`
    echo "Current usage: $cur | Object usage: $threshold"
    if [ "$cur" -gt "$threshold" ]; then
        break;
    fi
    cp template $i
    echo " $i Duplication complete!"
    ((i++))
done

cd .. #TestDir

echo "Script finished!"

exit 0

四、cleaner.sh

这个脚本用于清除前两个脚本在系统中留下的痕迹

#!/bin/sh

#本脚本用于清空脚本initializer.sh和duplicator.sh留下的痕迹

#检查文件是否存在
if [ ! -x "./TestDir" ]; then
    echo "文件 ./TestDir 不存在，无需清除"
    exit 0
fi

#用户确认后清除文件
echo "真的要清除全部数据吗？ (y/n)"
read input
case "$input" in
    y* | Y* )
        rm -rf ./TestDir
        echo "数据删除完毕";;
    n* | N* )
        echo "放弃删除数据";;
    * )
        echo "输入未识别";;
esac

exit 0

五、调用效果截图