Merge pull request #2419 from atlanhq/nb/hasLineage

[stag] Nb/has lineage Logging
atlanhq · Oct 25, 2023 · c8735ce · c8735ce
2 parents de3c62a + e31dc25
commit c8735ce
Showing 1 changed file with 7 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()) {
@@ -2493,7 +2497,10 @@ private void repairHasLineageForAsset(AtlasHasLineageRequest request) {
             }
 
             if (!foundActiveRel) {
+                metricRecorder = RequestContext.get().startMetricRecord("repairHasLineageForRequiredAsset");
                 AtlasGraphUtilsV2.setEncodedProperty(assetVertex, HAS_LINEAGE, false);
+                LOG.info("repairHasLineage: repairHasLineageForAsset: Repaired {}", request.getAssetGuid());
+                RequestContext.get().endMetricRecord(metricRecorder);
             }
         }
     }