Skip to content
Prev Previous commit
Next Next commit
Use Option.
  • Loading branch information
JoshRosen committed Jun 8, 2017
commit f36fbaaa4d0a69de163e22f9a7514adc02e837b5
8 changes: 4 additions & 4 deletions core/src/main/scala/org/apache/spark/util/ListenerBus.scala
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ import org.apache.spark.internal.Logging
*/
private[spark] trait ListenerBus[L <: AnyRef, E] extends Logging {

private[this] val listenersPlusTimers = new CopyOnWriteArrayList[(L, Timer)]
private[this] val listenersPlusTimers = new CopyOnWriteArrayList[(L, Option[Timer])]

// Marked `private[spark]` for access in tests.
private[spark] def listeners = listenersPlusTimers.asScala.map(_._1).asJava
Expand All @@ -47,7 +47,7 @@ private[spark] trait ListenerBus[L <: AnyRef, E] extends Logging {
* Add a listener to listen events. This method is thread-safe and can be called in any thread.
*/
final def addListener(listener: L): Unit = {
listenersPlusTimers.add((listener, getTimer(listener).orNull))
listenersPlusTimers.add((listener, getTimer(listener)))
}

/**
Expand All @@ -73,8 +73,8 @@ private[spark] trait ListenerBus[L <: AnyRef, E] extends Logging {
val listenerAndMaybeTimer = iter.next()
val listener = listenerAndMaybeTimer._1
val maybeTimer = listenerAndMaybeTimer._2
val maybeTimerContext = if (maybeTimer != null) {
maybeTimer.time()
val maybeTimerContext = if (maybeTimer.isDefined) {
maybeTimer.get.time()
} else {
null
}
Expand Down