feat: Allow specifying Data Product URN via UI

datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/exception/DataHubDataFetcherExceptionHandler.java

@@ -24,23 +24,33 @@ public DataFetcherExceptionHandlerResult onException(
     DataHubGraphQLErrorCode errorCode = DataHubGraphQLErrorCode.SERVER_ERROR;
     String message = "An unknown error occurred.";
 
-    // note: make sure to access the true error message via `getCause()`
-    if (exception.getCause() instanceof IllegalArgumentException) {
+    IllegalArgumentException illException =
+        findFirstThrowableCauseOfClass(exception, IllegalArgumentException.class);
+    if (illException != null) {
+      log.error("Illegal Argument Exception");
       errorCode = DataHubGraphQLErrorCode.BAD_REQUEST;
-      message = exception.getCause().getMessage();
+      message = illException.getMessage();
     }
 
-    if (exception instanceof DataHubGraphQLException) {
-      errorCode = ((DataHubGraphQLException) exception).errorCode();
-      message = exception.getMessage();
-    }
-
-    if (exception.getCause() instanceof DataHubGraphQLException) {
-      errorCode = ((DataHubGraphQLException) exception.getCause()).errorCode();
-      message = exception.getCause().getMessage();
+    DataHubGraphQLException graphQLException =
+        findFirstThrowableCauseOfClass(exception, DataHubGraphQLException.class);
+    if (graphQLException != null) {
+      errorCode = graphQLException.errorCode();
+      message = graphQLException.getMessage();
     }
 
     DataHubGraphQLError error = new DataHubGraphQLError(message, path, sourceLocation, errorCode);
     return DataFetcherExceptionHandlerResult.newResult().error(error).build();
   }
+
+  <T extends Throwable> T findFirstThrowableCauseOfClass(Throwable throwable, Class<T> clazz) {
+    while (throwable != null && throwable.getCause() != null) {
+      if (throwable.getCause().getClass().equals(clazz)) {
+        return (T) throwable.getCause();
+      } else {
+        throwable = throwable.getCause();
+      }
+    }
+    return null;
+  }
 }

datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/resolvers/dataproduct/CreateDataProductResolver.java

@@ -47,6 +47,7 @@ public CompletableFuture<DataProduct> get(final DataFetchingEnvironment environm
           try {
             final Urn dataProductUrn =
                 _dataProductService.createDataProduct(
+                    input.getId(),
                     input.getProperties().getName(),
                     input.getProperties().getDescription(),
                     authentication);

datahub-graphql-core/src/main/resources/entity.graphql

@@ -11054,6 +11054,10 @@ input CreateDataProductInput {
     The primary key of the Domain
     """
     domainUrn: String!
+    """
+    An id for the DataProduct
+    """
+    id : String
 }
 
 """

datahub-web-react/src/app/entity/domain/DataProductsTab/CreateDataProductModal.tsx

@@ -32,6 +32,7 @@ export default function CreateDataProductModal({ domain, onCreateDataProduct, on
             variables: {
                 input: {
                     domainUrn: domain.urn,
+                    id: builderState.id || '',
                     properties: {
                         name: builderState.name,
                         description: builderState.description || undefined,
@@ -49,10 +50,10 @@ export default function CreateDataProductModal({ domain, onCreateDataProduct, on
                     onClose();
                 }
             })
-            .catch(() => {
+            .catch(( error ) => {
                 onClose();
                 message.destroy();
-                message.error({ content: 'Failed to create Data Product. An unexpected error occurred' });
+                message.error({ content: `Failed to create Data Product: ${error.message}.` });
             });
     }
 

datahub-web-react/src/app/entity/domain/DataProductsTab/DataProductAdvanceOption.tsx

@@ -0,0 +1,68 @@
+import React from "react";
+import { Collapse, Form, Input, Typography } from "antd";
+import styled from "styled-components";
+import { validateCustomUrnId } from '../../../shared/textUtil';
+import { DataProductBuilderFormProps } from "./types";
+
+
+const FormItem = styled(Form.Item)`
+    .ant-form-item-label {
+        padding-bottom: 2px;
+    }
+`;
+
+const FormItemWithMargin = styled(FormItem)`
+    margin-bottom: 16px;
+`;
+
+const FormItemNoMargin = styled(FormItem)`
+    margin-bottom: 0;
+`;
+
+const AdvancedLabel = styled(Typography.Text)`
+    color: #373d44;
+`;
+
+export function DataProductAdvanceOption({builderState, updateBuilderState }:DataProductBuilderFormProps){
+
+    function updateDataProductId(id: string) {
+        updateBuilderState({
+            ...builderState,
+            id,
+        });
+    }
+
+    return (
+        <Collapse ghost>
+            <Collapse.Panel header={<AdvancedLabel>Advanced Options</AdvancedLabel>} key="1">
+                <FormItemWithMargin
+                    label={<Typography.Text strong>Data Product Id</Typography.Text>}
+                    help="By default, a random UUID will be generated to uniquely identify this data product. If
+                        you'd like to provide a custom id instead to more easily keep track of this data product,
+                        you may provide it here. Be careful, you cannot easily change the data product id after
+                        creation."
+                >
+                    <FormItemNoMargin
+                        rules={[
+                            () => ({
+                                validator(_, value) {
+                                    if (value && validateCustomUrnId(value)) {
+                                        return Promise.resolve();
+                                    }
+                                    return Promise.reject(new Error('Please enter a valid Data product id'));
+                                },
+                            }),
+                        ]}
+                    >
+                        <Input 
+                            data-testid="data-product-id" 
+                            placeholder="engineering" 
+                            value={builderState.id}
+                            onChange={(e) => updateDataProductId(e.target.value)}
+                        />
+                    </FormItemNoMargin>
+                </FormItemWithMargin>
+            </Collapse.Panel>
+        </Collapse>
+    )
+}

datahub-web-react/src/app/entity/domain/DataProductsTab/DataProductBuilderForm.tsx

@@ -3,18 +3,14 @@ import React from 'react';
 import styled from 'styled-components';
 import { Editor as MarkdownEditor } from '../../shared/tabs/Documentation/components/editor/Editor';
 import { ANTD_GRAY } from '../../shared/constants';
-import { DataProductBuilderState } from './types';
+import { DataProductBuilderFormProps } from './types';
+import { DataProductAdvanceOption } from './DataProductAdvanceOption';
 
 const StyledEditor = styled(MarkdownEditor)`
     border: 1px solid ${ANTD_GRAY[4]};
 `;
 
-type Props = {
-    builderState: DataProductBuilderState;
-    updateBuilderState: (newState: DataProductBuilderState) => void;
-};
-
-export default function DataProductBuilderForm({ builderState, updateBuilderState }: Props) {
+export default function DataProductBuilderForm({ builderState, updateBuilderState }: DataProductBuilderFormProps) {
     function updateName(name: string) {
         updateBuilderState({
             ...builderState,
@@ -47,6 +43,7 @@ export default function DataProductBuilderForm({ builderState, updateBuilderStat
             <Form.Item label={<Typography.Text strong>Description</Typography.Text>}>
                 <StyledEditor doNotFocus content={builderState.description} onChange={updateDescription} />
             </Form.Item>
+            <DataProductAdvanceOption builderState={builderState} updateBuilderState={updateBuilderState}/>
         </Form>
     );
 }

datahub-web-react/src/app/entity/domain/DataProductsTab/types.ts

@@ -1,4 +1,10 @@
 export type DataProductBuilderState = {
     name: string;
+    id?: string;
     description?: string;
 };
+
+export type DataProductBuilderFormProps = {
+    builderState: DataProductBuilderState;
+    updateBuilderState: (newState: DataProductBuilderState) => void;
+};

metadata-service/services/src/main/java/com/linkedin/metadata/service/DataProductService.java

@@ -1,5 +1,7 @@
 package com.linkedin.metadata.service;
 
+import static com.linkedin.metadata.Constants.DATA_PRODUCT_ENTITY_NAME;
+
 import com.datahub.authentication.Authentication;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableSet;
@@ -22,6 +24,7 @@
 import com.linkedin.metadata.graph.GraphClient;
 import com.linkedin.metadata.query.filter.RelationshipDirection;
 import com.linkedin.metadata.utils.EntityKeyUtils;
+import com.linkedin.r2.RemoteInvocationException;
 import java.util.List;
 import java.util.Objects;
 import java.util.UUID;
@@ -58,11 +61,26 @@ public DataProductService(@Nonnull EntityClient entityClient, @Nonnull GraphClie
    * @return the urn of the newly created DataProduct
    */
   public Urn createDataProduct(
-      @Nullable String name, @Nullable String description, @Nonnull Authentication authentication) {
+      @Nullable String id,
+      @Nullable String name,
+      @Nullable String description,
+      @Nonnull Authentication authentication) {
 
     // 1. Generate a unique id for the new DataProduct.
     final DataProductKey key = new DataProductKey();
-    key.setId(UUID.randomUUID().toString());
+    if (id != null && !id.isBlank()) {
+      key.setId(id);
+    } else {
+      key.setId(UUID.randomUUID().toString());
+    }
+    try {
+      if (_entityClient.exists(
+          EntityKeyUtils.convertEntityKeyToUrn(key, DATA_PRODUCT_ENTITY_NAME), authentication)) {
+        throw new IllegalArgumentException("This Data product already exists!");
+      }
+    } catch (RemoteInvocationException e) {
+      throw new RuntimeException("Unable to check for existence of Data Product!");
+    }
 
     // 2. Create a new instance of DataProductProperties
     final DataProductProperties properties = new DataProductProperties();