如何通用获取值的已占用内存？含NF求值内存获取需求

在Haskell生态里，确实有工具能实现你想要的这类内存测量需求，下面是具体的实现方式：

核心实现：借助GHC官方堆库 #

GHC提供的ghc-heap库是直接操作堆结构的底层工具，能满足你对求值状态和共享结构的考量：

• 当前状态内存大小：用getClosureData获取值的堆结构，再通过heapSize计算其占用内存——这个过程不会强制求值，只会统计当前已求值部分的大小，对应你说的getSize。
• 范式状态内存大小：先结合Control.DeepSeq里的force把值强制求值到范式（NF），再用同样的堆结构分析方法计算大小，对应你说的getNfSize。

而且GHC的堆遍历会自动识别共享结构：如果同一个值被多次引用，只会计算一次内存，完全符合你提到的“共享影响结果”的要求。

这里是具体的代码示例：

import GHC.Heap
import Control.DeepSeq

-- 计算值当前状态的内存占用（字节）
getSize :: a -> IO Int
getSize x = heapSize <$> getClosureData x

-- 先求值到范式，再计算内存占用
getNfSize :: NFData a => a -> IO Int
getNfSize x = do
  let !x' = force x  -- 强制求值到NF，感叹号确保严格求值
  heapSize <$> getClosureData x'

注意这些函数需要放在IO里，因为访问GHC运行时堆属于IO操作。

更易用的第三方工具 #

如果你不想直接操作底层堆结构，可以用weigh库——它封装了ghc-heap的逻辑，提供了更友好的API，能轻松对比不同求值状态、共享结构下的内存占用，还会输出详细的统计信息。

几点注意 #

• 内存计算结果会受GHC版本、编译优化级别的影响，不同环境下可能有细微差异。
• 对于未求值的thunk（惰性表达式），getSize只会统计thunk本身的大小，而不是它最终求值后的值的大小。

内容的提问来源于stack exchange，提问作者Cigarette Smoking Man