IGNITE-24446 Implement utilities necessary for zone assignments calcu…

…lation (#5200)
apache · Feb 12, 2025 · 6761ca9 · 6761ca9
1 parent 63fb728
commit 6761ca9
Show file tree

Hide file tree

Showing 3 changed files with 465 additions and 282 deletions.
diff --git a/...c/main/java/org/apache/ignite/internal/distributionzones/rebalance/ZoneRebalanceUtil.java b/...c/main/java/org/apache/ignite/internal/distributionzones/rebalance/ZoneRebalanceUtil.java
@@ -659,4 +659,35 @@ private static CompletableFuture<Map<Integer, Assignments>> zoneAssignments(
                     return numberOfMsPartitions == 0 ? Map.of() : result;
                 });
     }
+
+    /**
+     * Returns zone pending assignments for all zone partitions from meta storage locally.
+     *
+     * @param metaStorageManager Meta storage manager.
+     * @param zoneId Zone id.
+     * @param numberOfPartitions Number of partitions.
+     * @param revision Revision.
+     * @return Future with zone assignments as a value.
+     */
+    public static List<Assignments> zonePendingAssignmentsGetLocally(
+            MetaStorageManager metaStorageManager,
+            int zoneId,
+            int numberOfPartitions,
+            long revision
+    ) {
+        return IntStream.range(0, numberOfPartitions)
+                .mapToObj(partitionId -> getLocalAssignments(metaStorageManager, zoneId, partitionId, revision))
+                .collect(toList());
+    }
+
+    private static @Nullable Assignments getLocalAssignments(
+            MetaStorageManager metaStorageManager,
+            int zoneId,
+            int partitionId,
+            long revision
+    ) {
+        Entry entry = metaStorageManager.getLocally(pendingPartAssignmentsKey(new ZonePartitionId(zoneId, partitionId)), revision);
+
+        return entry != null ? Assignments.fromBytes(entry.value()) : null;
+    }
 }