scala - 在 Scala 中添加和删除回调

Question

这更多是一个“风格”问题......但它可能很有趣。

我写了以下特征:

trait EventProducer[B] {

  private var listeners: List[B => Unit] = List()

  def addListener(newListener: B => Unit) {
    listeners = listeners :+ newListener
  }

  protected def fireEvent(event: B) {
    listeners.map(listener => {
      listener(event)
    })
  }
}

到目前为止一切顺利。现在我想添加一个removeListener 函数...但是，嘿，我该怎么做呢？在 Java 中这很容易，因为我不使用匿名函数，而是使用一个类，然后我可以按值在列表中找到它并将其删除。但在这里我没有引用该函数(除非我保留它并且它不实用)。那么实现这个的好方法是什么？

Answer 1

看看我的微图书馆Model 。正如 @sschaef 指出的，您需要保留对该函数的引用。所以我从 addListener 返回它(简化):

class Model[U] {
  type Listener = PartialFunction[U, Unit]

  private var listeners = Vector.empty[Listener]

  def addListener(pf: Listener): pf.type = {
    listeners :+= pf
    pf
  }

  def removeListener(pf: Listener): Unit = {
    val idx = listeners.indexOf(pf)
    if (idx >=0) {
      listeners = listeners.patch(idx, Nil, 1)
    }
  }

  // ...
  def isEmpty = listeners.isEmpty
}

val m = new Model[String]
val l = m.addListener {
  case "handshake" => println("launch rockets")
}
m.removeListener(l)

---

需要非常小心的一件事是 eta 扩展，即使用方法代替函数。

def foo(s: String) = println(s"Observed $s")

m.addListener(foo)
m.removeListener(foo)
assert(m.isEmpty)  // fails!

这最后一点实际上无法编译，因为 eta 扩展仅适用于函数，而不适用于部分函数。在您的代码中，您使用了函数 B => Unit，因此请注意这个陷阱。