@itwin/core-backend 5.1.0-dev.43 → 5.1.0-dev.45

package/lib/esm/test/ecdb/CTE.test.js

@@ -24,17 +24,17 @@ describe("Common table expression support in ECSQL", () => {
         imodel1.close();
     });
     it("collect base properties recursively", async () => {
-        const query = `
-        WITH RECURSIVE
-            base_classes (aId, aParentId, aPath, aDepth) AS (
-                SELECT c.ECInstanceId, null, c.Name, 0  FROM meta.ECClassDef c WHERE c.Name=?
-                UNION ALL
-                SELECT c.ECInstanceId, cbc.TargetECInstanceId, aPath || '/' || c.Name, aDepth + 1
-                    FROM meta.ECClassDef c
-                        JOIN meta.ClassHasBaseClasses cbc ON cbc.SourceECInstanceId = c.ECInstanceId
-                        JOIN base_classes  ON aId = cbc.TargetECInstanceId
-                ORDER BY 1
-            )
+        const query = `
+        WITH RECURSIVE
+            base_classes (aId, aParentId, aPath, aDepth) AS (
+                SELECT c.ECInstanceId, null, c.Name, 0  FROM meta.ECClassDef c WHERE c.Name=?
+                UNION ALL
+                SELECT c.ECInstanceId, cbc.TargetECInstanceId, aPath || '/' || c.Name, aDepth + 1
+                    FROM meta.ECClassDef c
+                        JOIN meta.ClassHasBaseClasses cbc ON cbc.SourceECInstanceId = c.ECInstanceId
+                        JOIN base_classes  ON aId = cbc.TargetECInstanceId
+                ORDER BY 1
+            )
         SELECT p.Name prop from base_classes join meta.ECPropertyDef p on p.Class.id = aId GROUP BY p.Name`;
         const rows = await executeQuery(imodel1, query, ["Element"]);
         const expected = ["Angle", "BaseModel", "BBoxHigh", "BBoxLow", "Border", "BorderTemplate", "Category", "CategorySelector", "CodeScope", "CodeSpec", "CodeValue", "Data", "Description", "DisplayStyle", "DrawingModel", "Enabled", "Extents", "EyePoint", "FederationGuid", "Flags", "FocusDistance", "Format", "GeometryStream", "Height", "InSpatialIndex", "IsCameraOn", "IsPrivate", "JsonProperties", "LastMod", "LensAngle", "Model", "ModelSelector", "Name", "Origin", "PaletteName", "Parent", "Pitch", "Properties", "Rank", "Recipe", "RepositoryGuid", "Roll", "Rotation", "RotationAngle", "Scale", "SheetTemplate", "Type", "TypeDefinition", "Url", "UserLabel", "View", "ViewAttachment", "Width", "Yaw"];
@@ -42,58 +42,58 @@ describe("Common table expression support in ECSQL", () => {
         assert.sameOrderedMembers(actual, expected);
     });
     it("generate mandelbrot set", async () => {
-        const rows = await executeQuery(imodel1, `
-      WITH RECURSIVE
-        [xaxis]([x]) AS(
-          VALUES (- 2.0)
-          UNION ALL
-          SELECT [x] + 0.05
-          FROM   [xaxis]
-          WHERE  [x] < 1.2
-        ),
-        [yaxis]([y]) AS(
-          VALUES (- 1.0)
-          UNION ALL
-          SELECT [y] + 0.1
-          FROM   [yaxis]
-          WHERE  [y] < 1.0
-        ),
-        [m]([iter], [cx], [cy], [x], [y]) AS(
-          SELECT
-                0,
-                [x],
-                [y],
-                0.0,
-                0.0
-          FROM   [xaxis],
-                [yaxis]
-          UNION ALL
-          SELECT
-                [iter] + 1,
-                [cx],
-                [cy],
-                [x] * [x] - [y] * [y] + [cx],
-                2.0 * [x] * [y] + [cy]
-          FROM   [m]
-          WHERE  ([x] * [x] + [y] * [y]) < 4.0 AND [iter] < 28
-        ),
-        [m2]([iter], [cx], [cy]) AS(
-          SELECT
-                MAX ([iter]),
-                [cx],
-                [cy]
-          FROM   [m]
-          GROUP  BY
-                    [cx],
-                    [cy]
-        ),
-        [a]([t]) AS(
-          SELECT GROUP_CONCAT (SUBSTR (' .+*#', 1 + (CASE WHEN [iter] / 7 > 4 THEN 4 ELSE [iter] / 7 END), 1), '')
-          FROM   [m2]
-          GROUP  BY [cy]
-        )
-      SELECT GROUP_CONCAT (RTRIM ([t]), CHAR (0xa)) mandelbrot_set
-      FROM   [a];
+        const rows = await executeQuery(imodel1, `
+      WITH RECURSIVE
+        [xaxis]([x]) AS(
+          VALUES (- 2.0)
+          UNION ALL
+          SELECT [x] + 0.05
+          FROM   [xaxis]
+          WHERE  [x] < 1.2
+        ),
+        [yaxis]([y]) AS(
+          VALUES (- 1.0)
+          UNION ALL
+          SELECT [y] + 0.1
+          FROM   [yaxis]
+          WHERE  [y] < 1.0
+        ),
+        [m]([iter], [cx], [cy], [x], [y]) AS(
+          SELECT
+                0,
+                [x],
+                [y],
+                0.0,
+                0.0
+          FROM   [xaxis],
+                [yaxis]
+          UNION ALL
+          SELECT
+                [iter] + 1,
+                [cx],
+                [cy],
+                [x] * [x] - [y] * [y] + [cx],
+                2.0 * [x] * [y] + [cy]
+          FROM   [m]
+          WHERE  ([x] * [x] + [y] * [y]) < 4.0 AND [iter] < 28
+        ),
+        [m2]([iter], [cx], [cy]) AS(
+          SELECT
+                MAX ([iter]),
+                [cx],
+                [cy]
+          FROM   [m]
+          GROUP  BY
+                    [cx],
+                    [cy]
+        ),
+        [a]([t]) AS(
+          SELECT GROUP_CONCAT (SUBSTR (' .+*#', 1 + (CASE WHEN [iter] / 7 > 4 THEN 4 ELSE [iter] / 7 END), 1), '')
+          FROM   [m2]
+          GROUP  BY [cy]
+        )
+      SELECT GROUP_CONCAT (RTRIM ([t]), CHAR (0xa)) mandelbrot_set
+      FROM   [a];
     `);
         const expected = "                                    ....#\n" +
             "                                   ..#*..\n" +
@@ -121,34 +121,34 @@ describe("Common table expression support in ECSQL", () => {
     });
     it("basic cte test", async () => {
         let rows = [];
-        rows = await executeQuery(imodel1, `
-      WITH RECURSIVE
-        cnt (x,y) AS (
-            SELECT 100, 200
-            UNION ALL
-            SELECT x+1, 200 FROM cnt WHERE x<210
-        )
+        rows = await executeQuery(imodel1, `
+      WITH RECURSIVE
+        cnt (x,y) AS (
+            SELECT 100, 200
+            UNION ALL
+            SELECT x+1, 200 FROM cnt WHERE x<210
+        )
       SELECT * from cnt`);
         assert(rows.length === 111);
-        rows = await executeQuery(imodel1, `
-      WITH RECURSIVE
-        cnt (x,y) AS (
-            SELECT 100, 200
-        )
+        rows = await executeQuery(imodel1, `
+      WITH RECURSIVE
+        cnt (x,y) AS (
+            SELECT 100, 200
+        )
       SELECT * from cnt`);
         let slm = new SequentialLogMatcher();
         // these two are generated by sqlite
         slm.append().error().category("ECDb").message(/BE_SQLITE_ERROR duplicate WITH table name/gm);
         assert(rows.length === 1);
         try {
-            rows = await executeQuery(imodel1, `
-        WITH
-          cte_1 (a,b,c) AS (
-            SELECT 100, 400, 300
-          ),
-          cte_1 (a,b,c) AS (
-            SELECT 100, 400, 300
-          )
+            rows = await executeQuery(imodel1, `
+        WITH
+          cte_1 (a,b,c) AS (
+            SELECT 100, 400, 300
+          ),
+          cte_1 (a,b,c) AS (
+            SELECT 100, 400, 300
+          )
         SELECT * from cte_1`);
             assert(false);
         }
@@ -160,11 +160,11 @@ describe("Common table expression support in ECSQL", () => {
         // these two are generated by ECSQL. Its not clear why this message is logged twice.
         slm.append().error().category("ECDb").message(/Common table 'cte_1' has 3 values for columns 2/gm);
         try {
-            rows = await executeQuery(imodel1, `
-        WITH
-        cte_1 (a,b,c) AS (
-          SELECT 100, 400
-        )
+            rows = await executeQuery(imodel1, `
+        WITH
+        cte_1 (a,b,c) AS (
+          SELECT 100, 400
+        )
         SELECT * from cte_1`);
             assert(false);
         }

package/lib/esm/test/ecdb/CTE.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"CTE.test.js","sourceRoot":"","sources":["../../../../src/test/ecdb/CTE.test.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAC/F,OAAO,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACjE,OAAO,EAAY,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAE/D,8BAA8B;AAE9B,KAAK,UAAU,YAAY,CAAC,MAAgB,EAAE,KAAa,EAAE,QAAyB,EAAE,eAAyB;IAC/G,MAAM,IAAI,GAAU,EAAE,CAAC;IACvB,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,MAAM,CAAC,iBAAiB,CAAC,KAAK,EAAE,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,cAAc,CAAC,kBAAkB,EAAE,eAAe,EAAE,CAAC,EAAE,CAAC;QAC5J,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;IAC9B,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,QAAQ,CAAC,0CAA0C,EAAE,GAAG,EAAE;IACxD,IAAI,OAAmB,CAAC;IAExB,MAAM,CAAC,KAAK,IAAI,EAAE;QAChB,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,eAAe,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,KAAK,IAAI,EAAE;QACf,OAAO,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,KAAK,GAAG;;;;;;;;;;;2GAWyF,CAAE;QACzG,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QAC7D,MAAM,QAAQ,GAAG,CAAC,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,gBAAgB,EAAE,UAAU,EAAE,kBAAkB,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,EAAE,aAAa,EAAE,cAAc,EAAE,cAAc,EAAE,SAAS,EAAE,SAAS,EAAE,UAAU,EAAE,gBAAgB,EAAE,OAAO,EAAE,eAAe,EAAE,QAAQ,EAAE,gBAAgB,EAAE,QAAQ,EAAE,gBAAgB,EAAE,YAAY,EAAE,WAAW,EAAE,gBAAgB,EAAE,SAAS,EAAE,WAAW,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,EAAE,UAAU,EAAE,eAAe,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,gBAAgB,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,gBAAgB,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QAC1rB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,CAAC,kBAAkB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KAoDxC,CAAC,CAAC;QAEH,MAAM,QAAQ,GACZ,6CAA6C;YAC7C,6CAA6C;YAC7C,8CAA8C;YAC9C,oDAAoD;YACpD,qDAAqD;YACrD,qDAAqD;YACrD,sDAAsD;YACtD,sDAAsD;YACtD,sDAAsD;YACtD,qDAAqD;YACrD,qDAAqD;YACrD,qDAAqD;YACrD,sDAAsD;YACtD,sDAAsD;YACtD,sDAAsD;YACtD,qDAAqD;YACrD,qDAAqD;YACrD,oDAAoD;YACpD,8CAA8C;YAC9C,6CAA6C;YAC7C,6CAA6C;YAC7C,wCAAwC,CAAC;QAC3C,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,cAAc,KAAK,QAAQ,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE;QAC9B,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,IAAI,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;;;;;;;wBAOf,CAAC,CAAC;QACtB,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,GAAG,CAAC,CAAC;QAE5B,IAAI,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;;;;;wBAKf,CAAC,CAAC;QAEtB,IAAI,GAAG,GAAG,IAAI,oBAAoB,EAAE,CAAC;QACrC,oCAAoC;QACpC,GAAG,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,6CAA6C,CAAC,CAAC;QAC7F,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC;QAC1B,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;;;;;;;;4BAQb,CAAC,CAAC;YACxB,MAAM,CAAC,KAAK,CAAC,CAAC;QAChB,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,sDAAsD;QACtE,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtC,GAAG,GAAG,IAAI,oBAAoB,EAAE,CAAC;QACjC,oFAAoF;QACpF,GAAG,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,mDAAmD,CAAC,CAAC;QACnG,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;;;;;4BAKb,CAAC,CAAC;YACxB,MAAM,CAAC,KAAK,CAAC,CAAC;QAChB,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,oDAAoD;QACpE,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n* See LICENSE.md in the project root for license terms and full copyright notice.\r\n*--------------------------------------------------------------------------------------------*/\r\nimport { assert } from \"chai\";\r\nimport { QueryBinder, QueryRowFormat } from \"@itwin/core-common\";\r\nimport { IModelDb, SnapshotDb } from \"../../core-backend\";\r\nimport { IModelTestUtils } from \"../IModelTestUtils\";\r\nimport { SequentialLogMatcher } from \"../SequentialLogMatcher\";\r\n\r\n// cspell:ignore mirukuru ibim\r\n\r\nasync function executeQuery(iModel: IModelDb, ecsql: string, bindings?: any[] | object, abbreviateBlobs?: boolean): Promise<any[]> {\r\n  const rows: any[] = [];\r\n  for await (const queryRow of iModel.createQueryReader(ecsql, QueryBinder.from(bindings), { rowFormat: QueryRowFormat.UseJsPropertyNames, abbreviateBlobs })) {\r\n    rows.push(queryRow.toRow());\r\n  }\r\n  return rows;\r\n}\r\n\r\ndescribe(\"Common table expression support in ECSQL\", () => {\r\n  let imodel1: SnapshotDb;\r\n\r\n  before(async () => {\r\n    imodel1 = SnapshotDb.openFile(IModelTestUtils.resolveAssetFile(\"test.bim\"));\r\n  });\r\n\r\n  after(async () => {\r\n    imodel1.close();\r\n  });\r\n  it(\"collect base properties recursively\", async () => {\r\n    const query = `\r\n        WITH RECURSIVE\r\n            base_classes (aId, aParentId, aPath, aDepth) AS (\r\n                SELECT c.ECInstanceId, null, c.Name, 0  FROM meta.ECClassDef c WHERE c.Name=?\r\n                UNION ALL\r\n                SELECT c.ECInstanceId, cbc.TargetECInstanceId, aPath || '/' || c.Name, aDepth + 1\r\n                    FROM meta.ECClassDef c\r\n                        JOIN meta.ClassHasBaseClasses cbc ON cbc.SourceECInstanceId = c.ECInstanceId\r\n                        JOIN base_classes  ON aId = cbc.TargetECInstanceId\r\n                ORDER BY 1\r\n            )\r\n        SELECT p.Name prop from base_classes join meta.ECPropertyDef p on p.Class.id = aId GROUP BY p.Name` ;\r\n    const rows = await executeQuery(imodel1, query, [\"Element\"]);\r\n    const expected = [\"Angle\", \"BaseModel\", \"BBoxHigh\", \"BBoxLow\", \"Border\", \"BorderTemplate\", \"Category\", \"CategorySelector\", \"CodeScope\", \"CodeSpec\", \"CodeValue\", \"Data\", \"Description\", \"DisplayStyle\", \"DrawingModel\", \"Enabled\", \"Extents\", \"EyePoint\", \"FederationGuid\", \"Flags\", \"FocusDistance\", \"Format\", \"GeometryStream\", \"Height\", \"InSpatialIndex\", \"IsCameraOn\", \"IsPrivate\", \"JsonProperties\", \"LastMod\", \"LensAngle\", \"Model\", \"ModelSelector\", \"Name\", \"Origin\", \"PaletteName\", \"Parent\", \"Pitch\", \"Properties\", \"Rank\", \"Recipe\", \"RepositoryGuid\", \"Roll\", \"Rotation\", \"RotationAngle\", \"Scale\", \"SheetTemplate\", \"Type\", \"TypeDefinition\", \"Url\", \"UserLabel\", \"View\", \"ViewAttachment\", \"Width\", \"Yaw\"];\r\n    const actual = rows.map((r) => r.prop);\r\n    assert.sameOrderedMembers(actual, expected);\r\n  });\r\n\r\n  it(\"generate mandelbrot set\", async () => {\r\n    const rows = await executeQuery(imodel1, `\r\n      WITH RECURSIVE\r\n        [xaxis]([x]) AS(\r\n          VALUES (- 2.0)\r\n          UNION ALL\r\n          SELECT [x] + 0.05\r\n          FROM   [xaxis]\r\n          WHERE  [x] < 1.2\r\n        ),\r\n        [yaxis]([y]) AS(\r\n          VALUES (- 1.0)\r\n          UNION ALL\r\n          SELECT [y] + 0.1\r\n          FROM   [yaxis]\r\n          WHERE  [y] < 1.0\r\n        ),\r\n        [m]([iter], [cx], [cy], [x], [y]) AS(\r\n          SELECT\r\n                0,\r\n                [x],\r\n                [y],\r\n                0.0,\r\n                0.0\r\n          FROM   [xaxis],\r\n                [yaxis]\r\n          UNION ALL\r\n          SELECT\r\n                [iter] + 1,\r\n                [cx],\r\n                [cy],\r\n                [x] * [x] - [y] * [y] + [cx],\r\n                2.0 * [x] * [y] + [cy]\r\n          FROM   [m]\r\n          WHERE  ([x] * [x] + [y] * [y]) < 4.0 AND [iter] < 28\r\n        ),\r\n        [m2]([iter], [cx], [cy]) AS(\r\n          SELECT\r\n                MAX ([iter]),\r\n                [cx],\r\n                [cy]\r\n          FROM   [m]\r\n          GROUP  BY\r\n                    [cx],\r\n                    [cy]\r\n        ),\r\n        [a]([t]) AS(\r\n          SELECT GROUP_CONCAT (SUBSTR (' .+*#', 1 + (CASE WHEN [iter] / 7 > 4 THEN 4 ELSE [iter] / 7 END), 1), '')\r\n          FROM   [m2]\r\n          GROUP  BY [cy]\r\n        )\r\n      SELECT GROUP_CONCAT (RTRIM ([t]), CHAR (0xa)) mandelbrot_set\r\n      FROM   [a];\r\n    `);\r\n\r\n    const expected =\r\n      \"                                    ....#\\n\" +\r\n      \"                                   ..#*..\\n\" +\r\n      \"                                 ..+####+.\\n\" +\r\n      \"                            .......+####....   +\\n\" +\r\n      \"                           ..##+*##########+.++++\\n\" +\r\n      \"                          .+.##################+.\\n\" +\r\n      \"              .............+###################+.+\\n\" +\r\n      \"              ..++..#.....*#####################+.\\n\" +\r\n      \"             ...+#######++#######################.\\n\" +\r\n      \"          ....+*################################.\\n\" +\r\n      \" #############################################...\\n\" +\r\n      \"          ....+*################################.\\n\" +\r\n      \"             ...+#######++#######################.\\n\" +\r\n      \"              ..++..#.....*#####################+.\\n\" +\r\n      \"              .............+###################+.+\\n\" +\r\n      \"                          .+.##################+.\\n\" +\r\n      \"                           ..##+*##########+.++++\\n\" +\r\n      \"                            .......+####....   +\\n\" +\r\n      \"                                 ..+####+.\\n\" +\r\n      \"                                   ..#*..\\n\" +\r\n      \"                                    ....#\\n\" +\r\n      \"                                    +.\";\r\n    assert(rows[0].mandelbrot_set === expected);\r\n  });\r\n\r\n  it(\"basic cte test\", async () => {\r\n    let rows = [];\r\n    rows = await executeQuery(imodel1, `\r\n      WITH RECURSIVE\r\n        cnt (x,y) AS (\r\n            SELECT 100, 200\r\n            UNION ALL\r\n            SELECT x+1, 200 FROM cnt WHERE x<210\r\n        )\r\n      SELECT * from cnt`);\r\n    assert(rows.length === 111);\r\n\r\n    rows = await executeQuery(imodel1, `\r\n      WITH RECURSIVE\r\n        cnt (x,y) AS (\r\n            SELECT 100, 200\r\n        )\r\n      SELECT * from cnt`);\r\n\r\n    let slm = new SequentialLogMatcher();\r\n    // these two are generated by sqlite\r\n    slm.append().error().category(\"ECDb\").message(/BE_SQLITE_ERROR duplicate WITH table name/gm);\r\n    assert(rows.length === 1);\r\n    try {\r\n      rows = await executeQuery(imodel1, `\r\n        WITH\r\n          cte_1 (a,b,c) AS (\r\n            SELECT 100, 400, 300\r\n          ),\r\n          cte_1 (a,b,c) AS (\r\n            SELECT 100, 400, 300\r\n          )\r\n        SELECT * from cte_1`);\r\n      assert(false);\r\n    } catch {\r\n      assert(true); // should fail as cte_1 is used for two ct expression.\r\n    }\r\n    assert.isTrue(slm.finishAndDispose());\r\n    slm = new SequentialLogMatcher();\r\n    // these two are generated by ECSQL. Its not clear why this message is logged twice.\r\n    slm.append().error().category(\"ECDb\").message(/Common table 'cte_1' has 3 values for columns 2/gm);\r\n    try {\r\n      rows = await executeQuery(imodel1, `\r\n        WITH\r\n        cte_1 (a,b,c) AS (\r\n          SELECT 100, 400\r\n        )\r\n        SELECT * from cte_1`);\r\n      assert(false);\r\n    } catch {\r\n      assert(true); // number are to ct expression does not match select\r\n    }\r\n    assert.isTrue(slm.finishAndDispose());\r\n  });\r\n});\r\n"]}
+{"version":3,"file":"CTE.test.js","sourceRoot":"","sources":["../../../../src/test/ecdb/CTE.test.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAC/F,OAAO,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACjE,OAAO,EAAY,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,oBAAoB,EAAE,MAAM,yBAAyB,CAAC;AAE/D,8BAA8B;AAE9B,KAAK,UAAU,YAAY,CAAC,MAAgB,EAAE,KAAa,EAAE,QAAyB,EAAE,eAAyB;IAC/G,MAAM,IAAI,GAAU,EAAE,CAAC;IACvB,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,MAAM,CAAC,iBAAiB,CAAC,KAAK,EAAE,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,cAAc,CAAC,kBAAkB,EAAE,eAAe,EAAE,CAAC,EAAE,CAAC;QAC5J,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;IAC9B,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,QAAQ,CAAC,0CAA0C,EAAE,GAAG,EAAE;IACxD,IAAI,OAAmB,CAAC;IAExB,MAAM,CAAC,KAAK,IAAI,EAAE;QAChB,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,eAAe,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC,CAAC;IAC9E,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,KAAK,IAAI,EAAE;QACf,OAAO,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC,CAAC,CAAC;IACH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;QACnD,MAAM,KAAK,GAAG;;;;;;;;;;;2GAWyF,CAAE;QACzG,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QAC7D,MAAM,QAAQ,GAAG,CAAC,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,gBAAgB,EAAE,UAAU,EAAE,kBAAkB,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,EAAE,aAAa,EAAE,cAAc,EAAE,cAAc,EAAE,SAAS,EAAE,SAAS,EAAE,UAAU,EAAE,gBAAgB,EAAE,OAAO,EAAE,eAAe,EAAE,QAAQ,EAAE,gBAAgB,EAAE,QAAQ,EAAE,gBAAgB,EAAE,YAAY,EAAE,WAAW,EAAE,gBAAgB,EAAE,SAAS,EAAE,WAAW,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,EAAE,UAAU,EAAE,eAAe,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,gBAAgB,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,gBAAgB,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QAC1rB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,CAAC,kBAAkB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KAoDxC,CAAC,CAAC;QAEH,MAAM,QAAQ,GACZ,6CAA6C;YAC7C,6CAA6C;YAC7C,8CAA8C;YAC9C,oDAAoD;YACpD,qDAAqD;YACrD,qDAAqD;YACrD,sDAAsD;YACtD,sDAAsD;YACtD,sDAAsD;YACtD,qDAAqD;YACrD,qDAAqD;YACrD,qDAAqD;YACrD,sDAAsD;YACtD,sDAAsD;YACtD,sDAAsD;YACtD,qDAAqD;YACrD,qDAAqD;YACrD,oDAAoD;YACpD,8CAA8C;YAC9C,6CAA6C;YAC7C,6CAA6C;YAC7C,wCAAwC,CAAC;QAC3C,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,cAAc,KAAK,QAAQ,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE;QAC9B,IAAI,IAAI,GAAG,EAAE,CAAC;QACd,IAAI,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;;;;;;;wBAOf,CAAC,CAAC;QACtB,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,GAAG,CAAC,CAAC;QAE5B,IAAI,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;;;;;wBAKf,CAAC,CAAC;QAEtB,IAAI,GAAG,GAAG,IAAI,oBAAoB,EAAE,CAAC;QACrC,oCAAoC;QACpC,GAAG,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,6CAA6C,CAAC,CAAC;QAC7F,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC;QAC1B,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;;;;;;;;4BAQb,CAAC,CAAC;YACxB,MAAM,CAAC,KAAK,CAAC,CAAC;QAChB,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,sDAAsD;QACtE,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtC,GAAG,GAAG,IAAI,oBAAoB,EAAE,CAAC;QACjC,oFAAoF;QACpF,GAAG,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,mDAAmD,CAAC,CAAC;QACnG,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE;;;;;4BAKb,CAAC,CAAC;YACxB,MAAM,CAAC,KAAK,CAAC,CAAC;QAChB,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,oDAAoD;QACpE,CAAC;QACD,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n* Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n* See LICENSE.md in the project root for license terms and full copyright notice.\n*--------------------------------------------------------------------------------------------*/\nimport { assert } from \"chai\";\nimport { QueryBinder, QueryRowFormat } from \"@itwin/core-common\";\nimport { IModelDb, SnapshotDb } from \"../../core-backend\";\nimport { IModelTestUtils } from \"../IModelTestUtils\";\nimport { SequentialLogMatcher } from \"../SequentialLogMatcher\";\n\n// cspell:ignore mirukuru ibim\n\nasync function executeQuery(iModel: IModelDb, ecsql: string, bindings?: any[] | object, abbreviateBlobs?: boolean): Promise<any[]> {\n  const rows: any[] = [];\n  for await (const queryRow of iModel.createQueryReader(ecsql, QueryBinder.from(bindings), { rowFormat: QueryRowFormat.UseJsPropertyNames, abbreviateBlobs })) {\n    rows.push(queryRow.toRow());\n  }\n  return rows;\n}\n\ndescribe(\"Common table expression support in ECSQL\", () => {\n  let imodel1: SnapshotDb;\n\n  before(async () => {\n    imodel1 = SnapshotDb.openFile(IModelTestUtils.resolveAssetFile(\"test.bim\"));\n  });\n\n  after(async () => {\n    imodel1.close();\n  });\n  it(\"collect base properties recursively\", async () => {\n    const query = `\n        WITH RECURSIVE\n            base_classes (aId, aParentId, aPath, aDepth) AS (\n                SELECT c.ECInstanceId, null, c.Name, 0  FROM meta.ECClassDef c WHERE c.Name=?\n                UNION ALL\n                SELECT c.ECInstanceId, cbc.TargetECInstanceId, aPath || '/' || c.Name, aDepth + 1\n                    FROM meta.ECClassDef c\n                        JOIN meta.ClassHasBaseClasses cbc ON cbc.SourceECInstanceId = c.ECInstanceId\n                        JOIN base_classes  ON aId = cbc.TargetECInstanceId\n                ORDER BY 1\n            )\n        SELECT p.Name prop from base_classes join meta.ECPropertyDef p on p.Class.id = aId GROUP BY p.Name` ;\n    const rows = await executeQuery(imodel1, query, [\"Element\"]);\n    const expected = [\"Angle\", \"BaseModel\", \"BBoxHigh\", \"BBoxLow\", \"Border\", \"BorderTemplate\", \"Category\", \"CategorySelector\", \"CodeScope\", \"CodeSpec\", \"CodeValue\", \"Data\", \"Description\", \"DisplayStyle\", \"DrawingModel\", \"Enabled\", \"Extents\", \"EyePoint\", \"FederationGuid\", \"Flags\", \"FocusDistance\", \"Format\", \"GeometryStream\", \"Height\", \"InSpatialIndex\", \"IsCameraOn\", \"IsPrivate\", \"JsonProperties\", \"LastMod\", \"LensAngle\", \"Model\", \"ModelSelector\", \"Name\", \"Origin\", \"PaletteName\", \"Parent\", \"Pitch\", \"Properties\", \"Rank\", \"Recipe\", \"RepositoryGuid\", \"Roll\", \"Rotation\", \"RotationAngle\", \"Scale\", \"SheetTemplate\", \"Type\", \"TypeDefinition\", \"Url\", \"UserLabel\", \"View\", \"ViewAttachment\", \"Width\", \"Yaw\"];\n    const actual = rows.map((r) => r.prop);\n    assert.sameOrderedMembers(actual, expected);\n  });\n\n  it(\"generate mandelbrot set\", async () => {\n    const rows = await executeQuery(imodel1, `\n      WITH RECURSIVE\n        [xaxis]([x]) AS(\n          VALUES (- 2.0)\n          UNION ALL\n          SELECT [x] + 0.05\n          FROM   [xaxis]\n          WHERE  [x] < 1.2\n        ),\n        [yaxis]([y]) AS(\n          VALUES (- 1.0)\n          UNION ALL\n          SELECT [y] + 0.1\n          FROM   [yaxis]\n          WHERE  [y] < 1.0\n        ),\n        [m]([iter], [cx], [cy], [x], [y]) AS(\n          SELECT\n                0,\n                [x],\n                [y],\n                0.0,\n                0.0\n          FROM   [xaxis],\n                [yaxis]\n          UNION ALL\n          SELECT\n                [iter] + 1,\n                [cx],\n                [cy],\n                [x] * [x] - [y] * [y] + [cx],\n                2.0 * [x] * [y] + [cy]\n          FROM   [m]\n          WHERE  ([x] * [x] + [y] * [y]) < 4.0 AND [iter] < 28\n        ),\n        [m2]([iter], [cx], [cy]) AS(\n          SELECT\n                MAX ([iter]),\n                [cx],\n                [cy]\n          FROM   [m]\n          GROUP  BY\n                    [cx],\n                    [cy]\n        ),\n        [a]([t]) AS(\n          SELECT GROUP_CONCAT (SUBSTR (' .+*#', 1 + (CASE WHEN [iter] / 7 > 4 THEN 4 ELSE [iter] / 7 END), 1), '')\n          FROM   [m2]\n          GROUP  BY [cy]\n        )\n      SELECT GROUP_CONCAT (RTRIM ([t]), CHAR (0xa)) mandelbrot_set\n      FROM   [a];\n    `);\n\n    const expected =\n      \"                                    ....#\\n\" +\n      \"                                   ..#*..\\n\" +\n      \"                                 ..+####+.\\n\" +\n      \"                            .......+####....   +\\n\" +\n      \"                           ..##+*##########+.++++\\n\" +\n      \"                          .+.##################+.\\n\" +\n      \"              .............+###################+.+\\n\" +\n      \"              ..++..#.....*#####################+.\\n\" +\n      \"             ...+#######++#######################.\\n\" +\n      \"          ....+*################################.\\n\" +\n      \" #############################################...\\n\" +\n      \"          ....+*################################.\\n\" +\n      \"             ...+#######++#######################.\\n\" +\n      \"              ..++..#.....*#####################+.\\n\" +\n      \"              .............+###################+.+\\n\" +\n      \"                          .+.##################+.\\n\" +\n      \"                           ..##+*##########+.++++\\n\" +\n      \"                            .......+####....   +\\n\" +\n      \"                                 ..+####+.\\n\" +\n      \"                                   ..#*..\\n\" +\n      \"                                    ....#\\n\" +\n      \"                                    +.\";\n    assert(rows[0].mandelbrot_set === expected);\n  });\n\n  it(\"basic cte test\", async () => {\n    let rows = [];\n    rows = await executeQuery(imodel1, `\n      WITH RECURSIVE\n        cnt (x,y) AS (\n            SELECT 100, 200\n            UNION ALL\n            SELECT x+1, 200 FROM cnt WHERE x<210\n        )\n      SELECT * from cnt`);\n    assert(rows.length === 111);\n\n    rows = await executeQuery(imodel1, `\n      WITH RECURSIVE\n        cnt (x,y) AS (\n            SELECT 100, 200\n        )\n      SELECT * from cnt`);\n\n    let slm = new SequentialLogMatcher();\n    // these two are generated by sqlite\n    slm.append().error().category(\"ECDb\").message(/BE_SQLITE_ERROR duplicate WITH table name/gm);\n    assert(rows.length === 1);\n    try {\n      rows = await executeQuery(imodel1, `\n        WITH\n          cte_1 (a,b,c) AS (\n            SELECT 100, 400, 300\n          ),\n          cte_1 (a,b,c) AS (\n            SELECT 100, 400, 300\n          )\n        SELECT * from cte_1`);\n      assert(false);\n    } catch {\n      assert(true); // should fail as cte_1 is used for two ct expression.\n    }\n    assert.isTrue(slm.finishAndDispose());\n    slm = new SequentialLogMatcher();\n    // these two are generated by ECSQL. Its not clear why this message is logged twice.\n    slm.append().error().category(\"ECDb\").message(/Common table 'cte_1' has 3 values for columns 2/gm);\n    try {\n      rows = await executeQuery(imodel1, `\n        WITH\n        cte_1 (a,b,c) AS (\n          SELECT 100, 400\n        )\n        SELECT * from cte_1`);\n      assert(false);\n    } catch {\n      assert(true); // number are to ct expression does not match select\n    }\n    assert.isTrue(slm.finishAndDispose());\n  });\n});\n"]}

package/lib/esm/test/ecdb/ConcurrentQuery.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=ConcurrentQuery.test.d.ts.map

package/lib/esm/test/ecdb/ConcurrentQuery.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConcurrentQuery.test.d.ts","sourceRoot":"","sources":["../../../../src/test/ecdb/ConcurrentQuery.test.ts"],"names":[],"mappings":""}

package/lib/esm/test/ecdb/ConcurrentQuery.test.js

@@ -0,0 +1,164 @@
+import { DbResponseStatus } from "@itwin/core-common";
+import { expect } from "chai";
+import { ConcurrentQuery } from "../../ConcurrentQuery";
+import { SnapshotDb } from "../../IModelDb";
+import { _nativeDb } from "../../core-backend";
+import { IModelTestUtils } from "../IModelTestUtils";
+async function delay(ms) {
+    return new Promise(resolve => {
+        setTimeout(resolve, ms);
+    });
+}
+describe("ConcurrentQuery", () => {
+    it("default config", () => {
+        const testFile = IModelTestUtils.resolveAssetFile("test.bim");
+        const db = SnapshotDb.openFile(testFile);
+        const defaultConfig = {
+            autoShutdownWhenIdleForSeconds: 1800,
+            doNotUsePrimaryConnToPrepare: false,
+            globalQuota: { time: 60, memory: 8388608 },
+            ignoreDelay: true,
+            ignorePriority: false,
+            memoryMapFileSize: 0,
+            monitorPollInterval: 5000,
+            progressOpCount: 5000,
+            requestQueueSize: 2000,
+            statementCacheSizePerWorker: 40,
+            workerThreads: 4,
+        };
+        const config = ConcurrentQuery.resetConfig(db[_nativeDb], {});
+        expect(config).deep.eq(defaultConfig);
+        db.close();
+    });
+    it("modify config", () => {
+        const testFile = IModelTestUtils.resolveAssetFile("test.bim");
+        const db = SnapshotDb.openFile(testFile);
+        const modifiedConfig = {
+            autoShutdownWhenIdleForSeconds: 100,
+            doNotUsePrimaryConnToPrepare: true,
+            globalQuota: { time: 10, memory: 1000000 },
+            ignoreDelay: false,
+            ignorePriority: true,
+            memoryMapFileSize: 100,
+            monitorPollInterval: 2000,
+            progressOpCount: 6000,
+            requestQueueSize: 1000,
+            statementCacheSizePerWorker: 20,
+            workerThreads: 3,
+        };
+        const config = ConcurrentQuery.resetConfig(db[_nativeDb], modifiedConfig);
+        expect(config).deep.eq(modifiedConfig);
+        db.close();
+    });
+    it("time limit check", async () => {
+        const testFile = IModelTestUtils.resolveAssetFile("test.bim");
+        const db = SnapshotDb.openFile(testFile);
+        ConcurrentQuery.resetConfig(db[_nativeDb], { globalQuota: { time: 1, memory: 100000 }, progressOpCount: 1000 });
+        // await runSingleRequest(db, `SELECT 1`);
+        const req = {
+            query: `WITH sequence(n,k) AS (
+                SELECT  1,1 UNION ALL SELECT n + 1, random() FROM sequence WHERE n < 10000000
+              ) SELECT COUNT(*) FROM sequence s`
+        };
+        const resp = await ConcurrentQuery.executeQueryRequest(db[_nativeDb], req);
+        expect(resp.status).equals(DbResponseStatus.Partial);
+        expect(resp.stats.timeLimit).equals(1000);
+        expect(resp.stats.memLimit).equals(100000);
+        expect(resp.stats.cpuTime).to.be.closeTo(1000970, 500000);
+        expect(resp.stats.totalTime).to.be.closeTo(1001, 100);
+        expect(resp.stats.memUsed).to.be.closeTo(2, 3);
+        expect(resp.stats.prepareTime).to.be.closeTo(0, 2);
+        db.close();
+    });
+    it("memory limit check", async () => {
+        const testFile = IModelTestUtils.resolveAssetFile("test.bim");
+        const db = SnapshotDb.openFile(testFile);
+        ConcurrentQuery.resetConfig(db[_nativeDb], { globalQuota: { time: 60, memory: 1000 }, progressOpCount: 1000 });
+        // await runSingleRequest(db, `SELECT 1`);
+        const req = {
+            query: `WITH sequence(n) AS (
+                SELECT  1
+                UNION ALL
+                SELECT n + 1 FROM sequence WHERE n < 10000000
+              )
+              SELECT 'xxxxxxxxxx-xxxxxxxxxx-xxxxxxxxxx' FROM sequence s`
+        };
+        const resp = await ConcurrentQuery.executeQueryRequest(db[_nativeDb], req);
+        expect(resp.status).equals(DbResponseStatus.Partial);
+        expect(resp.stats.timeLimit).equals(60000);
+        expect(resp.stats.memLimit).equals(1000);
+        expect(resp.stats.memUsed).to.be.closeTo(1037, 100);
+        db.close();
+    });
+    it("prepare error", async () => {
+        const testFile = IModelTestUtils.resolveAssetFile("test.bim");
+        const db = SnapshotDb.openFile(testFile);
+        const req = {
+            query: `xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx`
+        };
+        const resp = await ConcurrentQuery.executeQueryRequest(db[_nativeDb], req);
+        expect(resp.status).equals(DbResponseStatus.Error_ECSql_PreparedFailed);
+        db.close();
+    });
+    it("restart query", async () => {
+        const testFile = IModelTestUtils.resolveAssetFile("test.bim");
+        const db = SnapshotDb.openFile(testFile);
+        ConcurrentQuery.resetConfig(db[_nativeDb], { globalQuota: { time: 60, memory: 10000000 }, progressOpCount: 1000 });
+        const req0 = {
+            query: `WITH sequence(n) AS (
+                SELECT  1 UNION ALL SELECT n + 1 FROM sequence WHERE n < 10000000
+              ) SELECT n FROM sequence s`,
+            restartToken: "Blah",
+        };
+        const req1 = {
+            query: `WITH sequence(n) AS (
+                SELECT  1 UNION ALL SELECT n + 1 FROM sequence WHERE n < 1000
+              ) SELECT n FROM sequence s`,
+            restartToken: "Blah",
+        };
+        const resp1 = ConcurrentQuery.executeQueryRequest(db[_nativeDb], req0);
+        await delay(1);
+        const resp2 = ConcurrentQuery.executeQueryRequest(db[_nativeDb], req1);
+        const resp = await Promise.all([resp1, resp2]);
+        expect(resp[0].status).equals(DbResponseStatus.Cancel);
+        expect(resp[1].status).equals(DbResponseStatus.Done);
+        db.close();
+    });
+    it("queue limit check", async () => {
+        const testFile = IModelTestUtils.resolveAssetFile("test.bim");
+        const db = SnapshotDb.openFile(testFile);
+        ConcurrentQuery.resetConfig(db[_nativeDb], { requestQueueSize: 40, globalQuota: { time: 5, memory: 100000 } });
+        const req = {
+            query: `WITH sequence(n) AS (
+                SELECT  1 UNION ALL SELECT n + 1 FROM sequence WHERE n < 10000000
+              ) SELECT n FROM sequence s`,
+        };
+        const responsePromises = [];
+        for (let i = 0; i < 60; ++i) {
+            responsePromises.push(ConcurrentQuery.executeQueryRequest(db[_nativeDb], req));
+        }
+        const responses = await Promise.all(responsePromises);
+        const queueResponses = Array.from(responses.filter((x) => x.status === DbResponseStatus.QueueFull));
+        expect(queueResponses.length).to.be.greaterThanOrEqual(10);
+        db.close();
+    });
+    it("timeout check", async () => {
+        const testFile = IModelTestUtils.resolveAssetFile("test.bim");
+        const db = SnapshotDb.openFile(testFile);
+        ConcurrentQuery.resetConfig(db[_nativeDb], { monitorPollInterval: 1, globalQuota: { time: 5, memory: 10000000 } });
+        const req = {
+            query: `WITH sequence(n) AS (
+                SELECT  1 UNION ALL SELECT n + 1 FROM sequence WHERE n < 10000000
+              ) SELECT n FROM sequence s`,
+        };
+        const responsePromises = [];
+        for (let i = 0; i < 100; ++i) {
+            responsePromises.push(ConcurrentQuery.executeQueryRequest(db[_nativeDb], req));
+        }
+        const responses = await Promise.all(responsePromises);
+        const queueResponses = Array.from(responses.filter((x) => x.status === DbResponseStatus.Timeout));
+        expect(queueResponses.length).to.be.greaterThanOrEqual(10);
+        db.close();
+    });
+});
+//# sourceMappingURL=ConcurrentQuery.test.js.map

package/lib/esm/test/ecdb/ConcurrentQuery.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ConcurrentQuery.test.js","sourceRoot":"","sources":["../../../../src/test/ecdb/ConcurrentQuery.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAmC,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACvF,OAAO,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAC9B,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAErD,KAAK,UAAU,KAAK,CAAC,EAAU;IAC7B,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE;QAC3B,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;AACL,CAAC;AAED,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,gBAAgB,EAAE,GAAG,EAAE;QACxB,MAAM,QAAQ,GAAG,eAAe,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QAC9D,MAAM,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,aAAa,GAAG;YACpB,8BAA8B,EAAE,IAAI;YACpC,4BAA4B,EAAE,KAAK;YACnC,WAAW,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE;YAC1C,WAAW,EAAE,IAAI;YACjB,cAAc,EAAE,KAAK;YACrB,iBAAiB,EAAE,CAAC;YACpB,mBAAmB,EAAE,IAAI;YACzB,eAAe,EAAE,IAAI;YACrB,gBAAgB,EAAE,IAAI;YACtB,2BAA2B,EAAE,EAAE;YAC/B,aAAa,EAAE,CAAC;SACjB,CAAC;QACF,MAAM,MAAM,GAAG,eAAe,CAAC,WAAW,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC;QAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC;QACtC,EAAE,CAAC,KAAK,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,eAAe,EAAE,GAAG,EAAE;QACvB,MAAM,QAAQ,GAAG,eAAe,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QAC9D,MAAM,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,cAAc,GAAG;YACrB,8BAA8B,EAAE,GAAG;YACnC,4BAA4B,EAAE,IAAI;YAClC,WAAW,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE;YAC1C,WAAW,EAAE,KAAK;YAClB,cAAc,EAAE,IAAI;YACpB,iBAAiB,EAAE,GAAG;YACtB,mBAAmB,EAAE,IAAI;YACzB,eAAe,EAAE,IAAI;YACrB,gBAAgB,EAAE,IAAI;YACtB,2BAA2B,EAAE,EAAE;YAC/B,aAAa,EAAE,CAAC;SACjB,CAAC;QACF,MAAM,MAAM,GAAG,eAAe,CAAC,WAAW,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,cAAc,CAAC,CAAC;QAC1E,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,cAAc,CAAC,CAAC;QACvC,EAAE,CAAC,KAAK,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE;QAChC,MAAM,QAAQ,GAAG,eAAe,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QAC9D,MAAM,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACzC,eAAe,CAAC,WAAW,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;QAChH,0CAA0C;QAC1C,MAAM,GAAG,GAAmB;YAC1B,KAAK,EAAE;;gDAEmC;SAC3C,CAAC;QAEF,MAAM,IAAI,GAAG,MAAM,eAAe,CAAC,mBAAmB,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,CAAC;QAC3E,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC3C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC1D,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QACtD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACnD,EAAE,CAAC,KAAK,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;QAClC,MAAM,QAAQ,GAAG,eAAe,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QAC9D,MAAM,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACzC,eAAe,CAAC,WAAW,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/G,0CAA0C;QAC1C,MAAM,GAAG,GAAmB;YAC1B,KAAK,EAAE;;;;;wEAK2D;SACnE,CAAC;QAEF,MAAM,IAAI,GAAG,MAAM,eAAe,CAAC,mBAAmB,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,CAAC;QAC3E,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC3C,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QACpD,EAAE,CAAC,KAAK,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE;QAC7B,MAAM,QAAQ,GAAG,eAAe,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QAC9D,MAAM,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACzC,MAAM,GAAG,GAAmB;YAC1B,KAAK,EAAE,gCAAgC;SACxC,CAAC;QAEF,MAAM,IAAI,GAAG,MAAM,eAAe,CAAC,mBAAmB,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,CAAC;QAC3E,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,0BAA0B,CAAC,CAAC;QACxE,EAAE,CAAC,KAAK,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE;QAC7B,MAAM,QAAQ,GAAG,eAAe,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QAC9D,MAAM,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACzC,eAAe,CAAC,WAAW,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;QACnH,MAAM,IAAI,GAAmB;YAC3B,KAAK,EAAE;;yCAE4B;YACnC,YAAY,EAAE,MAAM;SACrB,CAAC;QAEF,MAAM,IAAI,GAAmB;YAC3B,KAAK,EAAE;;yCAE4B;YACnC,YAAY,EAAE,MAAM;SACrB,CAAC;QAEF,MAAM,KAAK,GAAG,eAAe,CAAC,mBAAmB,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,CAAC;QACvE,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC;QACf,MAAM,KAAK,GAAG,eAAe,CAAC,mBAAmB,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,CAAC;QACvE,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;QACvD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QACrD,EAAE,CAAC,KAAK,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;QACjC,MAAM,QAAQ,GAAG,eAAe,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QAC9D,MAAM,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACzC,eAAe,CAAC,WAAW,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;QAC/G,MAAM,GAAG,GAAmB;YAC1B,KAAK,EAAE;;yCAE4B;SACpC,CAAC;QAEF,MAAM,gBAAgB,GAA+B,EAAE,CAAC;QACxD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;YAC5B,gBAAgB,CAAC,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;QACjF,CAAC;QACD,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACtD,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,gBAAgB,CAAC,SAAS,CAAC,CAAC,CAAC;QACpG,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;QAC3D,EAAE,CAAC,KAAK,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,eAAe,EAAE,KAAK,IAAI,EAAE;QAC7B,MAAM,QAAQ,GAAG,eAAe,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QAC9D,MAAM,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACzC,eAAe,CAAC,WAAW,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,EAAE,mBAAmB,EAAE,CAAC,EAAE,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;QACnH,MAAM,GAAG,GAAmB;YAC1B,KAAK,EAAE;;yCAE4B;SACpC,CAAC;QAEF,MAAM,gBAAgB,GAA+B,EAAE,CAAC;QACxD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC;YAC7B,gBAAgB,CAAC,IAAI,CAAC,eAAe,CAAC,mBAAmB,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;QACjF,CAAC;QACD,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACtD,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;QAClG,MAAM,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;QAC3D,EAAE,CAAC,KAAK,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { DbQueryRequest, DbQueryResponse, DbResponseStatus } from \"@itwin/core-common\";\nimport { expect } from \"chai\";\nimport { ConcurrentQuery } from \"../../ConcurrentQuery\";\nimport { SnapshotDb } from \"../../IModelDb\";\nimport { _nativeDb } from \"../../core-backend\";\nimport { IModelTestUtils } from \"../IModelTestUtils\";\n\nasync function delay(ms: number): Promise<void> {\n  return new Promise(resolve => {\n    setTimeout(resolve, ms);\n  });\n}\n\ndescribe(\"ConcurrentQuery\", () => {\n  it(\"default config\", () => {\n    const testFile = IModelTestUtils.resolveAssetFile(\"test.bim\");\n    const db = SnapshotDb.openFile(testFile);\n    const defaultConfig = {\n      autoShutdownWhenIdleForSeconds: 1800,\n      doNotUsePrimaryConnToPrepare: false,\n      globalQuota: { time: 60, memory: 8388608 },\n      ignoreDelay: true,\n      ignorePriority: false,\n      memoryMapFileSize: 0,\n      monitorPollInterval: 5000,\n      progressOpCount: 5000,\n      requestQueueSize: 2000,\n      statementCacheSizePerWorker: 40,\n      workerThreads: 4,\n    };\n    const config = ConcurrentQuery.resetConfig(db[_nativeDb], {});\n    expect(config).deep.eq(defaultConfig);\n    db.close();\n  });\n\n  it(\"modify config\", () => {\n    const testFile = IModelTestUtils.resolveAssetFile(\"test.bim\");\n    const db = SnapshotDb.openFile(testFile);\n    const modifiedConfig = {\n      autoShutdownWhenIdleForSeconds: 100,\n      doNotUsePrimaryConnToPrepare: true,\n      globalQuota: { time: 10, memory: 1000000 },\n      ignoreDelay: false,\n      ignorePriority: true,\n      memoryMapFileSize: 100,\n      monitorPollInterval: 2000,\n      progressOpCount: 6000,\n      requestQueueSize: 1000,\n      statementCacheSizePerWorker: 20,\n      workerThreads: 3,\n    };\n    const config = ConcurrentQuery.resetConfig(db[_nativeDb], modifiedConfig);\n    expect(config).deep.eq(modifiedConfig);\n    db.close();\n  });\n\n  it(\"time limit check\", async () => {\n    const testFile = IModelTestUtils.resolveAssetFile(\"test.bim\");\n    const db = SnapshotDb.openFile(testFile);\n    ConcurrentQuery.resetConfig(db[_nativeDb], { globalQuota: { time: 1, memory: 100000 }, progressOpCount: 1000 });\n    // await runSingleRequest(db, `SELECT 1`);\n    const req: DbQueryRequest = {\n      query: `WITH sequence(n,k) AS (\n                SELECT  1,1 UNION ALL SELECT n + 1, random() FROM sequence WHERE n < 10000000\n              ) SELECT COUNT(*) FROM sequence s`\n    };\n\n    const resp = await ConcurrentQuery.executeQueryRequest(db[_nativeDb], req);\n    expect(resp.status).equals(DbResponseStatus.Partial);\n    expect(resp.stats.timeLimit).equals(1000);\n    expect(resp.stats.memLimit).equals(100000);\n    expect(resp.stats.cpuTime).to.be.closeTo(1000970, 500000);\n    expect(resp.stats.totalTime).to.be.closeTo(1001, 100);\n    expect(resp.stats.memUsed).to.be.closeTo(2, 3);\n    expect(resp.stats.prepareTime).to.be.closeTo(0, 2);\n    db.close();\n  });\n\n  it(\"memory limit check\", async () => {\n    const testFile = IModelTestUtils.resolveAssetFile(\"test.bim\");\n    const db = SnapshotDb.openFile(testFile);\n    ConcurrentQuery.resetConfig(db[_nativeDb], { globalQuota: { time: 60, memory: 1000 }, progressOpCount: 1000 });\n    // await runSingleRequest(db, `SELECT 1`);\n    const req: DbQueryRequest = {\n      query: `WITH sequence(n) AS (\n                SELECT  1\n                UNION ALL\n                SELECT n + 1 FROM sequence WHERE n < 10000000\n              )\n              SELECT 'xxxxxxxxxx-xxxxxxxxxx-xxxxxxxxxx' FROM sequence s`\n    };\n\n    const resp = await ConcurrentQuery.executeQueryRequest(db[_nativeDb], req);\n    expect(resp.status).equals(DbResponseStatus.Partial);\n    expect(resp.stats.timeLimit).equals(60000);\n    expect(resp.stats.memLimit).equals(1000);\n    expect(resp.stats.memUsed).to.be.closeTo(1037, 100);\n    db.close();\n  });\n\n  it(\"prepare error\", async () => {\n    const testFile = IModelTestUtils.resolveAssetFile(\"test.bim\");\n    const db = SnapshotDb.openFile(testFile);\n    const req: DbQueryRequest = {\n      query: `xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx`\n    };\n\n    const resp = await ConcurrentQuery.executeQueryRequest(db[_nativeDb], req);\n    expect(resp.status).equals(DbResponseStatus.Error_ECSql_PreparedFailed);\n    db.close();\n  });\n\n  it(\"restart query\", async () => {\n    const testFile = IModelTestUtils.resolveAssetFile(\"test.bim\");\n    const db = SnapshotDb.openFile(testFile);\n    ConcurrentQuery.resetConfig(db[_nativeDb], { globalQuota: { time: 60, memory: 10000000 }, progressOpCount: 1000 });\n    const req0: DbQueryRequest = {\n      query: `WITH sequence(n) AS (\n                SELECT  1 UNION ALL SELECT n + 1 FROM sequence WHERE n < 10000000\n              ) SELECT n FROM sequence s`,\n      restartToken: \"Blah\",\n    };\n\n    const req1: DbQueryRequest = {\n      query: `WITH sequence(n) AS (\n                SELECT  1 UNION ALL SELECT n + 1 FROM sequence WHERE n < 1000\n              ) SELECT n FROM sequence s`,\n      restartToken: \"Blah\",\n    };\n\n    const resp1 = ConcurrentQuery.executeQueryRequest(db[_nativeDb], req0);\n    await delay(1);\n    const resp2 = ConcurrentQuery.executeQueryRequest(db[_nativeDb], req1);\n    const resp = await Promise.all([resp1, resp2]);\n    expect(resp[0].status).equals(DbResponseStatus.Cancel);\n    expect(resp[1].status).equals(DbResponseStatus.Done);\n    db.close();\n  });\n\n  it(\"queue limit check\", async () => {\n    const testFile = IModelTestUtils.resolveAssetFile(\"test.bim\");\n    const db = SnapshotDb.openFile(testFile);\n    ConcurrentQuery.resetConfig(db[_nativeDb], { requestQueueSize: 40, globalQuota: { time: 5, memory: 100000 } });\n    const req: DbQueryRequest = {\n      query: `WITH sequence(n) AS (\n                SELECT  1 UNION ALL SELECT n + 1 FROM sequence WHERE n < 10000000\n              ) SELECT n FROM sequence s`,\n    };\n\n    const responsePromises: Promise<DbQueryResponse>[] = [];\n    for (let i = 0; i < 60; ++i) {\n      responsePromises.push(ConcurrentQuery.executeQueryRequest(db[_nativeDb], req));\n    }\n    const responses = await Promise.all(responsePromises);\n    const queueResponses = Array.from(responses.filter((x) => x.status === DbResponseStatus.QueueFull));\n    expect(queueResponses.length).to.be.greaterThanOrEqual(10);\n    db.close();\n  });\n\n  it(\"timeout check\", async () => {\n    const testFile = IModelTestUtils.resolveAssetFile(\"test.bim\");\n    const db = SnapshotDb.openFile(testFile);\n    ConcurrentQuery.resetConfig(db[_nativeDb], { monitorPollInterval: 1, globalQuota: { time: 5, memory: 10000000 } });\n    const req: DbQueryRequest = {\n      query: `WITH sequence(n) AS (\n                SELECT  1 UNION ALL SELECT n + 1 FROM sequence WHERE n < 10000000\n              ) SELECT n FROM sequence s`,\n    };\n\n    const responsePromises: Promise<DbQueryResponse>[] = [];\n    for (let i = 0; i < 100; ++i) {\n      responsePromises.push(ConcurrentQuery.executeQueryRequest(db[_nativeDb], req));\n    }\n    const responses = await Promise.all(responsePromises);\n    const queueResponses = Array.from(responses.filter((x) => x.status === DbResponseStatus.Timeout));\n    expect(queueResponses.length).to.be.greaterThanOrEqual(10);\n    db.close();\n  });\n});\n"]}

package/lib/esm/test/ecdb/ConcurrentQueryLoad.test.js

@@ -6,10 +6,10 @@ import { _nativeDb } from "../../core-backend";
 import { IModelTestUtils } from "../IModelTestUtils";
 class LoadSimulator {
     db;
-    senario;
-    constructor(db, senario) {
+    scenario;
+    constructor(db, scenario) {
         this.db = db;
-        this.senario = senario;
+        this.scenario = scenario;
     }
     async runQueryTask(reader) {
         try {
@@ -22,23 +22,23 @@ class LoadSimulator {
     }
     async run() {
         ConcurrentQuery.shutdown(this.db[_nativeDb]);
-        if (this.senario.config) {
-            const config = ConcurrentQuery.resetConfig(this.db[_nativeDb], this.senario.config);
+        if (this.scenario.config) {
+            const config = ConcurrentQuery.resetConfig(this.db[_nativeDb], this.scenario.config);
             // eslint-disable-next-line no-console
             console.log(config);
         }
-        const overalTime = new StopWatch();
-        overalTime.start();
+        const overallTime = new StopWatch();
+        overallTime.start();
         const results = [];
-        for (let i = 0; i < this.senario.totalBatches; ++i) {
+        for (let i = 0; i < this.scenario.totalBatches; ++i) {
             const promises = [];
-            const readerTasks = Array(this.senario.taskPerBatch).fill(undefined).map(() => this.senario.createReader(this.db));
+            const readerTasks = Array(this.scenario.taskPerBatch).fill(undefined).map(() => this.scenario.createReader(this.db));
             readerTasks.forEach((reader) => {
                 promises.push(this.runQueryTask(reader));
             });
             results.push(...await Promise.all(promises));
         }
-        overalTime.stop();
+        overallTime.stop();
         const errors = results.filter((x) => x.error !== undefined);
         const errorsMap = new Map();
         errors.forEach((x) => {
@@ -94,7 +94,7 @@ class LoadSimulator {
                 retryCount,
                 prepareTime,
             },
-            overalTimeInSec: overalTime.currentSeconds,
+            overallTimeInSec: overallTime.currentSeconds,
             errorCount,
             totalQueries: results.length,
             errorMap: errorsMap
@@ -105,58 +105,48 @@ describe.skip("ConcurrentQueryLoad", () => {
     it("should run", async () => {
         Logger.initializeToConsole();
         Logger.setLevel("ECDb.ConcurrentQuery", LogLevel.Trace);
-        // {
-        //   workerThreads: 4,
-        //   requestQueueSize: 2000,
-        //   ignorePriority: false,
-        //   ignoreDelay: true,
-        //   doNotUsePrimaryConnToPrepare: false,
-        //   autoShutdowWhenIdlelForSeconds: 300,
-        //   statementCacheSizePerWorker: 40,
-        //   monitorPollInterval: 1000,
-        //   memoryMapFileSize: 0,
-        //   allowTestingArgs: false,
-        //   globalQuota: { time: 60, memory: 8388608 }
-        // }
-        const senario = {
+        const scenario = {
             name: "ConcurrentQueryLoad",
-            config: {},
+            config: {
+                globalQuota: { time: 1, memory: 8388608 },
+                workerThreads: 1,
+            },
             totalBatches: 1,
             taskPerBatch: 1,
             createReader: (dbs) => {
-                const quries = [
+                const queries = [
                     {
-                        sql: `
-            WITH sequence(n) AS (
-              SELECT  1
-              UNION ALL
-              SELECT n + 1 FROM sequence WHERE n < 10000
-            )
-            SELECT  COUNT(*)
-            FROM bis.SpatialIndex i, sequence s
-            WHERE i.ECInstanceId MATCH  iModel_spatial_overlap_aabb(
+                        sql: `
+            WITH sequence(n) AS (
+              SELECT  1
+              UNION ALL
+              SELECT n + 1 FROM sequence WHERE n < 10000000
+            )
+            SELECT  COUNT(*)
+            FROM bis.SpatialIndex i, sequence s
+            WHERE i.ECInstanceId MATCH  iModel_spatial_overlap_aabb(
               iModel_bbox(random(), random(), random(), random(),random(), random()))`
                     },
                     {
-                        sql: `
-            WITH sequence(n) AS (
-              SELECT  1
-              UNION ALL
-              SELECT n + 1 FROM sequence WHERE n < 1000000
-            )
+                        sql: `
+            WITH sequence(n) AS (
+              SELECT  1
+              UNION ALL
+              SELECT n + 1 FROM sequence WHERE n < 10000000
+            )
             SELECT  COUNT(*) FROM sequence`
                     },
                     {
                         sql: "SELECT $ FROM bis.Element LIMIT 10000"
                     }
                 ];
-                const idx = Math.floor(Math.random() * quries.length);
-                return dbs.createQueryReader(quries[idx].sql);
+                const idx = Math.floor(Math.random() * queries.length);
+                return dbs.createQueryReader(queries[idx].sql);
             }
         };
         const verySmallFile = IModelTestUtils.resolveAssetFile("test.bim");
         const db = SnapshotDb.openFile(verySmallFile);
-        const simulator = new LoadSimulator(db, senario);
+        const simulator = new LoadSimulator(db, scenario);
         const result = await simulator.run();
         // eslint-disable-next-line no-console
         console.log(result);