@finos/legend-graph 32.3.29 → 32.3.31

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

@@ -32,6 +32,7 @@ import {
   type QueryExecutionContextInfo,
   QueryExplicitExecutionContextInfo,
   QueryDataSpaceExecutionContextInfo,
+  QueryDataProductExecutionContext,
 } from '../../../../../graph-manager/action/query/Query.js';
 import {
   type V1_LightQuery,
@@ -40,6 +41,7 @@ import {
   V1_QueryExplicitExecutionContext,
   V1_QueryDataSpaceExecutionContext,
   type V1_QueryExecutionContext,
+  V1_QueryDataProductExecutionContext,
 } from './query/V1_Query.js';
 import type { PureModel } from '../../../../../graph/PureModel.js';
 import { DEPRECATED__ServiceTestResult } from '../../../../../graph-manager/action/service/DEPRECATED__ServiceTestResult.js';
@@ -124,6 +126,7 @@ export const V1_buildLightQuery = (
     `Query 'artifactId' field is missing`,
   );
   metamodel.owner = protocol.owner;
+  metamodel.description = protocol.description;
   metamodel.lastUpdatedAt = protocol.lastUpdatedAt;
   metamodel.createdAt = protocol.createdAt;
   metamodel.lastOpenAt = protocol.lastOpenAt;
@@ -337,6 +340,11 @@ export const V1_transformQueryExecutionContext = (
     protocol.dataSpacePath = execContext.dataSpacePath;
     protocol.executionKey = execContext.executionKey;
     return protocol;
+  } else if (execContext instanceof QueryDataProductExecutionContext) {
+    const protocol = new V1_QueryDataProductExecutionContext();
+    protocol.dataProductPath = execContext.dataProductPath;
+    protocol.executionKey = execContext.executionKey;
+    return protocol;
   }
   throw new UnsupportedOperationError('Unsupported query execution context');
 };

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

@@ -65,6 +65,7 @@ export class V1_QueryExecutionContext {}
 export enum V1_QueryExecutionContextType {
   QUERY_EXPLICIT_EXECUTION_CONTEXT = 'explicitExecutionContext',
   QUERY_DATASAPCE_EXECUTION_CONTEXT = 'dataSpaceExecutionContext',
+  QUERY_DATAPRODUCT_EXECUTION_CONTEXT = 'dataProductExecutionContext',
 }
 
 export class V1_QueryExplicitExecutionContext extends V1_QueryExecutionContext {
@@ -103,6 +104,24 @@ export class V1_QueryDataSpaceExecutionContext extends V1_QueryExecutionContext
   );
 }
 
+export class V1_QueryDataProductExecutionContext extends V1_QueryExecutionContext {
+  dataProductPath!: string;
+  executionKey: string | undefined;
+
+  static readonly serialization = new SerializationFactory(
+    createModelSchema(V1_QueryDataProductExecutionContext, {
+      _type: usingConstantValueSchema(
+        V1_QueryExecutionContextType.QUERY_DATAPRODUCT_EXECUTION_CONTEXT,
+      ),
+      dataProductPath: primitive(),
+      executionKey: optional(primitive()),
+    }),
+    {
+      deserializeNullAsUndefined: true,
+    },
+  );
+}
+
 export const V1_deserializeQueryExecutionContext = (
   json: PlainObject<V1_QueryExecutionContext>,
 ): V1_QueryExecutionContext => {
@@ -117,6 +136,11 @@ export const V1_deserializeQueryExecutionContext = (
         V1_QueryDataSpaceExecutionContext.serialization.schema,
         json,
       );
+    case V1_QueryExecutionContextType.QUERY_DATAPRODUCT_EXECUTION_CONTEXT:
+      return deserialize(
+        V1_QueryDataProductExecutionContext.serialization.schema,
+        json,
+      );
     default: {
       throw new UnsupportedOperationError(
         `Can't deserialize authentication strategy of type '${json._type}': no compatible deserializer available from plugins`,
@@ -138,6 +162,11 @@ export const V1_serializeQueryExecutionContext = (
       V1_QueryDataSpaceExecutionContext.serialization.schema,
       protocol,
     );
+  } else if (protocol instanceof V1_QueryDataProductExecutionContext) {
+    return serialize(
+      V1_QueryDataProductExecutionContext.serialization.schema,
+      protocol,
+    );
   }
   throw new UnsupportedOperationError(
     `Can't serialize authentication strategy: no compatible serializer available from plugins`,
@@ -217,6 +246,7 @@ export class V1_LightQuery {
   artifactId!: string;
   versionId!: string;
   originalVersionId?: string | undefined;
+  description?: string | undefined;
   lastUpdatedAt?: number | undefined;
   createdAt?: number | undefined;
   lastOpenAt?: number | undefined;
@@ -231,6 +261,7 @@ export class V1_LightQuery {
       lastOpenAt: optional(primitive()),
       name: primitive(),
       owner: optional(primitive()),
+      description: optional(primitive()),
       versionId: primitive(),
       originalVersionId: optional(primitive()),
     }),

package/src/graph-manager/protocol/pure/v1/model/packageableElements/function/test/V1_FunctionTest.ts

@@ -48,6 +48,7 @@ export class V1_FunctionTest extends V1_AtomicTest {
       this.id,
       this.doc ?? '',
       this.parameters ? hashArray(this.parameters) : '',
+      hashArray(this.assertions),
     ]);
   }
 }

package/src/graph-manager/protocol/pure/v1/transformation/pureGraph/from/V1_DataProductTransformer.ts

@@ -167,7 +167,7 @@ export const V1_transformDataProduct = (
       );
     }
     nativeModelAccess.defaultExecutionContext =
-      metamodelNativeModelAccess.defaultExecutionContext;
+      metamodelNativeModelAccess.defaultExecutionContext.key;
     nativeModelAccess.nativeModelExecutionContexts =
       metamodelNativeModelAccess.nativeModelExecutionContexts.map(
         (executionContext) => {

package/src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/helpers/V1_DataProductBuilder.ts

@@ -310,12 +310,16 @@ export const V1_buildNativeModelAccess = (
   context: V1_GraphBuilderContext,
 ): NativeModelAccess => {
   const metamodelNativeModelAccess = new NativeModelAccess();
-  metamodelNativeModelAccess.defaultExecutionContext =
-    nativeModelAccess.defaultExecutionContext;
   metamodelNativeModelAccess.nativeModelExecutionContexts =
     nativeModelAccess.nativeModelExecutionContexts.map((executionContext) =>
       V1_buildNativeModelExecutionContext(executionContext, context),
     );
+  metamodelNativeModelAccess.defaultExecutionContext = guaranteeNonNullable(
+    metamodelNativeModelAccess.nativeModelExecutionContexts.find(
+      (e) => e.key === nativeModelAccess.defaultExecutionContext,
+    ),
+    `Default execution context ${nativeModelAccess.defaultExecutionContext} not found among native model execution contexts`,
+  );
   if (nativeModelAccess.sampleQueries) {
     metamodelNativeModelAccess.sampleQueries =
       nativeModelAccess.sampleQueries.map((sampleQuery) =>

package/src/index.ts

@@ -399,6 +399,7 @@ export {
 } from './graph-manager/protocol/pure/v1/lakehouse/subscriptions/V1_ConsumerSubscriptions.js';
 export {
   V1_DataProductArtifact,
+  V1_NativeModelExecutionContextInfo,
   V1_AccessPointImplementation,
   V1_AccessPointGroupInfo,
   V1_SampleQueryInfo,
@@ -781,8 +782,6 @@ export {
   SupportInfo,
   ModelAccessPointGroup,
   InLineSampleQuery,
-  NativeModelAccess,
-  NativeModelExecutionContext,
   PackageableElementSampleQuery,
   SampleQuery,
   DataProductRuntimeInfo,
@@ -794,6 +793,8 @@ export {
   Expertise,
   ExternalDataProductType,
   type DataProductElement,
+  NativeModelAccess,
+  NativeModelExecutionContext,
   DataProductOperationalMetadata,
 } from './graph/metamodel/pure/dataProduct/DataProduct.js';
 export * from './graph-manager/action/changeDetection/DataProductObserveHelper.js';
@@ -972,6 +973,7 @@ export {
   DatasetEntitlementAccessApprovedReport,
   DatasetEntitlementUnsupportedReport,
 } from './graph-manager/action/analytics/StoreEntitlementAnalysis.js';
+export * from './graph-manager/helpers/DataProductHelper.js';
 export {
   FunctionAnalysisInfo,
   FunctionAnalysisParameterInfo,

package/tsconfig.json

@@ -435,6 +435,7 @@
     "./src/graph-manager/extensions/Testable_PureGraphManagerPlugin_Extension.ts",
     "./src/graph-manager/helpers/DSL_Data_GraphManagerHelper.ts",
     "./src/graph-manager/helpers/DSL_ExternalFormat_GraphManagerHelper.ts",
+    "./src/graph-manager/helpers/DataProductHelper.ts",
     "./src/graph-manager/helpers/ValueSpecificationGraphManagerHelper.ts",
     "./src/graph-manager/protocol/pure/PureGraphManagerBuilder.ts",
     "./src/graph-manager/protocol/pure/PureProtocolProcessorPlugin.ts",