feat(dashboards): add datasets field to DashboardInfo aspect

datahub-project · Jun 16, 2022 · 216129e · 216129e
1 parent f4674c3
commit 216129e
Show file tree

Hide file tree

Showing 8 changed files with 107 additions and 1 deletion.
diff --git a/datahub-web-react/src/app/entity/dashboard/DashboardEntity.tsx b/datahub-web-react/src/app/entity/dashboard/DashboardEntity.tsx
@@ -13,6 +13,7 @@ import { SidebarAboutSection } from '../shared/containers/profile/sidebar/Sideba
 import { SidebarTagsSection } from '../shared/containers/profile/sidebar/SidebarTagsSection';
 import { DocumentationTab } from '../shared/tabs/Documentation/DocumentationTab';
 import { DashboardChartsTab } from '../shared/tabs/Entity/DashboardChartsTab';
+import { DashboardDatasetsTab } from '../shared/tabs/Entity/DashboardDatasetsTab';
 import { PropertiesTab } from '../shared/tabs/Properties/PropertiesTab';
 import { GenericEntityProperties } from '../shared/types';
 import { DashboardPreview } from './preview/DashboardPreview';
@@ -87,10 +88,20 @@ export class DashboardEntity implements Entity<Dashboard> {
                     name: 'Charts',
                     component: DashboardChartsTab,
                     display: {
-                        visible: (_, _1) => true,
+                        visible: (_, dashboard: GetDashboardQuery) =>
+                            (dashboard?.dashboard?.charts?.total || 0) > 0 ||
+                            (dashboard?.dashboard?.datasets?.total || 0) === 0,
                         enabled: (_, dashboard: GetDashboardQuery) => (dashboard?.dashboard?.charts?.total || 0) > 0,
                     },
                 },
+                {
+                    name: 'Datasets',
+                    component: DashboardDatasetsTab,
+                    display: {
+                        visible: (_, dashboard: GetDashboardQuery) => (dashboard?.dashboard?.datasets?.total || 0) > 0,
+                        enabled: (_, dashboard: GetDashboardQuery) => (dashboard?.dashboard?.datasets?.total || 0) > 0,
+                    },
+                },
             ]}
             sidebarSections={[
                 {

diff --git a/datahub-web-react/src/app/entity/shared/tabs/Entity/DashboardDatasetsTab.tsx b/datahub-web-react/src/app/entity/shared/tabs/Entity/DashboardDatasetsTab.tsx
@@ -0,0 +1,19 @@
+import React from 'react';
+import { useBaseEntity } from '../../EntityContext';
+import { EntityType } from '../../../../../types.generated';
+import { EntityList } from './components/EntityList';
+import { useEntityRegistry } from '../../../../useEntityRegistry';
+
+export const DashboardDatasetsTab = () => {
+    const entity = useBaseEntity() as any;
+    const dashboard = entity && entity.dashboard;
+    const datasets = dashboard?.datasets?.relationships.map((relationship) => relationship.entity);
+    const entityRegistry = useEntityRegistry();
+    const totalDatasets = dashboard?.datasets?.total || 0;
+    const title = `Consumes ${totalDatasets} ${
+        totalDatasets === 1
+            ? entityRegistry.getEntityName(EntityType.Dataset)
+            : entityRegistry.getCollectionName(EntityType.Dataset)
+    }`;
+    return <EntityList title={title} type={EntityType.Dataset} entities={datasets || []} />;
+};
diff --git a/datahub-web-react/src/graphql/dashboard.graphql b/datahub-web-react/src/graphql/dashboard.graphql
@@ -7,6 +7,9 @@ query getDashboard($urn: String!) {
         charts: relationships(input: { types: ["Contains"], direction: OUTGOING, start: 0, count: 100 }) {
             ...fullRelationshipResults
         }
+        datasets: relationships(input: { types: ["Consumes"], direction: OUTGOING, start: 0, count: 100 }) {
+            ...fullRelationshipResults
+        }
         upstream: lineage(input: { direction: UPSTREAM, start: 0, count: 100 }) {
             ...fullLineageResults
         }

diff --git a/metadata-models/src/main/pegasus/com/linkedin/dashboard/DashboardInfo.pdl b/metadata-models/src/main/pegasus/com/linkedin/dashboard/DashboardInfo.pdl
@@ -3,6 +3,7 @@ namespace com.linkedin.dashboard
 import com.linkedin.common.AccessLevel
 import com.linkedin.common.ChangeAuditStamps
 import com.linkedin.common.ChartUrn
+import com.linkedin.common.DatasetUrn
 import com.linkedin.common.Time
 import com.linkedin.common.Url
 import com.linkedin.common.CustomProperties
@@ -47,6 +48,18 @@ record DashboardInfo includes CustomProperties, ExternalReference {
   }
   charts: array[ChartUrn] = [ ]
 
+  /**
+   * Datasets consumed by a dashboard
+   */
+  @Relationship = {
+    "/*": {
+      "name": "Consumes",
+      "entityTypes": [ "dataset" ],
+      "isLineage": true
+    }
+  }
+  datasets: array[DatasetUrn] = [ ]
+
   /**
    * Captures information about who created/last modified/deleted this dashboard and when
    */

diff --git a/metadata-service/restli-api/src/main/snapshot/com.linkedin.entity.aspects.snapshot.json b/metadata-service/restli-api/src/main/snapshot/com.linkedin.entity.aspects.snapshot.json
@@ -1200,6 +1200,21 @@
           "name" : "Contains"
         }
       }
+    }, {
+      "name" : "datasets",
+      "type" : {
+        "type" : "array",
+        "items" : "com.linkedin.common.DatasetUrn"
+      },
+      "doc" : "Datasets consumed by a dashboard",
+      "default" : [ ],
+      "Relationship" : {
+        "/*" : {
+          "entityTypes" : [ "dataset" ],
+          "isLineage" : true,
+          "name" : "Consumes"
+        }
+      }
     }, {
       "name" : "lastModified",
       "type" : "com.linkedin.common.ChangeAuditStamps",

diff --git a/metadata-service/restli-api/src/main/snapshot/com.linkedin.entity.entities.snapshot.json b/metadata-service/restli-api/src/main/snapshot/com.linkedin.entity.entities.snapshot.json
@@ -1217,6 +1217,21 @@
           "name" : "Contains"
         }
       }
+    }, {
+      "name" : "datasets",
+      "type" : {
+        "type" : "array",
+        "items" : "com.linkedin.common.DatasetUrn"
+      },
+      "doc" : "Datasets consumed by a dashboard",
+      "default" : [ ],
+      "Relationship" : {
+        "/*" : {
+          "entityTypes" : [ "dataset" ],
+          "isLineage" : true,
+          "name" : "Consumes"
+        }
+      }
     }, {
       "name" : "lastModified",
       "type" : "com.linkedin.common.ChangeAuditStamps",

diff --git a/metadata-service/restli-api/src/main/snapshot/com.linkedin.entity.runs.snapshot.json b/metadata-service/restli-api/src/main/snapshot/com.linkedin.entity.runs.snapshot.json
@@ -960,6 +960,21 @@
           "name" : "Contains"
         }
       }
+    }, {
+      "name" : "datasets",
+      "type" : {
+        "type" : "array",
+        "items" : "com.linkedin.common.DatasetUrn"
+      },
+      "doc" : "Datasets consumed by a dashboard",
+      "default" : [ ],
+      "Relationship" : {
+        "/*" : {
+          "entityTypes" : [ "dataset" ],
+          "isLineage" : true,
+          "name" : "Consumes"
+        }
+      }
     }, {
       "name" : "lastModified",
       "type" : "com.linkedin.common.ChangeAuditStamps",

diff --git a/metadata-service/restli-api/src/main/snapshot/com.linkedin.platform.platform.snapshot.json b/metadata-service/restli-api/src/main/snapshot/com.linkedin.platform.platform.snapshot.json
@@ -1217,6 +1217,21 @@
           "name" : "Contains"
         }
       }
+    }, {
+      "name" : "datasets",
+      "type" : {
+        "type" : "array",
+        "items" : "com.linkedin.common.DatasetUrn"
+      },
+      "doc" : "Datasets consumed by a dashboard",
+      "default" : [ ],
+      "Relationship" : {
+        "/*" : {
+          "entityTypes" : [ "dataset" ],
+          "isLineage" : true,
+          "name" : "Consumes"
+        }
+      }
     }, {
       "name" : "lastModified",
       "type" : "com.linkedin.common.ChangeAuditStamps",