add remote address when reuse connection

qiniu · Nov 13, 2020 · d0a5ab5 · d0a5ab5
1 parent af316fe
commit d0a5ab5
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/main/java/com/qiniu/http/Client.java b/src/main/java/com/qiniu/http/Client.java
@@ -79,6 +79,23 @@ public void connectStart(Call call, InetSocketAddress inetSocketAddress, Proxy p
                 tag.ip = inetSocketAddress + "";
             }
         });
+        builder.addNetworkInterceptor(new Interceptor() {
+            @Override
+            public okhttp3.Response intercept(Chain chain) throws IOException {
+                Request request = chain.request();
+                okhttp3.Response response = chain.proceed(request);
+                IpTag tag = (IpTag) request.tag();
+                if (tag.ip != null && tag.ip.length() > 5) {
+                    return response;
+                }
+                try {
+                    tag.ip = chain.connection().socket().getRemoteSocketAddress().toString();
+                } catch (Exception e) {
+                    // ingore
+                }
+                return response;
+            }
+        });
         builder.addInterceptor(new Interceptor() {
             @Override
             public okhttp3.Response intercept(Chain chain) throws IOException {