Kafka 分区操作详解：Leader 平衡与分区迁移
Kafka 分区是实现高可用和高吞吐的核心单元，分区的 Leader 副本负责处理读写请求，而副本分布直接影响集群负载均衡。当集群发生节点故障、扩容或缩容时，需通过分区操作（如 Leader 平衡、分区迁移）确保集群稳定高效。本文将详细介绍分区 Leader 平衡和分区迁移的操作方法、原理及实践场景。
分区 Leader 平衡
Kafka 主题的每个分区包含多个副本，其中Leader 副本负责处理读写请求，Follower 副本仅同步数据。理想情况下，Leader 副本应均匀分布在集群节点上，避免单个节点负载过高。当节点故障恢复后，原 Leader 可能未自动复位，导致负载不均衡，此时需通过Leader 平衡重新分配。
自动平衡
通过配置参数启用自动平衡，Kafka 控制器会定期检测并调整 Leader 分布。
（1）核心配置
auto.leader.rebalance.enable：是否启用自动 Leader 平衡（默认 true）。leader.imbalance.check.interval.seconds：检查间隔（默认 300 秒，即 5 分钟）。leader.imbalance.per.broker.percentage：触发平衡的不均衡阈值（默认 10%，即某节点 Leader 占比超过平均 10% 时触发）。
（2）原理
- 控制器定期检查各节点的 Leader 数量，计算不均衡比例。
- 当比例超过阈值时，将优先副本（Preferred Replica） 选举为新 Leader（优先副本是创建分区时指定的第一个副本，通常分布均衡）。
手动平衡
自动平衡可能因阈值或间隔设置导致响应不及时，此时需手动触发平衡，使用 kafka-preferred-replica-election.sh 脚本。
（1）操作步骤
① 定义需平衡的分区
创建 JSON 文件（如 preferred-replica.json），指定目标主题和分区：
{
"partitions": [
{"topic": "test-kafka-cluster", "partition": 0},
{"topic": "test-kafka-cluster", "partition": 1},
{"topic": "test-kafka-cluster", "partition": 2}
]
}
- 若需平衡所有分区，可省略
partition 字段（不推荐，可能影响集群性能）。
② 执行手动平衡
./kafka-preferred-replica-election.sh \
--bootstrap-server localhost:9092,localhost:9093 \
--path-to-json-file preferred-replica.json
kafka-preferred-replica-election.bat ^
--bootstrap-server localhost:9092,localhost:9093 ^
--path-to-json-file preferred-replica.json
（2）适用场景
- 集群刚重启，所有 Leader 集中在单个节点。
- 自动平衡未触发（如阈值未达），但负载已明显不均衡。
分区迁移
分区迁移用于调整副本的节点分布，适用于集群扩容（将旧节点分区迁移到新节点）、节点下线（将待下线节点的分区迁移到其他节点）或增加副本数（提升可靠性）。通过 kafka-reassign-partitions.sh 脚本实现。
核心流程
分区迁移分为三步：生成迁移方案 → 执行迁移 → 验证结果，核心是通过 JSON 文件定义迁移规则。
节点下线场景
若需下线 Broker ID=2 的节点，需先将其负责的分区迁移到其他节点（如 0 和 1）。
（1）生成迁移方案
① 定义待迁移主题
创建 JSON 文件（如 topics-to-move.json），指定目标主题：
{
"topics": [{"topic": "test-replication"}],
"version": 1
}
② 生成分区分配方案
./kafka-reassign-partitions.sh \
--zookeeper localhost:2181 \
--topics-to-move-json-file topics-to-move.json \
--broker-list "0,1"
--generate
输出示例：
# 当前分配方案（可备份，用于回滚）
Current partition replica assignment
{"version":1,"partitions":[{"topic":"test-replication","partition":0,"replicas":[2,0],...}]}
# 建议的新分配方案（无节点2）
Proposed partition reassignment configuration
{"version":1,"partitions":[{"topic":"test-replication","partition":0,"replicas":[0,1],...}]}
（2）执行迁移
将建议的方案保存为 reassignment.json，执行迁移：
./kafka-reassign-partitions.sh \
--zookeeper localhost:2181 \
--reassignment-json-file reassignment.json \
--execute
- 原理：目标节点创建分区目录 → 复制原分区数据 → 同步完成后切换 Leader → 删除原节点数据。
（3）验证迁移进度
./kafka-reassign-partitions.sh \
--zookeeper localhost:2181 \
--reassignment-json-file reassignment.json \
--verify
输出 Status of partition reassignment: Completed 表示迁移完成。
集群扩容场景
新增节点后，旧主题的分区不会自动迁移到新节点，需手动触发迁移，步骤与节点下线类似，但 --broker-list 需包含新节点 ID。
示例：新增节点 ID=3，将部分分区迁移到 3：
# reassignment.json（示例）
{
"version": 1,
"partitions": [
{"topic": "test-replication", "partition": 0, "replicas": [3, 0]},
{"topic": "test-replication", "partition": 1, "replicas": [3, 1]}
]
}
增加副本数
通过迁移可增加分区的副本数（如从 2 副本增至 3 副本），需在 reassignment.json 中指定新增的副本节点。
示例：为分区 0 增加副本至节点 2：
{
"version": 1,
"partitions": [
{"topic": "test-replication", "partition": 0, "replicas": [0, 1, 2]} # 原2副本→3副本
]
}
迁移限流
迁移过程可能占用大量网络带宽，需限制复制速率，避免影响集群服务。
（1）通过脚本参数限流
./kafka-reassign-partitions.sh \
--zookeeper localhost:2181 \
--reassignment-json-file reassignment.json \
--execute \
--throttle 1048576
（2）通过动态配置限流
./kafka-configs.sh \
--zookeeper localhost:2181 \
--entity-type topics \
--entity-name test-replication \
--alter \
--add-config leader.replication.throttled.rate=1048576,follower.replication.throttled.rate=1048576
最佳实践
- Leader 平衡：
- 生产环境建议启用自动平衡，同时监控 Leader 分布（通过
kafka-topics --describe）。
- 手动平衡仅在紧急情况下使用（如集群重启后）。
- 分区迁移：
- 迁移应在业务低峰期执行，避免影响读写性能。
- 迁移前备份当前分配方案，以便回滚。
- 限流值根据集群带宽调整（如 10-50MB/s），避免过度限制导致迁移耗时过长。
- 副本分布：
- 新增主题时确保副本均匀分布在所有节点（通过
--replication-factor 和分区数控制）。
- 节点下线前务必完成分区迁移，避免数据丢失