是否有一个泛型函数,它接受数据结构并返回其中的所有int?

时间:2015-07-03 19:07:59

标签: haskell generic-programming

我认为类型签名看起来像f :: a -> [Int] 输入数据看起来像data NamedPoint = NamedPoint String Int Int

data Person = Name Int Int Int

在REPL中使用它看起来像这样:

>> let namedPoint = NamedPoint "hey" 1 2
>> let tommy = Person "Tommy" 1 2 3
>> f namedPoint
>> [1,2] 
>> f Tommy
>> [1,2,3]

我认为这对于记录的替代方法很有用,因为当你懒得为具有大量参数的数据编写getter时。

2 个答案:

答案 0 :(得分:10)

Data类能够做到这一点。我发现最简单的方法是使用template包中的lens遍历。这实际上允许您使用Data实例设置或获取任何内容。在ghci:

> import Data.Data
> import Control.Lens
> import Data.Data.Lens

> -- Data is a derivable class
> :set -XDeriveDataTypeable
> data NamedPoint = NamedPoint String Int Int deriving (Data, Show)
> data Person = Name String Int Int Int deriving (Data, Show)
> let namedPoint = NamedPoint "hey" 1 2
> let tommy = Name "Tommy" 1 2 3

> let ints = toListOf template :: Data a => a -> [Int]
> ints namedPoint
[1,2]
> ints tommy
[1,2,3]

因为template是遍历,您还可以映射值(但您可能需要指定类型):

> namedPoint & template *~ (10 :: Int)
NamedPoint "hey" 10 20
> import Data.Char
> tommy & template %~ toUpper
Name "TOMMY" 1 2 3

答案 1 :(得分:9)

使用您描述的类型签名功能无法做到这一点。考虑一下f :: a -> [Int]的含义:f应该是一个带有任何可能类型值的函数,并返回Int的列表。如何定义这样的功能?唯一可能的定义是它忽略参数并返回一个常量值,如

f :: a -> [Int]
f _ = [0]

如果您知道a将会是什么,为什么不使用该类型?像这样:

f :: NamedPoint -> [Int]
f (NamedPoint _ a b) = [a, b]

如果你想要一些"一般"函数从数据类型返回所有​​Int,一个选项是定义类型类

class IntContainer a where
    f :: a -> [Int]

然后为您感兴趣的数据类型定义实例

instance IntContainer NamedPoint where
    f (NamedPoint _ a b) = [a, b]