51zero / 51zero/eel-sdk

limit parallelism of ParquetSource to reduce memory footprint

未关闭
#382 0 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看
主要语言
Scala
星标
147
派生
32
PR 合并指标
30 天内没有已合并 PR

描述

Hi!

I set up a ParquetSource with a JDBCSink and ran into memory issues.
The parquet files are stored in an s3 bucket and have been written by spark (snappy-compressed ~500 MByte).

Spark writes one file per partition (default=200). This causes eel to use lots of memory since many subscriptions are submitted to the executor, although I set up the stream like this `source.toDataStream().to(sink, parallelism = 1)`

In the code I found that you initialize the executor like this `val executor = Executors.newCachedThreadPool()` which creates an unbounded ThreadPool.

I did some experiments and repartitioned the spark dataframe to 1 and stored it again. Here's the comparison (see screenshots below):
memory usage of the 200 files parquet source: >1.2 Gbyte*
memory usage of the 1 file parquet source: 83 MByte constantly.

(*) I let it run on my local machine with normal DSL internet connection. On the server it ran oom pretty quickly - meaning it used more than 2GByte (my XmX setting for the app).

![image](https://user-images.githubusercontent.com/39078/40673447-1f702fae-6372-11e8-99a2-828d96b66a1f.png)

![image](https://user-images.githubusercontent.com/39078/40673462-2a1367dc-6372-11e8-8269-6ff53f90fa3e.png)

Can you think of a way to limit the amount of parallelism? Happy to provide a merge request if you point me in the right direction.

贡献指南

这个仓库没有索引到贡献指南

调研方向

Look at the ParquetSource implementation to find where the Executors.newCachedThreadPool() is used. Understand how the source handles multiple files and subscriptions. The goal is to modify the thread pool initialization to allow limiting parallelism, perhaps by adding a configurable parameter. Check existing tests for ParquetSource to see how to validate changes.

由索引模型根据 Issue 内容生成。

评估

技术栈
hadoop, scala, spark
领域
data-engineering, stream-processing
Issue 类型
功能
难度
3/5
预计耗时
1-2 天
活跃度
停滞
描述清晰度
描述清楚
新手友好度
45/100

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。