并发解析hea,转为pdf格式

由于每次解析一个heap需要时间有点久,就写了一个自动解析程pdf的一个脚本。

down_lib.sh是需要自己写的哦,主要是用于下载自己所需程序的库,用于解析heap。

bash 复制代码
#!/bin/bash

# 优化版通用解析脚本(并发加速):批量生成 heap profile 的 PDF 报告
# down_lib.sh, heapjx.sh 放到同一目录下。
# 用法示例: ./heapjx.sh /app/heap
START_TIME=$(date +%Y-%m-%d_%H:%M:%S)
echo "开始生成 heap profile PDF 报告... 时间戳:${START_TIME}"

set -euo pipefail

# 自动查找 pprof_x86 路径
PPROF="$(find ~/ -name pprof_x86 2>/dev/null | head -n1)"
if [[ -z "${PPROF}" ]]; then
    echo "未找到 pprof_x86 工具,请检查环境!"
    exit 1
fi

EXEC="${EXEC:-./ndm_locmap_lib/ndm_locmap}"
HEAP_DIR="${HEAP_DIR:-./heap}"
PDF_DIR="${PDF_DIR:-./pdf}"
NDM_LOCMAP_DIR="${NDM_LOCMAP_DIR:-./ndm_locmap_lib}"
LIB_PREFIX="${LIB_PREFIX:-./ndm_locmap_lib}"
OPTS="--pdf --show_bytes --lib_prefix=${LIB_PREFIX}"
USERNAME="${USERNAME:-root}"
HOST="${HOST:-192.168.2.62}"
YCHEAP_DIR="${YCHEAP_DIR:-/app/heap/heap/}"

# 自动创建必要目录
mkdir -p "${PDF_DIR}" "${NDM_LOCMAP_DIR}"

# 拉取heap文件目录
# 如果当前目录下没有 heap 目录,才拉取
echo "62板子上的heap目录: $1"
if [[ ! -d ./heap ]]; then
    echo "未检测到 heap 目录,执行拉取..."
    scp -r "${USERNAME}@${HOST}:$1" .
else
    echo "heap 目录已存在,跳过拉取"
fi

# 判断 ndm_locmap_lib 目录下是否有文件,无文件才拉库
if find ./ndm_locmap_lib -maxdepth 1 -type f | grep -q .; then
    echo "ndm_locmap_lib 目录下已有文件,跳过拉取依赖库步骤"
else
    if [[ -x ./down_lib.sh ]]; then
        echo "拉取依赖库..."
        bash ./down_lib.sh ndm_locmap root 192.168.2.62 ./ndm_locmap_lib || \
        echo "警告: down_lib.sh 执行失败,跳过拉取依赖库步骤"
    else
        echo "警告: down_lib.sh 未找到或不可执行,跳过拉取依赖库步骤"
    fi
fi

# 处理 heap 文件
shopt -s nullglob
heap_files=("${HEAP_DIR}"/ndm_locmap_*.heap)
if [[ ${#heap_files[@]} -eq 0 ]]; then
    echo "未在 ${HEAP_DIR} 目录下找到 ndm_locmap_*.heap 文件"
    exit 1
fi

# 获取CPU核心数(并发数),允许通过环境变量指定
CPU_CORES="${CPU_CORES:-$(nproc)}"

echo "开始并发生成 PDF,每次最多并发 ${CPU_CORES} 个任务..."

export PPROF EXEC OPTS PDF_DIR

gen_pdf() {
    heapfile="$1"
    fname=$(basename "$heapfile")
    base="${fname%.heap}" # 去掉结尾的 .heap
    outpdf="${PDF_DIR}/${base}.heap.pdf"
    echo "生成 ${outpdf} ..."
    "${PPROF}" "${EXEC}" "$heapfile" ${OPTS} > "${outpdf}"
    if [[ $? -eq 0 ]]; then
        echo "✔ ${outpdf} 生成成功"
    else
        echo "✗ ${outpdf} 生成失败"
    fi
}

export -f gen_pdf

# GNU parallel优先,否则用xargs(大多数Linux系统已自带)
if command -v parallel &> /dev/null; then
    printf "%s\n" "${heap_files[@]}" | parallel -j "${CPU_CORES}" gen_pdf {}
else
    printf "%s\n" "${heap_files[@]}" | xargs -n1 -P "${CPU_CORES}" -I{} bash -c 'gen_pdf "$@"' _ {}
fi

ENDTIME=$(date +%Y-%m-%d_%H:%M:%S)
echo "全部并发生成完成!时间戳:${ENDTIME}"
相关推荐
FreeBuf_6 小时前
微软365 PDF导出功能存在本地文件包含漏洞,可泄露敏感服务器数据
服务器·microsoft·pdf
muyun280011 小时前
安全访问云端内部应用:用frp的stcp功能解决SSH转发的痛点
运维·安全·ssh·frp
熊出没15 小时前
Vue前端导出页面为PDF文件
前端·vue.js·pdf
qinyia16 小时前
Wisdom SSH:探索AI助手在复杂运维任务中的卓越表现
运维·人工智能·ssh
最懒的菜鸟16 小时前
MinerU将PDF转成md文件,并分拣图片
人工智能·pdf
开开心心_Every1 天前
便捷的电脑自动关机辅助工具
开发语言·人工智能·pdf·c#·电脑·音视频·sublime text
用什么都重名1 天前
MinerU:高效智能PDF文档解析工具完全指南
人工智能·python·pdf·mineru·makedown
Xiaouuuuua1 天前
一个简单的脚本,让pdf开启夜间模式
java·前端·pdf
小Lu的开源日常1 天前
在 macOS 上设置 SSH 和 Git
git·macos·ssh
tao3556671 天前
树莓派免密登录(vs code/cursor)
linux·嵌入式硬件·ssh