@finos/legend-graph 32.3.39 → 32.4.0

package/src/graph-manager/protocol/pure/v1/engine/V1_EngineHelper.ts

@@ -32,7 +32,10 @@ import {
   type QueryExecutionContextInfo,
   QueryExplicitExecutionContextInfo,
   QueryDataSpaceExecutionContextInfo,
-  QueryDataProductExecutionContext,
+  QueryDataProductNativeExecutionContext,
+  QueryDataProductModelAccessExecutionContext,
+  QueryDataProductNativeExecutionContextInfo,
+  QueryDataProductModelAccessExecutionContextInfo,
 } from '../../../../../graph-manager/action/query/Query.js';
 import {
   type V1_LightQuery,
@@ -41,7 +44,8 @@ import {
   V1_QueryExplicitExecutionContext,
   V1_QueryDataSpaceExecutionContext,
   type V1_QueryExecutionContext,
-  V1_QueryDataProductExecutionContext,
+  V1_DataProductNativeExecutionContext,
+  V1_DataProductModelAccessExecutionContext,
 } from './query/V1_Query.js';
 import type { PureModel } from '../../../../../graph/PureModel.js';
 import { DEPRECATED__ServiceTestResult } from '../../../../../graph-manager/action/service/DEPRECATED__ServiceTestResult.js';
@@ -202,6 +206,20 @@ export const V1_buildExecutionContext = (
     exec.dataSpacePath = protocolExecContext.dataSpacePath;
     exec.executionKey = protocolExecContext.executionKey;
     return exec;
+  } else if (
+    protocolExecContext instanceof V1_DataProductNativeExecutionContext
+  ) {
+    const exec = new QueryDataProductNativeExecutionContext();
+    exec.dataProductPath = protocolExecContext.dataProductPath;
+    exec.executionKey = protocolExecContext.executionKey;
+    return exec;
+  } else if (
+    protocolExecContext instanceof V1_DataProductModelAccessExecutionContext
+  ) {
+    const exec = new QueryDataProductModelAccessExecutionContext();
+    exec.dataProductPath = protocolExecContext.dataProductPath;
+    exec.accessPointGroupId = protocolExecContext.accessPointGroupId;
+    return exec;
   }
   throw new UnsupportedOperationError('Unsupported query execution context');
 };
@@ -233,6 +251,18 @@ export const V1_buildExecutionContextInfo = (
     exec.dataSpacePath = v1_execContext.dataSpacePath;
     exec.executionKey = v1_execContext.executionKey;
     return exec;
+  } else if (v1_execContext instanceof V1_DataProductNativeExecutionContext) {
+    const exec = new QueryDataProductNativeExecutionContextInfo();
+    exec.dataProductPath = v1_execContext.dataProductPath;
+    exec.executionKey = v1_execContext.executionKey;
+    return exec;
+  } else if (
+    v1_execContext instanceof V1_DataProductModelAccessExecutionContext
+  ) {
+    const exec = new QueryDataProductModelAccessExecutionContextInfo();
+    exec.dataProductPath = v1_execContext.dataProductPath;
+    exec.accessPointGroupId = v1_execContext.accessPointGroupId;
+    return exec;
   }
   throw new UnsupportedOperationError('Unsupported query execution context');
 };
@@ -340,11 +370,18 @@ export const V1_transformQueryExecutionContext = (
     protocol.dataSpacePath = execContext.dataSpacePath;
     protocol.executionKey = execContext.executionKey;
     return protocol;
-  } else if (execContext instanceof QueryDataProductExecutionContext) {
-    const protocol = new V1_QueryDataProductExecutionContext();
+  } else if (execContext instanceof QueryDataProductNativeExecutionContext) {
+    const protocol = new V1_DataProductNativeExecutionContext();
     protocol.dataProductPath = execContext.dataProductPath;
     protocol.executionKey = execContext.executionKey;
     return protocol;
+  } else if (
+    execContext instanceof QueryDataProductModelAccessExecutionContext
+  ) {
+    const protocol = new V1_DataProductModelAccessExecutionContext();
+    protocol.dataProductPath = execContext.dataProductPath;
+    protocol.accessPointGroupId = execContext.accessPointGroupId;
+    return protocol;
   }
   throw new UnsupportedOperationError('Unsupported query execution context');
 };

package/src/graph-manager/protocol/pure/v1/engine/query/V1_Query.ts

@@ -65,7 +65,8 @@ export class V1_QueryExecutionContext {}
 export enum V1_QueryExecutionContextType {
   QUERY_EXPLICIT_EXECUTION_CONTEXT = 'explicitExecutionContext',
   QUERY_DATASAPCE_EXECUTION_CONTEXT = 'dataSpaceExecutionContext',
-  QUERY_DATAPRODUCT_EXECUTION_CONTEXT = 'dataProductExecutionContext',
+  QUERY_DATAPRODUCT_NATIVE_EXECUTION_CONTEXT = 'dataProductNativeExecutionContext',
+  QUERY_DATAPRODUCT_MODEL_ACCESS_EXECUTION_CONTEXT = 'dataProductModelAccessExecutionContext',
 }
 
 export class V1_QueryExplicitExecutionContext extends V1_QueryExecutionContext {
@@ -104,17 +105,37 @@ export class V1_QueryDataSpaceExecutionContext extends V1_QueryExecutionContext
   );
 }
 
-export class V1_QueryDataProductExecutionContext extends V1_QueryExecutionContext {
+export abstract class V1_QueryDataProductExecutionContext extends V1_QueryExecutionContext {
   dataProductPath!: string;
-  executionKey: string | undefined;
+}
+
+export class V1_DataProductNativeExecutionContext extends V1_QueryDataProductExecutionContext {
+  executionKey!: string;
 
   static readonly serialization = new SerializationFactory(
-    createModelSchema(V1_QueryDataProductExecutionContext, {
+    createModelSchema(V1_DataProductNativeExecutionContext, {
       _type: usingConstantValueSchema(
-        V1_QueryExecutionContextType.QUERY_DATAPRODUCT_EXECUTION_CONTEXT,
+        V1_QueryExecutionContextType.QUERY_DATAPRODUCT_NATIVE_EXECUTION_CONTEXT,
       ),
       dataProductPath: primitive(),
-      executionKey: optional(primitive()),
+      executionKey: primitive(),
+    }),
+    {
+      deserializeNullAsUndefined: true,
+    },
+  );
+}
+
+export class V1_DataProductModelAccessExecutionContext extends V1_QueryDataProductExecutionContext {
+  accessPointGroupId!: string;
+
+  static readonly serialization = new SerializationFactory(
+    createModelSchema(V1_DataProductModelAccessExecutionContext, {
+      _type: usingConstantValueSchema(
+        V1_QueryExecutionContextType.QUERY_DATAPRODUCT_MODEL_ACCESS_EXECUTION_CONTEXT,
+      ),
+      dataProductPath: primitive(),
+      accessPointGroupId: primitive(),
     }),
     {
       deserializeNullAsUndefined: true,
@@ -136,9 +157,14 @@ export const V1_deserializeQueryExecutionContext = (
         V1_QueryDataSpaceExecutionContext.serialization.schema,
         json,
       );
-    case V1_QueryExecutionContextType.QUERY_DATAPRODUCT_EXECUTION_CONTEXT:
+    case V1_QueryExecutionContextType.QUERY_DATAPRODUCT_NATIVE_EXECUTION_CONTEXT:
       return deserialize(
-        V1_QueryDataProductExecutionContext.serialization.schema,
+        V1_DataProductNativeExecutionContext.serialization.schema,
+        json,
+      );
+    case V1_QueryExecutionContextType.QUERY_DATAPRODUCT_MODEL_ACCESS_EXECUTION_CONTEXT:
+      return deserialize(
+        V1_DataProductModelAccessExecutionContext.serialization.schema,
         json,
       );
     default: {
@@ -162,9 +188,14 @@ export const V1_serializeQueryExecutionContext = (
       V1_QueryDataSpaceExecutionContext.serialization.schema,
       protocol,
     );
-  } else if (protocol instanceof V1_QueryDataProductExecutionContext) {
+  } else if (protocol instanceof V1_DataProductNativeExecutionContext) {
+    return serialize(
+      V1_DataProductNativeExecutionContext.serialization.schema,
+      protocol,
+    );
+  } else if (protocol instanceof V1_DataProductModelAccessExecutionContext) {
     return serialize(
-      V1_QueryDataProductExecutionContext.serialization.schema,
+      V1_DataProductModelAccessExecutionContext.serialization.schema,
       protocol,
     );
   }

package/src/graph-manager/protocol/pure/v1/lakehouse/deploy/V1_DataProductArtifact.ts

@@ -45,6 +45,7 @@ import { V1_relationElementModelSchema } from '../../transformation/pureProtocol
 import { V1_pureModelContextDataPropSchema } from '../../transformation/pureProtocol/V1_PureProtocolSerialization.js';
 import type { V1_PureModelContextData } from '../../model/context/V1_PureModelContextData.js';
 import type { V1_Multiplicity } from '../../model/packageableElements/domain/V1_Multiplicity.js';
+import { V1_MappedEntity } from '../../engine/analytics/V1_MappingModelCoverageAnalysis.js';
 import { V1_multiplicityModelSchema } from '../../transformation/pureProtocol/serializationHelpers/V1_CoreSerializationHelper.js';
 
 export enum V1_DataProductTypeEnum {
@@ -275,11 +276,15 @@ export class V1_DiagramInfo {
 export class V1_MappingGenerationInfo {
   path!: string;
   model!: V1_PureModelContextData;
+  mappedEntities: V1_MappedEntity[] = [];
 
   static readonly serialization = new SerializationFactory(
     createModelSchema(V1_MappingGenerationInfo, {
       path: primitive(),
       model: V1_pureModelContextDataPropSchema,
+      mappedEntities: list(
+        usingModelSchema(V1_MappedEntity.serialization.schema),
+      ),
     }),
   );
 }

package/src/index.ts

@@ -729,6 +729,7 @@ export {
   type DataProductElement,
   NativeModelAccess,
   NativeModelExecutionContext,
+  DataProductAccessType,
   DataProductOperationalMetadata,
 } from './graph/metamodel/pure/dataProduct/DataProduct.js';
 export * from './graph-manager/action/changeDetection/DataProductObserveHelper.js';
@@ -900,6 +901,10 @@ export {
   EntityMappedProperty,
   EnumMappedProperty,
 } from './graph-manager/action/analytics/MappingModelCoverageAnalysis.js';
+export {
+  DataProductAnalysis,
+  DataProductAnalysisQueryResult,
+} from './graph-manager/action/analytics/data-product/DataProductAnalysis.js';
 export {
   DatasetSpecification,
   DatasetEntitlementReport,

package/tsconfig.json

@@ -380,6 +380,7 @@
     "./src/graph-manager/action/TEMPORARY__AbstractEngineConfig.ts",
     "./src/graph-manager/action/analytics/MappingModelCoverageAnalysis.ts",
     "./src/graph-manager/action/analytics/StoreEntitlementAnalysis.ts",
+    "./src/graph-manager/action/analytics/data-product/DataProductAnalysis.ts",
     "./src/graph-manager/action/changeDetection/CoreObserverHelper.ts",
     "./src/graph-manager/action/changeDetection/DSL_Data_ObserverHelper.ts",
     "./src/graph-manager/action/changeDetection/DSL_ExecutionEnvironment_ObseverHelper.ts",