Merge branch 'master' into feat/PRD-789-new

Salman-Apptware · Jan 9, 2024 · 333dde3 · 333dde3
2 parents 735983d + 5bfd674
commit 333dde3
Show file tree

Hide file tree

Showing 136 changed files with 4,147 additions and 1,249 deletions.
diff --git a/build.gradle b/build.gradle
@@ -231,7 +231,8 @@ project.ext.externalDependency = [
     'common': 'commons-io:commons-io:2.7',
     'jline':'jline:jline:1.4.1',
     'jetbrains':' org.jetbrains.kotlin:kotlin-stdlib:1.6.0',
-    'annotationApi': 'javax.annotation:javax.annotation-api:1.3.2'
+    'annotationApi': 'javax.annotation:javax.annotation-api:1.3.2',
+    'classGraph': 'io.github.classgraph:classgraph:4.8.165',
 ]
 
 allprojects {

diff --git a/buildSrc/src/main/java/com/linkedin/metadata/aspect/plugins/config b/buildSrc/src/main/java/com/linkedin/metadata/aspect/plugins/config
@@ -0,0 +1 @@
+../../../../../../../../../entity-registry/src/main/java/com/linkedin/metadata/aspect/plugins/config
diff --git a/...ore/src/main/java/com/linkedin/datahub/graphql/resolvers/domain/CreateDomainResolver.java b/...ore/src/main/java/com/linkedin/datahub/graphql/resolvers/domain/CreateDomainResolver.java
@@ -2,7 +2,6 @@
 
 import static com.linkedin.datahub.graphql.resolvers.ResolverUtils.*;
 import static com.linkedin.datahub.graphql.resolvers.mutate.MutationUtils.*;
-import static com.linkedin.datahub.graphql.resolvers.mutate.util.OwnerUtils.*;
 import static com.linkedin.metadata.Constants.*;
 
 import com.linkedin.common.AuditStamp;
@@ -16,7 +15,6 @@
 import com.linkedin.datahub.graphql.exception.DataHubGraphQLException;
 import com.linkedin.datahub.graphql.generated.CreateDomainInput;
 import com.linkedin.datahub.graphql.generated.OwnerEntityType;
-import com.linkedin.datahub.graphql.generated.OwnershipType;
 import com.linkedin.datahub.graphql.resolvers.mutate.util.DomainUtils;
 import com.linkedin.datahub.graphql.resolvers.mutate.util.OwnerUtils;
 import com.linkedin.domain.DomainProperties;
@@ -100,14 +98,8 @@ public CompletableFuture<String> get(DataFetchingEnvironment environment) throws
 
             String domainUrn =
                 _entityClient.ingestProposal(proposal, context.getAuthentication(), false);
-            OwnershipType ownershipType = OwnershipType.TECHNICAL_OWNER;
-            if (!_entityService.exists(
-                UrnUtils.getUrn(mapOwnershipTypeToEntity(ownershipType.name())))) {
-              log.warn("Technical owner does not exist, defaulting to None ownership.");
-              ownershipType = OwnershipType.NONE;
-            }
             OwnerUtils.addCreatorAsOwner(
-                context, domainUrn, OwnerEntityType.CORP_USER, ownershipType, _entityService);
+                context, domainUrn, OwnerEntityType.CORP_USER, _entityService);
             return domainUrn;
           } catch (DataHubGraphQLException e) {
             throw e;

diff --git a/...main/java/com/linkedin/datahub/graphql/resolvers/glossary/CreateGlossaryNodeResolver.java b/...main/java/com/linkedin/datahub/graphql/resolvers/glossary/CreateGlossaryNodeResolver.java
@@ -2,7 +2,6 @@
 
 import static com.linkedin.datahub.graphql.resolvers.ResolverUtils.bindArgument;
 import static com.linkedin.datahub.graphql.resolvers.mutate.MutationUtils.*;
-import static com.linkedin.datahub.graphql.resolvers.mutate.util.OwnerUtils.*;
 import static com.linkedin.metadata.Constants.*;
 
 import com.linkedin.common.urn.GlossaryNodeUrn;
@@ -13,7 +12,6 @@
 import com.linkedin.datahub.graphql.exception.AuthorizationException;
 import com.linkedin.datahub.graphql.generated.CreateGlossaryEntityInput;
 import com.linkedin.datahub.graphql.generated.OwnerEntityType;
-import com.linkedin.datahub.graphql.generated.OwnershipType;
 import com.linkedin.datahub.graphql.resolvers.mutate.util.GlossaryUtils;
 import com.linkedin.datahub.graphql.resolvers.mutate.util.OwnerUtils;
 import com.linkedin.entity.client.EntityClient;
@@ -72,19 +70,8 @@ public CompletableFuture<String> get(DataFetchingEnvironment environment) throws
               String glossaryNodeUrn =
                   _entityClient.ingestProposal(proposal, context.getAuthentication(), false);
 
-              OwnershipType ownershipType = OwnershipType.TECHNICAL_OWNER;
-              if (!_entityService.exists(
-                  UrnUtils.getUrn(mapOwnershipTypeToEntity(ownershipType.name())))) {
-                log.warn("Technical owner does not exist, defaulting to None ownership.");
-                ownershipType = OwnershipType.NONE;
-              }
-
               OwnerUtils.addCreatorAsOwner(
-                  context,
-                  glossaryNodeUrn,
-                  OwnerEntityType.CORP_USER,
-                  ownershipType,
-                  _entityService);
+                  context, glossaryNodeUrn, OwnerEntityType.CORP_USER, _entityService);
               return glossaryNodeUrn;
             } catch (Exception e) {
               log.error(

diff --git a/...main/java/com/linkedin/datahub/graphql/resolvers/glossary/CreateGlossaryTermResolver.java b/...main/java/com/linkedin/datahub/graphql/resolvers/glossary/CreateGlossaryTermResolver.java
@@ -2,7 +2,6 @@
 
 import static com.linkedin.datahub.graphql.resolvers.ResolverUtils.bindArgument;
 import static com.linkedin.datahub.graphql.resolvers.mutate.MutationUtils.*;
-import static com.linkedin.datahub.graphql.resolvers.mutate.util.OwnerUtils.*;
 import static com.linkedin.metadata.Constants.*;
 
 import com.linkedin.common.urn.GlossaryNodeUrn;
@@ -14,7 +13,6 @@
 import com.linkedin.datahub.graphql.exception.AuthorizationException;
 import com.linkedin.datahub.graphql.generated.CreateGlossaryEntityInput;
 import com.linkedin.datahub.graphql.generated.OwnerEntityType;
-import com.linkedin.datahub.graphql.generated.OwnershipType;
 import com.linkedin.datahub.graphql.resolvers.mutate.util.GlossaryUtils;
 import com.linkedin.datahub.graphql.resolvers.mutate.util.OwnerUtils;
 import com.linkedin.entity.EntityResponse;
@@ -88,19 +86,9 @@ public CompletableFuture<String> get(DataFetchingEnvironment environment) throws
 
               String glossaryTermUrn =
                   _entityClient.ingestProposal(proposal, context.getAuthentication(), false);
-              OwnershipType ownershipType = OwnershipType.TECHNICAL_OWNER;
-              if (!_entityService.exists(
-                  UrnUtils.getUrn(mapOwnershipTypeToEntity(ownershipType.name())))) {
-                log.warn("Technical owner does not exist, defaulting to None ownership.");
-                ownershipType = OwnershipType.NONE;
-              }
 
               OwnerUtils.addCreatorAsOwner(
-                  context,
-                  glossaryTermUrn,
-                  OwnerEntityType.CORP_USER,
-                  ownershipType,
-                  _entityService);
+                  context, glossaryTermUrn, OwnerEntityType.CORP_USER, _entityService);
               return glossaryTermUrn;
             } catch (Exception e) {
               log.error(

diff --git a/...ql-core/src/main/java/com/linkedin/datahub/graphql/resolvers/mutate/AddOwnerResolver.java b/...ql-core/src/main/java/com/linkedin/datahub/graphql/resolvers/mutate/AddOwnerResolver.java
@@ -6,7 +6,6 @@
 import com.linkedin.common.urn.CorpuserUrn;
 import com.linkedin.common.urn.Urn;
 import com.linkedin.datahub.graphql.QueryContext;
-import com.linkedin.datahub.graphql.exception.AuthorizationException;
 import com.linkedin.datahub.graphql.generated.AddOwnerInput;
 import com.linkedin.datahub.graphql.generated.OwnerInput;
 import com.linkedin.datahub.graphql.generated.ResourceRefInput;
@@ -40,10 +39,7 @@ public CompletableFuture<Boolean> get(DataFetchingEnvironment environment) throw
     }
 
     OwnerInput ownerInput = ownerInputBuilder.build();
-    if (!OwnerUtils.isAuthorizedToUpdateOwners(environment.getContext(), targetUrn)) {
-      throw new AuthorizationException(
-          "Unauthorized to perform this action. Please contact your DataHub administrator.");
-    }
+    OwnerUtils.validateAuthorizedToUpdateOwners(environment.getContext(), targetUrn);
 
     return CompletableFuture.supplyAsync(
         () -> {

diff --git a/...l-core/src/main/java/com/linkedin/datahub/graphql/resolvers/mutate/AddOwnersResolver.java b/...l-core/src/main/java/com/linkedin/datahub/graphql/resolvers/mutate/AddOwnersResolver.java
@@ -6,7 +6,6 @@
 import com.linkedin.common.urn.CorpuserUrn;
 import com.linkedin.common.urn.Urn;
 import com.linkedin.datahub.graphql.QueryContext;
-import com.linkedin.datahub.graphql.exception.AuthorizationException;
 import com.linkedin.datahub.graphql.generated.AddOwnersInput;
 import com.linkedin.datahub.graphql.generated.OwnerInput;
 import com.linkedin.datahub.graphql.generated.ResourceRefInput;
@@ -34,10 +33,7 @@ public CompletableFuture<Boolean> get(DataFetchingEnvironment environment) throw
 
     return CompletableFuture.supplyAsync(
         () -> {
-          if (!OwnerUtils.isAuthorizedToUpdateOwners(environment.getContext(), targetUrn)) {
-            throw new AuthorizationException(
-                "Unauthorized to perform this action. Please contact your DataHub administrator.");
-          }
+          OwnerUtils.validateAuthorizedToUpdateOwners(environment.getContext(), targetUrn);
 
           OwnerUtils.validateAddOwnerInput(owners, targetUrn, _entityService);
           try {

diff --git a/...e/src/main/java/com/linkedin/datahub/graphql/resolvers/mutate/BatchAddOwnersResolver.java b/...e/src/main/java/com/linkedin/datahub/graphql/resolvers/mutate/BatchAddOwnersResolver.java
@@ -5,7 +5,6 @@
 import com.linkedin.common.urn.Urn;
 import com.linkedin.common.urn.UrnUtils;
 import com.linkedin.datahub.graphql.QueryContext;
-import com.linkedin.datahub.graphql.exception.AuthorizationException;
 import com.linkedin.datahub.graphql.generated.BatchAddOwnersInput;
 import com.linkedin.datahub.graphql.generated.OwnerInput;
 import com.linkedin.datahub.graphql.generated.ResourceRefInput;
@@ -74,10 +73,7 @@ private void validateInputResource(ResourceRefInput resource, QueryContext conte
           "Malformed input provided: owners cannot be applied to subresources.");
     }
 
-    if (!OwnerUtils.isAuthorizedToUpdateOwners(context, resourceUrn)) {
-      throw new AuthorizationException(
-          "Unauthorized to perform this action. Please contact your DataHub administrator.");
-    }
+    OwnerUtils.validateAuthorizedToUpdateOwners(context, resourceUrn);
     LabelUtils.validateResource(
         resourceUrn, resource.getSubResource(), resource.getSubResourceType(), _entityService);
   }

diff --git a/...rc/main/java/com/linkedin/datahub/graphql/resolvers/mutate/BatchRemoveOwnersResolver.java b/...rc/main/java/com/linkedin/datahub/graphql/resolvers/mutate/BatchRemoveOwnersResolver.java
@@ -5,7 +5,6 @@
 import com.linkedin.common.urn.Urn;
 import com.linkedin.common.urn.UrnUtils;
 import com.linkedin.datahub.graphql.QueryContext;
-import com.linkedin.datahub.graphql.exception.AuthorizationException;
 import com.linkedin.datahub.graphql.generated.BatchRemoveOwnersInput;
 import com.linkedin.datahub.graphql.generated.ResourceRefInput;
 import com.linkedin.datahub.graphql.resolvers.mutate.util.LabelUtils;
@@ -14,7 +13,6 @@
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
 import java.util.List;
-import java.util.Optional;
 import java.util.concurrent.CompletableFuture;
 import java.util.stream.Collectors;
 import lombok.RequiredArgsConstructor;
@@ -32,10 +30,10 @@ public CompletableFuture<Boolean> get(DataFetchingEnvironment environment) throw
         bindArgument(environment.getArgument("input"), BatchRemoveOwnersInput.class);
     final List<String> owners = input.getOwnerUrns();
     final List<ResourceRefInput> resources = input.getResources();
-    final Optional<Urn> maybeOwnershipTypeUrn =
+    final Urn ownershipTypeUrn =
         input.getOwnershipTypeUrn() == null
-            ? Optional.empty()
-            : Optional.of(Urn.createFromString(input.getOwnershipTypeUrn()));
+            ? null
+            : Urn.createFromString(input.getOwnershipTypeUrn());
     final QueryContext context = environment.getContext();
 
     return CompletableFuture.supplyAsync(
@@ -46,7 +44,7 @@ public CompletableFuture<Boolean> get(DataFetchingEnvironment environment) throw
 
           try {
             // Then execute the bulk remove
-            batchRemoveOwners(owners, maybeOwnershipTypeUrn, resources, context);
+            batchRemoveOwners(owners, ownershipTypeUrn, resources, context);
             return true;
           } catch (Exception e) {
             log.error(
@@ -71,24 +69,21 @@ private void validateInputResource(ResourceRefInput resource, QueryContext conte
           "Malformed input provided: owners cannot be removed from subresources.");
     }
 
-    if (!OwnerUtils.isAuthorizedToUpdateOwners(context, resourceUrn)) {
-      throw new AuthorizationException(
-          "Unauthorized to perform this action. Please contact your DataHub administrator.");
-    }
+    OwnerUtils.validateAuthorizedToUpdateOwners(context, resourceUrn);
     LabelUtils.validateResource(
         resourceUrn, resource.getSubResource(), resource.getSubResourceType(), _entityService);
   }
 
   private void batchRemoveOwners(
       List<String> ownerUrns,
-      Optional<Urn> maybeOwnershipTypeUrn,
+      Urn ownershipTypeUrn,
       List<ResourceRefInput> resources,
       QueryContext context) {
     log.debug("Batch removing owners. owners: {}, resources: {}", ownerUrns, resources);
     try {
       OwnerUtils.removeOwnersFromResources(
           ownerUrns.stream().map(UrnUtils::getUrn).collect(Collectors.toList()),
-          maybeOwnershipTypeUrn,
+          ownershipTypeUrn,
           resources,
           UrnUtils.getUrn(context.getActorUrn()),
           _entityService);

diff --git a/...core/src/main/java/com/linkedin/datahub/graphql/resolvers/mutate/RemoveOwnerResolver.java b/...core/src/main/java/com/linkedin/datahub/graphql/resolvers/mutate/RemoveOwnerResolver.java
@@ -6,14 +6,12 @@
 import com.linkedin.common.urn.CorpuserUrn;
 import com.linkedin.common.urn.Urn;
 import com.linkedin.datahub.graphql.QueryContext;
-import com.linkedin.datahub.graphql.exception.AuthorizationException;
 import com.linkedin.datahub.graphql.generated.RemoveOwnerInput;
 import com.linkedin.datahub.graphql.generated.ResourceRefInput;
 import com.linkedin.datahub.graphql.resolvers.mutate.util.OwnerUtils;
 import com.linkedin.metadata.entity.EntityService;
 import graphql.schema.DataFetcher;
 import graphql.schema.DataFetchingEnvironment;
-import java.util.Optional;
 import java.util.concurrent.CompletableFuture;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -31,15 +29,12 @@ public CompletableFuture<Boolean> get(DataFetchingEnvironment environment) throw
 
     Urn ownerUrn = Urn.createFromString(input.getOwnerUrn());
     Urn targetUrn = Urn.createFromString(input.getResourceUrn());
-    Optional<Urn> maybeOwnershipTypeUrn =
+    Urn ownershipTypeUrn =
         input.getOwnershipTypeUrn() == null
-            ? Optional.empty()
-            : Optional.of(Urn.createFromString(input.getOwnershipTypeUrn()));
+            ? null
+            : Urn.createFromString(input.getOwnershipTypeUrn());
 
-    if (!OwnerUtils.isAuthorizedToUpdateOwners(environment.getContext(), targetUrn)) {
-      throw new AuthorizationException(
-          "Unauthorized to perform this action. Please contact your DataHub administrator.");
-    }
+    OwnerUtils.validateAuthorizedToUpdateOwners(environment.getContext(), targetUrn);
 
     return CompletableFuture.supplyAsync(
         () -> {
@@ -50,7 +45,7 @@ public CompletableFuture<Boolean> get(DataFetchingEnvironment environment) throw
                     ((QueryContext) environment.getContext()).getActorUrn());
             OwnerUtils.removeOwnersFromResources(
                 ImmutableList.of(ownerUrn),
-                maybeOwnershipTypeUrn,
+                ownershipTypeUrn,
                 ImmutableList.of(new ResourceRefInput(input.getResourceUrn(), null, null)),
                 actor,
                 _entityService);
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		../../../../../../../../../entity-registry/src/main/java/com/linkedin/metadata/aspect/plugins/config