Merge pull request #3404 from atlanhq/DG-1720

DG-1720 Support Product lineage with current lineage APIs
atlanhq · Aug 21, 2024 · ca4fda5 · ca4fda5
2 parents fae5357 + f5726f6
commit ca4fda5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/repository/src/main/java/org/apache/atlas/discovery/EntityLineageService.java b/repository/src/main/java/org/apache/atlas/discovery/EntityLineageService.java
@@ -699,17 +699,17 @@ private void setHasDownstream(AtlasLineageOnDemandContext atlasLineageOnDemandCo
         List<AtlasEdge> filteredEdges = getFilteredAtlasEdges(inVertex, IN, lineageInputLabel, atlasLineageOnDemandContext);
         if (!filteredEdges.isEmpty()) {
             inLineageInfo.setHasDownstream(true);
-            inLineageInfo.setTotalOutputRelationsCount(filteredEdges.size());
         }
+        inLineageInfo.setTotalOutputRelationsCount(filteredEdges.size());
     }
 
     private void setHasUpstream(AtlasLineageOnDemandContext atlasLineageOnDemandContext, AtlasVertex outVertex, LineageInfoOnDemand outLineageInfo) {
         String lineageOutputLabel = RequestContext.get().getLineageOutputLabel();
         List<AtlasEdge> filteredEdges = getFilteredAtlasEdges(outVertex, IN, lineageOutputLabel, atlasLineageOnDemandContext);
         if (!filteredEdges.isEmpty()) {
             outLineageInfo.setHasUpstream(true);
-            outLineageInfo.setTotalInputRelationsCount(filteredEdges.size());
         }
+        outLineageInfo.setTotalInputRelationsCount(filteredEdges.size());
     }
 
     private List<AtlasEdge> getFilteredAtlasEdges(AtlasVertex outVertex, AtlasEdgeDirection direction, String processEdgeLabel, AtlasLineageOnDemandContext atlasLineageOnDemandContext) {