动态获取某个类型的数据构造函数列表

Question

我有这样的数据类型：

data ABCS = A Int | B Int | ... | Z Int deriving (Data, Typeable)

在测试中，我想动态提取所有构造函数，从每个构造函数中创建一个实例，然后运行测试。

我一直在查看 Data.Typeable 和 Data.Data，但我还没有看到/理解如何仅从类型 (ABC) 开始执行此操作。

非常感谢您的帮助。

Answer 1

如果您可以使用Data.Data，它适用于这个用例，但由于Int 参数，它有点笨拙。

{-# LANGUAGE ScopedTypeVariables #-}
import Data.Data
import Data.Typeable

allCtors :: forall a. Data a => [Int -> a]
allCtors = map observeCtor $ dataTypeConstrs $ dataTypeOf (undefined :: a)
  where
    observeCtor :: Constr -> Int -> a
    observeCtor c i = fromJust $ fromConstrM (cast i) c

然后我们有例如

λ data ABC = A Int | B Int | C Int deriving (Show, Data, Typeable)
data ABC = A Int | B Int | C Int
λ map ($ 2) allCtors :: [ABC]
[A 2,B 2,C 2]

如果您不想使用Data.Data，您可以使用GHC.Generics 和-XDefaultSignatures 来做到这一点

---

FWIW，如果您可以重构 ABC 以便 A、B、C 标记是它们自己的类型，那么您就不必处理任何这些...

data ABCTagged = ABCTagged ABC Int deriving Show

data ABC = A | B | C deriving (Show, Eq, Ord, Enum. Bounded)

...然后只需使用enumFrom minBound :: [ABC] 即可获取整个列表。简单的！不过，不确定这对您来说是否可行。