28 lines
1.6 KiB
Bash
Executable File
28 lines
1.6 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# 同一时间最多 N 个 NativeAOT 编译（ILCompiler 加链接每个要占几 GB 内存）。dev-image 里后端和三个 RID 的 Agent 同时开编译时，
|
||
# 内存不够，系统杀掉 MSBuild 的工作进程（MSB4166 Child node exited prematurely），整条流水线失败。
|
||
# 用法：with-aot-slot.sh <命令...> 环境：MOSHIRO_AOT_SLOTS（默认 4：dev-image 里后端加三个 RID 的 Agent 正好各占一个）、MOSHIRO_AOT_SLOT_DIR、MOSHIRO_AOT_SLOT_WAIT（等空位的秒数，默认 3000）
|
||
# flock --close：命令自己拿不到锁的文件描述符，锁只由 flock 进程持有，所以命令遗留的后台进程（比如 MSBuild 复用的节点）不会让空位一直占着。
|
||
set -euo pipefail
|
||
[[ $# -gt 0 ]] || { echo "usage: with-aot-slot.sh <command> [args...]" >&2; exit 2; }
|
||
slots="${MOSHIRO_AOT_SLOTS:-4}"
|
||
dir="${MOSHIRO_AOT_SLOT_DIR:-/home/justme/woodpecker/cache/aot-slots}"
|
||
wait_seconds="${MOSHIRO_AOT_SLOT_WAIT:-3000}"
|
||
mkdir -p "$dir"
|
||
deadline=$((SECONDS + wait_seconds))
|
||
announced=false
|
||
while :; do
|
||
for ((slot = 0; slot < slots; slot++)); do
|
||
status=0
|
||
flock --close --nonblock --conflict-exit-code 200 "$dir/slot-$slot" "$@" || status=$?
|
||
# 200 = 这个空位被占着，换下一个；其他返回码是命令自己的结果
|
||
[[ "$status" != 200 ]] && exit "$status"
|
||
done
|
||
if ((SECONDS >= deadline)); then
|
||
echo "no NativeAOT slot became free within ${wait_seconds}s" >&2
|
||
exit 1
|
||
fi
|
||
[[ "$announced" == true ]] || { echo "waiting for a free NativeAOT slot ($slots in use at most)" >&2; announced=true; }
|
||
sleep 3
|
||
done
|