Add server address and port for Spymemcached

Author: steverao

instrumentation/spymemcached-2.12/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/spymemcached/SpymemcachedNetworkAttributesGetter.java

@@ -0,0 +1,50 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.instrumentation.spymemcached;
+
+import io.opentelemetry.instrumentation.api.semconv.network.ServerAttributesGetter;
+import java.net.InetSocketAddress;
+import java.net.SocketAddress;
+import java.util.Collection;
+import javax.annotation.Nullable;
+import net.spy.memcached.MemcachedConnection;
+import net.spy.memcached.MemcachedNode;
+
+final class SpymemcachedNetworkAttributesGetter
+    implements ServerAttributesGetter<SpymemcachedRequest> {
+
+  @Nullable
+  @Override
+  public String getServerAddress(SpymemcachedRequest request) {
+    MemcachedConnection connection = request.getConnection();
+    if (connection != null) {
+      Collection<MemcachedNode> nodes = connection.getLocator().getAll();
+      if (!nodes.isEmpty()) {
+        SocketAddress socketAddress = nodes.iterator().next().getSocketAddress();
+        if (socketAddress instanceof InetSocketAddress) {
+          return ((InetSocketAddress) socketAddress).getHostString();
+        }
+      }
+    }
+    return null;
+  }
+
+  @Nullable
+  @Override
+  public Integer getServerPort(SpymemcachedRequest request) {
+    MemcachedConnection connection = request.getConnection();
+    if (connection != null) {
+      Collection<MemcachedNode> nodes = connection.getLocator().getAll();
+      if (!nodes.isEmpty()) {
+        SocketAddress socketAddress = nodes.iterator().next().getSocketAddress();
+        if (socketAddress instanceof InetSocketAddress) {
+          return ((InetSocketAddress) socketAddress).getPort();
+        }
+      }
+    }
+    return null;
+  }
+}

instrumentation/spymemcached-2.12/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/spymemcached/SpymemcachedSingletons.java

@@ -11,6 +11,7 @@
 import io.opentelemetry.instrumentation.api.incubator.semconv.db.DbClientSpanNameExtractor;
 import io.opentelemetry.instrumentation.api.instrumenter.Instrumenter;
 import io.opentelemetry.instrumentation.api.instrumenter.SpanKindExtractor;
+import io.opentelemetry.instrumentation.api.semconv.network.ServerAttributesExtractor;
 
 public final class SpymemcachedSingletons {
   private static final String INSTRUMENTATION_NAME = "io.opentelemetry.spymemcached-2.12";
@@ -19,13 +20,15 @@ public final class SpymemcachedSingletons {
 
   static {
     SpymemcachedAttributesGetter dbAttributesGetter = new SpymemcachedAttributesGetter();
+    SpymemcachedNetworkAttributesGetter netAttributesGetter = new SpymemcachedNetworkAttributesGetter();
 
     INSTRUMENTER =
         Instrumenter.builder(
                 GlobalOpenTelemetry.get(),
                 INSTRUMENTATION_NAME,
                 DbClientSpanNameExtractor.create(dbAttributesGetter))
             .addAttributesExtractor(DbClientAttributesExtractor.create(dbAttributesGetter))
+            .addAttributesExtractor(ServerAttributesExtractor.create(netAttributesGetter))
             .addOperationMetrics(DbClientMetrics.get())
             .buildInstrumenter(SpanKindExtractor.alwaysClient());
   }