我可以使用模板haskell来定义缺失的函数吗？

Question

我有一种情况需要在不同的机器上编译一些Haskell代码.这些机器中至少有一台有一个相当旧版本的Control.Concurrent.STM,它不知道modifyTVar.我目前的解决方法是从较新版本的软件包中复制modifyTVar的代码.这让我想知道,是否可以使用模板Haskell来检查函数是否已经定义并且只定义它,如果它丢失了.我知道正确的解决方案可能是获得更新的软件包,但这种情况让我很好奇.

Answer 1

似乎有可能如下.首先是一个辅助模块:

{-# LANGUAGE TemplateHaskell #-}

module AddFn where

import Language.Haskell.TH

-- | Add a function if it doesn't exist.
addFn :: String -> Q [Dec] -> Q [Dec]
addFn name decl = do
    r <- lookupValueName name
    case r of
        Just l -> return []
        Nothing -> report False ("adding missing " ++ name) >> decl

并使用它作为

{-# LANGUAGE TemplateHaskell #-}

module Main where

import AddFn
import qualified Data.Traversable as T

$(addFn "mapM"
    [d| mapM :: (Monad m) => (a -> m b) -> [a] -> m [b]
        mapM = T.mapM
    |])

$(addFn "mapM1"
    [d| mapM1 :: (Monad m) => (a -> m b) -> [a] -> m [b]
        mapM1 = T.mapM
    |])

缺点是它正在使用lookupValueName,这只是在TH的最新版本中,因此在处理旧安装时,这可能无济于事.或许可能的解决方案是调用reify给定的名称,并recover在缺少名称时使用来处理这种情况.

更新:使用reify而不是lookupValueName工作的版本:

-- | Add a function if it doesn't exist.
addFn :: String -> Q [Dec] -> Q [Dec]
addFn name decl = recover decl (reify (mkName name) >> return [])