124 lines
4.8 KiB
Python
124 lines
4.8 KiB
Python
|
"""
|
|||
|
|
Nuitka 打包资源档位 / 链接器注入 测试（不触发真实编译）
|
|||
|
|
|
|||
|
|
覆盖 build.py 里 nuitka 打包的三处防坑逻辑：
|
|||
|
|
1. 链接器：装了 ld.lld 时自动经 LDFLAGS 注入 -fuse-ld=lld（默认 GNU ld 链接大工程
|
|||
|
|
峰值数 GB，小内存机器/WSL 会被 OOM 杀掉："clang: unable to execute command: Killed"）；
|
|||
|
|
已有 LDFLAGS 时不覆盖（用户在 CHONGMING_NUITKA_PGO 之外自行指定链接器优先）
|
|||
|
|
2. 内存档位：可用内存 ≥ NUITKA_LTO_MIN_GB 完整优化（LTO）；低于则去 LTO（LTO 链接
|
|||
|
|
峰值 3 GB+，实测 ld/lld 都被 OOM 杀过）；更低（或 CHONGMING_NUITKA_LOW_MEM=1）
|
|||
|
|
进一步 --low-memory + 降并发；CHONGMING_NUITKA_FULL=1 强制完整优化
|
|||
|
|
3. PGO 缺省关闭：--pgo-c 与 standalone/onefile 不兼容且 profiling 运行会等程序退出
|
|||
|
|
（常驻服务卡死），只能由 CHONGMING_NUITKA_PGO=1 显式打开——
|
|||
|
|
这里用源码断言兜住"默认关"这条（防止回潮）
|
|||
|
|
|
|||
|
|
运行: uv run python test/test_nuitka_pack_flags.py
|
|||
|
|
"""
|
|||
|
|
import io
|
|||
|
|
import contextlib
|
|||
|
|
import os
|
|||
|
|
import shutil
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
ROOT = Path(__file__).parent.parent
|
|||
|
|
sys.path.insert(0, str(ROOT / "utils" / "cli" / "src"))
|
|||
|
|
|
|||
|
|
from chongming_cli import build as b # noqa: E402
|
|||
|
|
|
|||
|
|
failures = []
|
|||
|
|
|
|||
|
|
|
|||
|
|
def check(name, cond, detail=""):
|
|||
|
|
status = "PASS" if cond else "FAIL"
|
|||
|
|
print(f"[{status}] {name}" + (f" — {detail}" if detail else ""))
|
|||
|
|
if not cond:
|
|||
|
|
failures.append(name)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def quiet_note():
|
|||
|
|
"""调 _nuitka_env_and_mem_note 时屏蔽它打印的降档提示。"""
|
|||
|
|
with contextlib.redirect_stdout(io.StringIO()):
|
|||
|
|
return b._nuitka_env_and_mem_note()
|
|||
|
|
|
|||
|
|
|
|||
|
|
saved = {k: os.environ.pop(k, None) for k in
|
|||
|
|
(b.NUITKA_LOW_MEM_ENV, b.NUITKA_FULL_ENV, b.NUITKA_PGO_ENV)}
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# ========== 1. 链接器注入 ==========
|
|||
|
|
saved_ldflags = os.environ.pop("LDFLAGS", None)
|
|||
|
|
env, note, profile = quiet_note()
|
|||
|
|
if sys.platform.startswith("linux") and shutil.which("ld.lld"):
|
|||
|
|
check("装了 lld 自动注入 -fuse-ld=lld", env.get("LDFLAGS", "").endswith("-fuse-ld=lld"), env.get("LDFLAGS"))
|
|||
|
|
check("成功信息回显链接器", "lld" in note, note)
|
|||
|
|
else:
|
|||
|
|
check("无 lld 时不注入（本机无 lld，跳过）", "-fuse-ld" not in (env.get("LDFLAGS") or ""))
|
|||
|
|
|
|||
|
|
os.environ["LDFLAGS"] = "-Wl,--as-needed"
|
|||
|
|
env, _, _ = quiet_note()
|
|||
|
|
check("已有 LDFLAGS 时追加而不覆盖", env["LDFLAGS"].startswith("-Wl,--as-needed"), env["LDFLAGS"])
|
|||
|
|
if saved_ldflags is None:
|
|||
|
|
os.environ.pop("LDFLAGS", None)
|
|||
|
|
else:
|
|||
|
|
os.environ["LDFLAGS"] = saved_ldflags
|
|||
|
|
|
|||
|
|
# ========== 2. 内存档位 ==========
|
|||
|
|
mem = b._mem_available_gb()
|
|||
|
|
check("_mem_available_gb 可读（Linux）", mem > 0 or not sys.platform.startswith("linux"), f"{mem:.1f} GB")
|
|||
|
|
|
|||
|
|
env, note, profile = quiet_note()
|
|||
|
|
if mem >= 0:
|
|||
|
|
expect = (
|
|||
|
|
"squeeze" if mem < b.NUITKA_SQUEEZE_GB
|
|||
|
|
else "no-lto" if mem < b.NUITKA_LTO_MIN_GB
|
|||
|
|
else "full"
|
|||
|
|
)
|
|||
|
|
check(
|
|||
|
|
"默认档位与内存阈值一致",
|
|||
|
|
profile == expect,
|
|||
|
|
f"{mem:.1f} GB → {profile}（阈值 {b.NUITKA_SQUEEZE_GB}/{b.NUITKA_LTO_MIN_GB} GB）",
|
|||
|
|
)
|
|||
|
|
check(
|
|||
|
|
"非 full 档在提示里说明原因",
|
|||
|
|
(profile == "full" and "LTO" not in note) or (
|
|||
|
|
profile == "no-lto" and "未启用 LTO" in note
|
|||
|
|
) or (profile == "squeeze" and "--low-memory" in note),
|
|||
|
|
note,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
os.environ[b.NUITKA_LOW_MEM_ENV] = "1"
|
|||
|
|
env, note, profile = quiet_note()
|
|||
|
|
check(f"{b.NUITKA_LOW_MEM_ENV}=1 强制 squeeze 档", profile == "squeeze" and "--low-memory" in note, note)
|
|||
|
|
|
|||
|
|
os.environ.pop(b.NUITKA_LOW_MEM_ENV)
|
|||
|
|
os.environ[b.NUITKA_FULL_ENV] = "1"
|
|||
|
|
env, note, profile = quiet_note()
|
|||
|
|
check(f"{b.NUITKA_FULL_ENV}=1 强制完整优化（覆盖内存自动判断）", profile == "full" and "--low-memory" not in note, note)
|
|||
|
|
os.environ.pop(b.NUITKA_FULL_ENV)
|
|||
|
|
|
|||
|
|
# ========== 3. PGO 缺省关闭（源码兜底，防回潮） ==========
|
|||
|
|
src = (ROOT / "utils" / "cli" / "src" / "chongming_cli" / "build.py").read_text(encoding="utf-8")
|
|||
|
|
check(
|
|||
|
|
"PGO 参数只在 NUITKA_PGO_ENV 开关下出现",
|
|||
|
|
f'os.environ.get(NUITKA_PGO_ENV) == "1"' in src and b.NUITKA_PGO_ENV == "CHONGMING_NUITKA_PGO",
|
|||
|
|
)
|
|||
|
|
check(
|
|||
|
|
"Nuitka 缺 patchelf 时提前失败（链接后才检查，白跑一整轮）",
|
|||
|
|
"patchelf" in src and "缺少 patchelf" in src,
|
|||
|
|
)
|
|||
|
|
check(
|
|||
|
|
"Nuitka 失败会抛错而不是静默返回（构建要非零退出）",
|
|||
|
|
'raise RuntimeError("Nuitka 编译失败' in src,
|
|||
|
|
)
|
|||
|
|
finally:
|
|||
|
|
for k, v in saved.items():
|
|||
|
|
if v is not None:
|
|||
|
|
os.environ[k] = v
|
|||
|
|
|
|||
|
|
print("\n" + ("=" * 40))
|
|||
|
|
if failures:
|
|||
|
|
print(f"❌ {len(failures)} 项未通过: {failures}")
|
|||
|
|
sys.exit(1)
|
|||
|
|
print("✅ 全部通过")
|