Avoid instrumenting already-instrumented RDDs, as this causes metrics…

… to be recorded multiple times
bigdatagenomics · Jan 18, 2015 · d8e02d9 · d8e02d9
1 parent a11e07e
commit d8e02d9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/adam-core/src/main/scala/org/apache/spark/rdd/InstrumentedRDDFunctions.scala b/adam-core/src/main/scala/org/apache/spark/rdd/InstrumentedRDDFunctions.scala
@@ -24,8 +24,9 @@ import scala.reflect.ClassTag
  * calling the `unInstrument` method. For more details and usage instructions see the [[MetricsContext]] class.
  */
 class InstrumentedRDDFunctions[T: ClassTag](self: RDD[T]) {
-  def instrument(): RDD[T] = {
-    InstrumentedRDD.instrument(self)
+  def instrument(): RDD[T] = self match {
+    case instrumentedRDD: InstrumentedRDD[T] => self
+    case _                                   => InstrumentedRDD.instrument(self)
   }
   def unInstrument(): RDD[T] = self match {
     case instrumentedRDD: InstrumentedRDD[T] => instrumentedRDD.decoratedRDD