Modify SpanBuilder

Signed-off-by: Shreyansh Ray <rayshrey@amazon.com>
rayshrey · Oct 19, 2023 · bbb4a42 · bbb4a42
1 parent 9feb78c
commit bbb4a42
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/server/src/main/java/org/opensearch/telemetry/tracing/SpanBuilder.java b/server/src/main/java/org/opensearch/telemetry/tracing/SpanBuilder.java
@@ -157,14 +157,14 @@ private static Attributes buildSpanAttributes(String action, TcpChannel tcpChann
     }
 
     private static Attributes buildSpanAttributes(String nodeId, ReplicatedWriteRequest request) {
-        Attributes attributes = Attributes.create().addAttribute(AttributeNames.NODE_ID, nodeId);
+        Attributes attributes = Attributes.create().addAttribute(AttributeNames.NODE_ID, nodeId)
+            .addAttribute(AttributeNames.REFRESH_POLICY, request.getRefreshPolicy().getValue());
+        if (request.shardId() != null) {
+            attributes.addAttribute(AttributeNames.INDEX, request.shardId().getIndexName())
+                .addAttribute(AttributeNames.SHARD_ID, request.shardId().getId());
+        }
         if (request instanceof BulkShardRequest) {
-            attributes.addAttribute(AttributeNames.BULK_REQUEST_ITEMS, ((BulkShardRequest) request).items().length)
-                .addAttribute(AttributeNames.REFRESH_POLICY, request.getRefreshPolicy().getValue());
-            if (request.shardId() != null) {
-                attributes.addAttribute(AttributeNames.INDEX, request.shardId().getIndexName())
-                    .addAttribute(AttributeNames.SHARD_ID, request.shardId().getId());
-            }
+            attributes.addAttribute(AttributeNames.BULK_REQUEST_ITEMS, ((BulkShardRequest) request).items().length);
         }
         return attributes;
     }