Haskell预定义visitor模式迭代器内提取及累积结果的实现方法

这类绑定外部C库的IO迭代接口，最直接的实现方式是使用IORef（IO上下文下的可变引用）存储遍历的中间状态，不需要额外嵌套State Monad，完全适配现有接口的类型约束。

示例1：统计数据库总条目数 #

import Data.IORef
import Database.KyotoCabinet

countTotalEntries :: IO Int
countTotalEntries = do
  db <- openTree "testdatabase/mydb.kct" defaultLoggingOptions (Writer [] [])
  -- 初始化计数可变引用，初始值为0
  countRef <- newIORef 0
  let visitor _ _ = do
        -- 严格修改引用，避免惰性导致的内存溢出
        modifyIORef' countRef (+1)
        return (Left NoOperation)
  iterate db visitor False
  close db
  -- 遍历结束后读取引用中的最终结果
  readIORef countRef

示例2：收集所有以'a'开头的键 #

import Data.IORef
import Data.Char (ord)
import qualified Data.ByteString as B
import Database.KyotoCabinet

collectAKeys :: IO [B.ByteString]
collectAKeys = do
  db <- openTree "testdatabase/mydb.kct" defaultLoggingOptions (Writer [] [])
  -- 初始化存储键列表的可变引用
  keysRef <- newIORef []
  let visitor k _ = do
        if B.head k == fromIntegral (ord 'a')
          -- 匹配成功就把键插入列表头部
          then modifyIORef' keysRef (k :)
          else return ()
        return (Left NoOperation)
  iterate db visitor False
  close db
  -- 头插会导致顺序反转，最终读取后反转回原顺序
  reverse <$> readIORef keysRef

实现说明 #

• 库提供的iterate要求visitor必须返回IO动作，IORef的读写修改本身就是IO操作，完全符合类型约束
• 遍历过程中每个条目触发visitor调用时，都可以对IORef存储的状态做修改，状态会在多次visitor调用之间自动留存
• 遍历量大的场景必须使用严格版修改函数modifyIORef'，避免惰性求值累积大量未计算的thunk导致内存泄漏

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