[WFCORE-6873] Remoting: use subsystem type endpoint for outbound comm…

…unication
wildfly · Jul 1, 2024 · d0e4cc5 · d0e4cc5
1 parent 6a02058
commit d0e4cc5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/remoting/subsystem/src/main/java/org/jboss/as/remoting/EndpointService.java b/remoting/subsystem/src/main/java/org/jboss/as/remoting/EndpointService.java
@@ -29,6 +29,7 @@ public class EndpointService implements Service {
 
     protected final String endpointName;
     protected volatile Endpoint endpoint;
+    private final EndpointType type;
     protected final OptionMap optionMap;
     private final Consumer<Endpoint> endpointConsumer;
     private final Supplier<XnioWorker> workerSupplier;
@@ -40,6 +41,7 @@ public EndpointService(final Consumer<Endpoint> endpointConsumer, final Supplier
             nodeName = "remote";
         }
         endpointName = type == EndpointType.SUBSYSTEM ? nodeName : nodeName + ":" + type;
+        this.type = type;
         this.optionMap = optionMap;
     }
 
@@ -58,6 +60,9 @@ public void start(final StartContext context) throws StartException {
         // Reuse the options for the remote connection factory for now
         this.endpoint = endpoint;
         endpointConsumer.accept(endpoint);
+        if (type == EndpointType.SUBSYSTEM) {
+            Endpoint.ENDPOINT_CONTEXT_MANAGER.setGlobalDefault(endpoint);
+        }
     }
 
     /** {@inheritDoc} */
@@ -73,6 +78,9 @@ public void stop(final StopContext context) {
                 context.complete();
             });
         }
+        if (type == EndpointType.SUBSYSTEM) {
+            Endpoint.ENDPOINT_CONTEXT_MANAGER.setGlobalDefault(null);
+        }
     }
 
     public enum EndpointType {