Convert elasticsearch tests to java (#12300)

open-telemetry · Sep 24, 2024 · e4752ab · e4752ab
1 parent 6d94506
commit e4752ab
Show file tree

Hide file tree

Showing 44 changed files with 2,575 additions and 2,906 deletions.
diff --git a/.../elasticsearch/transport/v5_0/Elasticsearch5TransportExperimentalAttributesExtractor.java b/.../elasticsearch/transport/v5_0/Elasticsearch5TransportExperimentalAttributesExtractor.java
@@ -5,6 +5,7 @@
 
 package io.opentelemetry.javaagent.instrumentation.elasticsearch.transport.v5_0;
 
+import io.opentelemetry.api.common.AttributeKey;
 import io.opentelemetry.api.common.AttributesBuilder;
 import io.opentelemetry.context.Context;
 import io.opentelemetry.javaagent.instrumentation.elasticsearch.transport.ElasticTransportRequest;
@@ -14,6 +15,11 @@
 public class Elasticsearch5TransportExperimentalAttributesExtractor
     extends ElasticsearchTransportExperimentalAttributesExtractor {
 
+  private static final AttributeKey<String> ELASTICSEARCH_REQUEST_WRITE_TYPE =
+      AttributeKey.stringKey("elasticsearch.request.write.type");
+  private static final AttributeKey<String> ELASTICSEARCH_REQUEST_WRITE_ROUTING =
+      AttributeKey.stringKey("elasticsearch.request.write.routing");
+
   @Override
   public void onStart(
       AttributesBuilder attributes,
@@ -24,8 +30,8 @@ public void onStart(
     Object request = transportRequest.getRequest();
     if (request instanceof DocumentRequest) {
       DocumentRequest<?> req = (DocumentRequest<?>) request;
-      attributes.put("elasticsearch.request.write.type", req.type());
-      attributes.put("elasticsearch.request.write.routing", req.routing());
+      attributes.put(ELASTICSEARCH_REQUEST_WRITE_TYPE, req.type());
+      attributes.put(ELASTICSEARCH_REQUEST_WRITE_ROUTING, req.routing());
     }
   }
 }
diff --git a/...elasticsearch-transport-5.0/javaagent/src/test/groovy/Elasticsearch5NodeClientTest.groovy b/...elasticsearch-transport-5.0/javaagent/src/test/groovy/Elasticsearch5NodeClientTest.groovy