diff options
| author | Ali Ghodsi <alig@cs.berkeley.edu> | 2013-08-15 11:04:38 -0700 |
|---|---|---|
| committer | Ali Ghodsi <alig@cs.berkeley.edu> | 2013-08-20 16:13:37 -0700 |
| commit | 4f99be1ffd2f5237566b143b317e3677b50d06e8 (patch) | |
| tree | 8c638c9d2e904490918968f117aa06b733359d42 /core | |
| parent | f67753cdfcb08b3786320c45af98e097a8ed5d39 (diff) | |
| download | spark-4f99be1ffd2f5237566b143b317e3677b50d06e8.tar.gz spark-4f99be1ffd2f5237566b143b317e3677b50d06e8.tar.bz2 spark-4f99be1ffd2f5237566b143b317e3677b50d06e8.zip | |
use count rather than foreach
Diffstat (limited to 'core')
| -rw-r--r-- | core/src/main/scala/spark/rdd/CoalescedRDD.scala | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/core/src/main/scala/spark/rdd/CoalescedRDD.scala b/core/src/main/scala/spark/rdd/CoalescedRDD.scala index bfd0cf257e..450d8b33a9 100644 --- a/core/src/main/scala/spark/rdd/CoalescedRDD.scala +++ b/core/src/main/scala/spark/rdd/CoalescedRDD.scala @@ -42,8 +42,7 @@ private[spark] case class CoalescedRDDPartition( * @return locality of this coalesced partition between 0 and 1 */ def localFraction :Double = { - var loc: Int = 0 - parents.foreach(p => if (rdd.preferredLocations(p).contains(preferredLocation)) loc += 1) + val loc = parents.count(p => rdd.preferredLocations(p).contains(preferredLocation)) if (parents.size == 0) 0.0 else (loc.toDouble / parents.size.toDouble) } } |