@finos/legend-graph 32.1.40 → 32.1.42

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

@@ -45,7 +45,24 @@ export const V1_buildEngineRuntime = (
   if (runtime instanceof V1_SingleConnectionEngineRuntime) {
     runtimeValue = new SingleConnectionRuntime();
   } else if (runtime instanceof V1_LakehouseRuntime) {
-    runtimeValue = new LakehouseRuntime(runtime.ingestEnv, runtime.warehouse);
+    let conPointer: ConnectionPointer | undefined;
+    if (runtime.connectionPointer) {
+      const connection = V1_buildConnection(
+        runtime.connectionPointer,
+        context,
+        undefined,
+      );
+      conPointer = guaranteeType(
+        connection,
+        ConnectionPointer,
+        `Connection in Connection store expected to be connection pointer`,
+      );
+    }
+    runtimeValue = new LakehouseRuntime(
+      runtime.environment,
+      runtime.warehouse,
+      conPointer,
+    );
   } else {
     runtimeValue = new EngineRuntime();
   }

package/src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_DatabaseSerializationHelper.ts

@@ -575,7 +575,7 @@ const V1_setupRelationalDatabaseConnectionModelSchema = (
       (val) => V1_deserializeDatasourceSpecification(val, plugins),
     ),
     localMode: optional(primitive()),
-    store: alias('element', primitive()),
+    store: optional(alias('element', primitive())),
     queryTimeOutInSeconds: optional(primitive()),
     quoteIdentifiers: optional(primitive()),
     timeZone: optional(primitive()),

package/src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_RuntimeSerializationHelper.ts

@@ -22,6 +22,7 @@ import {
   serialize,
   deserialize,
   object,
+  optional,
 } from 'serializr';
 import {
   type PlainObject,
@@ -56,7 +57,7 @@ export enum V1_RuntimeType {
   LEGACY_RUNTIME = 'legacyRuntime',
   ENGINE_RUNTIME = 'engineRuntime',
   SINGLE_ENGINE_RUNTIME = 'localEngineRuntime',
-  LAKEHOUSE_RUNTIME = 'LakeRuntime',
+  LAKEHOUSE_RUNTIME = 'LakehouseRuntime',
 }
 
 export const V1_runtimePointerModelSchema = createModelSchema(
@@ -119,8 +120,11 @@ export const V1_setupEngineRuntimeSerialization = (
     connectionStores: list(object(V1_ConnectionStores)),
     connections: list(object(V1_StoreConnections)),
     mappings: list(usingModelSchema(V1_packageableElementPointerModelSchema)),
-    ingestEnv: primitive(),
-    warehouse: primitive(),
+    environment: optional(primitive()),
+    warehouse: optional(primitive()),
+    connectionPointer: optional(
+      usingModelSchema(V1_connectionPointerModelSchema),
+    ),
   });
 };
 

package/src/index.ts

@@ -322,6 +322,7 @@ export {
   V1_IngestEnvironment,
   V1_AWSSnowflakeIngestEnvironment,
 } from './graph-manager/protocol/pure/v1/lakehouse/ingest/V1_LakehouseIngestEnvironment.js';
+export { V1_observe_DataContract } from './graph-manager/protocol/pure/v1/lakehouse/helpers/V1_ConsumerEntitlementsObserver.js';
 export {
   V1_CreateSubscriptionInputModelSchema,
   V1_dataSubscriptionModelSchema,

package/tsconfig.json

@@ -499,6 +499,7 @@
     "./src/graph-manager/protocol/pure/v1/lakehouse/entitlements/V1_ConsumerEntitlements.ts",
     "./src/graph-manager/protocol/pure/v1/lakehouse/entitlements/V1_CoreEntitlements.ts",
     "./src/graph-manager/protocol/pure/v1/lakehouse/entitlements/V1_EntitlementsTasks.ts",
+    "./src/graph-manager/protocol/pure/v1/lakehouse/helpers/V1_ConsumerEntitlementsObserver.ts",
     "./src/graph-manager/protocol/pure/v1/lakehouse/ingest/V1_LakehouseIngestEnvironment.ts",
     "./src/graph-manager/protocol/pure/v1/lakehouse/platform/V1_LakehouseDiscovery.ts",
     "./src/graph-manager/protocol/pure/v1/lakehouse/subscriptions/V1_ConsumerSubscriptions.ts",