如何在Haskell中用MVar实现非阻塞入队的线程安全队列？

你的需求核心是入队操作不阻塞，仅清空队列的操作可按需处理。MVar确实不是最适配的原语——它只有“空/满”两种状态，无法在不阻塞的前提下原子完成“空则插入、有则追加”的操作：要么会在MVar满时阻塞（putMVar），要么在MVar空时阻塞（modifyMVar）。

推荐使用**STM（软件事务内存）**中的TVar实现，STM操作天然具备原子性且不会无意义阻塞，完美匹配你的需求：

定义队列类型 #

import Control.Concurrent.STM (TVar, newTVarIO, readTVar, writeTVar, modifyTVar', atomically)
import Data.List (reverse)

newtype Queue a = Queue (TVar [a])

用TVar [a]存储队列元素，元素以逆序存于列表（入队时往头部追加，保证O(1)操作效率），清空时再反转得到正确顺序。

实现核心操作 #

1. 移除队列中所有元素 #

这里实现非阻塞版本（队列为空时直接返回空列表）：

drainQueue :: Queue a -> IO [a]
drainQueue (Queue tvar) = atomically $ do
  xs <- readTVar tvar
  writeTVar tvar []
  return $ reverse xs

如果需要阻塞版本（队列为空时等待直到有元素），可添加retry逻辑：

drainQueueBlocking :: Queue a -> IO [a]
drainQueueBlocking (Queue tvar) = atomically $ do
  xs <- readTVar tvar
  if null xs
    then retry  -- 队列为空时等待，直到有元素入队
    else do
      writeTVar tvar []
      return $ reverse xs

2. 向队列中添加一个元素（完全非阻塞） #

enqueue :: Queue a -> a -> IO ()
enqueue (Queue tvar) x = atomically $ modifyTVar' tvar (x:)

modifyTVar'是原子修改TVar的操作，无论队列是空还是已有元素，都能瞬间完成，不会产生任何阻塞。

初始化队列 #

newQueue :: IO (Queue a)
newQueue = Queue <$> newTVarIO []

为什么不选MVar？ #

MVar是针对“单个值同步”设计的原语，原子操作仅围绕“取走/放入单个值”展开。若尝试用tryTakeMVar+putMVar组合实现非阻塞入队，会出现竞态问题：当两个线程同时在MVar为空时入队，第一个线程putMVar后，第二个线程的putMVar会阻塞，违反入队不阻塞的要求。而STM操作是事务性的，所有修改要么全部成功要么全部回滚，不会出现这类问题。

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