@finos/legend-graph 32.1.12 → 32.1.14

package/src/graph-manager/action/changeDetection/STO_Relational_ObserverHelper.ts

@@ -132,6 +132,10 @@ import type { DEPRECATED__RelationalInputData } from '../../../graph/metamodel/p
 import { INTERNAL__UnknownPostProcessor } from '../../../graph/metamodel/pure/packageableElements/store/relational/connection/postprocessor/INTERNAL__UnknownPostProcessor.js';
 import { INTERNAL__UnknownDatasourceSpecification } from '../../../graph/metamodel/pure/packageableElements/store/relational/connection/INTERNAL__UnknownDatasourceSpecification.js';
 import { INTERNAL__UnknownAuthenticationStrategy } from '../../../graph/metamodel/pure/packageableElements/store/relational/connection/INTERNAL__UnknownAuthenticationStrategy.js';
+import {
+  type RelationalQueryGenerationConfig,
+  GenerationFeaturesConfig,
+} from '../../../graph/metamodel/pure/packageableElements/store/relational/connection/RelationalQueryGenerationConfig.js';
 
 // ------------------------------------- Operation -------------------------------------
 
@@ -1194,6 +1198,36 @@ export const observe_PostProcessor = (
   return metamodel;
 };
 
+export const observe_Abstract_RelationalQueryGenerationConfig = (
+  metamodel: RelationalQueryGenerationConfig,
+): void => {
+  makeObservable(metamodel, {
+    hashCode: computed,
+  });
+};
+
+export const observe_GenerationFeaturesConfig = (
+  metamodel: GenerationFeaturesConfig,
+): GenerationFeaturesConfig => {
+  observe_Abstract_RelationalQueryGenerationConfig(metamodel);
+
+  makeObservable(metamodel, {
+    enabled: observable,
+    disabled: observable,
+  });
+
+  return metamodel;
+};
+
+export const observe_RelationalQueryGenerationConfig = (
+  metamodel: RelationalQueryGenerationConfig,
+): RelationalQueryGenerationConfig => {
+  if (metamodel instanceof GenerationFeaturesConfig) {
+    return observe_GenerationFeaturesConfig(metamodel);
+  }
+  return metamodel;
+};
+
 const observe_Abstract_DatabaseConnection = (
   metamodel: DatabaseConnection,
 ): void => {
@@ -1204,6 +1238,11 @@ const observe_Abstract_DatabaseConnection = (
     timeZone: observable,
     quoteIdentifiers: observable,
     queryTimeOutInSeconds: observable,
+    queryGenerationConfigs: observable,
+  });
+
+  metamodel.queryGenerationConfigs.forEach((config) => {
+    observe_RelationalQueryGenerationConfig(config);
   });
 };
 

package/src/graph-manager/protocol/pure/v1/model/packageableElements/store/relational/connection/V1_RelationalDatabaseConnection.ts

@@ -23,6 +23,7 @@ import {
 import type { V1_DatasourceSpecification } from './V1_DatasourceSpecification.js';
 import type { V1_AuthenticationStrategy } from './V1_AuthenticationStrategy.js';
 import type { V1_PostProcessor } from './postprocessor/V1_PostProcessor.js';
+import type { V1_RelationalQueryGenerationConfig } from './V1_RelationalQueryGenerationConfig.js';
 
 /**
  * TODO: to be modularized or handled differently
@@ -59,6 +60,7 @@ export abstract class V1_DatabaseConnection extends V1_Connection {
   quoteIdentifiers?: boolean | undefined;
   queryTimeOutInSeconds?: number | undefined;
   postProcessorWithParameter: unknown[] = [];
+  queryGenerationConfigs: V1_RelationalQueryGenerationConfig[] = [];
 }
 
 export class V1_RelationalDatabaseConnection
@@ -86,6 +88,7 @@ export class V1_RelationalDatabaseConnection
       this.localMode?.toString() ?? '',
       this.queryTimeOutInSeconds?.toString() ?? '',
       hashArray(this.postProcessors),
+      hashArray(this.queryGenerationConfigs),
     ]);
   }
 

package/src/graph-manager/protocol/pure/v1/model/packageableElements/store/relational/connection/V1_RelationalQueryGenerationConfig.ts

@@ -0,0 +1,38 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { CORE_HASH_STRUCTURE } from '../../../../../../../../../graph/Core_HashUtils.js';
+import { type Hashable, hashArray } from '@finos/legend-shared';
+
+export abstract class V1_RelationalQueryGenerationConfig implements Hashable {
+  abstract get hashCode(): string;
+}
+
+export class V1_GenerationFeaturesConfig
+  extends V1_RelationalQueryGenerationConfig
+  implements Hashable
+{
+  enabled: string[] = [];
+  disabled: string[] = [];
+
+  get hashCode(): string {
+    return hashArray([
+      CORE_HASH_STRUCTURE.GENERATION_FEATURES_CONFIG,
+      hashArray(this.enabled),
+      hashArray(this.disabled),
+    ]);
+  }
+}

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

@@ -99,6 +99,7 @@ import { INTERNAL__UnknownDatasourceSpecification } from '../../../../../../../g
 import { INTERNAL__UnknownAuthenticationStrategy } from '../../../../../../../graph/metamodel/pure/packageableElements/store/relational/connection/INTERNAL__UnknownAuthenticationStrategy.js';
 import { V1_INTERNAL__UnknownDatasourceSpecification } from '../../../model/packageableElements/store/relational/connection/V1_INTERNAL__UnknownDatasourceSpecification.js';
 import { V1_INTERNAL__UnknownAuthenticationStrategy } from '../../../model/packageableElements/store/relational/connection/V1_INTERNAL__UnknownAuthenticationStrategy.js';
+import { V1_transformQueryGenerationConfig } from './V1_QueryGenerationConfigTransformer.js';
 
 const transformStaticDatasourceSpecification = (
   metamodel: StaticDatasourceSpecification,
@@ -369,6 +370,11 @@ export const V1_transformRelationalDatabaseConnection = (
       V1_transformPostProcessor(postprocessor, context),
     );
   }
+  if (metamodel.queryGenerationConfigs.length) {
+    connection.queryGenerationConfigs = metamodel.queryGenerationConfigs.map(
+      (config) => V1_transformQueryGenerationConfig(config, context),
+    );
+  }
   return connection;
 };
 

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

@@ -0,0 +1,43 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { UnsupportedOperationError } from '@finos/legend-shared';
+import {
+  type RelationalQueryGenerationConfig,
+  GenerationFeaturesConfig,
+} from '../../../../../../../graph/metamodel/pure/packageableElements/store/relational/connection/RelationalQueryGenerationConfig.js';
+import {
+  type V1_RelationalQueryGenerationConfig,
+  V1_GenerationFeaturesConfig,
+} from '../../../model/packageableElements/store/relational/connection/V1_RelationalQueryGenerationConfig.js';
+import type { V1_GraphTransformerContext } from './V1_GraphTransformerContext.js';
+
+export const V1_transformQueryGenerationConfig = (
+  metamodel: RelationalQueryGenerationConfig,
+  context: V1_GraphTransformerContext,
+): V1_RelationalQueryGenerationConfig => {
+  if (metamodel instanceof GenerationFeaturesConfig) {
+    const protocol = new V1_GenerationFeaturesConfig();
+    protocol.enabled = metamodel.enabled;
+    protocol.disabled = metamodel.disabled;
+    return protocol;
+  }
+
+  throw new UnsupportedOperationError(
+    `Can't transform query generation config: no compatible transformer available`,
+    metamodel,
+  );
+};

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

@@ -60,6 +60,7 @@ import type { V1_ModelChainConnection } from '../../../../model/packageableEleme
 import { V1_buildPostProcessor } from './V1_PostProcessorBuilderHelper.js';
 import type { V1_INTERNAL__UnknownConnection } from '../../../../model/packageableElements/connection/V1_INTERNAL__UnknownConnection.js';
 import { INTERNAL__UnknownConnection } from '../../../../../../../../graph/metamodel/pure/packageableElements/connection/INTERNAL__UnknownConnection.js';
+import { V1_buildQueryGenerationConfig } from './V1_QueryGenerationConfigBuilderHelper.js';
 
 class V1_ConnectionBuilder implements V1_ConnectionVisitor<Connection> {
   context: V1_GraphBuilderContext;
@@ -293,6 +294,9 @@ class V1_ConnectionBuilder implements V1_ConnectionVisitor<Connection> {
       V1_buildPostProcessor(p, this.context),
     );
     val.postProcessorWithParameter = connection.postProcessorWithParameter;
+    val.queryGenerationConfigs = connection.queryGenerationConfigs.map(
+      (config) => V1_buildQueryGenerationConfig(config, this.context),
+    );
     return val;
   }
 }

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

@@ -0,0 +1,43 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { UnsupportedOperationError } from '@finos/legend-shared';
+import {
+  type RelationalQueryGenerationConfig,
+  GenerationFeaturesConfig,
+} from '../../../../../../../../graph/metamodel/pure/packageableElements/store/relational/connection/RelationalQueryGenerationConfig.js';
+import {
+  type V1_RelationalQueryGenerationConfig,
+  V1_GenerationFeaturesConfig,
+} from '../../../../model/packageableElements/store/relational/connection/V1_RelationalQueryGenerationConfig.js';
+import type { V1_GraphBuilderContext } from '../V1_GraphBuilderContext.js';
+
+export const V1_buildQueryGenerationConfig = (
+  protocol: V1_RelationalQueryGenerationConfig,
+  context: V1_GraphBuilderContext,
+): RelationalQueryGenerationConfig => {
+  if (protocol instanceof V1_GenerationFeaturesConfig) {
+    const metamodel = new GenerationFeaturesConfig();
+    metamodel.enabled = protocol.enabled;
+    metamodel.disabled = protocol.disabled;
+    return metamodel;
+  }
+
+  throw new UnsupportedOperationError(
+    `Can't build query generation config: no compatible builder available`,
+    protocol,
+  );
+};

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

@@ -81,6 +81,7 @@ import {
   V1_deserializeMilestoning,
   V1_serializeMilestoning,
 } from './V1_MilestoningSerializationHelper.js';
+import type { V1_RelationalQueryGenerationConfig } from '../../../model/packageableElements/store/relational/connection/V1_RelationalQueryGenerationConfig.js';
 import type { V1_Milestoning } from '../../../model/packageableElements/store/relational/model/milestoning/V1_Milestoning.js';
 import type { PureProtocolProcessorPlugin } from '../../../../PureProtocolProcessorPlugin.js';
 import { V1_RelationalDatabaseConnection } from '../../../model/packageableElements/store/relational/connection/V1_RelationalDatabaseConnection.js';
@@ -105,6 +106,10 @@ import {
 } from './V1_CoreSerializationHelper.js';
 import { PackageableElementPointerType } from '../../../../../../../graph/MetaModelConst.js';
 import { V1_TabularFunction } from '../../../model/packageableElements/store/relational/model/V1_TabularFunction.js';
+import {
+  V1_serializeQueryGenerationConfig,
+  V1_deserializeQueryGenerationConfig,
+} from './V1_QueryGenerationConfigSerializationHelper.js';
 
 export const V1_DATABASE_ELEMENT_PROTOCOL_TYPE = 'relational';
 
@@ -582,6 +587,11 @@ const V1_setupRelationalDatabaseConnectionModelSchema = (
     postProcessorWithParameter: customEquivalentList({
       INTERNAL__forceReturnEmptyInTest: true,
     }),
+    queryGenerationConfigs: customList(
+      (value: V1_RelationalQueryGenerationConfig) =>
+        V1_serializeQueryGenerationConfig(value, plugins),
+      (value) => V1_deserializeQueryGenerationConfig(value, plugins),
+    ),
     type: primitive(),
   });
 };

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

@@ -0,0 +1,77 @@
+/**
+ * Copyright (c) 2020-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import {
+  type V1_RelationalQueryGenerationConfig,
+  V1_GenerationFeaturesConfig,
+} from '../../../model/packageableElements/store/relational/connection/V1_RelationalQueryGenerationConfig.js';
+import {
+  createModelSchema,
+  deserialize,
+  serialize,
+  primitive,
+  list,
+} from 'serializr';
+import {
+  type PlainObject,
+  usingConstantValueSchema,
+  UnsupportedOperationError,
+} from '@finos/legend-shared';
+import type { PureProtocolProcessorPlugin } from '../../../../PureProtocolProcessorPlugin.js';
+
+enum V1_QueryGenerationConfigType {
+  GENERATION_FEATURES = 'generationFeaturesConfig',
+}
+
+const V1_generationFeaturesConfigModelSchema = createModelSchema(
+  V1_GenerationFeaturesConfig,
+  {
+    _type: usingConstantValueSchema(
+      V1_QueryGenerationConfigType.GENERATION_FEATURES,
+    ),
+    disabled: list(primitive()),
+    enabled: list(primitive()),
+  },
+);
+
+export const V1_serializeQueryGenerationConfig = (
+  value: V1_RelationalQueryGenerationConfig,
+  plugins: PureProtocolProcessorPlugin[],
+): PlainObject<V1_RelationalQueryGenerationConfig> => {
+  if (value instanceof V1_GenerationFeaturesConfig) {
+    return serialize(V1_generationFeaturesConfigModelSchema, value);
+  }
+
+  throw new UnsupportedOperationError(
+    `Can't serialize query generation config: no compatible serializer available`,
+    value,
+  );
+};
+
+export const V1_deserializeQueryGenerationConfig = (
+  json: PlainObject<V1_RelationalQueryGenerationConfig>,
+  plugins: PureProtocolProcessorPlugin[],
+): V1_RelationalQueryGenerationConfig => {
+  switch (json._type) {
+    case V1_QueryGenerationConfigType.GENERATION_FEATURES:
+      return deserialize(V1_generationFeaturesConfigModelSchema, json);
+    default: {
+      throw new UnsupportedOperationError(
+        `Can't deserialize query generation config of type '${json._type}'`,
+      );
+    }
+  }
+};

package/tsconfig.json

@@ -288,6 +288,7 @@
     "./src/graph/metamodel/pure/packageableElements/store/relational/connection/INTERNAL__UnknownAuthenticationStrategy.ts",
     "./src/graph/metamodel/pure/packageableElements/store/relational/connection/INTERNAL__UnknownDatasourceSpecification.ts",
     "./src/graph/metamodel/pure/packageableElements/store/relational/connection/RelationalDatabaseConnection.ts",
+    "./src/graph/metamodel/pure/packageableElements/store/relational/connection/RelationalQueryGenerationConfig.ts",
     "./src/graph/metamodel/pure/packageableElements/store/relational/connection/postprocessor/INTERNAL__UnknownPostProcessor.ts",
     "./src/graph/metamodel/pure/packageableElements/store/relational/connection/postprocessor/Mapper.ts",
     "./src/graph/metamodel/pure/packageableElements/store/relational/connection/postprocessor/MapperPostProcessor.ts",
@@ -650,6 +651,7 @@
     "./src/graph-manager/protocol/pure/v1/model/packageableElements/store/relational/connection/V1_INTERNAL__UnknownAuthenticationStrategy.ts",
     "./src/graph-manager/protocol/pure/v1/model/packageableElements/store/relational/connection/V1_INTERNAL__UnknownDatasourceSpecification.ts",
     "./src/graph-manager/protocol/pure/v1/model/packageableElements/store/relational/connection/V1_RelationalDatabaseConnection.ts",
+    "./src/graph-manager/protocol/pure/v1/model/packageableElements/store/relational/connection/V1_RelationalQueryGenerationConfig.ts",
     "./src/graph-manager/protocol/pure/v1/model/packageableElements/store/relational/connection/postprocessor/V1_INTERNAL__UnknownPostProcessor.ts",
     "./src/graph-manager/protocol/pure/v1/model/packageableElements/store/relational/connection/postprocessor/V1_Mapper.ts",
     "./src/graph-manager/protocol/pure/v1/model/packageableElements/store/relational/connection/postprocessor/V1_MapperPostProcessor.ts",
@@ -775,6 +777,7 @@
     "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/from/V1_MilestoningTransformer.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/from/V1_PackageableElementTransformer.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/from/V1_PostProcessorTransformer.ts",
+    "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/from/V1_QueryGenerationConfigTransformer.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/from/V1_RawValueSpecificationTransformer.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/from/V1_RuntimeTransformer.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/from/V1_SectionIndexTransformer.ts",
@@ -814,6 +817,7 @@
     "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/helpers/V1_MilestoningBuilderHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/helpers/V1_PostProcessorBuilderHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/helpers/V1_ProcessingContext.ts",
+    "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/helpers/V1_QueryGenerationConfigBuilderHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/helpers/V1_RelationalClassMappingBuilderHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/helpers/V1_RelationalConnectionBuilderHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/helpers/V1_RelationalPropertyMappingBuilder.ts",
@@ -840,6 +844,7 @@
     "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_MappingSerializationHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_MilestoningSerializationHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_PostProcessorSerializationHelper.ts",
+    "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_QueryGenerationConfigSerializationHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_RawValueSpecificationSerializationHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_RuntimeSerializationHelper.ts",
     "./src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_SectionIndexSerializationHelper.ts",