本文章记录使用kafka-reassign-partitions.sh脚本来重新分配节点分区。kafka节点扩容后,已经存在的旧的topic的分区是不会伴随节点的增加而变动的,所以需要我们手动的去调整分区。下面提供了一套脚本,可以实现自动化重分区,欢迎大家斧正:
脚本目录和文件目录要注意,这是我的集群的目录,大家可以根据自己的目录进行调整
主脚本:
需要传两个参数
$1:zookeeper链接地址
$2:brokerid的列表
atuoReassign.sh:
#!/bin/bash
zkServer=$1
brokerIdList=$2
if [ -z "$zkServer" ]
then
echo "zkServer can not be empty"
exit 1
else
echo "zkServer:"$zkServer
fi
if [ -z "$brokerIdList" ]
then
echo "brokerIdList can not be empty"
exit 2
else
echo "brokerIdList:" $brokerIdList
fi
echo "###########################start generate##############################"
sh /var/lib/kafka/autoAssign.sh $zkServer
echo "###########################end generate##############################"
echo "###########################start execute##############################"
sh /var/lib/kafka/autoRebalance.sh $zkServer $brokerIdList
echo "###########################end execute##############################"
echo "###########################delete generate json##############################"
rm -rf /opt/kafka/bin/topicMove.json
rm -rf /opt/kafka/bin/ressgintopic.json
echo "###########################end ressgin##############################"
autoAssign.sh:
#!/bin/bash
zkServer=$1
topicArr=`sh /opt/kafka/bin/kafka-topics.sh --zookeeper $zkServer --list`
topictext=''
symbol='"'
for topic in ${topicArr[@]}
do
topictext=$topictext','{$symbol'topic'$symbol:$symbol$topic$symbol}
done
assigntext='{"topics":['${topictext:1}'],"version":1}'
echo $assigntext >/opt/kafka/bin/topicMove.json
autoRebalance.sh:
#!/bin/bash
source /etc/profile
zkServer=$1
brokerIdList=$2
content=`sh /opt/kafka/bin/kafka-reassign-partitions.sh --zookeeper $zkServer --topics-to-move-json-file /opt/kafka/bin/topicMove.json --broker-list $brokerIdList --generate`
content=`echo $content | awk -F 'Proposed partition reassignment configuration' '{print $2}'`
echo $content
echo $content > /opt/kafka/bin/ressgintopic.json
sh /opt/kafka/bin/kafka-reassign-partitions.sh --zookeeper $zkServer$clusterName --reassignment-json-file /opt/kafka/bin/ressgintopic.json --execute