@finos/legend-query-builder 0.2.2 → 0.3.0

package/src/stores/fetch-structure/tds/olapGroupBy/QueryBuilderOlapGroupByValueSpecificationBuilder.ts

@@ -0,0 +1,148 @@
+/**
+ * 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 LambdaFunction,
+  CollectionInstanceValue,
+  extractElementNameFromPath,
+  GenericType,
+  GenericTypeExplicitReference,
+  PrimitiveInstanceValue,
+  PRIMITIVE_TYPE,
+  SimpleFunctionExpression,
+  TYPICAL_MULTIPLICITY_TYPE,
+  VariableExpression,
+} from '@finos/legend-graph';
+import { guaranteeNonNullable } from '@finos/legend-shared';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../graphManager/QueryBuilderSupportedFunctions.js';
+import { buildGenericLambdaFunctionInstanceValue } from '../../../QueryBuilderValueSpecificationHelper.js';
+import { getFunctionNameFromTDSSortColumn } from '../QueryBuilderTDSHelper.js';
+import {
+  type QueryBuilderOlapGroupByColumnState,
+  type QueryBuilderOlapGroupByState,
+  QueryBuilderTDSOlapAggreationOperatorState,
+} from './QueryBuilderOlapGroupByState.js';
+
+const appendOlapGroupByColumnState = (
+  olapGroupByColumnState: QueryBuilderOlapGroupByColumnState,
+  lambda: LambdaFunction,
+): LambdaFunction => {
+  const graph =
+    olapGroupByColumnState.olapState.tdsState.queryBuilderState
+      .graphManagerState.graph;
+  const multiplicityOne = graph.getTypicalMultiplicity(
+    TYPICAL_MULTIPLICITY_TYPE.ONE,
+  );
+  const typeString = graph.getPrimitiveType(PRIMITIVE_TYPE.STRING);
+
+  // create window cols expression
+  const windowColumns = olapGroupByColumnState.windowColumns.map((column) => {
+    const stringInstance = new PrimitiveInstanceValue(
+      GenericTypeExplicitReference.create(new GenericType(typeString)),
+      multiplicityOne,
+    );
+    stringInstance.values = [column.columnName];
+    return stringInstance;
+  });
+
+  const windowCollectionExpression = new CollectionInstanceValue(
+    graph.getMultiplicity(windowColumns.length, windowColumns.length),
+  );
+  windowCollectionExpression.values = windowColumns;
+
+  // create sortBy expression
+  let sortByFunction: SimpleFunctionExpression | undefined;
+  if (olapGroupByColumnState.sortByState) {
+    const sortByState = olapGroupByColumnState.sortByState;
+    sortByFunction = new SimpleFunctionExpression(
+      getFunctionNameFromTDSSortColumn(sortByState.sortType),
+      multiplicityOne,
+    );
+    const sortColInstance = new PrimitiveInstanceValue(
+      GenericTypeExplicitReference.create(new GenericType(typeString)),
+      multiplicityOne,
+    );
+    sortColInstance.values = [sortByState.columnState.columnName];
+    sortByFunction.parametersValues[0] = sortColInstance;
+  }
+
+  // create olap operation expression
+  const operationState = olapGroupByColumnState.operationState;
+  const olapFunc = extractElementNameFromPath(operationState.operator.pureFunc);
+  const olapFuncExpression = new SimpleFunctionExpression(
+    olapFunc,
+    multiplicityOne,
+  );
+  olapFuncExpression.parametersValues = [
+    new VariableExpression(operationState.lambdaParameterName, multiplicityOne),
+  ];
+  const olapLambdaFuncInstance = buildGenericLambdaFunctionInstanceValue(
+    operationState.lambdaParameterName,
+    [olapFuncExpression],
+    graph,
+  );
+  let olapAggregationExpression: SimpleFunctionExpression | undefined;
+  if (operationState instanceof QueryBuilderTDSOlapAggreationOperatorState) {
+    // column param
+    const olapAggregateColumn = new PrimitiveInstanceValue(
+      GenericTypeExplicitReference.create(new GenericType(typeString)),
+      multiplicityOne,
+    );
+    olapAggregateColumn.values = [operationState.columnState.columnName];
+    // build `meta::pure::tds::func`
+    olapAggregationExpression = new SimpleFunctionExpression(
+      extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.TDS_FUNC),
+      multiplicityOne,
+    );
+    olapAggregationExpression.parametersValues = [
+      olapAggregateColumn,
+      olapLambdaFuncInstance,
+    ];
+  }
+  const olapOperationExpression =
+    olapAggregationExpression ?? olapLambdaFuncInstance;
+
+  // OLAP column nam expression
+  const olapColumn = new PrimitiveInstanceValue(
+    GenericTypeExplicitReference.create(new GenericType(typeString)),
+    multiplicityOne,
+  );
+  olapColumn.values = [olapGroupByColumnState.columnName];
+
+  // create main expression
+  const olapExpression = new SimpleFunctionExpression(
+    extractElementNameFromPath(QUERY_BUILDER_SUPPORTED_FUNCTIONS.OLAP_GROUPBY),
+    multiplicityOne,
+  );
+  const currentExpression = guaranteeNonNullable(lambda.expressionSequence[0]);
+  olapExpression.parametersValues = [
+    ...[currentExpression, windowCollectionExpression],
+    ...(sortByFunction ? [sortByFunction] : []),
+    ...[olapOperationExpression, olapColumn],
+  ];
+  lambda.expressionSequence[0] = olapExpression;
+  return lambda;
+};
+
+export const appendOlapGroupByState = (
+  olapGroupByState: QueryBuilderOlapGroupByState,
+  lambda: LambdaFunction,
+): LambdaFunction => {
+  olapGroupByState.olapColumns.forEach((c) =>
+    appendOlapGroupByColumnState(c, lambda),
+  );
+  return lambda;
+};

package/src/stores/fetch-structure/tds/olapGroupBy/operators/QueryBuilderTDSOlapOperator_.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 { PRIMITIVE_TYPE, type PureModel, type Type } from '@finos/legend-graph';
+import { type Hashable, uuid } from '@finos/legend-shared';
+import type { QueryBuilderTDSColumnState } from '../../QueryBuilderTDSColumnState_.js';
+
+export abstract class QueryBuilderTDSOlapOperator implements Hashable {
+  readonly uuid = uuid();
+
+  abstract getLabel(): string;
+
+  abstract get hashCode(): string;
+
+  abstract get pureFunc(): string;
+
+  abstract isCompatibleWithType(type: Type | undefined): boolean;
+
+  isColumnAggregator(): boolean {
+    return false;
+  }
+
+  getOperatorReturnType(graph: PureModel): Type {
+    return graph.getPrimitiveType(PRIMITIVE_TYPE.NUMBER);
+  }
+
+  isCompatibleWithColumn(column: QueryBuilderTDSColumnState): boolean {
+    return this.isCompatibleWithType(column.getColumnType());
+  }
+}

package/src/stores/fetch-structure/tds/olapGroupBy/operators/QueryBuilderTDSOlapOperator_Average.ts

@@ -0,0 +1,56 @@
+/**
+ * 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 { hashArray } from '@finos/legend-shared';
+import { type Type, PRIMITIVE_TYPE } from '@finos/legend-graph';
+import { QueryBuilderTDSOlapOperator } from './QueryBuilderTDSOlapOperator_.js';
+import { QUERY_BUILDER_HASH_STRUCTURE } from '../../../../../graphManager/QueryBuilderHashUtils.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+
+export class QueryBuilderTDSOlapOperator_Average extends QueryBuilderTDSOlapOperator {
+  override isColumnAggregator(): boolean {
+    return true;
+  }
+
+  getLabel(): string {
+    return 'avg';
+  }
+  get pureFunc(): string {
+    return QUERY_BUILDER_SUPPORTED_FUNCTIONS.AVERAGE;
+  }
+
+  get hashCode(): string {
+    return hashArray([QUERY_BUILDER_HASH_STRUCTURE.OLAP_OPERATOR_AVERAGE]);
+  }
+
+  isCompatibleWithType(type: Type | undefined): boolean {
+    if (type) {
+      return (
+        [
+          PRIMITIVE_TYPE.BOOLEAN,
+          PRIMITIVE_TYPE.NUMBER,
+          PRIMITIVE_TYPE.INTEGER,
+          PRIMITIVE_TYPE.DECIMAL,
+          PRIMITIVE_TYPE.FLOAT,
+          PRIMITIVE_TYPE.DATE,
+          PRIMITIVE_TYPE.STRICTDATE,
+          PRIMITIVE_TYPE.DATETIME,
+        ] as string[]
+      ).includes(type.path);
+    }
+    return true;
+  }
+}

package/src/stores/fetch-structure/tds/olapGroupBy/operators/QueryBuilderTDSOlapOperator_Count.ts

@@ -0,0 +1,57 @@
+/**
+ * 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 { hashArray } from '@finos/legend-shared';
+import { type Type, PRIMITIVE_TYPE } from '@finos/legend-graph';
+import { QueryBuilderTDSOlapOperator } from './QueryBuilderTDSOlapOperator_.js';
+import { QUERY_BUILDER_HASH_STRUCTURE } from '../../../../../graphManager/QueryBuilderHashUtils.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+
+export class QueryBuilderTDSOlapOperator_Count extends QueryBuilderTDSOlapOperator {
+  override isColumnAggregator(): boolean {
+    return true;
+  }
+
+  getLabel(): string {
+    return 'count';
+  }
+  get pureFunc(): string {
+    return QUERY_BUILDER_SUPPORTED_FUNCTIONS.COUNT;
+  }
+
+  get hashCode(): string {
+    return hashArray([QUERY_BUILDER_HASH_STRUCTURE.OLAP_OPERATOR_COUNT]);
+  }
+
+  isCompatibleWithType(type: Type | undefined): boolean {
+    if (type) {
+      return (
+        [
+          PRIMITIVE_TYPE.STRING,
+          PRIMITIVE_TYPE.BOOLEAN,
+          PRIMITIVE_TYPE.NUMBER,
+          PRIMITIVE_TYPE.INTEGER,
+          PRIMITIVE_TYPE.DECIMAL,
+          PRIMITIVE_TYPE.FLOAT,
+          PRIMITIVE_TYPE.DATE,
+          PRIMITIVE_TYPE.STRICTDATE,
+          PRIMITIVE_TYPE.DATETIME,
+        ] as string[]
+      ).includes(type.path);
+    }
+    return true;
+  }
+}

package/src/stores/fetch-structure/tds/olapGroupBy/operators/QueryBuilderTDSOlapOperator_Max.ts

@@ -0,0 +1,56 @@
+/**
+ * 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 { hashArray } from '@finos/legend-shared';
+import { type Type, PRIMITIVE_TYPE } from '@finos/legend-graph';
+import { QueryBuilderTDSOlapOperator } from './QueryBuilderTDSOlapOperator_.js';
+import { QUERY_BUILDER_HASH_STRUCTURE } from '../../../../../graphManager/QueryBuilderHashUtils.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+
+export class QueryBuilderTDSOlapOperator_Max extends QueryBuilderTDSOlapOperator {
+  override isColumnAggregator(): boolean {
+    return true;
+  }
+
+  getLabel(): string {
+    return 'max';
+  }
+  get pureFunc(): string {
+    return QUERY_BUILDER_SUPPORTED_FUNCTIONS.MAX;
+  }
+
+  get hashCode(): string {
+    return hashArray([QUERY_BUILDER_HASH_STRUCTURE.OLAP_OPERATOR_MAX]);
+  }
+
+  isCompatibleWithType(type: Type | undefined): boolean {
+    if (type) {
+      return (
+        [
+          PRIMITIVE_TYPE.BOOLEAN,
+          PRIMITIVE_TYPE.NUMBER,
+          PRIMITIVE_TYPE.INTEGER,
+          PRIMITIVE_TYPE.DECIMAL,
+          PRIMITIVE_TYPE.FLOAT,
+          PRIMITIVE_TYPE.DATE,
+          PRIMITIVE_TYPE.STRICTDATE,
+          PRIMITIVE_TYPE.DATETIME,
+        ] as string[]
+      ).includes(type.path);
+    }
+    return true;
+  }
+}

package/src/stores/fetch-structure/tds/olapGroupBy/operators/QueryBuilderTDSOlapOperator_Min.ts

@@ -0,0 +1,56 @@
+/**
+ * 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 { hashArray } from '@finos/legend-shared';
+import { type Type, PRIMITIVE_TYPE } from '@finos/legend-graph';
+import { QueryBuilderTDSOlapOperator } from './QueryBuilderTDSOlapOperator_.js';
+import { QUERY_BUILDER_HASH_STRUCTURE } from '../../../../../graphManager/QueryBuilderHashUtils.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+
+export class QueryBuilderTDSOlapOperator_Min extends QueryBuilderTDSOlapOperator {
+  override isColumnAggregator(): boolean {
+    return true;
+  }
+
+  getLabel(): string {
+    return 'min';
+  }
+  get pureFunc(): string {
+    return QUERY_BUILDER_SUPPORTED_FUNCTIONS.MIN;
+  }
+
+  get hashCode(): string {
+    return hashArray([QUERY_BUILDER_HASH_STRUCTURE.OLAP_OPERATOR_MIN]);
+  }
+
+  isCompatibleWithType(type: Type | undefined): boolean {
+    if (type) {
+      return (
+        [
+          PRIMITIVE_TYPE.BOOLEAN,
+          PRIMITIVE_TYPE.NUMBER,
+          PRIMITIVE_TYPE.INTEGER,
+          PRIMITIVE_TYPE.DECIMAL,
+          PRIMITIVE_TYPE.FLOAT,
+          PRIMITIVE_TYPE.DATE,
+          PRIMITIVE_TYPE.STRICTDATE,
+          PRIMITIVE_TYPE.DATETIME,
+        ] as string[]
+      ).includes(type.path);
+    }
+    return true;
+  }
+}

package/src/stores/fetch-structure/tds/olapGroupBy/operators/QueryBuilderTDSOlapOperator_Sum.ts

@@ -0,0 +1,57 @@
+/**
+ * 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 { hashArray } from '@finos/legend-shared';
+import { type Type, PRIMITIVE_TYPE } from '@finos/legend-graph';
+import { QueryBuilderTDSOlapOperator } from './QueryBuilderTDSOlapOperator_.js';
+import { QUERY_BUILDER_HASH_STRUCTURE } from '../../../../../graphManager/QueryBuilderHashUtils.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+
+export class QueryBuilderTDSOlapOperator_Sum extends QueryBuilderTDSOlapOperator {
+  override isColumnAggregator(): boolean {
+    return true;
+  }
+
+  getLabel(): string {
+    return 'sum';
+  }
+  get pureFunc(): string {
+    return QUERY_BUILDER_SUPPORTED_FUNCTIONS.SUM;
+  }
+
+  get hashCode(): string {
+    return hashArray([QUERY_BUILDER_HASH_STRUCTURE.OLAP_OPERATOR_SUM]);
+  }
+
+  isCompatibleWithType(type: Type | undefined): boolean {
+    if (type) {
+      return (
+        [
+          PRIMITIVE_TYPE.STRING,
+          PRIMITIVE_TYPE.BOOLEAN,
+          PRIMITIVE_TYPE.NUMBER,
+          PRIMITIVE_TYPE.INTEGER,
+          PRIMITIVE_TYPE.DECIMAL,
+          PRIMITIVE_TYPE.FLOAT,
+          PRIMITIVE_TYPE.DATE,
+          PRIMITIVE_TYPE.STRICTDATE,
+          PRIMITIVE_TYPE.DATETIME,
+        ] as string[]
+      ).includes(type.path);
+    }
+    return true;
+  }
+}

package/src/stores/fetch-structure/tds/olapGroupBy/operators/QueryBuilderTDSOlapRankOperator_AverageRank_.ts

@@ -0,0 +1,54 @@
+/**
+ * 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 { PRIMITIVE_TYPE, type Type } from '@finos/legend-graph';
+import { hashArray } from '@finos/legend-shared';
+import { QUERY_BUILDER_HASH_STRUCTURE } from '../../../../../graphManager/QueryBuilderHashUtils.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+import { QueryBuilderTDSOlapOperator } from './QueryBuilderTDSOlapOperator_.js';
+
+export class QueryBuilderTDSOlapRankOperator_AverageRank extends QueryBuilderTDSOlapOperator {
+  getLabel(): string {
+    return 'average rank';
+  }
+
+  get pureFunc(): string {
+    return QUERY_BUILDER_SUPPORTED_FUNCTIONS.OLAP_AVERAGE_RANK;
+  }
+
+  get hashCode(): string {
+    return hashArray([QUERY_BUILDER_HASH_STRUCTURE.OLAP_OPERATOR_DENSE_RANK]);
+  }
+
+  isCompatibleWithType(type: Type | undefined): boolean {
+    if (type) {
+      return (
+        [
+          PRIMITIVE_TYPE.STRING,
+          PRIMITIVE_TYPE.BOOLEAN,
+          PRIMITIVE_TYPE.NUMBER,
+          PRIMITIVE_TYPE.INTEGER,
+          PRIMITIVE_TYPE.DECIMAL,
+          PRIMITIVE_TYPE.FLOAT,
+          PRIMITIVE_TYPE.DATE,
+          PRIMITIVE_TYPE.STRICTDATE,
+          PRIMITIVE_TYPE.DATETIME,
+        ] as string[]
+      ).includes(type.path);
+    }
+    return true;
+  }
+}

package/src/stores/fetch-structure/tds/olapGroupBy/operators/QueryBuilderTDSOlapRankOperator_DenseRank_.ts

@@ -0,0 +1,54 @@
+/**
+ * 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 { PRIMITIVE_TYPE, type Type } from '@finos/legend-graph';
+import { hashArray } from '@finos/legend-shared';
+import { QUERY_BUILDER_HASH_STRUCTURE } from '../../../../../graphManager/QueryBuilderHashUtils.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+import { QueryBuilderTDSOlapOperator } from './QueryBuilderTDSOlapOperator_.js';
+
+export class QueryBuilderTDSOlapRankOperator_DenseRank extends QueryBuilderTDSOlapOperator {
+  getLabel(): string {
+    return 'dense rank';
+  }
+
+  get pureFunc(): string {
+    return QUERY_BUILDER_SUPPORTED_FUNCTIONS.OLAP_DENSE_RANK;
+  }
+
+  get hashCode(): string {
+    return hashArray([QUERY_BUILDER_HASH_STRUCTURE.OLAP_OPERATOR_DENSE_RANK]);
+  }
+
+  isCompatibleWithType(type: Type | undefined): boolean {
+    if (type) {
+      return (
+        [
+          PRIMITIVE_TYPE.STRING,
+          PRIMITIVE_TYPE.BOOLEAN,
+          PRIMITIVE_TYPE.NUMBER,
+          PRIMITIVE_TYPE.INTEGER,
+          PRIMITIVE_TYPE.DECIMAL,
+          PRIMITIVE_TYPE.FLOAT,
+          PRIMITIVE_TYPE.DATE,
+          PRIMITIVE_TYPE.STRICTDATE,
+          PRIMITIVE_TYPE.DATETIME,
+        ] as string[]
+      ).includes(type.path);
+    }
+    return true;
+  }
+}

package/src/stores/fetch-structure/tds/olapGroupBy/operators/QueryBuilderTDSOlapRankOperator_Rank_.ts

@@ -0,0 +1,54 @@
+/**
+ * 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 { PRIMITIVE_TYPE, type Type } from '@finos/legend-graph';
+import { hashArray } from '@finos/legend-shared';
+import { QUERY_BUILDER_HASH_STRUCTURE } from '../../../../../graphManager/QueryBuilderHashUtils.js';
+import { QUERY_BUILDER_SUPPORTED_FUNCTIONS } from '../../../../../graphManager/QueryBuilderSupportedFunctions.js';
+import { QueryBuilderTDSOlapOperator } from './QueryBuilderTDSOlapOperator_.js';
+
+export class QueryBuilderTDSOlapRankOperator_Rank extends QueryBuilderTDSOlapOperator {
+  getLabel(): string {
+    return 'rank';
+  }
+
+  get pureFunc(): string {
+    return QUERY_BUILDER_SUPPORTED_FUNCTIONS.OLAP_RANK;
+  }
+
+  get hashCode(): string {
+    return hashArray([QUERY_BUILDER_HASH_STRUCTURE.OLAP_OPERATOR_RANK]);
+  }
+
+  isCompatibleWithType(type: Type | undefined): boolean {
+    if (type) {
+      return (
+        [
+          PRIMITIVE_TYPE.STRING,
+          PRIMITIVE_TYPE.BOOLEAN,
+          PRIMITIVE_TYPE.NUMBER,
+          PRIMITIVE_TYPE.INTEGER,
+          PRIMITIVE_TYPE.DECIMAL,
+          PRIMITIVE_TYPE.FLOAT,
+          PRIMITIVE_TYPE.DATE,
+          PRIMITIVE_TYPE.STRICTDATE,
+          PRIMITIVE_TYPE.DATETIME,
+        ] as string[]
+      ).includes(type.path);
+    }
+    return true;
+  }
+}