Merge pull request #2416 from atlanhq/nb/hasLineage

Logging
atlanhq · Oct 23, 2023 · dee70fe · dee70fe
2 parents d6d6714 + a7710b4
commit dee70fe
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityStoreV2.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityStoreV2.java
@@ -2478,10 +2478,14 @@ public void repairHasLineage(AtlasHasLineageRequests requests) throws AtlasBaseE
     private void repairHasLineageForAsset(AtlasHasLineageRequest request) {
         //only supports repairing scenario mentioned here - https://atlanhq.atlassian.net/browse/DG-128?focusedCommentId=20652
 
+        AtlasPerfMetrics.MetricRecorder metricRecorder = RequestContext.get().startMetricRecord("repairHasLineageForAssetGetById");
         AtlasVertex assetVertex = AtlasGraphUtilsV2.findByGuid(this.graph, request.getAssetGuid());
+        RequestContext.get().endMetricRecord(metricRecorder);
 
         if (getEntityHasLineage(assetVertex)) {
+            metricRecorder = RequestContext.get().startMetricRecord("repairHasLineageForAssetGetRelations");
             Iterator<AtlasEdge> lineageEdges = assetVertex.getEdges(AtlasEdgeDirection.BOTH, PROCESS_EDGE_LABELS).iterator();
+            RequestContext.get().endMetricRecord(metricRecorder);
             boolean foundActiveRel = false;
 
             while (lineageEdges.hasNext()) {