Haskell State monad：在序列后获取更新状态

Question

作为编译器课程的一部分，我正在使用 haskell 构建 C# 编译器。 我正在使用状态单子，问题在于块的代码。我正在使用状态来包装声明的变量的环境。解析块时，我想扩展此状态（因为块内的声明），但之后返回原始块（因为声明不会超出块）。但是，我想先知道新更新状态的大小。所以我有下面的代码：

type EnvState = State Env (Int, Code)
type Env = M.Map String Int

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

环境是 Data.Map 类型。

我的问题是 newEnv 不是序列后更新的 env，而是等于 origEnv。因此，返回的大小 100% 取决于原始 env 的大小，无论在序列中插入什么都不会改变。 （我已经测试了插入方法并且它有效）。

这是由于懒惰的评估吗？奇怪的执行顺序？或者这应该给新的，更新的环境，我在其他地方做错了吗？感谢您的帮助。

Answer 1

正如 Justin L. 所暗示的，问题可能是 block 没有更改 Env。这是您的示例代码，未更改，添加了一些测试代码以制作完整的程序。函数setValue 产生一个EnvState，它改变了Env。在对fStatBlock 的调用中使用setValue确实会导致比基于原始env 的大小大一。

完整程序

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

type Code = [Int]


-- begin original code ------

type EnvState = State Env (Int, Code)
type Env = M.Map String Int

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

-- end original code -------


setValue :: String -> Int -> EnvState
setValue name value = state (\env -> ((0,[]), insert name value env))

main = do
    let env = fromList [("x", 5), ("y", 10)]
        fsb = fStatBlock [setValue "a" 15] 
    print $ fst $ fst $ runState fsb env

结果

$ ./main 
3