157 lines
4.5 KiB
Bash
Executable File
157 lines
4.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
LC_ALL=C
|
|
|
|
DIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
|
|
cd "$DIR"
|
|
|
|
gcc -O2 -pthread -o thread_recursive thread_recursive.c -lm
|
|
|
|
echo "Compiled thread_recursive"
|
|
|
|
# Usage:
|
|
# ./run_thread_recursive.sh [branch] [depth] [hold_seconds]
|
|
# Example:
|
|
# ./run_thread_recursive.sh 6 6 6
|
|
|
|
BRANCH=${1:-6}
|
|
DEPTH=${2:-6}
|
|
HOLD=${3:-6}
|
|
|
|
RUNS_DIR="$DIR/runs"
|
|
mkdir -p "$RUNS_DIR"
|
|
RUN_ID=$(date +"%Y%m%d_%H%M%S")
|
|
RUN_OUT="$RUNS_DIR/$RUN_ID"
|
|
mkdir -p "$RUN_OUT"
|
|
|
|
LOG_FILE="$RUN_OUT/activity.log"
|
|
CSV="$RUN_OUT/thread_recursive_scaling.csv"
|
|
|
|
log() {
|
|
echo "[$(date -Iseconds)] $*" | tee -a "$LOG_FILE"
|
|
}
|
|
|
|
count_processes() {
|
|
ls -d /proc/[0-9]* 2>/dev/null | wc -l
|
|
}
|
|
|
|
count_threads_total() {
|
|
awk '/^Threads:/ {sum+=$2} END {print sum+0}' /proc/[0-9]*/status 2>/dev/null || echo 0
|
|
}
|
|
|
|
get_mem_kb() {
|
|
awk '/^MemTotal:/ {t=$2} /^MemAvailable:/ {a=$2} END {print t" "a}' /proc/meminfo
|
|
}
|
|
|
|
sample_process_csv() {
|
|
local pid="$1"
|
|
local csv="$2"
|
|
local start_uptime="$3"
|
|
local max_seconds="$4"
|
|
|
|
local have_pidstat=0
|
|
if command -v pidstat >/dev/null 2>&1; then
|
|
have_pidstat=1
|
|
else
|
|
log "pidstat not found; install sysstat for accurate CPU sampling"
|
|
fi
|
|
|
|
local prev_cpu_total=0
|
|
local prev_cpu_idle=0
|
|
if [ -r /proc/stat ]; then
|
|
read -r _ c_user c_nice c_system c_idle c_iowait c_irq c_softirq c_steal _ _ < /proc/stat
|
|
prev_cpu_total=$((c_user + c_nice + c_system + c_idle + c_iowait + c_irq + c_softirq + c_steal))
|
|
prev_cpu_idle=$((c_idle + c_iowait))
|
|
fi
|
|
|
|
while kill -0 "$pid" 2>/dev/null; do
|
|
local now_uptime elapsed_s
|
|
now_uptime=$(awk '{print $1}' /proc/uptime)
|
|
elapsed_s=$(awk -v s="$start_uptime" -v n="$now_uptime" 'BEGIN{printf "%.3f", (n-s)}')
|
|
|
|
if [ "$max_seconds" -gt 0 ]; then
|
|
if awk -v e="$elapsed_s" -v m="$max_seconds" 'BEGIN{exit (e>m)?0:1}'; then
|
|
log "Timeout reached (${max_seconds}s), stopping PID=$pid"
|
|
kill -TERM "-$pid" 2>/dev/null || true
|
|
sleep 2
|
|
kill -KILL "-$pid" 2>/dev/null || true
|
|
break
|
|
fi
|
|
fi
|
|
|
|
local proc_count thread_total
|
|
proc_count=$(count_processes || echo 0)
|
|
thread_total=$(count_threads_total || echo 0)
|
|
|
|
local load1 load5 load15
|
|
read -r load1 load5 load15 _ < /proc/loadavg
|
|
|
|
local mem_total mem_avail mem_used
|
|
read -r mem_total mem_avail < <(get_mem_kb || echo "0 0")
|
|
mem_used=$((mem_total - mem_avail))
|
|
|
|
local cpu rss vsz nlwp
|
|
rss=$(awk '/^VmRSS:/ {print $2}' "/proc/$pid/status" 2>/dev/null || echo 0)
|
|
vsz=$(awk '/^VmSize:/ {print $2}' "/proc/$pid/status" 2>/dev/null || echo 0)
|
|
nlwp=$(awk '/^Threads:/ {print $2}' "/proc/$pid/status" 2>/dev/null || echo 0)
|
|
if [ "$have_pidstat" -eq 1 ]; then
|
|
cpu=$(pidstat -p "$pid" 1 1 2>/dev/null | awk -v p="$pid" '$1 ~ /^[0-9]/ && $3==p {print $8; exit}')
|
|
cpu=${cpu:-0.00}
|
|
else
|
|
cpu=0.00
|
|
fi
|
|
local stk=0 heap=0
|
|
|
|
local sys_cpu_pct=0.00
|
|
if [ -r /proc/stat ]; then
|
|
read -r _ c_user c_nice c_system c_idle c_iowait c_irq c_softirq c_steal _ _ < /proc/stat
|
|
local cpu_total=$((c_user + c_nice + c_system + c_idle + c_iowait + c_irq + c_softirq + c_steal))
|
|
local cpu_idle=$((c_idle + c_iowait))
|
|
local dt_total=$((cpu_total - prev_cpu_total))
|
|
local dt_idle=$((cpu_idle - prev_cpu_idle))
|
|
sys_cpu_pct=$(awk -v t="$dt_total" -v i="$dt_idle" 'BEGIN{ if (t<=0) printf "0.00"; else printf "%.2f", (100.0*(t-i))/t }')
|
|
prev_cpu_total=$cpu_total
|
|
prev_cpu_idle=$cpu_idle
|
|
fi
|
|
|
|
printf "%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s\n" \
|
|
"$(date -Iseconds)" "$elapsed_s" "$load1" "$load5" "$load15" \
|
|
"$mem_total" "$mem_avail" "$mem_used" "$proc_count" "$thread_total" \
|
|
"$pid" "$cpu" "$sys_cpu_pct" "${rss:-0}" "${vsz:-0}" "${stk:-0}" "${heap:-0}" "${nlwp:-0}" \
|
|
>> "$csv"
|
|
|
|
sleep 0.2
|
|
done
|
|
}
|
|
|
|
CMD="cd '$DIR' && exec ./thread_recursive -b $BRANCH -d $DEPTH -s $HOLD"
|
|
|
|
log "Starting thread_recursive_scaling: $CMD"
|
|
|
|
echo "timestamp,elapsed_s,load1,load5,load15,mem_total_kb,mem_available_kb,mem_used_kb,proc_count,thread_count_total,pid,cpu_pct,sys_cpu_pct,rss_kb,vsz_kb,stack_kb,heap_kb,proc_threads" > "$CSV"
|
|
|
|
START_UPTIME=$(awk '{print $1}' /proc/uptime)
|
|
|
|
if command -v setsid >/dev/null 2>&1; then
|
|
setsid bash -c "$CMD" &
|
|
PID=$!
|
|
else
|
|
bash -c "$CMD" &
|
|
PID=$!
|
|
fi
|
|
|
|
sample_process_csv "$PID" "$CSV" "$START_UPTIME" 0
|
|
|
|
set +e
|
|
wait "$PID"
|
|
EXIT_CODE=$?
|
|
set -e
|
|
|
|
log "Finished thread_recursive_scaling (exit_code=$EXIT_CODE)"
|
|
|
|
echo
|
|
echo "Run complete"
|
|
echo " samples: $CSV"
|
|
echo " log: $LOG_FILE"
|
|
echo " exit_code=$EXIT_CODE"
|