add processing time

atlanhq · Nov 28, 2024 · bcc0af4 · bcc0af4
1 parent 60fc7de
commit bcc0af4
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/...sitory/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java b/...sitory/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java
@@ -1014,6 +1014,8 @@ private AtlasEntityHeader mapVertexToAtlasEntityHeader(AtlasVertex entityVertex)
 
     private AtlasEntityHeader mapVertexToAtlasEntityHeader(AtlasVertex entityVertex, Set<String> attributes) throws AtlasBaseException {
         AtlasPerfMetrics.MetricRecorder metricRecorder = RequestContext.get().startMetricRecord("mapVertexToAtlasEntityHeader");
+        AtlasPerfMetrics.MetricRecorder attributeProcessingRecorder = null;
+        boolean metricEnded = false;
         AtlasEntityHeader ret = new AtlasEntityHeader();
         try {
             //pre-fetching the properties
@@ -1068,6 +1070,7 @@ private AtlasEntityHeader mapVertexToAtlasEntityHeader(AtlasVertex entityVertex,
                 }
 
                 if (CollectionUtils.isNotEmpty(attributes)) {
+                    attributeProcessingRecorder = RequestContext.get().startMetricRecord("attributeProcessingTime");
                     for (String attrName : attributes) {
                         AtlasAttribute attribute = entityType.getAttribute(attrName);
 
@@ -1090,11 +1093,17 @@ private AtlasEntityHeader mapVertexToAtlasEntityHeader(AtlasVertex entityVertex,
                         if (attrValue != null) {
                             ret.setAttribute(attrName, attrValue);
                         }
+
                     }
+                    RequestContext.get().endMetricRecord(attributeProcessingRecorder);
+                    metricEnded = true;
                 }
             }
         }
         finally {
+            if (!metricEnded) {
+                RequestContext.get().endMetricRecord(attributeProcessingRecorder);
+            }
             RequestContext.get().endMetricRecord(metricRecorder);
         }
         return ret;