yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-10-17 20:08:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(entiy-client): handle null entityUrn case for restli (#11122)

Browse Source
This commit is contained in:
david-leifker 2024-08-08 13:54:14 -05:00 committed by GitHub
parent a4a887c866
commit 3d9a9541f1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
metadata-service/restli-client/src/main/java/com/linkedin/entity/client/RestliEntityClient.java
View File

@ -50,6 +50,7 @@ import com.linkedin.metadata.aspect.VersionedAspect;
import com.linkedin.metadata.browse.BrowseResult;
import com.linkedin.metadata.browse.BrowseResultV2;
import com.linkedin.metadata.graph.LineageDirection;
import com.linkedin.metadata.models.EntitySpec;
import com.linkedin.metadata.query.AutoCompleteResult;
import com.linkedin.metadata.query.LineageFlags;
import com.linkedin.metadata.query.ListResult;
@ -66,6 +67,7 @@ import com.linkedin.metadata.search.LineageScrollResult;
import com.linkedin.metadata.search.LineageSearchResult;
import com.linkedin.metadata.search.ScrollResult;
import com.linkedin.metadata.search.SearchResult;
import com.linkedin.metadata.utils.EntityKeyUtils;
import com.linkedin.mxe.MetadataChangeProposal;
import com.linkedin.mxe.MetadataChangeProposalArray;
import com.linkedin.mxe.PlatformEvent;
@ -1063,7 +1065,20 @@ public class RestliEntityClient extends BaseClient implements EntityClient {
String result =
sendClientRequest(requestBuilder, opContext.getSessionAuthentication()).getEntity();
return metadataChangeProposals.stream()
.map(proposal -> "success".equals(result) ? proposal.getEntityUrn().toString() : null)
.map(
proposal -> {
if ("success".equals(result)) {
if (proposal.getEntityUrn() != null) {
return proposal.getEntityUrn().toString();
} else {
EntitySpec entitySpec =
opContext.getEntityRegistry().getEntitySpec(proposal.getEntityType());
return EntityKeyUtils.getUrnFromProposal(proposal, entitySpec.getKeyAspectSpec())
.toString();
}
}
return null;
})
.collect(Collectors.toList());
}