首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >简单读写的spark OOM

简单读写的spark OOM
EN

Stack Overflow用户
提问于 2019-07-10 22:11:52
回答 1查看 306关注 0票数 1

我使用spark从postgres表中读取数据,并将其作为json转储到Google云存储。这张桌子很大,有100多GB。代码相对简单(请看下面),但它在OOM中失败了。似乎spark正在尝试在开始写入之前读取内存中的整个表。这是真的吗?如何更改行为,使其以流的方式进行读写?

谢谢。

代码语言:javascript
复制
SparkSession sparkSession = SparkSession
                .builder()
                .appName("01-Getting-Started")
                .getOrCreate();
Dataset<Row> dataset = sparkSession.read().jdbc("jdbc:postgresql://<ip>:<port>/<db>", "<table>", properties);
dataset.write().mode(SaveMode.Append).json("gs://some/path");
EN

回答 1

Stack Overflow用户

发布于 2019-07-11 05:48:22

有两个重载的DataFrameReader.jdbc()方法可用于拆分输入的JDBC数据。

jdbc(String url, String table, String[] predicates, java.util.Properties connectionProperties) -生成的DataFrame对每个给定的谓词都有一个分区,例如

代码语言:javascript
复制
String[] preds = {“state=‘Alabama’”, “state=‘Alaska’”, “state=‘Arkansas’”, …};
Dataset<Row> dataset = sparkSession.read().jdbc("jdbc:postgresql://<ip>:<port>/<db>", "<table>", preds, properties);

jdbc(String url, String table, String columnName, long lowerBound, long upperBound, int numPartitions, java.util.Properties connectionProperties) - Spark将基于数字列columnName将数据划分为numPartitions分区,包括lowerBoundupperBound,例如:

Dataset<Row> dataset = sparkSession.read().jdbc("jdbc:postgresql://<ip>:<port>/<db>", "<table>", “<idColumn>”, 1, 1000, 100, properties);

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/56972600

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档