使用镜头读取多个字段

时间:2014-11-03 20:19:26

标签: haskell lens

给出类型

data Prisoner = P { _name   :: String
                  , _rank   :: Int
                  , _cereal :: Cereal }

data Cereal = C { _number             :: Int
                , _percentDailyValue  :: Map String Float
                , _mascot             :: String }

我可以通过模式匹配来提取某人的姓名,等级和谷物数量:

getNameRankAndCerealNumber_0 :: Prisoner -> (String, Int, Int)
getNameRankAndCerealNumber_0 (P { _name=name
                                , _rank=rank
                                , _cereal = C { _number=cerealNumber }}
                             ) = (name, rank, cerealNumber)

或者,我可以使用镜头分别提取每个部分

makeLenses ''Cereal
makeLenses ''Prisoner

getNameRankAndCerealNumber_1 :: Prisoner -> (String, Int, Int)
getNameRankAndCerealNumber_1 p = (p ^. name, p ^. rank, p ^. cereal.number)

有没有办法在数据结构的单次遍历中同时提取所有三个?

某种方法可以合并Getter s,Getter s a -> Getter s b -> Getter s (a,b)

2 个答案:

答案 0 :(得分:22)

我们可以使用Control.Lens.ReifiedApplicative新类型的ReifiedGetter个实例:

runGetter $ (,) <$> Getter number <*> Getter mascot

一般来说,Control.Lens.Reified中的新类型为getter和folds提供了许多非常有用的实例。

注意#1 :请注意,我们将镜片组合为吸气剂,并获得吸气剂作为回报。你不能以这种方式获得复合镜头,因为如果它们的“焦点”重叠会有问题。在这种情况下,什么可能是正确的setter行为?

注意#2 alongside功能可让您将两个镜头组合在一起,获得可在产品的两半上工作的真正镜头。这与前一种情况不同,因为我们可以确定镜头不重叠。当你的类型是元组或者与元组具有同构时,alongside会派上用场。

答案 1 :(得分:5)

充实danidiaz's answer above,我能够使用Getter Prisoner (String, Int, Int)构建ReifiedGetter

getNameRankAndCerealNumber_2 :: Prisoner -> (String, Int, Int)
getNameRankAndCerealNumber_2 = p ^. nameRankAndCerealNumber_2

nameRankAndCerealNumber_2 :: Getter Prisoner (String, Int, Int)
nameRankAndCerealNumber_2 = runGetter ((,,) <$> Getter name <*> Getter rank <*> Getter (cereal.number))

Lens' Prisoner (String, Int, Int)使用alongside,但我必须在Iso'Prisoner之间以及HList [String, Int, Int]之间手动构建HList [a,b,c] (a,b,c)

getNameRankAndCerealNumber_3 :: Prisoner -> (String, Int, Int)
getNameRankAndCerealNumber_3 p = p ^. nameRankAndCerealNumber_3

setNameRankAndCerealNumber_3 :: (String, Int, Int) -> Prisoner -> Prisoner
setNameRankAndCerealNumber_3 t p = p & nameRankAndCerealNumber_3 .~ t

nameRankAndCerealNumber_3 :: Lens' Prisoner (String, Int, Int)
nameRankAndCerealNumber_3 = hlist . alongside id (alongside id number) . triple
  where triple :: Iso' (a,(b,c)) (a,b,c)
        triple = dimap (\(a,(b,c)) -> (a,b,c)) (fmap $ \(a,b,c) -> (a,(b,c)))
        hlist :: Iso' Prisoner (String, (Int, Cereal))
        hlist = dimap (\(P n r c) -> (n,(r,c))) (fmap $ \(n,(r,c)) -> P n r c)

可能有一种更简单的方法,但这是另一个问题。