构建Haskell版C#编译器时State Monad状态更新异常求助

你遇到的问题核心在于：执行sequence block后状态没有发生预期的更新，导致newEnv和origEnv完全一致。虽然你提到插入方法有效，但大概率是代码块中的EnvState动作没有正确修改State monad的状态。

为什么当前代码不生效？ #

在State monad的do记法中，每一步操作是连贯的状态传递：

1. origEnv <- get 获取初始状态
2. xs <- sequence block 执行所有块内的状态动作，理论上应该逐步更新状态
3. newEnv <- get 获取执行后的状态

如果newEnv和origEnv相同，说明block中的每个EnvState动作没有实际修改状态——比如你可能只计算了新环境，但没有调用put将其写入State中。

举个典型的错误写法例子：

declareVar :: String -> Int -> EnvState
declareVar name val = do
  env <- get
  let newEnv = M.insert name val env
  -- 这里漏掉了关键的 put newEnv！
  return (0, [])

这种情况下，虽然计算出了新环境，但从未更新State，所以执行后状态还是原来的样子。

正确的实现方向 #

首先要确保块内的每个状态动作都正确修改状态。比如变量声明的正确写法应该是：

declareVar :: String -> Int -> EnvState
declareVar name val = do
  env <- get
  let newEnv = M.insert name val env
  put newEnv  -- 关键步骤：将新环境写入State
  return (M.size newEnv, [])  -- 按你的需求调整返回值

你的fStatBlock逻辑本身是合理的：先保存原状态，执行块内动作，获取更新后的状态大小，再恢复原状态。只要块里的动作正确修改状态，这个逻辑就能正常工作。

排查步骤 #

1. 单独测试一个简单的EnvState动作，验证状态修改是否生效：

testAction :: EnvState
testAction = do
  orig <- get
  put (M.insert "x" 0 orig)
  new <- get
  return (M.size new, [])

-- 运行测试：runState testAction M.empty
-- 预期输出：((1, []), fromList [("x",0)])

如果这个测试正常，说明基础状态修改逻辑没问题，问题出在block中的动作组合。
2. 检查block中的每个动作是否都调用了put，确保状态确实被更新。

修正后的示例代码 #

假设变量声明动作正确，fStatBlock的逻辑可以正常工作：

import qualified Data.Map as M
import Control.Monad.State

type Env = M.Map String Int
type EnvState = State Env (Int, [String])  -- 用[String]代替Code简化示例

declareVar :: String -> Int -> EnvState
declareVar name val = do
  env <- get
  let newEnv = M.insert name val env
  put newEnv
  return (M.size newEnv, [name ++ " declared"])

fStatBlock :: [EnvState] -> EnvState
fStatBlock block = do
  origEnv <- get
  xs <- sequence block
  newEnv <- get
  put origEnv
  return (M.size newEnv, concatMap snd xs)

-- 测试运行
main :: IO ()
main = do
  let initialEnv = M.fromList [("y", 1)]
      block = [declareVar "x" 0, declareVar "z" 2]
      ((size, code), finalEnv) = runState (fStatBlock block) initialEnv
  putStrLn $ "New env size: " ++ show size  -- 预期输出3
  putStrLn $ "Generated code: " ++ show code  -- 预期输出["x declared","z declared"]
  putStrLn $ "Final env: " ++ show finalEnv  -- 预期输出初始的fromList [("y",1)]

这个示例中，fStatBlock执行后会正确返回块内变量声明后的环境大小（3），同时最终状态恢复为初始环境（只包含"y"）。

内容的提问来源于stack exchange，提问作者R. Bosman