ascherbakoff commented on a change in pull request #103:
URL: https://github.com/apache/ignite-3/pull/103#discussion_r619349629
##########
File path:
modules/affinity/src/main/java/org/apache/ignite/internal/affinity/AffinityManager.java
##########
@@ -52,9 +73,121 @@ public AffinityManager(
this.configurationMgr = configurationMgr;
this.metaStorageMgr = metaStorageMgr;
this.baselineMgr = baselineMgr;
+
+ String localNodeName =
configurationMgr.configurationRegistry().getConfiguration(NetworkConfiguration.KEY)
+ .name().value();
+
+
configurationMgr.configurationRegistry().getConfiguration(LocalConfiguration.KEY)
+ .metastorageMembers().listen(ctx -> {
+ if (ctx.newValue() != null) {
+ if (hasMetastorageLocally(localNodeName, ctx.newValue()))
+ subscribeToCalculateAssignment();
+ else
+ unsubscribeToCalculateAssignment();
+ }
+ return CompletableFuture.completedFuture(null);
+ });
+
+ String[] metastorageMembers =
configurationMgr.configurationRegistry().getConfiguration(LocalConfiguration.KEY)
+ .metastorageMembers().value();
+
+ if (hasMetastorageLocally(localNodeName, metastorageMembers))
+ subscribeToCalculateAssignment();
}
- // TODO: IGNITE-14237 Affinity function.
- // TODO: IGNITE-14238 Creating and destroying caches.
- // TODO: IGNITE-14235 Provide a minimal cache/table configuration.
+ /**
+ * Tests a node has a involved into Metastorage.
+ *
+ * @param localNodeName Local node uniq name.
+ * @param metastorageMembers Metastorage members names.
+ * @return True if the node has Metastorage, false otherwise.
+ */
+ private boolean hasMetastorageLocally(String localNodeName, String[]
metastorageMembers) {
+ boolean isLocalNodeHasMetasorage = false;
+
+ for (String name : metastorageMembers) {
+ if (name.equals(localNodeName)) {
+ isLocalNodeHasMetasorage = true;
+
+ break;
+ }
+ }
+ return isLocalNodeHasMetasorage;
+ }
+
+ /**
+ * Subscribes to metastorage members update.
+ */
+ private void subscribeToCalculateAssignment() {
+ assert affinityCalculateSubscriptionFut == null : "Affinity
calculation already subscribed";
+
+ String tableInternalPrefix = INTERNAL_PREFIX + "assignment.#";
+
+ affinityCalculateSubscriptionFut = metaStorageMgr.registerWatch(new
Key(tableInternalPrefix), new WatchListener() {
+ @Override public boolean onUpdate(@NotNull Iterable<WatchEvent>
events) {
+ for (WatchEvent evt : events) {
+ if (ArrayUtils.empty(evt.newEntry().value())) {
+ String keyTail =
evt.newEntry().key().toString().substring(INTERNAL_PREFIX.length());
+
+ String placeholderValue = keyTail.substring(0,
keyTail.indexOf('.'));
+
+ UUID tblId = UUID.fromString(placeholderValue);
+
+ try {
+ String name = new String(metaStorageMgr.get(
+ new Key(INTERNAL_PREFIX +
tblId.toString())).get()
+ .value(), StandardCharsets.UTF_8);
+
+ int partitions =
configurationMgr.configurationRegistry().getConfiguration(TablesConfiguration.KEY)
+ .tables().get(name).partitions().value();
+ int replicas =
configurationMgr.configurationRegistry().getConfiguration(TablesConfiguration.KEY)
+ .tables().get(name).replicas().value();
+
+ metaStorageMgr.put(evt.newEntry().key(),
IgniteUtils.toBytes(
Review comment:
Why affinity is not calculated on table creation ?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
