Alex Yang
56 lines
1.2 KiB
Bash
Executable File
56 lines
1.2 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# DNS压力测试脚本
|
|
SERVER="127.0.0.1"
|
|
DOMAIN="example.com"
|
|
THREADS=20
|
|
QUERIES=1000
|
|
|
|
# 创建临时文件存储进程ID
|
|
pids=()
|
|
|
|
echo "开始DNS压力测试..."
|
|
echo "服务器: $SERVER"
|
|
echo "域名: $DOMAIN"
|
|
echo "线程数: $THREADS"
|
|
echo "总查询数: $QUERIES"
|
|
echo "--------------------------------------"
|
|
|
|
# 记录开始时间（秒）
|
|
start_time=$SECONDS
|
|
|
|
# 运行并发查询
|
|
for ((i=1; i<=THREADS; i++)); do
|
|
for ((j=1; j<=$((QUERIES/THREADS)); j++)); do
|
|
dig @$SERVER $DOMAIN A +short > /dev/null &
|
|
pids+=($!)
|
|
done
|
|
# 每启动5个线程暂停一下，避免系统资源瞬间耗尽
|
|
if (( $i % 5 == 0 )); then
|
|
echo "线程 $i 已启动，已完成 $i/$THREADS 个线程..."
|
|
sleep 0.5
|
|
fi
|
|
done
|
|
|
|
echo "所有 $THREADS 个线程已启动，共执行 $QUERIES 个查询，等待完成..."
|
|
|
|
# 等待所有查询完成
|
|
for pid in "${pids[@]}"; do
|
|
wait $pid
|
|
done
|
|
|
|
# 计算执行时间
|
|
elapsed=$((SECONDS - start_time))
|
|
|
|
# 计算QPS（每秒查询数）
|
|
if [ $elapsed -eq 0 ]; then
|
|
elapsed=1 # 避免除以零
|
|
fi
|
|
qps=$((QUERIES / elapsed))
|
|
|
|
echo "--------------------------------------"
|
|
echo "所有查询已完成！"
|
|
echo "执行时间: $elapsed 秒"
|
|
echo "QPS: $qps 次/秒"
|
|
echo "--------------------------------------"
|