From dd8e1f16cdb8b8d546e01c7c4ba658a2d54ee128 Mon Sep 17 00:00:00 2001
From: John Joyce <john@acryl.io>
Date: Tue, 22 Feb 2022 19:40:17 -0800
Subject: [PATCH] Adding group ownership (#4219)

---
 .../graphql/types/corpgroup/mappers/CorpGroupMapper.java   | 7 +++++++
 datahub-graphql-core/src/main/resources/entity.graphql     | 5 +++++
 2 files changed, 12 insertions(+)

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 3b353be749..c9b8bcafd8 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<EntityResponse, CorpGroup> {
         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<EntityResponse, CorpGroup> {
     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 910ab9b7b7..5940cd2c24 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
     """