@finos/legend-extension-dsl-data-quality/src/graph-manager/action/changeDetection/DSL_DataQuality_ObserverHelper.ts

Legend extension for Data Quality

/**
 * 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 ObserverContext,
  observe_Abstract_PackageableElement,
  observe_Class,
  observe_PackageableElementReference,
  observe_PropertyReference,
  observe_RawLambda,
  skipObserved,
  skipObservedWithContext,
} from '@finos/legend-graph';
import { computed, makeObservable, observable, override } from 'mobx';
import {
  type DataQualityClassValidationsConfiguration,
  type DataQualityServiceValidationConfiguration,
  type DataQualityRelationValidation,
  type DataQualityRelationValidationConfiguration,
  type DataQualityRelationQueryLambda,
  type DataQualityRelationComparisonConfiguration,
  type ReconStrategy,
  MD5HashStrategy,
  DataSpaceDataQualityExecutionContext,
  MappingAndRuntimeDataQualityExecutionContext,
} from '../../../graph/metamodel/pure/packageableElements/data-quality/DataQualityValidationConfiguration.js';
import { observe_DataSpace } from '@finos/legend-extension-dsl-data-space/graph';
import {
  type DataQualityRootGraphFetchTree,
  DataQualityPropertyGraphFetchTree,
} from '../../../graph/metamodel/pure/packageableElements/data-quality/DataQualityGraphFetchTree.js';

export const observe_DataSpaceDataQualityExecutionContext = skipObserved(
  (
    metamodel: DataSpaceDataQualityExecutionContext,
  ): DataSpaceDataQualityExecutionContext => {
    makeObservable<DataSpaceDataQualityExecutionContext>(metamodel, {
      context: observable,
      dataSpace: observable,
    });
    observe_DataSpace(metamodel.dataSpace.value);
    return metamodel;
  },
);

export const observe_MappingAndRuntimeDataQualityExecutionContext =
  skipObserved(
    (
      metamodel: MappingAndRuntimeDataQualityExecutionContext,
    ): MappingAndRuntimeDataQualityExecutionContext => {
      makeObservable<MappingAndRuntimeDataQualityExecutionContext>(metamodel, {
        mapping: observable,
        runtime: observable,
      });
      observe_PackageableElementReference(metamodel.mapping);
      observe_PackageableElementReference(metamodel.runtime);
      return metamodel;
    },
  );

export const observe_DataQualityPropertyGraphFetchTree = skipObserved(
  (
    metamodel: DataQualityPropertyGraphFetchTree,
  ): DataQualityPropertyGraphFetchTree => {
    makeObservable<DataQualityPropertyGraphFetchTree>(metamodel, {
      constraints: observable,
      property: observable,
      subTrees: observable,
    });
    observe_PropertyReference(metamodel.property);
    metamodel.subTrees.forEach((subTree) => {
      if (subTree instanceof DataQualityPropertyGraphFetchTree) {
        observe_DataQualityPropertyGraphFetchTree(subTree);
      }
    });
    return metamodel;
  },
);

export const observe_DataQualityRootGraphFetchTree = skipObserved(
  (metamodel: DataQualityRootGraphFetchTree): DataQualityRootGraphFetchTree => {
    makeObservable<DataQualityRootGraphFetchTree>(metamodel, {
      class: observable,
      subTrees: observable,
      constraints: observable,
    });
    observe_Class(metamodel.class.value);
    metamodel.subTrees.forEach((subTree) => {
      if (subTree instanceof DataQualityPropertyGraphFetchTree) {
        observe_DataQualityPropertyGraphFetchTree(subTree);
      }
    });
    return metamodel;
  },
);
export const observe_DataQualityConstraintsConfiguration = skipObserved(
  (
    metamodel: DataQualityClassValidationsConfiguration,
  ): DataQualityClassValidationsConfiguration => {
    observe_Abstract_PackageableElement(metamodel);

    makeObservable<
      DataQualityClassValidationsConfiguration,
      '_elementHashCode'
    >(metamodel, {
      _elementHashCode: override,
      context: observable,
      filter: observable,
      dataQualityRootGraphFetchTree: observable,
    });
    if (metamodel.context instanceof DataSpaceDataQualityExecutionContext) {
      observe_DataSpaceDataQualityExecutionContext(metamodel.context);
    }
    if (
      metamodel.context instanceof MappingAndRuntimeDataQualityExecutionContext
    ) {
      observe_MappingAndRuntimeDataQualityExecutionContext(metamodel.context);
    }
    if (metamodel.filter) {
      observe_RawLambda(metamodel.filter);
    }
    if (metamodel.dataQualityRootGraphFetchTree) {
      observe_DataQualityRootGraphFetchTree(
        metamodel.dataQualityRootGraphFetchTree,
      );
    }
    return metamodel;
  },
);

export const observe_DataQualityServiceValidationConfiguration = skipObserved(
  (
    metamodel: DataQualityServiceValidationConfiguration,
  ): DataQualityServiceValidationConfiguration => {
    observe_Abstract_PackageableElement(metamodel);

    makeObservable<
      DataQualityServiceValidationConfiguration,
      '_elementHashCode'
    >(metamodel, {
      _elementHashCode: override,
      contextName: observable,
      serviceName: observable,
      dataQualityRootGraphFetchTree: observable,
    });
    return metamodel;
  },
);

export const observe_DataQualityRelationValidation = skipObserved(
  (metamodel: DataQualityRelationValidation): DataQualityRelationValidation => {
    makeObservable(metamodel, {
      name: observable,
      assertion: observable,
      description: observable,
      type: observable,
    });

    observe_RawLambda(metamodel.assertion);
    return metamodel;
  },
);

export const observe_DataQualityRelationQueryLambda = skipObserved(
  (metamodel: DataQualityRelationQueryLambda): DataQualityRelationQueryLambda =>
    makeObservable(metamodel, {
      body: observable.ref, // only observe the reference, the object itself is not observed
      parameters: observable,
      hashCode: computed,
    }),
);

export const observe_DataQualityRelationValidationConfiguration =
  skipObservedWithContext(
    (
      metamodel: DataQualityRelationValidationConfiguration,
      context: ObserverContext,
    ): DataQualityRelationValidationConfiguration => {
      observe_Abstract_PackageableElement(metamodel);

      makeObservable<
        DataQualityRelationValidationConfiguration,
        '_elementHashCode'
      >(metamodel, {
        _elementHashCode: override,
        query: observable,
        validations: observable,
      });
      metamodel.validations.forEach((value) =>
        observe_DataQualityRelationValidation(value),
      );
      observe_DataQualityRelationQueryLambda(metamodel.query);
      return metamodel;
    },
  );

export const observe_ReconStrategy = skipObserved(
  (metamodel: ReconStrategy): ReconStrategy => {
    if (metamodel instanceof MD5HashStrategy) {
      return makeObservable(metamodel, {
        sourceHashColumn: observable,
        targetHashColumn: observable,
        aggregatedHash: observable,
        hashCode: computed,
      });
    }
    return metamodel;
  },
);

export const observe_DataQualityRelationComparisonConfiguration =
  skipObservedWithContext(
    (
      metamodel: DataQualityRelationComparisonConfiguration,
    ): DataQualityRelationComparisonConfiguration => {
      observe_Abstract_PackageableElement(metamodel);
      makeObservable<
        DataQualityRelationComparisonConfiguration,
        '_elementHashCode'
      >(metamodel, {
        _elementHashCode: override,
        source: observable,
        target: observable,
        keys: observable,
        columnsToCompare: observable,
        strategy: observable,
        expectedMatch: observable,
      });
      observe_DataQualityRelationQueryLambda(metamodel.source);
      observe_DataQualityRelationQueryLambda(metamodel.target);
      observe_ReconStrategy(metamodel.strategy);
      return metamodel;
    },
  );