elasticsearch源码分析-02选举主节点

在Elasticsearch中，数据的写入和查询是在主节点上进行的，主节点的选举是集群健康运行的关键。在Elasticsearch中，主节点被称为master节点。主节点的选举是通过一个叫做Zen Discovery的模块来完成的，它负责集群中节点的发现以及主节点的选举。

以下是主节点选举的核心代码：

// 在Elasticsearch的ZenDiscovery模块中，有一个MasterFinder类负责发现和选举主节点
class MasterFinder {
    // 选举主节点
    public void findMaster(ClusterState state, DiscoveryNode localNode) {
        // 获取所有可能的主节点
        List<DiscoveryNode> possibleNodes = state.getNodes().getMasterNodes().values().stream()
            .filter(node -> node.isMasterEligible())
            .collect(Collectors.toList());
 
        // 如果本地节点有权限成为主节点，并且没有其他节点比本地节点更合适
        if (localNode.isMasterEligible() && possibleNodes.isEmpty()) {
            becomeMaster.run();
        } else {
            // 如果有合适的节点，则选举该节点为主节点
            DiscoveryNode masterNode = possibleNodes.stream()
                .filter(node -> node.getId().equals(state.nodes().getMasterNodeId()))
                .findFirst()
                .orElse(null);
 
            if (masterNode != null) {
                // 如果本地节点不是主节点，但是已经有一个主节点了，则本地节点成为它的follower
                if (localNode.equals(masterNode) == false) {
                    followerNode.run();
                }
            } else {
                // 如果没有主节点，并且本地节点没有权限成为主节点，则重新选举
                becomeMaster.run();
            }
        }
    }
}

在这段代码中，ClusterState 对象包含了集群状态的所有信息，包括所有节点的信息和当前的主节点。DiscoveryNode 是一个代表节点的类，包含了节点的所有信息，包括节点是否有资格成为主节点以及节点的ID等。

在选举主节点时，首先会筛选出所有有资格成为主节点的节点，然后判断是否需要进行主节点的选举。如果本地节点是唯一有资格的节点，且之前没有主节点，则本地节点会成为主节点。如果有其他节点有资格，则会根据集群状态中记录的主节点ID来决定是否需要重新选举。

这个过程是Elasticsearch保证集群健康运行和数据一致性的重要步骤。