diff --git a/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/types/corpgroup/mappers/CorpGroupMapper.java b/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/types/corpgroup/mappers/CorpGroupMapper.java index 3b353be7490..c9b8bcafd80 100644 --- a/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/types/corpgroup/mappers/CorpGroupMapper.java +++ b/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/types/corpgroup/mappers/CorpGroupMapper.java @@ -1,8 +1,10 @@ package com.linkedin.datahub.graphql.types.corpgroup.mappers; +import com.linkedin.common.Ownership; import com.linkedin.data.DataMap; import com.linkedin.datahub.graphql.generated.CorpGroup; import com.linkedin.datahub.graphql.generated.EntityType; +import com.linkedin.datahub.graphql.types.common.mappers.OwnershipMapper; import com.linkedin.datahub.graphql.types.common.mappers.util.MappingHelper; import com.linkedin.datahub.graphql.types.mappers.ModelMapper; import com.linkedin.entity.EntityResponse; @@ -38,6 +40,7 @@ public class CorpGroupMapper implements ModelMapper { mappingHelper.mapToResult(CORP_GROUP_KEY_ASPECT_NAME, this::mapCorpGroupKey); mappingHelper.mapToResult(CORP_GROUP_INFO_ASPECT_NAME, this::mapCorpGroupInfo); mappingHelper.mapToResult(CORP_GROUP_EDITABLE_INFO_ASPECT_NAME, this::mapCorpGroupEditableInfo); + mappingHelper.mapToResult(OWNERSHIP_ASPECT_NAME, this::mapOwnership); return mappingHelper.getResult(); } @@ -55,4 +58,8 @@ public class CorpGroupMapper implements ModelMapper { private void mapCorpGroupEditableInfo(@Nonnull CorpGroup corpGroup, @Nonnull DataMap dataMap) { corpGroup.setEditableProperties(CorpGroupEditablePropertiesMapper.map(new CorpGroupEditableInfo(dataMap))); } + + private void mapOwnership(@Nonnull CorpGroup corpGroup, @Nonnull DataMap dataMap) { + corpGroup.setOwnership(OwnershipMapper.map(new Ownership(dataMap))); + } } diff --git a/datahub-graphql-core/src/main/resources/entity.graphql b/datahub-graphql-core/src/main/resources/entity.graphql index 910ab9b7b76..5940cd2c247 100644 --- a/datahub-graphql-core/src/main/resources/entity.graphql +++ b/datahub-graphql-core/src/main/resources/entity.graphql @@ -2402,6 +2402,11 @@ type CorpGroup implements Entity { """ name: String! + """ + Ownership metadata of the Corp Group + """ + ownership: Ownership + """ Additional read only properties about the group """