AsyncHttpClient: extract net attributes from InetSocketAddress (#3882)

Author: laurit

instrumentation/async-http-client/async-http-client-2.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/asynchttpclient/v2_0/AsyncHttpClientNetAttributesExtractor.java

@@ -5,33 +5,26 @@
 
 package io.opentelemetry.javaagent.instrumentation.asynchttpclient.v2_0;
 
-import io.opentelemetry.instrumentation.api.instrumenter.net.NetAttributesExtractor;
+import io.opentelemetry.instrumentation.api.instrumenter.net.InetSocketAddressNetAttributesExtractor;
 import io.opentelemetry.semconv.trace.attributes.SemanticAttributes;
+import java.net.InetSocketAddress;
 import org.asynchttpclient.Request;
 import org.asynchttpclient.Response;
 import org.checkerframework.checker.nullness.qual.Nullable;
 
 final class AsyncHttpClientNetAttributesExtractor
-    extends NetAttributesExtractor<Request, Response> {
+    extends InetSocketAddressNetAttributesExtractor<Request, Response> {
 
   @Override
   public String transport(Request request) {
     return SemanticAttributes.NetTransportValues.IP_TCP;
   }
 
   @Override
-  public String peerName(Request request, @Nullable Response response) {
-    return request.getUri().getHost();
-  }
-
-  @Override
-  public Integer peerPort(Request request, @Nullable Response response) {
-    return request.getUri().getPort();
-  }
-
-  @Override
-  @Nullable
-  public String peerIp(Request request, @Nullable Response response) {
+  public @Nullable InetSocketAddress getAddress(Request request, @Nullable Response response) {
+    if (response != null && response.getRemoteAddress() instanceof InetSocketAddress) {
+      return (InetSocketAddress) response.getRemoteAddress();
+    }
     return null;
   }
 }