Load data qualty status from a source table from the correct connection.

dqops · Sep 30, 2024 · 91a6c6f · 91a6c6f
1 parent cd8d350
commit 91a6c6f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/dqops/src/main/java/com/dqops/metadata/lineage/lineageservices/TableLineageServiceImpl.java b/dqops/src/main/java/com/dqops/metadata/lineage/lineageservices/TableLineageServiceImpl.java
@@ -140,6 +140,11 @@ public TableCurrentDataQualityStatusModel collectUpstreamLineage(
             Set<DomainConnectionTableKey> upstreamSourceTables = tableLineageEntry.getUpstreamSourceTables();
             for (DomainConnectionTableKey upstreamTableKey : upstreamSourceTables) {
                 TableCurrentDataQualityStatusModel upstreamOnlyQualityStatus = this.tableStatusCache.getCurrentTableStatus(upstreamTableKey, null);
+                if (upstreamOnlyQualityStatus == null) {
+                    upstreamOnlyQualityStatus = new TableCurrentDataQualityStatusModel() {{
+                        setTableExist(false);
+                    }};
+                }
 
                 if (visitedTables.contains(upstreamTableKey)) {
                     if (!onStackTables.contains(upstreamTableKey)) {

diff --git a/dqops/src/main/java/com/dqops/rest/controllers/DataLineageController.java b/dqops/src/main/java/com/dqops/rest/controllers/DataLineageController.java
@@ -221,7 +221,7 @@ public Mono<ResponseEntity<Flux<TableLineageSourceListModel>>> getTableSourceTab
                     listModel.setSourceTableDataQualityStatus(notFoundTableStatus);
                     return;
                 }
-                TableWrapper sourceTableWrapper = connectionWrapper.getTables().getByObjectName(
+                TableWrapper sourceTableWrapper = sourceConnectionWrapper.getTables().getByObjectName(
                         new PhysicalTableName(listModel.getSourceSchema(), listModel.getSourceTable()), true);
                 if (sourceTableWrapper == null) {
                     listModel.setSourceTableDataQualityStatus(notFoundTableStatus);