Minor cleanup.

apache · Sep 13, 2017 · cf5c6ce · cf5c6ce
1 parent 2915a5e
commit cf5c6ce
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/core/src/main/scala/org/apache/spark/scheduler/LiveListenerBus.scala b/core/src/main/scala/org/apache/spark/scheduler/LiveListenerBus.scala
@@ -169,12 +169,8 @@ private[spark] class LiveListenerBus(conf: SparkConf) {
     }
   }
 
-  private[spark] def findListenersByClass[T <: SparkListenerInterface : ClassTag]():
-      Seq[T] = {
-    val c = implicitly[ClassTag[T]].runtimeClass
-    queues.asScala.flatMap { queue =>
-      queue.listeners.asScala.filter(_.getClass() == c).map(_.asInstanceOf[T])
-    }
+  private[spark] def findListenersByClass[T <: SparkListenerInterface : ClassTag](): Seq[T] = {
+    queues.asScala.flatMap { queue => queue.findListenersByClass[T]() }
   }
 
   private[spark] def listeners: JList[SparkListenerInterface] = {