improve logic

finos · Jun 10, 2024 · a194ffe · a194ffe
1 parent 41ccfc0
commit a194ffe
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/packages/legend-query-builder/src/stores/milestoning/QueryBuilderMilestoningHelper.ts b/packages/legend-query-builder/src/stores/milestoning/QueryBuilderMilestoningHelper.ts
@@ -340,4 +340,31 @@ export const validateMilestoningPropertyExpressionChain = (
       );
     }
   }
+  // we still need to do a loose milestoningState initialization for other cases e.g. the getAllVersion is used
+  else {
+    if (
+      propertyExpression.parametersValues.length === 3 &&
+      targetStereotype === MILESTONING_STEREOTYPE.BITEMPORAL
+    ) {
+      queryBuilderState.milestoningState.setProcessingDate(
+        propertyExpression.parametersValues[1],
+      );
+      queryBuilderState.milestoningState.setBusinessDate(
+        propertyExpression.parametersValues[2],
+      );
+    }
+    if (propertyExpression.parametersValues.length === 2) {
+      if (targetStereotype === MILESTONING_STEREOTYPE.BUSINESS_TEMPORAL) {
+        queryBuilderState.milestoningState.setBusinessDate(
+          propertyExpression.parametersValues[1],
+        );
+      } else if (
+        targetStereotype === MILESTONING_STEREOTYPE.PROCESSING_TEMPORAL
+      ) {
+        queryBuilderState.milestoningState.setProcessingDate(
+          propertyExpression.parametersValues[1],
+        );
+      }
+    }
+  }
 };