Dialogue channels use EndpointChannelFactory throughout

palantir · May 6, 2024 · f7174ec · f7174ec
1 parent 9536ba6
commit f7174ec
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/dialogue-clients/src/main/java/com/palantir/dialogue/clients/ReloadingClientFactory.java b/dialogue-clients/src/main/java/com/palantir/dialogue/clients/ReloadingClientFactory.java
@@ -482,10 +482,15 @@ public String toString() {
         public Supplier<Channel> stickyChannels() {
             return () -> this;
         }
+
+        @Override
+        public EndpointChannel endpoint(Endpoint _endpoint) {
+            return _request -> Futures.immediateFailedFuture(exceptionSupplier.get());
+        }
     }
 
     /* Abstracts away DialogueChannel so that we can handle no-service/no-uri case in #getInternalDialogueChannel. */
-    private interface InternalDialogueChannel extends Channel {
+    private interface InternalDialogueChannel extends Channel, EndpointChannelFactory {
         Supplier<Channel> stickyChannels();
     }
 
@@ -502,6 +507,11 @@ public ListenableFuture<Response> execute(Endpoint endpoint, Request request) {
             return dialogueChannel.execute(endpoint, request);
         }
 
+        @Override
+        public EndpointChannel endpoint(Endpoint endpoint) {
+            return dialogueChannel.endpoint(endpoint);
+        }
+
         @Override
         public Supplier<Channel> stickyChannels() {
             return dialogueChannel.stickyChannels();