我有一个简单的 Spark 作业,从 5 节点 Cassandra 集群读取 500m 行,该集群始终运行 6 个任务,由于每个任务的大小,这会导致写入问题。我试过调整input_split_size,好像没有效果。目前我被迫重新分区表扫描,这并不理想,因为它很昂贵。
阅读了几篇文章后,我尝试在启动脚本(如下)中增加 num-executors,尽管这没有效果。
如果没有办法在 Cassandra 表扫描中设置任务数量,那很好,我会做的。但我有一种持续的琐碎感觉,我在这里遗漏了一些东西。
Spark 工作人员生活在 C* 节点上,这些节点是 8 核、64GB 服务器,每个服务器配备 2TB SSD。
...
val conf = new SparkConf(true).set("spark.cassandra.connection.host",
cassandraHost).setAppName("rowMigration")
conf.set("spark.shuffle.memoryFraction", "0.4")
conf.set("spark.serializer", "org.apache.spark.serializer.KryoSerializer")
conf.set("spark.executor.memory", "15G")
conf.set("spark.cassandra.input.split.size_in_mb", "32") //default 64mb
conf.set("spark.cassandra.output.batch.size.bytes", "1000") //default
conf.set("spark.cassandra.output.concurrent.writes", "5") //default
val sc = new SparkContext(conf)
val rawEvents = sc.cassandraTable(cassandraKeyspace, eventTable)
.select("accountid", "userid", "eventname", "eventid", "eventproperties")
.filter(row=>row.getString("accountid").equals("someAccount"))
.repartition(100)
val object = rawEvents
.map(ele => (ele.getString("userid"),
UUID.randomUUID(),
UUID.randomUUID(),
ele.getUUID("eventid"),
ele.getString("eventname"),
"event type",
UUIDs.unixTimestamp(ele.getUUID("eventid")),
ele.getMap[String, String]("eventproperties"),
Map[String, String](),
Map[String, String](),
Map[String, String]()))
.map(row=>MyObject(row))
Object.saveToCassandra(targetCassandraKeyspace,eventTable)
启动脚本:
#!/bin/bash
export SHADED_JAR="Migrate.jar"
export SPARKHOME="${SPARKHOME:-/opt/spark}"
export SPARK_CLASSPATH="$SHADED_JAR:$SPARK_CLASSPATH"
export CLASS=com.migration.migrate
"${SPARKHOME}/bin/spark-submit" \
--class "${CLASS}" \
--jars $SHADED_JAR,$SHADED_JAR \
--master spark://cas-1-5:7077 \
--num-executors 15 \
--executor-memory 20g \
--executor-cores 4 "$SHADED_JAR" \
--worker-cores 20 \
-Dcassandra.connection.host=10.1.20.201 \
-Dzookeeper.host=10.1.20.211:2181 \
编辑 - 遵循 Piotr 的回答:
我在 sc.cassandraTable 上设置了 ReadConf.splitCount,如下所示,但这不会改变生成的任务数,这意味着我仍然需要重新分区表扫描。我开始认为我正在考虑这个错误并且重新分区是必要的。目前这项工作大约需要 1.5 小时,将表扫描重新划分为 1000 个任务,每个任务大约 10MB,从而将写入时间减少到几分钟。
val cassReadConfig = new ReadConf {
ReadConf.apply(splitCount = Option(1000)
)
}
val sc = new SparkContext(conf)
val rawEvents = sc.cassandraTable(cassandraKeyspace, eventTable)
.withReadConf(readConf = cassReadConfig)