为将集合元素添加到Map [K，Set [V]]的创建状态转换的更好方法

Question

我有Map[K, Set[V]] ，正在使用Scalaz Lenses和State向其中添加元素。

到目前为止，我看到自己反复这样做：

myMapLens.member(key) %= {
    case Some(vals) => Some(vals + newValue)
    case None       => Some(Set(newValue))
}

有没有更好的方法使用Scalaz做到这一点？ 每次将我的值转换为Some(...)似乎都是浪费。

具体来说，有没有一种方法可以组成Scalaz MapLens和SetLens来实现这一目标？

Answer 1

您可以编写一个适配器以“展平”该Option ：

import scalaz._, Scalaz._

def noneZero[A: Monoid]: Lens[Option[A], A] = Lens.lensg(_ => Some(_), _.orZero)

这比您需要的通用一些，但是对于您的用例具有相同的行为：

val myMapLens = Lens.lensId[Map[String, Set[Int]]]

val myLens = myMapLens.member("foo").andThen(noneZero).contains(1)

您当然可以在SetLens上使用其他任何方法- contains一个很好的演示：

scala> myLens.get(Map("foo" -> Set(1)))
res0: Boolean = true

scala> myLens.get(Map("bar" -> Set(1)))
res1: Boolean = false

scala> myLens.set(Map("foo" -> Set(2)), true)
res2: Map[String,Set[Int]] = Map(foo -> Set(2, 1))

scala> myLens.set(Map("bar" -> Set(2)), true)
res3: Map[String,Set[Int]] = Map(bar -> Set(2), foo -> Set(1))

scala> myLens.set(Map("foo" -> Set(1)), false)
res4: Map[String,Set[Int]] = Map(foo -> Set())

可以说，以下是编写适配器的原则性更强的方法：

def noneZero[A: Monoid: Equal]: Lens[Option[A], A] = Lens.lensg(
  _ => a => a.ifEmpty[Option[A]](none)(some(a)),
  _.orZero
)

除了取消设置集合中的最后一个值将其从地图中删除外 ，这的行为相同。

scala> myLens.set(Map("foo" -> Set(1)), false)
res5: Map[String,Set[Int]] = Map()

但是，这可能不是您想要的。

Answer 2

香草

myMap + (key -> myMap.get(key).fold(Set(newValue))(_ + newValue))

似乎更容易。

编写扩展方法也是如此，为避免不必要的重构，值得做一些额外的工作：

implicit class MapsToSetsCanAdd[K,V](map: Map[K, Set[V]]) {
  def setAdd(key: K, value: V) = map.get(key) match {
    case Some(set) => if (set contains value) map else map + (key -> (set + value))
    case None => map + (key -> Set(value))
  }
}

现在，您可以愉快地myMap setAdd (key, newValue) 。