Kafka-服务端-PartitionStateMachine

PartitionStateMachine是Controller Leader用于维护分区状态的状态机。分区的状态是通过PartitionState接口定义的，它有四个子类分别代表了分区四种可能的状态，如表所示。

分区各个PartitionState之间的转换如图所示。

下面分析各个状态之间转换时，需要完成的相关操作。

• NonExistentPartition →NewPartition

从ZooKeeper中加载分区的AR集合到ControllerContext的partitionReplicaAssignment集合中。

• NewPartition →OnlinePartition

首先将Leader副本和ISR集合的信息写入到ZooKeeper中，这里会将分区的AR集合中第一个可用的副本选举为Leader副本，并将分区的所有可用副本作为ISR集合。

之后，向所有可用的副本发送LeaderAndIsrRequest,指导这些副本进行Leader/Follower的角色切换，并向所有可用的Broker发送UpdateMetadataRequest来更新其上的MetadataCache。

• OnlinePartitio/OffinePartition →OnlinePartition

为分区选择新的Leader副本和ISR集合，并将结果写入ZooKeeper。之后，向需要进行角色切换的副本发送LeaderAndIsrRequest,指导这些副本进行Leader/Follower的角色切换，并向所有可用的Broker发送UpdateMetadataRequest来更新其上的MetadataCache。

• NewPartition,OnlinePartition →OfflinePartition

只进行状态转换，并没有其他的操作。

• OfinePartition →NonExistentPartition

只进行状态转换，并没有其他的操作。

PartitionStateMachine中的各个字段含义和作用如下所述。

• controllerContext:ControllerContext对象，用于维护KafkaController的上下文信息。
• zkUtils:ZooKeeper的客户端，用于与ZooKeeper服务器交互。
• partitionState:Map[TopicAndPartition,PartitionState]类型，记录了每个分区对应的PartitionState状态。
• brokerRequestBatch:ControllerBrokerRequestBatch对象，用于向指定的Broker批量发送请求。

noOpPartitionLeaderSelector: 默认的Leader副本选举类器， 继承了PartitionLeaderSelector。NoOpLeaderSelector实现并没有真正进行Leader副本的选举，其实现是返回当前的Leader副本、ISR集合和AR集合。

• topicChangeListener:ZooKeeper的监听器，用于监听Topic的变化。
• deleteTopicsListener:ZooKeeper的监听器，用于监听Topic的删除。
• partitionModificationsListeners:用于监听分区的修改。

PartitionStateMachine启动时会对partitionState集合进行初始化，并调用triggerOnlinePartitionStateChange方法将NewPartition和OfflinePartition状态的分区转换成OnlinePartition状态。

PartitionStateMachine.handleStateChange()方法是管理分区状态的核心方法，该方法控制着PartitionState的转换。这里需要注意该方法的第三个参数，它指定了用来选举Leader副本的PartitionLeaderSelector对象。

PartitionState由NewPartition切换为OnlinePartition时，调用了initializeLeaderAndIsrForPartition方法，其操作的主要步骤是：

1. 从ControllerContext.partitionReplicaAssignment集合中选择第一个可用的副本作为Leader副本，其余的副本构成ISR集合。
2. 将Leader副本和ISR集合的信息写入到ZooKeeper。
3. 更新ControllerContext.partitionLeadershipInfo中缓存的Leader副本、ISR集合等信息。
4. 将上述步骤中确定的Leader副本、ISR集合、AR集合等信息添加到ControllerBrokerRequestBatch,之后会封装成LeaderAndlsrRequest发送给相关的Broker。