如果它符合某些条件,则更改列表元素,否则使用Data.Lens添加新元素

Question

我有一个记录列表,需要一个函数,在列表中搜索具有给定名称的记录并修改此记录的值,或者如果没有记录匹配,则将新记录附加到结果列表中.到目前为止,这是我的代码:

import Control.Lens
import Control.Applicative ((<$>), pure)
import Data.List (any)

data SomeRec = SomeRec { _name :: String, _val :: Int }
$(makeLenses ''SomeRec)

_find :: (a -> Bool) -> Simple Traversal [a] a
_find _ _ [] = pure []
_find pred f (a:as) = if pred a
                        then (: as) <$> f a
                        else (a:) <$> (_find pred f as)

changeOrCreate :: [SomeRec] -> String -> (Int -> Int) -> [SomeRec]
changeOrCreate recs nameToSearch valModifier = 
  if (any (\r -> r^.name == nameToSearch) recs)
    then over (_find (\r -> r^.name == nameToSearch)) (over val valModifier) recs
    else recs ++ [SomeRec nameToSearch (valModifier 0)]

它工作正常,但我想知道是否有更直接的方式来编写这个Data.Lens(没有if-construct)？此外,我是否必须编写_find函数或库中是否有相应的东西？

更新:以下是实验来源的要点:https://gist.github.com/SKoschnicke/5795863

Answer 1

怎么样：

changeOrCreate :: String -> (Int -> Int) -> [SomeRec] -> [SomeRec]
changeOrCreate nameToSearch valModifier = 
  pos . val %~ valModifier
  & outside (filtered (not . has pos)) %~ (. newRec)
  where
    pos :: Traversal' [SomeRec] SomeRec
    pos = taking 1 (traversed . filtered (anyOf name (== nameToSearch)))
    newRec = (SomeRec nameToSearch 0 :)