使用TMVar创建基于时间的唯一ID生成器

时间:2018-01-05 08:18:09

标签: haskell stm

我正在制作一个天真的基于时间的唯一ID生成器来学习并发Haskell中的一些概念。

next应返回唯一的连续POSIX时间,但此时间戳不必与我下次调用的确切时刻相匹配。

此代码显然不会生成唯一的时间戳,我不明白为什么:

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

getTime :: (Integral b, RealFrac a) => a -> IO b
getTime precision =
  round . (* precision) . fromRational . toRational <$> getPOSIXTime

next :: Integral b => TMVar b -> IO b
next s = do
  t <- getTime 1 -- one second precision
  atomically $ do
    v <- takeTMVar s
    let t' = if t == v then t + 1 else t
    putTMVar s t'
    return t'

main = do
  next' <- next <$> newTMVarIO 0
  res <- mapConcurrently id [next' | _ <- [1 .. 100]]
  print $ length $ nub res -- expected 100

1 个答案:

答案 0 :(得分:3)

假设我们在同一秒内呼叫next几次,每次getTime 1再次呼叫42。

第一次,t为42,v为0,t / = v,因此t'为42。

第二次,t是42,v是42,t == v所以t'是43,我们存储43。

第三次,t为42,v为43,t / = v所以t'为42,我们存储42。

第四次,t是42,v是42,t == v所以t'是43,我们存储43。

看到问题?