Add refresh policy as attribute

Signed-off-by: Shreyansh Ray <rayshrey@amazon.com>
rayshrey · Oct 18, 2023 · a6b4241 · a6b4241
1 parent 21fa83b
commit a6b4241
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/server/src/main/java/org/opensearch/telemetry/tracing/AttributeNames.java b/server/src/main/java/org/opensearch/telemetry/tracing/AttributeNames.java
@@ -89,4 +89,9 @@ private AttributeNames() {
      * Node ID
      */
     public static final String NODE_ID = "node_id";
+
+    /**
+     * Refresh Policy
+     */
+    public static final String REFRESH_POLICY = "refresh_policy";
 }
diff --git a/server/src/main/java/org/opensearch/telemetry/tracing/SpanBuilder.java b/server/src/main/java/org/opensearch/telemetry/tracing/SpanBuilder.java
@@ -133,7 +133,6 @@ private static Attributes buildSpanAttributes(String action, Transport.Connectio
         return attributes;
     }
 
-<<<<<<< HEAD
     /**
      * Creates {@link SpanCreationContext} from Inbound Handler.
      * @param action action.
@@ -159,7 +158,8 @@ private static Attributes buildSpanAttributes(String action, TcpChannel tcpChann
     private static Attributes buildSpanAttributes(String nodeId, BulkShardRequest bulkShardRequest) {
         Attributes attributes = Attributes.create()
             .addAttribute(AttributeNames.NODE_ID, nodeId)
-            .addAttribute(AttributeNames.BULK_REQUEST_ITEMS, bulkShardRequest.items().length);
+            .addAttribute(AttributeNames.BULK_REQUEST_ITEMS, bulkShardRequest.items().length)
+            .addAttribute(AttributeNames.REFRESH_POLICY, bulkShardRequest.getRefreshPolicy().getValue());
         if (bulkShardRequest.shardId() != null) {
             attributes.addAttribute(AttributeNames.INDEX, bulkShardRequest.shardId().getIndexName())
                 .addAttribute(AttributeNames.SHARD_ID, bulkShardRequest.shardId().getId());