fix(graphql): Support querying Posts and Queries (#10502)

datahub-project · May 15, 2024 · 7c4faf4 · 7c4faf4
1 parent c55c12c
commit 7c4faf4
Showing 1 changed file with 25 additions and 23 deletions.
diff --git a/...ql-core/src/main/java/com/linkedin/datahub/graphql/types/entitytype/EntityTypeMapper.java b/...ql-core/src/main/java/com/linkedin/datahub/graphql/types/entitytype/EntityTypeMapper.java
@@ -15,37 +15,39 @@ public class EntityTypeMapper {
 
   static final Map<EntityType, String> ENTITY_TYPE_TO_NAME =
       ImmutableMap.<EntityType, String>builder()
-          .put(EntityType.DATASET, "dataset")
-          .put(EntityType.ROLE, "role")
-          .put(EntityType.CORP_USER, "corpuser")
-          .put(EntityType.CORP_GROUP, "corpGroup")
-          .put(EntityType.DATA_PLATFORM, "dataPlatform")
-          .put(EntityType.DASHBOARD, "dashboard")
-          .put(EntityType.CHART, "chart")
-          .put(EntityType.TAG, "tag")
-          .put(EntityType.DATA_FLOW, "dataFlow")
-          .put(EntityType.DATA_JOB, "dataJob")
+          .put(EntityType.DATASET, Constants.DATASET_ENTITY_NAME)
+          .put(EntityType.ROLE, Constants.ROLE_ENTITY_NAME)
+          .put(EntityType.CORP_USER, Constants.CORP_USER_ENTITY_NAME)
+          .put(EntityType.CORP_GROUP, Constants.CORP_GROUP_ENTITY_NAME)
+          .put(EntityType.DATA_PLATFORM, Constants.DATA_PLATFORM_ENTITY_NAME)
+          .put(EntityType.DASHBOARD, Constants.DASHBOARD_ENTITY_NAME)
+          .put(EntityType.CHART, Constants.CHART_ENTITY_NAME)
+          .put(EntityType.TAG, Constants.TAG_ENTITY_NAME)
+          .put(EntityType.DATA_FLOW, Constants.DATA_FLOW_ENTITY_NAME)
+          .put(EntityType.DATA_JOB, Constants.DATA_JOB_ENTITY_NAME)
           .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")
-          .put(EntityType.MLMODEL_GROUP, "mlModelGroup")
-          .put(EntityType.MLFEATURE_TABLE, "mlFeatureTable")
-          .put(EntityType.MLFEATURE, "mlFeature")
-          .put(EntityType.MLPRIMARY_KEY, "mlPrimaryKey")
-          .put(EntityType.CONTAINER, "container")
-          .put(EntityType.DOMAIN, "domain")
-          .put(EntityType.NOTEBOOK, "notebook")
-          .put(EntityType.DATA_PLATFORM_INSTANCE, "dataPlatformInstance")
-          .put(EntityType.TEST, "test")
+          .put(EntityType.GLOSSARY_TERM, Constants.GLOSSARY_TERM_ENTITY_NAME)
+          .put(EntityType.GLOSSARY_NODE, Constants.GLOSSARY_NODE_ENTITY_NAME)
+          .put(EntityType.MLMODEL, Constants.ML_MODEL_ENTITY_NAME)
+          .put(EntityType.MLMODEL_GROUP, Constants.ML_MODEL_GROUP_ENTITY_NAME)
+          .put(EntityType.MLFEATURE_TABLE, Constants.ML_FEATURE_TABLE_ENTITY_NAME)
+          .put(EntityType.MLFEATURE, Constants.ML_FEATURE_ENTITY_NAME)
+          .put(EntityType.MLPRIMARY_KEY, Constants.ML_PRIMARY_KEY_ENTITY_NAME)
+          .put(EntityType.CONTAINER, Constants.CONTAINER_ENTITY_NAME)
+          .put(EntityType.DOMAIN, Constants.DOMAIN_ENTITY_NAME)
+          .put(EntityType.NOTEBOOK, Constants.NOTEBOOK_ENTITY_NAME)
+          .put(EntityType.DATA_PLATFORM_INSTANCE, Constants.DATA_PLATFORM_INSTANCE_ENTITY_NAME)
+          .put(EntityType.TEST, Constants.TEST_ENTITY_NAME)
           .put(EntityType.ER_MODEL_RELATIONSHIP, Constants.ER_MODEL_RELATIONSHIP_ENTITY_NAME)
           .put(EntityType.DATAHUB_VIEW, Constants.DATAHUB_VIEW_ENTITY_NAME)
           .put(EntityType.DATA_PRODUCT, Constants.DATA_PRODUCT_ENTITY_NAME)
-          .put(EntityType.SCHEMA_FIELD, "schemaField")
+          .put(EntityType.SCHEMA_FIELD, Constants.SCHEMA_FIELD_ENTITY_NAME)
           .put(EntityType.STRUCTURED_PROPERTY, Constants.STRUCTURED_PROPERTY_ENTITY_NAME)
           .put(EntityType.ASSERTION, Constants.ASSERTION_ENTITY_NAME)
           .put(EntityType.RESTRICTED, Constants.RESTRICTED_ENTITY_NAME)
           .put(EntityType.BUSINESS_ATTRIBUTE, Constants.BUSINESS_ATTRIBUTE_ENTITY_NAME)
+          .put(EntityType.QUERY, Constants.QUERY_ENTITY_NAME)
+          .put(EntityType.POST, Constants.POST_ENTITY_NAME)
           .build();
 
   private static final Map<String, EntityType> ENTITY_NAME_TO_TYPE =