如何采用Writer Monad更具可读性地重写纯函数？

DevOps 小满

2026-5-19

嘿，这个问题我太有共鸣了——把干净利落的纯递归函数改成带Writer Monad的版本，一不小心就会把代码弄得乱糟糟的！别担心，咱们可以用几个实用技巧，在保留日志功能的同时，让代码可读性尽量贴近原纯函数。

核心思路：分离关注点，复刻原函数结构

Writer Monad的本质是把“核心计算”和“日志收集”两个逻辑分开，所以咱们要尽量避免把tell和递归逻辑揉成一团，而是复刻原纯函数的结构，只在合适的地方插入日志操作。

下面举几个具体技巧，结合例子来说明：

1. 把日志逻辑抽成辅助函数

不要在递归里直接写tell ["xxx"]，而是单独定义一个记录日志的小函数，让递归代码更清爽：

import Control.Monad.Writer

-- 专门的日志辅助函数，把日志格式和Writer操作封装起来
logStep :: String -> Writer [String] ()
logStep msg = tell [msg]

2. 严格复刻原递归函数的“形状”

假设你的原纯递归函数是这样的（举个简单例子）：

-- 原纯函数：递归处理列表，每个元素翻倍
doubleList :: [Int] -> [Int]
doubleList [] = []
doubleList (x:xs) = x*2 : doubleList xs

那咱们写monadic版本时，完全可以照搬这个结构，只在关键节点插入日志：

doubleListM :: [Int] -> Writer [String] [Int]
doubleListM [] = do
  logStep "处理空列表，返回空结果"
  return []
doubleListM (x:xs) = do
  logStep $ "处理元素：" ++ show x ++ "，将其翻倍"
  -- 保持原递归调用的逻辑，只是改成monadic的绑定
  rest <- doubleListM xs
  return (x*2 : rest)

你看，这个版本和原函数的结构几乎一模一样，只是多了日志步骤，可读性瞬间上来了。

3. 列表推导转monadic时，用`mapM`/`forM`保持简洁

如果原函数用了列表推导，比如：

-- 原纯函数：列表推导实现的递归阶乘（示例）
factorial :: Int -> Int
factorial n = product [1..n]

转成Writer版本时，可以把单个元素的处理抽成纯函数+日志，再用mapM批量处理：

-- 单个数字的处理（带日志）
multiplyWithLog :: Int -> Int -> Writer [String] Int
multiplyWithLog acc num = do
  logStep $ "当前累积值：" ++ show acc ++ "，乘以" ++ show num
  return (acc * num)

-- monadic版本的阶乘，用foldM（递归的monadic版）实现
factorialM :: Int -> Writer [String] Int
factorialM n = foldM multiplyWithLog 1 [1..n]

这样既保留了列表推导的批量处理逻辑，又清晰地加入了日志。

4. 用自定义辅助函数减少重复代码

如果经常需要“记录日志后返回某个值”，可以写个小工具函数：

logAndReturn :: String -> a -> Writer [String] a
logAndReturn msg x = logStep msg >> return x

这样在递归里就能简化成一行：

doubleListM [] = logAndReturn "处理空列表，返回空结果" []

关键总结

只要记住让核心递归逻辑的结构和原纯函数保持一致，把日志操作当成“附加的小步骤”而非核心逻辑的一部分，就能写出既带日志功能、又和原代码一样易读的Writer Monad版本。

内容的提问来源于stack exchange，提问作者Retired Writing Code for Fun