基于TMVar的唯一时间戳ID生成器问题：并发Haskell故障排查

你的代码之所以会生成重复的时间戳，核心问题出在时间戳的获取时机和STM事务内的逻辑判断上，我来一步步拆解：

问题根源 #

你在next函数里先在IO上下文获取时间戳t，再进入STM事务处理TMVar。这会导致多个并发线程可能在进入STM前就拿到了同一个t，而当后续线程进入STM时，TMVar里存储的上一个值v可能已经比t大了——这时候你的逻辑会直接返回t，而这个t很可能已经被之前的线程返回过，从而造成重复。

举个具体的执行流程例子：

• 初始TMVar值为0，当前时间戳t=100
• 线程1进入STM：取出v=0，因t≠v返回100，并把v更新为100
• 线程2进入STM：取出v=100，因t==v返回101，更新v为101
• 线程3拿到的t还是100（在STM外获取的），进入STM取出v=101，因t≠v直接返回100——这个值已经被线程1用过，重复就产生了！

修复方案 #

我们需要保证生成的每个ID都严格递增且不小于当前时间，所以修改STM内的逻辑：取当前时间t和v+1的最大值作为新ID。这样既保证ID尽量贴近实际时间，又能避免重复。

修正后的完整代码：

import Control.Concurrent.Async (mapConcurrently)
import Data.Time.Clock.POSIX (getPOSIXTime)
import Control.Concurrent.STM (TMVar, atomically, takeTMVar, putTMVar, newTMVarIO)
import Data.List (nub)

-- 简化getTime的实现，*1可以省略
getTime :: Integral b => IO b
getTime = round <$> getPOSIXTime

next :: Integral b => TMVar b -> IO b
next s = do
  t <- getTime
  atomically $ do
    v <- takeTMVar s
    -- 关键修改：取当前时间和上一个ID+1的最大值，确保ID严格递增
    let t' = max t (v + 1)
    putTMVar s t'
    return t'

main = do
  next' <- next <$> newTMVarIO 0
  res <- mapConcurrently id [next' | _ <- [1 .. 100]]
  print $ length $ nub res -- 现在会输出100，符合预期

为什么这个逻辑能解决问题？ #

• 如果当前时间t比上一个IDv大：直接用t作为新ID，保证ID贴近实际时间
• 如果当前时间t小于等于v：说明之前已经有线程在同一时间（或更早时间）生成过ID，我们就在v的基础上加1，确保新ID是严格递增的，不会重复

这样无论多少线程并发调用next，生成的ID都是唯一且连续的，完全符合你的需求。

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