@finos/legend-graph 28.0.2 → 28.1.0

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

@@ -156,6 +156,7 @@ const V1_pureModelContextCompositeModelSchema = createModelSchema(
   {
     _type: usingConstantValueSchema(V1_PureModelContextType.COMPOSITE),
     serializer: usingModelSchema(V1_Protocol.serialization.schema),
+    // TODO: use `V1_pureModelContextDataPropSchema`
     data: object(V1_PureModelContextData),
     pointer: usingModelSchema(V1_pureModelContextPointerModelSchema),
   },
@@ -194,7 +195,16 @@ export const V1_serializePureModelContext = (
   if (pureModelContext instanceof V1_PureModelContextPointer) {
     return serialize(V1_pureModelContextPointerModelSchema, pureModelContext);
   } else if (pureModelContext instanceof V1_PureModelContextData) {
-    return V1_serializePureModelContextData(pureModelContext);
+    const rawPMCD = V1_serializePureModelContextData(pureModelContext);
+    if (pureModelContext.INTERNAL__rawDependencyEntities.length) {
+      rawPMCD.elements = [
+        ...(rawPMCD as { elements: PlainObject[] }).elements,
+        ...pureModelContext.INTERNAL__rawDependencyEntities.map(
+          (e) => e.content,
+        ),
+      ];
+    }
+    return rawPMCD;
   } else if (pureModelContext instanceof V1_PureModelContextComposite) {
     return serialize(V1_pureModelContextCompositeModelSchema, pureModelContext);
   } else if (pureModelContext instanceof V1_PureModelContextText) {
@@ -206,6 +216,12 @@ export const V1_serializePureModelContext = (
   );
 };
 
+export const V1_pureModelContextDataPropSchema = custom(
+  (val: V1_PureModelContextData) => V1_serializePureModelContextData(val),
+  (val: PlainObject<V1_PureModelContextData>) =>
+    V1_deserializePureModelContextData(val),
+);
+
 export const V1_pureModelContextPropSchema = custom(
   (val: V1_PureModelContext) => V1_serializePureModelContext(val),
   // TODO: we will populate this when we need to `deserialize`

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

@@ -61,6 +61,7 @@ import {
   V1_Other,
   V1_Decimal,
   V1_SemiStructured,
+  V1_Json,
 } from '../../../model/packageableElements/store/relational/model/V1_RelationalDataType.js';
 import { V1_View } from '../../../model/packageableElements/store/relational/model/V1_View.js';
 import { V1_FilterPointer } from '../../../model/packageableElements/store/relational/mapping/V1_FilterPointer.js';
@@ -132,6 +133,7 @@ enum V1_RelationalDataTypeType {
   ARRAY = 'Array',
   OTHER = 'Other',
   SEMISTRUCTURED = 'SemiStructured',
+  JSON = 'Json',
 }
 
 const V1_otherRelationalDataTypeModelSchema = createModelSchema(V1_Other, {
@@ -217,6 +219,10 @@ const V1_semiStructuredRelationalDataTypeModelSchema = createModelSchema(
   },
 );
 
+const V1_jsonRelationalDataTypeModelSchema = createModelSchema(V1_Json, {
+  _type: usingConstantValueSchema(V1_RelationalDataTypeType.JSON),
+});
+
 const V1_serializeColType = (
   protocol: V1_RelationalDataType,
 ): PlainObject<V1_RelationalDataType> => {
@@ -256,6 +262,8 @@ const V1_serializeColType = (
     return serialize(V1_otherRelationalDataTypeModelSchema, protocol);
   } else if (protocol instanceof V1_SemiStructured) {
     return serialize(V1_semiStructuredRelationalDataTypeModelSchema, protocol);
+  } else if (protocol instanceof V1_Json) {
+    return serialize(V1_jsonRelationalDataTypeModelSchema, protocol);
   }
   throw new UnsupportedOperationError(
     `Can't serialize relational data type`,
@@ -305,6 +313,8 @@ const V1_deserializeColType = (
       return deserialize(V1_otherRelationalDataTypeModelSchema, json);
     case V1_RelationalDataTypeType.SEMISTRUCTURED:
       return deserialize(V1_semiStructuredRelationalDataTypeModelSchema, json);
+    case V1_RelationalDataTypeType.JSON:
+      return deserialize(V1_jsonRelationalDataTypeModelSchema, json);
     default:
       throw new UnsupportedOperationError(
         `Can't deserialize relational column data type '${json._type}'`,

package/src/index.ts

@@ -262,8 +262,10 @@ export { V1_Engine } from './graph-manager/protocol/pure/v1/engine/V1_Engine.js'
 export {
   V1_PureModelContextType,
   V1_pureModelContextPropSchema,
+  V1_pureModelContextDataPropSchema,
   V1_entitiesToPureModelContextData,
   V1_deserializePureModelContextData,
+  V1_serializePureModelContextData,
 } from './graph-manager/protocol/pure/v1/transformation/pureProtocol/V1_PureProtocolSerialization.js';
 export {
   V1_propertyPointerModelSchema,
@@ -511,6 +513,8 @@ export {
   V1_DatasetEntitlementAccessRequestedReport,
   V1_DatasetEntitlementAccessApprovedReport,
   V1_DatasetEntitlementUnsupportedReport,
+  V1_deserializeDatasetSpecification,
+  V1_buildDatasetSpecification,
 } from './graph-manager/protocol/pure/v1/engine/analytics/V1_StoreEntitlementAnalysis.js';
 
 // ------------------------------------- TO BE MODULARIZED --------------------------------------------