Question

我正在尝试学习一些模板Haskell。作为练习，我编写了一个可以生成isLeft和isRight（受this question启发）等功能的函数。这是我的谦卑尝试：

isA connam = do
    ConE nam <- connam
    nn <- newName "p"
    lamE [varP nn] $ caseE (varE nn) [
                       match (conP nam [wildP]) ( normalB [| True |] ) [],
                       match wildP ( normalB [| False |] ) [] 
                     ]

问题是它只适用于单参数构造函数。罪魁祸首是conP nam [wildP]模式。理想情况下，它应该看起来像conP nam (replicate (numArgs nam) wildP)，其中numArgs是一个函数，返回构造函数的参数个数。但是我该如何编写这样的功能呢？我想我需要访问相关的数据声明，但我不知道如何。

关于这个功能here还有另一个问题。

Answer 1

虽然您可以使用reify并检查类型以确定数据构造函数的arity，但使用记录模式生成与arity无关的代码要容易得多：

isFoo :: Bar -> Bool
isFoo p = case p of
    (Foo {}) -> True     -- Valid no matter what the arity of Foo is
    _        -> False

可以通过在代码中将conP替换为recP来完成此操作。

isA connam = do
    ConE nam <- connam
    nn <- newName "p"
    lamE [varP nn] $ caseE (varE nn) [
                       match (recP nam []) ( normalB [| True |] ) [],
                       match wildP ( normalB [| False |] ) [] 
                     ]

如何在Template Haskell中检查引用的数据构造函数名称？

1 个答案: