@itwin/core-backend 5.2.0-dev.8 → 5.2.0

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":["/*---------------------------------------------------------------------------------------------\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"]}
+{"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"]}

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

@@ -56,8 +56,8 @@ describe("ConcurrentQuery", () => {
         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
+            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);
@@ -76,11 +76,11 @@ describe("ConcurrentQuery", () => {
         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
-              )
+            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);
@@ -105,14 +105,14 @@ describe("ConcurrentQuery", () => {
         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
+            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
+            query: `WITH sequence(n) AS (
+                SELECT  1 UNION ALL SELECT n + 1 FROM sequence WHERE n < 1000
               ) SELECT n FROM sequence s`,
             restartToken: "Blah",
         };
@@ -129,8 +129,8 @@ describe("ConcurrentQuery", () => {
         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
+            query: `WITH sequence(n) AS (
+                SELECT  1 UNION ALL SELECT n + 1 FROM sequence WHERE n < 10000000
               ) SELECT n FROM sequence s`,
         };
         const responsePromises = [];
@@ -147,8 +147,8 @@ describe("ConcurrentQuery", () => {
         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
+            query: `WITH sequence(n) AS (
+                SELECT  1 UNION ALL SELECT n + 1 FROM sequence WHERE n < 10000000
               ) SELECT n FROM sequence s`,
         };
         const responsePromises = [];

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

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

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

@@ -116,24 +116,24 @@ describe.skip("ConcurrentQueryLoad", () => {
             createReader: (dbs) => {
                 const queries = [
                     {
-                        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(
+                        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 < 10000000
-            )
+                        sql: `
+            WITH sequence(n) AS (
+              SELECT  1
+              UNION ALL
+              SELECT n + 1 FROM sequence WHERE n < 10000000
+            )
             SELECT  COUNT(*) FROM sequence`
                     },
                     {

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

@@ -1 +1 @@
-{"version":3,"file":"ConcurrentQueryLoad.test.js","sourceRoot":"","sources":["../../../../src/test/ecdb/ConcurrentQueryLoad.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAElE,OAAO,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAC9B,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAExD,OAAO,EAAY,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACtD,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAerD,MAAM,aAAa;IACE;IAA4B;IAA/C,YAAmB,EAAmB,EAAS,QAAmB;QAA/C,OAAE,GAAF,EAAE,CAAiB;QAAS,aAAQ,GAAR,QAAQ,CAAW;IAAI,CAAC;IAC/D,KAAK,CAAC,YAAY,CAAC,MAAmB;QAC5C,IAAI,CAAC;YACH,OAAO,MAAM,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;YAC/B,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;QAC7C,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,GAAG;QACd,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QAC7C,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YACzB,MAAM,MAAM,GAAG,eAAe,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YACrF,sCAAsC;YACtC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACtB,CAAC;QACD,MAAM,WAAW,GAAG,IAAI,SAAS,EAAE,CAAC;QACpC,WAAW,CAAC,KAAK,EAAE,CAAC;QACpB,MAAM,OAAO,GAAkB,EAAE,CAAC;QAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC;YACpD,MAAM,QAAQ,GAA2B,EAAE,CAAC;YAC5C,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;YACrH,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBAC7B,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC;YAC3C,CAAC,CAAC,CAAC;YACH,OAAO,CAAC,IAAI,CAAC,GAAI,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAEhD,CAAC;QACD,WAAW,CAAC,IAAI,EAAE,CAAC;QACnB,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;QAC5D,MAAM,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACnB,IAAI,CAAC,CAAC,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC7B,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;oBACpC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBACpC,CAAC;qBAAM,CAAC;oBACN,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAE,GAAG,CAAC,CAAC,CAAC;gBACtE,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC5B,SAAS,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBAC5B,CAAC;qBAAM,CAAC;oBACN,SAAS,CAAC,GAAG,CAAC,OAAO,EAAE,SAAS,CAAC,GAAG,CAAC,OAAO,CAAE,GAAG,CAAC,CAAC,CAAC;gBACtD,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC;QACjC,IAAI,cAAc,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QACvC,IAAI,gBAAgB,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QACzC,IAAI,cAAc,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QACvC,IAAI,mBAAmB,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QAC5C,IAAI,SAAS,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QAClC,IAAI,UAAU,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,WAAW,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QAEpC,oBAAoB;QACpB,OAAO,CAAC,OAAO,CAAC,CAAC,CAAc,EAAE,EAAE;YACjC,cAAc,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YACjD,gBAAgB,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;YACrD,cAAc,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YACjD,mBAAmB,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;YAC3D,SAAS,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YACvC,UAAU,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YACzC,WAAW,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,cAAc,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,gBAAgB,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3C,cAAc,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,mBAAmB,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC9C,SAAS,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACpC,UAAU,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACrC,yCAAyC;QAEzC,OAAO;YACL,MAAM,EAAE;gBACN,cAAc;gBACd,gBAAgB;gBAChB,cAAc;gBACd,mBAAmB;gBACnB,SAAS;gBACT,UAAU;gBACV,WAAW;aACZ;YACD,gBAAgB,EAAE,WAAW,CAAC,cAAc;YAC5C,UAAU;YACV,YAAY,EAAE,OAAO,CAAC,MAAM;YAC5B,QAAQ,EAAE,SAAS;SACpB,CAAC;IAEJ,CAAC;CACF;AACD,QAAQ,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACxC,EAAE,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE;QAC1B,MAAM,CAAC,mBAAmB,EAAE,CAAC;QAC7B,MAAM,CAAC,QAAQ,CAAC,sBAAsB,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;QAExD,MAAM,QAAQ,GAAc;YAC1B,IAAI,EAAE,qBAAqB;YAC3B,MAAM,EAAE;gBACN,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE;gBACzC,aAAa,EAAE,CAAC;aACjB;YACD,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,CAAC,GAAoB,EAAE,EAAE;gBACrC,MAAM,OAAO,GAAG;oBACd;wBACE,GAAG,EAAE;;;;;;;;;sFASqE;qBAC3E;oBACD;wBACE,GAAG,EAAE;;;;;;2CAM0B;qBAChC;oBACD;wBACE,GAAG,EAAE,uCAAuC;qBAC7C;iBACF,CAAC;gBACF,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;gBACvD,OAAO,GAAG,CAAC,iBAAiB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;YACjD,CAAC;SACF,CAAC;QAEF,MAAM,aAAa,GAAG,eAAe,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QACnE,MAAM,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;QAC9C,MAAM,SAAS,GAAG,IAAI,aAAa,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC;QACrC,sCAAsC;QACtC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpB,EAAE,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { Logger, LogLevel, StopWatch } from \"@itwin/core-bentley\";\nimport { DbQueryConfig, ECSqlReader, QueryStats } from \"@itwin/core-common\";\nimport { expect } from \"chai\";\nimport { ConcurrentQuery } from \"../../ConcurrentQuery\";\nimport { ECDb } from \"../../ECDb\";\nimport { IModelDb, SnapshotDb } from \"../../IModelDb\";\nimport { _nativeDb } from \"../../core-backend\";\nimport { IModelTestUtils } from \"../IModelTestUtils\";\n\ninterface ITaskResult {\n  stats: QueryStats;\n  error?: any;\n}\n\ninterface IScenario {\n  name: string;\n  config?: DbQueryConfig;\n  totalBatches: number;\n  taskPerBatch: number;\n  createReader: (db: ECDb | IModelDb) => ECSqlReader;\n}\n\nclass LoadSimulator {\n  constructor(public db: ECDb | IModelDb, public scenario: IScenario) { }\n  private async runQueryTask(reader: ECSqlReader): Promise<ITaskResult> {\n    try {\n      while (await reader.step()) { }\n      return { stats: reader.stats };\n    } catch (err) {\n      return { stats: reader.stats, error: err };\n    }\n  }\n\n  public async run() {\n    ConcurrentQuery.shutdown(this.db[_nativeDb]);\n    if (this.scenario.config) {\n      const config = ConcurrentQuery.resetConfig(this.db[_nativeDb], this.scenario.config);\n      // eslint-disable-next-line no-console\n      console.log(config);\n    }\n    const overallTime = new StopWatch();\n    overallTime.start();\n    const results: ITaskResult[] = [];\n    for (let i = 0; i < this.scenario.totalBatches; ++i) {\n      const promises: Promise<ITaskResult>[] = [];\n      const readerTasks = Array(this.scenario.taskPerBatch).fill(undefined).map(() => this.scenario.createReader(this.db));\n      readerTasks.forEach((reader) => {\n        promises.push(this.runQueryTask(reader));\n      });\n      results.push(... await Promise.all(promises));\n\n    }\n    overallTime.stop();\n    const errors = results.filter((x) => x.error !== undefined);\n    const errorsMap = new Map<string, number>();\n    errors.forEach((x) => {\n      if (x.error instanceof Error) {\n        if (!errorsMap.has(x.error.message)) {\n          errorsMap.set(x.error.message, 1);\n        } else {\n          errorsMap.set(x.error.message, errorsMap.get(x.error.message)! + 1);\n        }\n      } else {\n        if (!errorsMap.has(\"error\")) {\n          errorsMap.set(\"error\", 1);\n        } else {\n          errorsMap.set(\"error\", errorsMap.get(\"error\")! + 1);\n        }\n      }\n    });\n    const errorCount = errors.length;\n    let backendCpuTime: bigint = BigInt(0);\n    let backendTotalTime: bigint = BigInt(0);\n    let backendMemUsed: bigint = BigInt(0);\n    let backendRowsReturned: bigint = BigInt(0);\n    let totalTime: bigint = BigInt(0);\n    let retryCount: bigint = BigInt(0);\n    let prepareTime: bigint = BigInt(0);\n\n    // Calculate average\n    results.forEach((r: ITaskResult) => {\n      backendCpuTime += BigInt(r.stats.backendCpuTime);\n      backendTotalTime += BigInt(r.stats.backendTotalTime);\n      backendMemUsed += BigInt(r.stats.backendMemUsed);\n      backendRowsReturned += BigInt(r.stats.backendRowsReturned);\n      totalTime += BigInt(r.stats.totalTime);\n      retryCount += BigInt(r.stats.retryCount);\n      prepareTime += BigInt(r.stats.prepareTime);\n    });\n\n    backendCpuTime /= BigInt(results.length);\n    backendTotalTime /= BigInt(results.length);\n    backendMemUsed /= BigInt(results.length);\n    backendRowsReturned /= BigInt(results.length);\n    totalTime /= BigInt(results.length);\n    retryCount /= BigInt(results.length);\n    // prepareTime /= BigInt(results.length);\n\n    return {\n      result: {\n        backendCpuTime,\n        backendTotalTime,\n        backendMemUsed,\n        backendRowsReturned,\n        totalTime,\n        retryCount,\n        prepareTime,\n      },\n      overallTimeInSec: overallTime.currentSeconds,\n      errorCount,\n      totalQueries: results.length,\n      errorMap: errorsMap\n    };\n\n  }\n}\ndescribe.skip(\"ConcurrentQueryLoad\", () => {\n  it(\"should run\", async () => {\n    Logger.initializeToConsole();\n    Logger.setLevel(\"ECDb.ConcurrentQuery\", LogLevel.Trace);\n\n    const scenario: IScenario = {\n      name: \"ConcurrentQueryLoad\",\n      config: {\n        globalQuota: { time: 1, memory: 8388608 },\n        workerThreads: 1,\n      },\n      totalBatches: 1,\n      taskPerBatch: 1,\n      createReader: (dbs: ECDb | IModelDb) => {\n        const queries = [\n          {\n            sql: `\n            WITH sequence(n) AS (\n              SELECT  1\n              UNION ALL\n              SELECT n + 1 FROM sequence WHERE n < 10000000\n            )\n            SELECT  COUNT(*)\n            FROM bis.SpatialIndex i, sequence s\n            WHERE i.ECInstanceId MATCH  iModel_spatial_overlap_aabb(\n              iModel_bbox(random(), random(), random(), random(),random(), random()))`\n          },\n          {\n            sql: `\n            WITH sequence(n) AS (\n              SELECT  1\n              UNION ALL\n              SELECT n + 1 FROM sequence WHERE n < 10000000\n            )\n            SELECT  COUNT(*) FROM sequence`\n          },\n          {\n            sql: \"SELECT $ FROM bis.Element LIMIT 10000\"\n          }\n        ];\n        const idx = Math.floor(Math.random() * queries.length);\n        return dbs.createQueryReader(queries[idx].sql);\n      }\n    };\n\n    const verySmallFile = IModelTestUtils.resolveAssetFile(\"test.bim\");\n    const db = SnapshotDb.openFile(verySmallFile);\n    const simulator = new LoadSimulator(db, scenario);\n    const result = await simulator.run();\n    // eslint-disable-next-line no-console\n    console.log(result);\n    db.close();\n    expect(result.errorCount).to.be.equal(0);\n  });\n});\n"]}
+{"version":3,"file":"ConcurrentQueryLoad.test.js","sourceRoot":"","sources":["../../../../src/test/ecdb/ConcurrentQueryLoad.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAElE,OAAO,EAAE,MAAM,EAAE,MAAM,MAAM,CAAC;AAC9B,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAExD,OAAO,EAAY,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACtD,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAerD,MAAM,aAAa;IACE;IAA4B;IAA/C,YAAmB,EAAmB,EAAS,QAAmB;QAA/C,OAAE,GAAF,EAAE,CAAiB;QAAS,aAAQ,GAAR,QAAQ,CAAW;IAAI,CAAC;IAC/D,KAAK,CAAC,YAAY,CAAC,MAAmB;QAC5C,IAAI,CAAC;YACH,OAAO,MAAM,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;YAC/B,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;QAC7C,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,GAAG;QACd,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QAC7C,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YACzB,MAAM,MAAM,GAAG,eAAe,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YACrF,sCAAsC;YACtC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACtB,CAAC;QACD,MAAM,WAAW,GAAG,IAAI,SAAS,EAAE,CAAC;QACpC,WAAW,CAAC,KAAK,EAAE,CAAC;QACpB,MAAM,OAAO,GAAkB,EAAE,CAAC;QAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC;YACpD,MAAM,QAAQ,GAA2B,EAAE,CAAC;YAC5C,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;YACrH,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBAC7B,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC;YAC3C,CAAC,CAAC,CAAC;YACH,OAAO,CAAC,IAAI,CAAC,GAAI,MAAM,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;QAEhD,CAAC;QACD,WAAW,CAAC,IAAI,EAAE,CAAC;QACnB,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;QAC5D,MAAM,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACnB,IAAI,CAAC,CAAC,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC7B,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;oBACpC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBACpC,CAAC;qBAAM,CAAC;oBACN,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAE,GAAG,CAAC,CAAC,CAAC;gBACtE,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC5B,SAAS,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;gBAC5B,CAAC;qBAAM,CAAC;oBACN,SAAS,CAAC,GAAG,CAAC,OAAO,EAAE,SAAS,CAAC,GAAG,CAAC,OAAO,CAAE,GAAG,CAAC,CAAC,CAAC;gBACtD,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QACH,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC;QACjC,IAAI,cAAc,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QACvC,IAAI,gBAAgB,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QACzC,IAAI,cAAc,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QACvC,IAAI,mBAAmB,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QAC5C,IAAI,SAAS,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QAClC,IAAI,UAAU,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,WAAW,GAAW,MAAM,CAAC,CAAC,CAAC,CAAC;QAEpC,oBAAoB;QACpB,OAAO,CAAC,OAAO,CAAC,CAAC,CAAc,EAAE,EAAE;YACjC,cAAc,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YACjD,gBAAgB,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;YACrD,cAAc,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YACjD,mBAAmB,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;YAC3D,SAAS,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;YACvC,UAAU,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YACzC,WAAW,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QAEH,cAAc,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,gBAAgB,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC3C,cAAc,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACzC,mBAAmB,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC9C,SAAS,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACpC,UAAU,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACrC,yCAAyC;QAEzC,OAAO;YACL,MAAM,EAAE;gBACN,cAAc;gBACd,gBAAgB;gBAChB,cAAc;gBACd,mBAAmB;gBACnB,SAAS;gBACT,UAAU;gBACV,WAAW;aACZ;YACD,gBAAgB,EAAE,WAAW,CAAC,cAAc;YAC5C,UAAU;YACV,YAAY,EAAE,OAAO,CAAC,MAAM;YAC5B,QAAQ,EAAE,SAAS;SACpB,CAAC;IAEJ,CAAC;CACF;AACD,QAAQ,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACxC,EAAE,CAAC,YAAY,EAAE,KAAK,IAAI,EAAE;QAC1B,MAAM,CAAC,mBAAmB,EAAE,CAAC;QAC7B,MAAM,CAAC,QAAQ,CAAC,sBAAsB,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC;QAExD,MAAM,QAAQ,GAAc;YAC1B,IAAI,EAAE,qBAAqB;YAC3B,MAAM,EAAE;gBACN,WAAW,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE;gBACzC,aAAa,EAAE,CAAC;aACjB;YACD,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,CAAC,GAAoB,EAAE,EAAE;gBACrC,MAAM,OAAO,GAAG;oBACd;wBACE,GAAG,EAAE;;;;;;;;;sFASqE;qBAC3E;oBACD;wBACE,GAAG,EAAE;;;;;;2CAM0B;qBAChC;oBACD;wBACE,GAAG,EAAE,uCAAuC;qBAC7C;iBACF,CAAC;gBACF,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;gBACvD,OAAO,GAAG,CAAC,iBAAiB,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;YACjD,CAAC;SACF,CAAC;QAEF,MAAM,aAAa,GAAG,eAAe,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;QACnE,MAAM,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;QAC9C,MAAM,SAAS,GAAG,IAAI,aAAa,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QAClD,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,GAAG,EAAE,CAAC;QACrC,sCAAsC;QACtC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpB,EAAE,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { Logger, LogLevel, StopWatch } from \"@itwin/core-bentley\";\r\nimport { DbQueryConfig, ECSqlReader, QueryStats } from \"@itwin/core-common\";\r\nimport { expect } from \"chai\";\r\nimport { ConcurrentQuery } from \"../../ConcurrentQuery\";\r\nimport { ECDb } from \"../../ECDb\";\r\nimport { IModelDb, SnapshotDb } from \"../../IModelDb\";\r\nimport { _nativeDb } from \"../../core-backend\";\r\nimport { IModelTestUtils } from \"../IModelTestUtils\";\r\n\r\ninterface ITaskResult {\r\n  stats: QueryStats;\r\n  error?: any;\r\n}\r\n\r\ninterface IScenario {\r\n  name: string;\r\n  config?: DbQueryConfig;\r\n  totalBatches: number;\r\n  taskPerBatch: number;\r\n  createReader: (db: ECDb | IModelDb) => ECSqlReader;\r\n}\r\n\r\nclass LoadSimulator {\r\n  constructor(public db: ECDb | IModelDb, public scenario: IScenario) { }\r\n  private async runQueryTask(reader: ECSqlReader): Promise<ITaskResult> {\r\n    try {\r\n      while (await reader.step()) { }\r\n      return { stats: reader.stats };\r\n    } catch (err) {\r\n      return { stats: reader.stats, error: err };\r\n    }\r\n  }\r\n\r\n  public async run() {\r\n    ConcurrentQuery.shutdown(this.db[_nativeDb]);\r\n    if (this.scenario.config) {\r\n      const config = ConcurrentQuery.resetConfig(this.db[_nativeDb], this.scenario.config);\r\n      // eslint-disable-next-line no-console\r\n      console.log(config);\r\n    }\r\n    const overallTime = new StopWatch();\r\n    overallTime.start();\r\n    const results: ITaskResult[] = [];\r\n    for (let i = 0; i < this.scenario.totalBatches; ++i) {\r\n      const promises: Promise<ITaskResult>[] = [];\r\n      const readerTasks = Array(this.scenario.taskPerBatch).fill(undefined).map(() => this.scenario.createReader(this.db));\r\n      readerTasks.forEach((reader) => {\r\n        promises.push(this.runQueryTask(reader));\r\n      });\r\n      results.push(... await Promise.all(promises));\r\n\r\n    }\r\n    overallTime.stop();\r\n    const errors = results.filter((x) => x.error !== undefined);\r\n    const errorsMap = new Map<string, number>();\r\n    errors.forEach((x) => {\r\n      if (x.error instanceof Error) {\r\n        if (!errorsMap.has(x.error.message)) {\r\n          errorsMap.set(x.error.message, 1);\r\n        } else {\r\n          errorsMap.set(x.error.message, errorsMap.get(x.error.message)! + 1);\r\n        }\r\n      } else {\r\n        if (!errorsMap.has(\"error\")) {\r\n          errorsMap.set(\"error\", 1);\r\n        } else {\r\n          errorsMap.set(\"error\", errorsMap.get(\"error\")! + 1);\r\n        }\r\n      }\r\n    });\r\n    const errorCount = errors.length;\r\n    let backendCpuTime: bigint = BigInt(0);\r\n    let backendTotalTime: bigint = BigInt(0);\r\n    let backendMemUsed: bigint = BigInt(0);\r\n    let backendRowsReturned: bigint = BigInt(0);\r\n    let totalTime: bigint = BigInt(0);\r\n    let retryCount: bigint = BigInt(0);\r\n    let prepareTime: bigint = BigInt(0);\r\n\r\n    // Calculate average\r\n    results.forEach((r: ITaskResult) => {\r\n      backendCpuTime += BigInt(r.stats.backendCpuTime);\r\n      backendTotalTime += BigInt(r.stats.backendTotalTime);\r\n      backendMemUsed += BigInt(r.stats.backendMemUsed);\r\n      backendRowsReturned += BigInt(r.stats.backendRowsReturned);\r\n      totalTime += BigInt(r.stats.totalTime);\r\n      retryCount += BigInt(r.stats.retryCount);\r\n      prepareTime += BigInt(r.stats.prepareTime);\r\n    });\r\n\r\n    backendCpuTime /= BigInt(results.length);\r\n    backendTotalTime /= BigInt(results.length);\r\n    backendMemUsed /= BigInt(results.length);\r\n    backendRowsReturned /= BigInt(results.length);\r\n    totalTime /= BigInt(results.length);\r\n    retryCount /= BigInt(results.length);\r\n    // prepareTime /= BigInt(results.length);\r\n\r\n    return {\r\n      result: {\r\n        backendCpuTime,\r\n        backendTotalTime,\r\n        backendMemUsed,\r\n        backendRowsReturned,\r\n        totalTime,\r\n        retryCount,\r\n        prepareTime,\r\n      },\r\n      overallTimeInSec: overallTime.currentSeconds,\r\n      errorCount,\r\n      totalQueries: results.length,\r\n      errorMap: errorsMap\r\n    };\r\n\r\n  }\r\n}\r\ndescribe.skip(\"ConcurrentQueryLoad\", () => {\r\n  it(\"should run\", async () => {\r\n    Logger.initializeToConsole();\r\n    Logger.setLevel(\"ECDb.ConcurrentQuery\", LogLevel.Trace);\r\n\r\n    const scenario: IScenario = {\r\n      name: \"ConcurrentQueryLoad\",\r\n      config: {\r\n        globalQuota: { time: 1, memory: 8388608 },\r\n        workerThreads: 1,\r\n      },\r\n      totalBatches: 1,\r\n      taskPerBatch: 1,\r\n      createReader: (dbs: ECDb | IModelDb) => {\r\n        const queries = [\r\n          {\r\n            sql: `\r\n            WITH sequence(n) AS (\r\n              SELECT  1\r\n              UNION ALL\r\n              SELECT n + 1 FROM sequence WHERE n < 10000000\r\n            )\r\n            SELECT  COUNT(*)\r\n            FROM bis.SpatialIndex i, sequence s\r\n            WHERE i.ECInstanceId MATCH  iModel_spatial_overlap_aabb(\r\n              iModel_bbox(random(), random(), random(), random(),random(), random()))`\r\n          },\r\n          {\r\n            sql: `\r\n            WITH sequence(n) AS (\r\n              SELECT  1\r\n              UNION ALL\r\n              SELECT n + 1 FROM sequence WHERE n < 10000000\r\n            )\r\n            SELECT  COUNT(*) FROM sequence`\r\n          },\r\n          {\r\n            sql: \"SELECT $ FROM bis.Element LIMIT 10000\"\r\n          }\r\n        ];\r\n        const idx = Math.floor(Math.random() * queries.length);\r\n        return dbs.createQueryReader(queries[idx].sql);\r\n      }\r\n    };\r\n\r\n    const verySmallFile = IModelTestUtils.resolveAssetFile(\"test.bim\");\r\n    const db = SnapshotDb.openFile(verySmallFile);\r\n    const simulator = new LoadSimulator(db, scenario);\r\n    const result = await simulator.run();\r\n    // eslint-disable-next-line no-console\r\n    console.log(result);\r\n    db.close();\r\n    expect(result.errorCount).to.be.equal(0);\r\n  });\r\n});\r\n"]}