京東面試:如何合理設(shè)置 Flink 并行度?有哪些優(yōu)化的點(diǎn)?
一、Flink并行度基礎(chǔ)概念
1. 什么是并行度
在Apache Flink中,并行度(Parallelism)是指一個(gè)Flink程序的并行執(zhí)行能力。一個(gè)Flink程序由多個(gè)任務(wù)(task)組成,這些任務(wù)可以并行執(zhí)行以提高處理效率。每個(gè)task包含多個(gè)并行執(zhí)行的實(shí)例,且每一個(gè)實(shí)例都處理task輸入數(shù)據(jù)的一個(gè)子集。一個(gè)task的并行實(shí)例數(shù)被稱為該task的并行度。
Flink的并行架構(gòu)由以下幾個(gè)關(guān)鍵組件組成:
- JobManager:協(xié)調(diào)分布式執(zhí)行,如調(diào)度任務(wù)、協(xié)調(diào)檢查點(diǎn)等
- TaskManager:執(zhí)行任務(wù)的工作節(jié)點(diǎn),提供內(nèi)存和處理能力
- Task Slot:TaskManager中的資源單位,每個(gè)slot可以執(zhí)行一個(gè)并行任務(wù)實(shí)例
2. 并行度的重要性
合理設(shè)置并行度對(duì)Flink作業(yè)的性能至關(guān)重要,原因如下:
- 資源利用率:適當(dāng)?shù)牟⑿卸仍O(shè)置可以充分利用集群資源,避免資源浪費(fèi)
- 處理吞吐量:更高的并行度通常意味著更高的數(shù)據(jù)處理吞吐量
- 延遲控制:合理的并行度可以減少數(shù)據(jù)處理的延遲
- 負(fù)載均衡:適當(dāng)?shù)牟⑿卸扔兄谠诩褐芯夥峙涔ぷ髫?fù)載
- 成本效益:優(yōu)化并行度可以在保證性能的同時(shí)降低資源成本
3. 影響并行度的因素
在設(shè)置Flink作業(yè)的并行度時(shí),需要考慮以下因素:
- 數(shù)據(jù)量:處理的數(shù)據(jù)量越大,可能需要更高的并行度
- 計(jì)算復(fù)雜性:計(jì)算邏輯越復(fù)雜,可能需要更高的并行度
- 可用資源:集群的可用資源(CPU、內(nèi)存等)限制了最大可能的并行度
- 數(shù)據(jù)傾斜:數(shù)據(jù)分布不均勻可能導(dǎo)致某些并行實(shí)例負(fù)載過重
- 狀態(tài)大?。河袪顟B(tài)操作的狀態(tài)大小會(huì)影響內(nèi)存使用和并行度選擇
- 網(wǎng)絡(luò)傳輸:過高的并行度可能導(dǎo)致過多的網(wǎng)絡(luò)傳輸開銷
二、并行度配置級(jí)別與方法
Flink提供了多個(gè)級(jí)別的并行度配置,從最具體到最一般依次為:
1. 算子級(jí)別(最高優(yōu)先級(jí))
可以為單個(gè)算子(operator)設(shè)置特定的并行度,這將覆蓋所有其他級(jí)別的設(shè)置:
// Java示例
DataStream<String> dataStream = env.fromElements("a","b","c");
// 為map算子設(shè)置并行度為2
dataStream.map(s -> s.toUpperCase()).setParallelism(2);
// 為keyBy/sum算子設(shè)置并行度為3
dataStream.keyBy(value -> value).sum(0).setParallelism(3);
# Python示例
data_stream = env.from_elements("a", "b", "c")
# 為map算子設(shè)置并行度為2
data_stream.map(lambda s: s.upper()).set_parallelism(2)
# 為keyBy/sum算子設(shè)置并行度為3
data_stream.key_by(lambda x: x).sum(0).set_parallelism(3)
2. 執(zhí)行環(huán)境級(jí)別
可以在StreamExecutionEnvironment中設(shè)置所有算子的默認(rèn)并行度:
// Java示例
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
env.setParallelism(4);// 設(shè)置默認(rèn)并行度為4
// 此處所有算子將使用并行度4,除非單獨(dú)指定
DataStream<String> dataStream = env.fromElements("a","b","c");
dataStream.map(s -> s.toUpperCase());
dataStream.keyBy(value -> value).sum(0);
# Python示例
env = StreamExecutionEnvironment.get_execution_environment()
env.set_parallelism(4) # 設(shè)置默認(rèn)并行度為4
# 此處所有算子將使用并行度4,除非單獨(dú)指定
data_stream = env.from_elements("a", "b", "c")
data_stream.map(lambda s: s.upper())
data_stream.key_by(lambda x: x).sum(0)
3. 客戶端級(jí)別
在提交作業(yè)時(shí),可以通過命令行參數(shù)指定并行度:
# 使用命令行參數(shù)設(shè)置并行度
bin/flink run -p 8 examples/streaming/WordCount.jar
4. 系統(tǒng)級(jí)別(最低優(yōu)先級(jí))
可以在Flink配置文件(flink-conf.yaml)中設(shè)置集群范圍的默認(rèn)并行度:
# 在flink-conf.yaml中設(shè)置
parallelism.default:2
5. 并行度配置優(yōu)先級(jí)
當(dāng)多個(gè)級(jí)別同時(shí)設(shè)置并行度時(shí),優(yōu)先級(jí)從高到低為:
- 算子級(jí)別 (setParallelism())
- 執(zhí)行環(huán)境級(jí)別(env.setParallelism()) 3
- 客戶端級(jí)別 (命令行 -p 參數(shù)) 4. 系統(tǒng)級(jí)別 (flink-conf.yaml)
三、自適應(yīng)并行度與自動(dòng)優(yōu)化
1. 自適應(yīng)批處理調(diào)度器
Flink引入了AdaptiveBatchScheduler調(diào)度器,該調(diào)度器能夠自動(dòng)調(diào)整批處理作業(yè)的并行度,無需手動(dòng)設(shè)置。它根據(jù)輸入數(shù)據(jù)量和可用資源自動(dòng)推導(dǎo)出最優(yōu)的并行度配置。
(1) 自動(dòng)推導(dǎo)算子并行度
AdaptiveBatchScheduler支持自動(dòng)推導(dǎo)算子并行度,主要優(yōu)勢(shì)包括:
- 推作業(yè)用戶可以從并行度調(diào)優(yōu)中解放出來
- 根據(jù)數(shù)據(jù)量自動(dòng)推導(dǎo)并行度可以更好地適應(yīng)數(shù)據(jù)變化
- SQL作業(yè)的算子也可以分配不同的并行度
(2) 啟用自動(dòng)并行度推導(dǎo)
要使用AdaptiveBatchScheduler自動(dòng)推導(dǎo)算子并行度,需要:
① 啟用自動(dòng)并行度推導(dǎo):
// Java示例
Configuration configuration=newConfiguration();
// 啟用自適應(yīng)批處理調(diào)度器的自動(dòng)并行度功能
configuration.set(ExecutionOptions.BATCH_ADAPTIVE_AUTO_PARALLELISM_ENABLED,true);
// 設(shè)置自動(dòng)并行度的最小值
configuration.set(ExecutionOptions.BATCH_ADAPTIVE_AUTO_PARALLELISM_MIN_PARALLELISM,1);
// 設(shè)置自動(dòng)并行度的最大值
configuration.set(ExecutionOptions.BATCH_ADAPTIVE_AUTO_PARALLELISM_MAX_PARALLELISM,64);
// 設(shè)置每個(gè)任務(wù)平均處理的數(shù)據(jù)量
configuration.set(ExecutionOptions.BATCH_ADAPTIVE_AUTO_PARALLELISM_AVG_DATA_VOLUME_PER_TASK, MemorySize.ofMebiBytes(8));
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment(configuration);
② 也可以通過配置文件(flink-conf.yaml)啟用:
execution.batch.adaptive.auto-parallelism.enabled:true
execution.batch.adaptive.auto-parallelism.min-parallelism:1
execution.batch.adaptive.auto-parallelism.max-parallelism:64
execution.batch.adaptive.auto-parallelism.avg-data-volume-per-task: 8mb
2. 自適應(yīng)數(shù)據(jù)分發(fā)
AdaptiveBatchScheduler還支持其他優(yōu)化功能:
(1) 自適應(yīng)Broadcast Join
對(duì)于廣播連接(Broadcast Join),調(diào)度器可以根據(jù)數(shù)據(jù)量自動(dòng)選擇最佳的廣播策略,減少不必要的數(shù)據(jù)傳輸。
(2) 自適應(yīng)Skewed Join優(yōu)化
對(duì)于數(shù)據(jù)傾斜的連接操作,調(diào)度器可以自動(dòng)檢測(cè)并優(yōu)化數(shù)據(jù)分布不均勻的情況,提高連接操作的性能。
四、性能調(diào)優(yōu)策略
1. 資源配置優(yōu)化
(1) TaskManager和Slot配置
TaskManager和Slot的合理配置對(duì)并行度優(yōu)化至關(guān)重要:
- TaskManager數(shù)量:通常與集群物理節(jié)點(diǎn)數(shù)相關(guān)
- 每個(gè)TaskManager的Slot數(shù)量:通常設(shè)置為每個(gè)TaskManager的CPU核心數(shù)
- 內(nèi)存配置:需要根據(jù)作業(yè)特性合理分配TaskManager的內(nèi)存
# TaskManager配置示例
taskmanager.numberOfTaskSlots:8
taskmanager.memory.process.size: 4096m
(2) 資源組(Resource Group)
資源組允許將相關(guān)的算子分組,以便它們?cè)谕粋€(gè)TaskManager上執(zhí)行,減少網(wǎng)絡(luò)傳輸:
// Java示例
// 定義資源組
ResourceSpec spec = ResourceSpec.newBuilder()
.setCpuCores(1.0)
.setTaskHeapMemoryMB(512)
.build();
// 將算子分配到資源組
dataStream.map(newMyMapper()).slotSharingGroup("group1").setResources(spec);
2. 算子鏈(Operator Chaining)
算子鏈?zhǔn)荈link的一項(xiàng)重要優(yōu)化,它將多個(gè)算子合并到一個(gè)任務(wù)中執(zhí)行,減少了任務(wù)間的數(shù)據(jù)傳輸開銷:
(1) 啟用/禁用算子鏈
// Java示例
// 全局禁用算子鏈
env.disableOperatorChaining();
// 為特定算子禁用鏈接
dataStream.map(newMyMapper()).disableChaining();
// 開始新的鏈
dataStream.map(newMyMapper()).startNewChain();
(2) 算子鏈最佳實(shí)踐
- 將計(jì)算密集型算子與IO密集型算子分開鏈接
- 避免將狀態(tài)較大的算子鏈接在一起
- 考慮將具有相似資源需求的算子鏈接在一起
3. 數(shù)據(jù)傾斜處理
數(shù)據(jù)傾斜是影響并行度效率的主要因素之一:
(1) 識(shí)別數(shù)據(jù)傾斜
- 使用Flink Web UI監(jiān)控任務(wù)執(zhí)行
- 觀察各個(gè)子任務(wù)的處理記錄數(shù)和處理時(shí)間
- 檢查背壓(backpressure)指標(biāo)
(2) 解決數(shù)據(jù)傾斜的策略
① 預(yù)聚合:在keyBy之前進(jìn)行局部聚合,減少數(shù)據(jù)量
// Java示例 - 兩階段聚合處理數(shù)據(jù)傾斜
dataStream
.map(newPreAggregateFunction()) // 第一階段:局部預(yù)聚合
.keyBy(value -> value.getKey())
.window(TumblingEventTimeWindows.of(Time.seconds(10)))
.reduce(newAggregateFunction()) // 第二階段:全局聚合
② Key重分區(qū):為熱點(diǎn)key添加隨機(jī)前綴,將一個(gè)熱點(diǎn)key分散到多個(gè)任務(wù)
// Java示例 - 使用隨機(jī)前綴重分區(qū)熱點(diǎn)key
dataStream
.map(event ->{
// 為熱點(diǎn)key添加隨機(jī)前綴
if(isHotKey(event.getKey())){
int randomPrefix = ThreadLocalRandom.current().nextInt(parallelism);
returnnew Tuple2<>(randomPrefix +"_"+event.getKey(), event.getValue());
}else{
returnnew Tuple2<>(event.getKey(), event.getValue());
}
})
.keyBy(tuple -> tuple.f0) // 使用新key進(jìn)行分區(qū)
.window(TumblingEventTimeWindows.of(Time.seconds(10)))
.reduce(newAggregateFunction())
// 最后移除前綴
.map(tuple ->new Tuple2<>(removePrefix(tuple.f0), tuple.f1));
4. 狀態(tài)管理優(yōu)化
對(duì)于有狀態(tài)的操作,狀態(tài)管理對(duì)并行度優(yōu)化也很重要:
(1) 狀態(tài)后端選擇
Flink提供了三種狀態(tài)后端,根據(jù)作業(yè)特性選擇合適的狀態(tài)后端:
- MemoryStateBackend:小狀態(tài),低延遲,不需要恢復(fù)
- FsStateBackend:大狀態(tài),低延遲,可靠恢復(fù)
- RocksDBStateBackend:超大狀態(tài),較高延遲,可增量檢查點(diǎn)
// Java示例 - 配置狀態(tài)后端
// 內(nèi)存狀態(tài)后端
env.setStateBackend(newMemoryStateBackend());
// 文件系統(tǒng)狀態(tài)后端
env.setStateBackend(newFsStateBackend("hdfs://namenode:9000/flink/checkpoints"));
// RocksDB狀態(tài)后端
env.setStateBackend(newRocksDBStateBackend("hdfs://namenode:9000/flink/checkpoints",true));
(2) 狀態(tài)大小與并行度的關(guān)系
- 增加并行度會(huì)將狀態(tài)分散到更多的任務(wù)實(shí)例中
- 過大的狀態(tài)可能導(dǎo)致內(nèi)存壓力,影響性能
- 考慮使用RocksDBStateBackend處理超大狀態(tài)