chore(hotpath): add samply symbol summary tools (#5875)

Co-authored-by: heihutu <heihutu@gmail.com>
This commit is contained in:
houseme
2026-08-09 09:25:29 +08:00
committed by GitHub
parent 3b9c67e79b
commit 6106cd3772
5 changed files with 615 additions and 0 deletions
+51
View File
@@ -0,0 +1,51 @@
#!/usr/bin/env bash
set -euo pipefail
repo_root=$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)
tmp_dir=$(mktemp -d)
trap 'rm -rf "$tmp_dir"' EXIT
mock_ssh="$tmp_dir/ssh"
cat >"$mock_ssh" <<'EOF'
#!/usr/bin/env bash
set -euo pipefail
if read -r consumed; then
printf 'ssh-consumed-stdin:%s\n' "$consumed" >>"$CALL_LOG"
exit 24
fi
printf 'ssh:%s\n' "$*" >>"$CALL_LOG"
echo "123 12.5 456789 00:01:02"
EOF
chmod +x "$mock_ssh"
export CALL_LOG="$tmp_dir/calls.log"
output=$("$repo_root/scripts/sample_remote_rustfs_rss.sh" \
--nodes vm004,vm005 \
--duration-secs 1 \
--interval-secs 1 \
--out "$tmp_dir/rss.tsv" \
--ssh-bin "$mock_ssh")
[[ "$output" == "rss_samples=2" ]]
[[ "$(wc -l <"$tmp_dir/rss.tsv")" -eq 3 ]]
grep -q $'^ts_utc\tnode\tpid\tpcpu\trss_kib\tetime$' "$tmp_dir/rss.tsv"
grep -q $'\tvm004\t123\t12.5\t456789\t00:01:02$' "$tmp_dir/rss.tsv"
grep -q $'\tvm005\t123\t12.5\t456789\t00:01:02$' "$tmp_dir/rss.tsv"
[[ "$(grep -c '^ssh:' "$CALL_LOG")" -eq 2 ]]
if grep -q '^ssh-consumed-stdin:' "$CALL_LOG"; then
echo "ssh consumed the sampling loop stdin" >&2
exit 1
fi
grep -q 'sudo su - root -c' "$CALL_LOG"
if "$repo_root/scripts/sample_remote_rustfs_rss.sh" \
--nodes 'vm004;rm' \
--duration-secs 1 \
--out "$tmp_dir/bad.tsv" \
--ssh-bin "$mock_ssh" >"$tmp_dir/bad.stdout" 2>"$tmp_dir/bad.stderr"; then
echo "unsafe node name was accepted" >&2
exit 1
fi
grep -q 'node contains unsafe characters' "$tmp_dir/bad.stderr"
echo "test_sample_remote_rustfs_rss: ok"