按键分组时Spark会耗尽内存

时间:2014-03-25 14:26:56

标签: scala amazon-ec2 apache-spark

我正在尝试使用this guide在EC2上使用Spark主机执行常见爬网数据的简单转换,我的代码如下所示:

package ccminer

import org.apache.spark.SparkContext
import org.apache.spark.SparkContext._

object ccminer {
  val english = "english|en|eng"
  val spanish = "es|esp|spa|spanish|espanol"
  val turkish = "turkish|tr|tur|turc"
  val greek = "greek|el|ell"
  val italian = "italian|it|ita|italien"
  val all = (english :: spanish :: turkish :: greek :: italian :: Nil).mkString("|")

  def langIndep(s: String) = s.toLowerCase().replaceAll(all, "*")

  def main(args: Array[String]): Unit = {
    if (args.length != 3) {
      System.err.println("Bad command line")
      System.exit(-1)
    }

    val cluster = "spark://???"
    val sc = new SparkContext(cluster, "Common Crawl Miner",
      System.getenv("SPARK_HOME"), Seq("/root/spark/ccminer/target/scala-2.10/cc-miner_2.10-1.0.jar"))

    sc.sequenceFile[String, String](args(0)).map {
      case (k, v) => (langIndep(k), v)
    }
    .groupByKey(args(2).toInt)
    .filter {
      case (_, vs) => vs.size > 1
    }
    .saveAsTextFile(args(1))
  }
}

我按照以下命令运行它:

sbt/sbt "run-main ccminer.ccminer s3n://aws-publicdatasets/common-crawl/parse-output/segment/1341690165636/textData-* s3n://parallelcorpus/out/ 2000"

但很快失败并出现以下错误

java.lang.OutOfMemoryError: Java heap space
at com.ning.compress.BufferRecycler.allocEncodingBuffer(BufferRecycler.java:59)
at com.ning.compress.lzf.ChunkEncoder.<init>(ChunkEncoder.java:93)
at com.ning.compress.lzf.impl.UnsafeChunkEncoder.<init>(UnsafeChunkEncoder.java:40)
at com.ning.compress.lzf.impl.UnsafeChunkEncoderLE.<init>(UnsafeChunkEncoderLE.java:13)
at com.ning.compress.lzf.impl.UnsafeChunkEncoders.createEncoder(UnsafeChunkEncoders.java:31)
at com.ning.compress.lzf.util.ChunkEncoderFactory.optimalInstance(ChunkEncoderFactory.java:44)
at com.ning.compress.lzf.LZFOutputStream.<init>(LZFOutputStream.java:61)
at org.apache.spark.io.LZFCompressionCodec.compressedOutputStream(CompressionCodec.scala:60)
at org.apache.spark.storage.BlockManager.wrapForCompression(BlockManager.scala:803)
at org.apache.spark.storage.BlockManager$$anonfun$5.apply(BlockManager.scala:471)
at org.apache.spark.storage.BlockManager$$anonfun$5.apply(BlockManager.scala:471)
at org.apache.spark.storage.DiskBlockObjectWriter.open(BlockObjectWriter.scala:117)
at org.apache.spark.storage.DiskBlockObjectWriter.write(BlockObjectWriter.scala:174)
at org.apache.spark.scheduler.ShuffleMapTask$$anonfun$runTask$1.apply(ShuffleMapTask.scala:164)
at org.apache.spark.scheduler.ShuffleMapTask$$anonfun$runTask$1.apply(ShuffleMapTask.scala:161)
at scala.collection.Iterator$class.foreach(Iterator.scala:727)
at scala.collection.AbstractIterator.foreach(Iterator.scala:1157)
at org.apache.spark.scheduler.ShuffleMapTask.runTask(ShuffleMapTask.scala:161)
at org.apache.spark.scheduler.ShuffleMapTask.runTask(ShuffleMapTask.scala:102)
at org.apache.spark.scheduler.Task.run(Task.scala:53)
at org.apache.spark.executor.Executor$TaskRunner$$anonfun$run$1.apply$mcV$sp(Executor.scala:213)
at org.apache.spark.deploy.SparkHadoopUtil.runAsUser(SparkHadoopUtil.scala:49)
at org.apache.spark.executor.Executor$TaskRunner.run(Executor.scala:178)
at java.util.concurrent.ThreadPoolExecutor.runWorker(ThreadPoolExecutor.java:1145)
at java.util.concurrent.ThreadPoolExecutor$Worker.run(ThreadPoolExecutor.java:615)
at java.lang.Thread.run(Thread.java:744)

所以我的基本问题是,编写一个可以按键分组并且几乎无限量输入而不会耗尽内存的Spark任务有什么必要?

2 个答案:

答案 0 :(得分:14)

shuffle任务中java.lang.OutOfMemoryError异常的最常见原因(例如groupByKey,reduceByKey等)是parallelism的低级别。

您可以通过在configuration中设置 spark.default.parallelism 属性来增加默认值。

答案 1 :(得分:4)

所以这说明你已经用完了JVM的已分配堆空间。您可能会增加堆大小,但这仍然受系统功能的限制(不能超过物理RAM的数量)。

另一方面,正如 homutov 所解释的那样,这种情况发生在大型收集行动中。例如groupByKey,reduceByKey,cartisien + mapToPair。这些操作会将RDD数据收集到一个地方,使JVM用完堆空间

你能做什么?

根据我的经验,当群集/系统资源有限时,您可以使用Spark tuning guide spark.default.parallelism 可以增加,直到你可以将任务带入你的集群/系统[我曾经通过调整并行性在我的笔记本电脑的虚拟机上运行了14000个实例,1024个特征数据集的KNN实现]。

Command line flag :   --conf spark.default.parallelism=4   ; 4 is the parallelism value

请记住,您需要 TUNE 这些功能才能实现最有效和失败避免(用完堆)设置,以便从Spark中获得最佳效果。

<强>另外

请记住使用原始数据类型而不是包装器。并使用数组而不是集合

 ex :  List<Integers> vs int[] ; int[] is better than List 

在Spark数组中可以节省许多宝贵的空间并提高性能。

还可以使用 BroadCast 变量代替笛卡尔积或任何大型组合任务。