ibessonov commented on a change in pull request #229:
URL: https://github.com/apache/ignite-3/pull/229#discussion_r673004869
##########
File path:
modules/network-api/src/main/java/org/apache/ignite/network/AbstractMessagingService.java
##########
@@ -17,26 +17,56 @@
package org.apache.ignite.network;
+import java.util.ArrayList;
import java.util.Collection;
-import java.util.Collections;
-import java.util.concurrent.CopyOnWriteArrayList;
+import java.util.List;
+import java.util.concurrent.atomic.AtomicReferenceArray;
+import org.apache.ignite.network.annotations.MessageGroup;
/**
* Base class for {@link MessagingService} implementations.
*/
public abstract class AbstractMessagingService implements MessagingService {
- /** */
- private final Collection<NetworkMessageHandler> messageHandlers = new
CopyOnWriteArrayList<>();
+ /** Mapping from group type (array index) to a list of registered message
handlers. */
+ private final AtomicReferenceArray<List<NetworkMessageHandler>>
handlersByGroupType =
+ new AtomicReferenceArray<>(Short.MAX_VALUE + 1);
/** {@inheritDoc} */
- @Override public void addMessageHandler(NetworkMessageHandler handler) {
- messageHandlers.add(handler);
+ @Override public void addMessageHandler(Class<?> messageGroup,
NetworkMessageHandler handler) {
+ handlersByGroupType.getAndUpdate(getMessageGroupType(messageGroup),
handlers -> {
+ if (handlers == null)
+ return List.of(handler);
+
+ var result = new ArrayList<NetworkMessageHandler>(handlers.size()
+ 1);
+
+ result.addAll(handlers);
+ result.add(handler);
+
+ return result;
+ });
+ }
+
+ /**
+ * Extracts the message group ID from a class annotated with {@link
MessageGroup}.
+ */
+ private static short getMessageGroupType(Class<?> messageGroup) {
Review comment:
You're right
--
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.
To unsubscribe, e-mail: notifications-unsubscr...@ignite.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
