http2 support context

dinstone · Feb 1, 2024 · b9e7ab6 · b9e7ab6
1 parent 716e25d
commit b9e7ab6
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/.../focus-transport-http2/src/main/java/com/dinstone/focus/transport/http2/Http2Channel.java b/.../focus-transport-http2/src/main/java/com/dinstone/focus/transport/http2/Http2Channel.java
@@ -16,6 +16,7 @@
 package com.dinstone.focus.transport.http2;
 
 import java.io.IOException;
+import java.net.InetSocketAddress;
 import java.util.concurrent.CompletableFuture;
 
 import com.dinstone.focus.compress.Compressor;
@@ -129,6 +130,14 @@ private byte[] encodeContent(Invocation invocation, ServiceConfig serviceConfig,
         return content;
     }
 
+    public InetSocketAddress getRemoteAddress() {
+        return (InetSocketAddress) channel.remoteAddress();
+    }
+
+    public InetSocketAddress getLocalAddress() {
+        return (InetSocketAddress) channel.localAddress();
+    }
+
     public static class StreamChannelHandler extends SimpleChannelInboundHandler<Http2StreamFrame> {
 
         private final CompletableFuture<Object> future;

diff --git a/...ocus-transport-http2/src/main/java/com/dinstone/focus/transport/http2/Http2Connector.java b/...ocus-transport-http2/src/main/java/com/dinstone/focus/transport/http2/Http2Connector.java
@@ -41,9 +41,13 @@ public CompletableFuture<Object> send(Invocation invocation, ServiceInstance ins
         // create connection
         if (instance.isEnableSsl()) {
             Http2Channel http2Channel = secureChannelFactory.create(instance.getInstanceAddress());
+            invocation.context().setRemoteAddress(http2Channel.getRemoteAddress());
+            invocation.context().setLocalAddress(http2Channel.getLocalAddress());
             return http2Channel.send(invocation);
         } else {
             Http2Channel http2Channel = commonChannelFactory.create(instance.getInstanceAddress());
+            invocation.context().setRemoteAddress(http2Channel.getRemoteAddress());
+            invocation.context().setLocalAddress(http2Channel.getLocalAddress());
             return http2Channel.send(invocation);
         }
     }

diff --git a/...ansport-http2/src/main/java/com/dinstone/focus/transport/http2/Http2MessageProcessor.java b/...ansport-http2/src/main/java/com/dinstone/focus/transport/http2/Http2MessageProcessor.java
@@ -16,6 +16,7 @@
 package com.dinstone.focus.transport.http2;
 
 import java.io.IOException;
+import java.net.InetSocketAddress;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.Executor;
 import java.util.function.Function;
@@ -73,6 +74,9 @@ private void invoke(Channel channel, Http2HeadersFrame headersFrame, Http2DataFr
             ByteBuf dataBuf = dataFrame == null ? null : dataFrame.content();
             Invocation invocation = decode(headers, dataBuf, serviceConfig, methodConfig);
 
+            invocation.context().setRemoteAddress((InetSocketAddress) channel.remoteAddress());
+            invocation.context().setLocalAddress((InetSocketAddress) channel.localAddress());
+
             // invoke invocation
             CompletableFuture<Object> replyFuture = serviceConfig.getHandler().handle(invocation);
             replyFuture.whenComplete((reply, error) -> {