如何通过Java API实现GridDB集群节点故障后的容器分区重分配与自动恢复

我之前维护3节点GridDB时序集群时，刚好碰到和你一模一样的需求——节点故障后要自动检测分区健康、触发重分配，结合实际踩过的坑和官方API的使用经验，给你梳理下可行的落地方案：

一、查询分区分布与健康状态 #

你提到用getPartitionInfo()拿不到健康状态，其实可以结合集群节点状态来判断分区是否正常：

首先通过GridStore.getClusterInfo()获取所有节点的活跃状态，再关联容器的分区信息，就能定位哪些分区因为节点下线处于异常状态：

GridClusterInfo clusterInfo = store.getClusterInfo();
// 遍历所有节点，获取在线状态
for (GridNode node : clusterInfo.getNodes()) {
    GridNodeStatus nodeStatus = node.getStatus();
    System.out.printf("节点 %s:%d 状态: %s%n", node.getHost(), node.getPort(), nodeStatus);
}

// 关联容器分区，查看每个分区的主/副本节点
Container<Point, TimeSeries> container = store.getContainer("sensor_data");
GridPartitionInfo partitionInfo = container.getPartitionInfo();

for (int partitionId = 0; partitionId < partitionInfo.getPartitionCount(); partitionId++) {
    GridPartitionNode primaryNode = partitionInfo.getPrimaryNode(partitionId);
    // 校验主节点是否活跃
    boolean isPrimaryActive = clusterInfo.getNode(primaryNode.getHost(), primaryNode.getPort()).getStatus() == GridNodeStatus.ACTIVE;
    
    List<GridPartitionNode> replicaNodes = partitionInfo.getReplicaNodes(partitionId);
    // 同理可以校验副本节点状态
    for (GridPartitionNode replica : replicaNodes) {
        boolean isReplicaActive = clusterInfo.getNode(replica.getHost(), replica.getPort()).getStatus() == GridNodeStatus.ACTIVE;
    }
}

通过这种方式，就能精准识别出哪些分区的主/副本节点处于离线状态，判断分区健康度。

二、节点恢复后触发分区重平衡 #

GridDB的Java API提供了GridStore.rebalancePartitions()方法，只要应用拥有集群管理权限（配置的用户角色为ADMIN），就能手动触发全集群的分区重分配，把离线节点上的分区迁移到活跃节点，或者节点恢复后重新分配：

try {
    // 触发分区重平衡，该方法是异步执行的
    store.rebalancePartitions();
    System.out.println("分区重平衡已启动，GridDB会自动完成分区迁移与数据同步");
} catch (GridStoreException e) {
    // 常见异常：权限不足、集群状态不稳定（比如还有节点在离线中）
    System.err.println("触发重平衡失败：" + e.getMessage());
    e.printStackTrace();
}

如果需要等待重平衡完成，可以通过轮询GridClusterInfo的节点状态，或者监听下面提到的分区变化事件来确认。

三、监听分区与节点状态变化，实现自动恢复 #

要实现自动触发恢复逻辑，不需要人工干预，可以通过添加集群事件监听器，监听节点状态变化和分区迁移事件：

store.addClusterEventListener(new GridClusterEventListener() {
    @Override
    public void onNodeStatusChanged(GridNode node, GridNodeStatus oldStatus, GridNodeStatus newStatus) {
        // 当节点从离线恢复为活跃状态时，触发重平衡
        if (oldStatus == GridNodeStatus.INACTIVE && newStatus == GridNodeStatus.ACTIVE) {
            System.out.printf("节点 %s 已恢复，启动分区重平衡...%n", node.getHost());
            try {
                store.rebalancePartitions();
            } catch (GridStoreException e) {
                e.printStackTrace();
            }
        }
    }

    @Override
    public void onPartitionChanged(GridPartitionEvent event) {
        // 分区迁移完成时触发，可以在这里做数据一致性校验
        if (event.getType() == GridPartitionEvent.Type.REBALANCE_COMPLETED) {
            System.out.printf("分区 %d 重平衡完成，当前主节点：%s%n", 
                event.getPartitionId(), 
                event.getNewPrimaryNode().getHost());
            // 可选：校验容器数据完整性，比如查询最新的时序数据点
            // TimeSeries latest = container.query("SELECT * ORDER BY timestamp DESC LIMIT 1").fetch();
        }
    }
});

这个监听器会自动捕捉节点状态变化，一旦故障节点恢复，就自动触发分区重分配，完全不需要人工介入。

四、数据一致性保障的注意事项 #

GridDB在分区重平衡过程中，会自动同步主副本之间的数据，确保数据不丢失、一致性。应用侧不需要额外做复杂的同步操作，但需要注意：

• 重平衡过程中，读写操作不会被中断，GridDB会自动路由到正确的节点
• 如果在节点故障期间有写入，GridDB的副本机制会在节点恢复后自动同步这些数据
• 若要确认数据完全一致，可以在分区重平衡完成后，对容器执行一次简单的查询，验证数据完整性

内容来源于stack exchange