Nim闭包子线程运行时无法更新捕获变量的原因探究

我编写了一个可捕获并更新外部变量的Nim闭包，在定义该变量的线程中直接调用时，能正常更新外部变量；但将该闭包传入子线程调用后，却无法按预期更新外部变量。想了解闭包传入线程时的运行机制，以及相关官方文档参考。

测试代码 #

# nim c -r --threads:on testClosure.nim
import strutils

template hexDumpAddrOf[T](v: T): string =
  # 输出地址大小与ByteAddress大小
  # echo fmt"size of addr: {sizeof(v.unsafeAddr)}, sizeof ByteAddress: {sizeof(ByteAddress)}"
  var p = cast[array[sizeof(ByteAddress), uint8]](v.unsafeAddr)
  var result = ""
  for x in p:
    result = x.toHex() & result
  result

proc closureThreadProcWrapper(closure: proc() ) =
  closure()

proc testClosureThread() = 
  var thr: Thread[proc()]
  var output = @["first"]  # 被子线程更新

  echo " output的原始地址:     ", hexDumpAddrOf(output) 

  proc localClosure() = 
    # 捕获的output地址在子线程中运行时与原地址不同
    echo " localClosure中output的地址: ", hexDumpAddrOf(output) 
    output.add "anything"

  localClosure()  # 打印与原始地址相同的地址
  echo "直接调用闭包后，外部变量已更新: ", output # 输出 @["first", "anything"]
  createThread(thr, closureThreadProcWrapper, localClosure) # 打印不同的output地址？？？

  thr.joinThread
  echo "子线程中调用闭包后，外部变量未按预期更新: ", output # 输出 @["first", "anything"]？？？

when isMainModule:
  testClosureThread()

运行输出 #

output的原始地址:     00007F63349C8060
 localClosure中output的地址: 00007F63349C8060
直接调用闭包后，外部变量已更新: @["first", "anything"]
 localClosure中output的地址: 00007F63348C9060
子线程中调用闭包后，外部变量未按预期更新: @["first", "anything"]

原因分析 #

• Nim闭包默认采用值捕获机制：当闭包被传递到子线程时，捕获的外部变量会被复制一份到子线程的独立内存空间，子线程内修改的是这份副本，而非原变量。
• 输出中地址不同的现象，正是因为子线程操作的是原变量的拷贝，所以主线程的原变量不会被修改。

解决方法 #

要实现跨线程修改原变量，需要让闭包捕获变量的引用/指针，而非值副本：

方法1：显式传递变量指针 #

修改闭包定义，接收变量的指针参数，调用时传入原变量的地址：

# 修改闭包定义
proc localClosure(outputPtr: ptr seq[string]) = 
  echo " localClosure中output的地址: ", hexDumpAddrOf(outputPtr[]) 
  outputPtr[].add "anything"

# 主线程调用
localClosure(addr output)
# 子线程调用时包装闭包，传入原变量地址
createThread(thr, closureThreadProcWrapper, proc() = localClosure(addr output))

方法2：使用引用类型包装变量 #

将原变量定义为ref类型（引用语义），这样闭包捕获的是引用本身，子线程和主线程操作的是同一个底层对象：

# 修改变量定义为引用类型
var output = newSeq[string](@["first"])

# 闭包无需修改，直接操作即可
proc localClosure() = 
  echo " localClosure中output的地址: ", hexDumpAddrOf(output[]) 
  output.add "anything"

官方文档参考 #

• Nim闭包的默认捕获规则：闭包默认以值语义捕获外部变量，若需共享状态，必须显式使用指针或引用类型。
• 线程内存模型：线程拥有独立的栈空间，跨线程传递闭包时，未显式处理的捕获变量会被拷贝，无法共享修改。

内容的提问来源于stack exchange，提问作者azyx