linux资源占用脚本

1. CPU占用

脚本名：cup.sh

#!/bin/bash
 
# 构造死循环代码占用一个CPU
endless_loop() {
  echo -ne "i=0; while true ;do i=i+100; i=100;done" | /usr/bin/bash &
}
 
# 校验参数个数
if [ $# != 2 ]; then
  echo "USAGE: $0 <cpus,sleep time>"
  exit 1
fi
 
# 根据第一个参数循环执行endless_loop占用CPU，并将进程id存入数组
for i in $(seq $1); do
  endless_loop
  pid_array[$i]=$!
done
 
# 界面给出提示，并将kill 进程id命令重定向到/tmp/cpufile.txt文件中
for i in "${pid_array[@]}"; do
  echo 'execute: kill' $i
  echo 'kill' $i >>/tmp/cpufile.txt
done
 
# 给出提示，本次脚本需要执行多少秒，如果想提前结束脚本执行ctrl+C，并且要手动去kill掉进程
echo "If executed ctrl+C,Please execute the above lines manually"
echo "Please wait $2 seconds"
sleep $2
 
# 执行完成后，自动从/tmp/cpufile.txt文件中读取命令kill掉占用CPU的进程
for i in $(awk '{print $2}' /tmp/cpufile.txt); do
  echo "kill $i"
  kill $i
done
 
# 清理文件
find /tmp -name cpufile.txt -delete

调用方式

# 占用3个核，占用100秒
sh cpu.sh 3 100

2. 内存占用

memory.sh

#!/bin/bash 
# 校验参数
if [ $# != 2 ]; then
  echo "USAGE: $0 <free,sleep time>"
  exit 1
fi
 
free -g
[ -d /tmp/memory ] && exit 1 || mkdir /tmp/memory
mount -t tmpfs -o size=$1G tmpfs /tmp/memory
dd if=/dev/zero of=/tmp/memory/block
free -g
echo "If executed ctrl+C,Please execute the following lines manually"
echo "execute: rm -rf /tmp/memory/block"
echo "execute: umount /tmp/memory"
echo "execute: rmdir /tmp/memory"
echo "Please wait $2 seconds"
sleep $2
find /tmp/memory -name block -delete
umount /tmp/memory
find /tmp -name memory -delte

调用方式

# 占用3G内存，占用100秒
sh memory.sh 3 100

crontab执行示例

# cpu占用3个核，100秒
*/15 19-23,0-7 * * * nohup /bin/bash /data/scripts/cpu.sh 3 100 > /dev/null &
# cpu占用1个核，80秒
*/30 8-18 * * * nohup /bin/bash /data/scripts/cpu.sh 1 80 > /dev/null &

# 内存占用5个G，300秒
*/20 19-23,0-7 * * * nohup /bin/bash /data/scripts/memory.sh 5 300 > /dev/null &