haskell - 是否有一个库可以检测设置了哪些 RTS 标志？

Question

当我运行 ./MyProgram +RTS -N4 时，我希望我的程序自动检测到它正在 4 个核心上运行，并创建 4 个 Spark 来处理数据集。现在，每次我想更改此选项时，我都要重新编译，这很烦人。

Answer 1

回答你原来的问题:运行时选项是 GHC 特定的，GHC 提供了一个模块，可以让你查看完整的参数列表(包括 +RTS 和 -RTS) 传递给程序:GHC.Environment .

这里有一个小演示:

import GHC.Environment (getFullArgs)
import System.Environment (getArgs)

main = do
    getFullArgs >>= mapM putStrLn
    putStrLn "=="
    getArgs >>= mapM putStrLn
    return ()

将程序编译为 ghc App.hs --make -rtsopts -threaded 执行时将产生以下结果:

./App hello +RTS -N4 -RTS ook ooks
hello
+RTS
-N4
-RTS
ook
ooks
==
hello
ook
ooks