yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-10-04 21:46:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(lineage) Fix possible null pointer exception in UpstreamLineagesMapper (#6147)

Browse Source
This commit is contained in:
Chris Collins 2022-10-07 10:38:38 -04:00 committed by GitHub
parent b8513ae492
commit 2e2ef536d8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/types/common/mappers/UpstreamLineagesMapper.java
View File

@ -30,14 +30,18 @@ public class UpstreamLineagesMapper {
for (FineGrainedLineage fineGrainedLineage : upstreamLineage.getFineGrainedLineages()) {
com.linkedin.datahub.graphql.generated.FineGrainedLineage resultEntry = new com.linkedin.datahub.graphql.generated.FineGrainedLineage();
resultEntry.setUpstreams(fineGrainedLineage.getUpstreams().stream()
.filter(entry -> entry.getEntityType().equals("schemaField"))
.map(entry -> mapDatasetSchemaField(entry)).collect(
Collectors.toList()));
resultEntry.setDownstreams(fineGrainedLineage.getDownstreams().stream()
.filter(entry -> entry.getEntityType().equals("schemaField"))
.map(entry -> mapDatasetSchemaField(entry)).collect(
Collectors.toList()));
if (fineGrainedLineage.hasUpstreams()) {
resultEntry.setUpstreams(fineGrainedLineage.getUpstreams().stream()
.filter(entry -> entry.getEntityType().equals("schemaField"))
.map(entry -> mapDatasetSchemaField(entry)).collect(
Collectors.toList()));
}
if (fineGrainedLineage.hasDownstreams()) {
resultEntry.setDownstreams(fineGrainedLineage.getDownstreams().stream()
.filter(entry -> entry.getEntityType().equals("schemaField"))
.map(entry -> mapDatasetSchemaField(entry)).collect(
Collectors.toList()));
}
result.add(resultEntry);
}
return result;