CentOS系统cpu、内存、磁盘占用脚本使用（内涵详解）

一、前言

  脚本仅用于压力测试不可用于其他不良用途，如若发现概不负责！

二、脚本

1.【内存脚本使用方法】：

jsjbM.sh脚本：

#!/bin/bash
################################################################
#       Mem Used Script
#       eg. ./mem.sh 10G & to start testing
#       eg. ./mem.sh stop  to stop testing and clear env
#       update: 2020-04-21  Nobita
################################################################

# 获取命令行参数
num=$1

# 获取当前用户
user=`whoami`

# 定义开始函数
start()
{
    # 检查是否已存在/tmp/memory目录
    if [ -d /tmp/memory ];then
        echo "目录 \"/tmp/memory\" 已存在！正在使用该目录。" >> mem.log
    else
        # 如果不存在，则创建/tmp/memory目录，并将其挂载为tmpfs文件系统
        sudo mkdir /tmp/memory
        mount -t tmpfs -o size=$num tmpfs /tmp/memory
    fi

    # 在/tmp/memory目录下创建一个名为block的文件，并将/dev/zero的内容写入其中
    dd if=/dev/zero of=/tmp/memory/block >> mem.log 2>&1
}

# 定义停止函数
stop()
{
    # 删除/tmp/memory目录下的block文件
    rm -rf /tmp/memory/block

    # 卸载/tmp/memory目录
    umount /tmp/memory

    # 删除/tmp/memory目录
    rmdir /tmp/memory

    # 检查是否成功删除/tmp/memory目录
    if [ -d /tmp/memory ];then
        echo "请勿删除目录 \"/tmp/memory\"，请检查。"
    else
        echo "清理环境完成！"
    fi
}

# 定义主函数
main()
{
    # 如果命令行参数为'stop'，则调用停止函数
    if [ $num == 'stop' ];then
        stop
    # 如果当前用户不是root用户，则输出提示信息并退出脚本
    elif [ $user != "root" ];then
        echo "请使用\"root\"用户执行脚本！"
        exit 1
    # 否则，调用开始函数
    else
        start
    fi
}

# 检查命令行参数个数是否为1或2，如果是，则调用主函数，否则输出用法提示信息
if [ $# = 2 -o $# = 1 ];then
    main
else
    echo '用法: <./mem.sh 10G &> 启动测试 或 <./mem.sh stop> 清理环境'
fi

使用方法：

sh jsjbM.sh 10G  \\使用10G内存
sh jsjbM.sh stop \\释放内存

2.【cpu固定占用脚本使用方法】：

jsjbC.sh脚本：

#CPU_NUM=$(cat /proc/cpuinfo |grep "processor" | wc -l) #查看本机CPU核数
CPU_NUM=$1 #查看本机CPU核数
echo $CPU_NUM
#每有一核CPU，启动一个dd进程，共启动CPU_NUM个dd进程
for i in `seq 1 $CPU_NUM`
do
	dd if=/dev/zero of=/dev/null & 
done

使用方法：

sh jsjbC.sh 2  \\占用两核cpu到达100%
pkill -9 dd    \\归还cpu

2.1【cpu动态占用脚本】：

cpu.sh脚本：

#!/bin/bash

read -p "请输入要占用的cpu使用率（写整数即可，别加%）：" numb

# 设置占用率参数（取值范围：0-100）
cpu_usage_percentage=$numb

# 计算要占用的 CPU 核心数量（使用 50% 的占用率，假设每个 CPU 核心有两个线程）
cpu_cores=$(($(nproc) * $cpu_usage_percentage / 100))

# 启动占用 CPU 的任务
for ((cpu=0; cpu<cpu_cores; cpu++))
do
    while : ; do
        :
    done &
done

echo "已启动动态占用 CPU，占用率为 $cpu_usage_percentage%"

# 按任意键停止占用 CPU 的任务
# read -n 1 -s -r -p "按任意键停止占用 CPU..."

# 结束占用 CPU 的任务
# killall -9 cpu.sh

# echo "已停止动态占用 CPU"

# exit 0

使用方法：

[root@bogon user1]# ./cpu.sh                            \\执行脚本
请输入要占用的cpu使用率（写整数即可，别加%）：50     \\输入需要占用的cpu使用率，例：50%
已启动动态占用 CPU，占用率为 50%
[root@bogon user1]# pkill -9 cpu.sh                      \\释放cpu

2.2【cpu随机触发动态脚本】：

dynamic_cpu_usage.sh脚本：

#!/bin/bash

# 最小和最大触发间隔时间（单位：秒）
MIN_INTERVAL=5
MAX_INTERVAL=30

# 最小和最大占用持续时间（单位：秒）
MIN_DURATION=60
MAX_DURATION=300

# 目标 CPU 使用率（百分比）
TARGET_CPU_USAGE=${1:-50}

cpu_load() {
  while true; do
    # 生成随机的触发间隔时间和占用持续时间
    interval=$(( RANDOM % (MAX_INTERVAL - MIN_INTERVAL + 1) + MIN_INTERVAL ))
    duration=$(( RANDOM % (MAX_DURATION - MIN_DURATION + 1) + MIN_DURATION ))

    # 计算占用 CPU 的进程数量
    cpu_cores=$(nproc)
    stress_processes=$(( (TARGET_CPU_USAGE * cpu_cores) / 100 ))

    # 打印信息
    echo "Next CPU usage will start in $interval seconds and last for $duration seconds."
    echo "Target CPU usage: $TARGET_CPU_USAGE%"

    # 等待触发间隔时间
    sleep $interval

    # 启动占用 CPU 的进程
    echo "Start CPU usage..."
    for (( i=0; i<stress_processes; i++ )); do
      (
        end_time=$(( $(date +%s) + duration ))
        while [[ $(date +%s) -lt $end_time ]]; do
          :
        done
      ) &
    done

    # 等待所有进程结束
    wait

    # 停止占用 CPU 的进程
    echo "Stop CPU usage..."
  done
}

cpu_load

使用方法：

[root@bogon user1]# nohup ./dynamic_cpu_usage.sh 50 &       \\执行随机触发动态占用50%cpu利用率脚本
[root@bogon user1]# tail -f nohup.out                       \\查看日志
[root@bogon user1]# killall -9 dynamic_cpu_usage.sh         \\停止脚本

3.【磁盘脚本使用方法】：

fill_disk.sh脚本：

#!/bin/bash

# 设定要控制的目标目录
TARGET_DIR="/data"

# 设定目标磁盘占用比例(比如 50 表示 50%)
TARGET_USAGE=$1

# 计算当前磁盘信息
CURRENT_USAGE=$(df "${TARGET_DIR}" | awk 'NR==2 {print $5}' | sed 's/%//')
TOTAL_SPACE=$(df "${TARGET_DIR}" | awk 'NR==2 {print $2}')
USED_SPACE=$(df "${TARGET_DIR}" | awk 'NR==2 {print $3}')

# 计算需要填充的空间
TARGET_SPACE=$(echo "scale=0; (${TOTAL_SPACE} * ${TARGET_USAGE}/100 - ${USED_SPACE})/1024" | bc)

# 创建填充文件以增加磁盘占用
if [ "${TARGET_SPACE}" -gt "0" ]; then
    echo "Creating filler file to occupy ${TARGET_SPACE}M"
    dd if=/dev/zero of="${TARGET_DIR}/filler_file" bs=1M count=${TARGET_SPACE}
else
    echo "Disk usage is already at or above target."
fi

使用方法：

# 修改需要操作的磁盘路径
vim  fill_disk.sh       \\修改磁盘路径
...
# 设定要控制的目标目录
TARGET_DIR="/data"      \\改成自己数据盘的路径，脚本执行完毕后会在当前路径生成所占大小的名为filler_file的文件
...
 
# 执行脚本
sh fill_disk.sh 10    \\占用磁盘10%
rm -rf filler_file   \\释放磁盘存储只需要删除filler_file文件即可

三、结束语

  本次脚本的使用教程就到此结束了，后期会给大家带来更多的脚本,感谢认真读完，如果觉得还可以，那么给作者点赞、收藏、关注吧！