add input dataset type (MarquezProject#2528)

Signed-off-by: sophiely <[email protected]>
Co-authored-by: Willy Lulciuc <[email protected]>

Author: sophiely

api/src/main/java/marquez/db/DatasetDao.java

@@ -87,7 +87,7 @@ LEFT JOIN (
                   JSONB_AGG(df.facet ORDER BY df.lineage_event_time ASC) AS facets
               FROM dataset_facets AS df
               WHERE df.facet IS NOT NULL AND
-               (df.type ILIKE 'dataset' OR df.type ILIKE 'unknown') AND
+               (df.type ILIKE 'dataset' OR df.type ILIKE 'unknown' OR df.type ILIKE 'input') AND
                 df.dataset_uuid = (SELECT uuid FROM datasets WHERE name = :datasetName AND namespace_name = :namespaceName)
               GROUP BY df.dataset_version_uuid
           ) f ON f.dataset_version_uuid = d.current_version_uuid
@@ -137,7 +137,7 @@ LEFT JOIN (
                   JSONB_AGG(df.facet ORDER BY df.lineage_event_time ASC) AS facets
                 FROM dataset_facets AS df
                 WHERE df.facet IS NOT NULL AND
-                 (df.type ILIKE 'dataset' OR df.type ILIKE 'unknown') AND
+                 (df.type ILIKE 'dataset' OR df.type ILIKE 'unknown' OR df.type ILIKE 'input') AND
                   df.dataset_uuid IN (SELECT uuid FROM datasets_view WHERE namespace_name = :namespaceName ORDER BY name LIMIT :limit OFFSET :offset)
               GROUP BY df.dataset_version_uuid
           ) f ON f.dataset_version_uuid = d.current_version_uuid

api/src/main/java/marquez/db/DatasetVersionDao.java

@@ -194,7 +194,7 @@ WITH selected_dataset_versions AS (
       ), selected_dataset_version_facets AS (
           SELECT dv.uuid, dv.dataset_name, dv.namespace_name, df.run_uuid, df.lineage_event_time, df.facet
           FROM selected_dataset_versions dv
-          LEFT JOIN dataset_facets_view df ON df.dataset_version_uuid = dv.uuid  AND (df.type ILIKE 'dataset' OR df.type ILIKE 'unknown')
+          LEFT JOIN dataset_facets_view df ON df.dataset_version_uuid = dv.uuid  AND (df.type ILIKE 'dataset' OR df.type ILIKE 'unknown' OR df.type ILIKE 'input')
       )
       SELECT d.type, d.name, d.physical_name, d.namespace_name, d.source_name, d.description, dv.lifecycle_state,\s
           dv.created_at, dv.version, dv.fields, dv.run_uuid AS createdByRunUuid, sv.schema_location,
@@ -264,7 +264,7 @@ LEFT JOIN (
 				facet as facets,lineage_event_time
 			FROM dataset_facets_view
 			WHERE
-				(type ILIKE 'dataset' OR type ILIKE 'unknown')
+				(type ILIKE 'dataset' OR type ILIKE 'unknown' OR type ILIKE 'input')
       	) f ON f.dataset_version_uuid = dv.uuid
       	WHERE dv.namespace_name = :namespaceName
             AND dv.dataset_name = :datasetName