From 6fe9d6faa55c4fe770b00390e6b64bf7ccb10fd7 Mon Sep 17 00:00:00 2001 From: Harshal Sheth Date: Mon, 2 Oct 2023 16:58:31 -0400 Subject: [PATCH] fix(graphql): support additional types in scrollAcrossEntities (#8891) --- .../linkedin/datahub/graphql/resolvers/EntityTypeMapper.java | 2 ++ 1 file changed, 2 insertions(+) diff --git a/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/resolvers/EntityTypeMapper.java b/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/resolvers/EntityTypeMapper.java index 3682b22825..b0f23e6317 100644 --- a/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/resolvers/EntityTypeMapper.java +++ b/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/resolvers/EntityTypeMapper.java @@ -17,6 +17,7 @@ public class EntityTypeMapper { ImmutableMap.builder() .put(EntityType.DATASET, "dataset") .put(EntityType.ROLE, "role") + .put(EntityType.ASSERTION, Constants.ASSERTION_ENTITY_NAME) .put(EntityType.CORP_USER, "corpuser") .put(EntityType.CORP_GROUP, "corpGroup") .put(EntityType.DATA_PLATFORM, "dataPlatform") @@ -25,6 +26,7 @@ public class EntityTypeMapper { .put(EntityType.TAG, "tag") .put(EntityType.DATA_FLOW, "dataFlow") .put(EntityType.DATA_JOB, "dataJob") + .put(EntityType.DATA_PROCESS_INSTANCE, Constants.DATA_PROCESS_INSTANCE_ENTITY_NAME) .put(EntityType.GLOSSARY_TERM, "glossaryTerm") .put(EntityType.GLOSSARY_NODE, "glossaryNode") .put(EntityType.MLMODEL, "mlModel")