@itwin/core-backend 5.1.0-dev.5 → 5.1.0-dev.51

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.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\";\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

@@ -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);

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;AAgBrD,MAAM,aAAa;IACE;IAA4B;IAA/C,YAAmB,EAAmB,EAAS,OAAiB;QAA7C,OAAE,GAAF,EAAE,CAAiB;QAAS,YAAO,GAAP,OAAO,CAAU;IAAI,CAAC;IAC7D,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,OAAO,CAAC,MAAM,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,eAAe,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACpF,sCAAsC;YACtC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACtB,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,SAAS,EAAE,CAAC;QACnC,UAAU,CAAC,KAAK,EAAE,CAAC;QACnB,MAAM,OAAO,GAAkB,EAAE,CAAC;QAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC;YACnD,MAAM,QAAQ,GAA2B,EAAE,CAAC;YAC5C,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;YACnH,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,UAAU,CAAC,IAAI,EAAE,CAAC;QAClB,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,eAAe,EAAE,UAAU,CAAC,cAAc;YAC1C,UAAU;YACV,YAAY,EAAE,OAAO,CAAC,MAAM;YAC5B,QAAQ,EAAE,SAAS;SACpB,CAAC;IAEJ,CAAC;CACF;AAED,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;QACxD,IAAI;QACJ,sBAAsB;QACtB,4BAA4B;QAC5B,2BAA2B;QAC3B,uBAAuB;QACvB,yCAAyC;QACzC,yCAAyC;QACzC,qCAAqC;QACrC,+BAA+B;QAC/B,0BAA0B;QAC1B,6BAA6B;QAC7B,+CAA+C;QAC/C,IAAI;QAEJ,MAAM,OAAO,GAAa;YACxB,IAAI,EAAE,qBAAqB;YAC3B,MAAM,EAAE,EAEP;YACD,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,CAAC,GAAoB,EAAE,EAAE;gBACrC,MAAM,MAAM,GAAG;oBACb;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,MAAM,CAAC,MAAM,CAAC,CAAC;gBACtD,OAAO,GAAG,CAAC,iBAAiB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;YAChD,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,OAAO,CAAC,CAAC;QACjD,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;AAEL,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 ISenario {\n  name: string;\n  config?: DbQueryConfig;\n  totalBatches: number;\n  taskPerBatch: number;\n  createReader: (db: ECDb | IModelDb) => ECSqlReader;\n}\n\n\nclass LoadSimulator {\n  constructor(public db: ECDb | IModelDb, public senario: ISenario) { }\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.senario.config) {\n      const config = ConcurrentQuery.resetConfig(this.db[_nativeDb], this.senario.config);\n      // eslint-disable-next-line no-console\n      console.log(config);\n    }\n    const overalTime = new StopWatch();\n    overalTime.start();\n    const results: ITaskResult[] = [];\n    for (let i = 0; i < this.senario.totalBatches; ++i) {\n      const promises: Promise<ITaskResult>[] = [];\n      const readerTasks = Array(this.senario.taskPerBatch).fill(undefined).map(() => this.senario.createReader(this.db));\n      readerTasks.forEach((reader) => {\n        promises.push(this.runQueryTask(reader));\n      });\n      results.push(... await Promise.all(promises));\n\n    }\n    overalTime.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      overalTimeInSec: overalTime.currentSeconds,\n      errorCount,\n      totalQueries: results.length,\n      errorMap: errorsMap\n    };\n\n  }\n}\n\ndescribe.skip(\"ConcurrentQueryLoad\", () => {\n  it(\"should run\", async () => {\n    Logger.initializeToConsole();\n    Logger.setLevel(\"ECDb.ConcurrentQuery\", LogLevel.Trace);\n    // {\n    //   workerThreads: 4,\n    //   requestQueueSize: 2000,\n    //   ignorePriority: false,\n    //   ignoreDelay: true,\n    //   doNotUsePrimaryConnToPrepare: false,\n    //   autoShutdowWhenIdlelForSeconds: 300,\n    //   statementCacheSizePerWorker: 40,\n    //   monitorPollInterval: 1000,\n    //   memoryMapFileSize: 0,\n    //   allowTestingArgs: false,\n    //   globalQuota: { time: 60, memory: 8388608 }\n    // }\n\n    const senario: ISenario = {\n      name: \"ConcurrentQueryLoad\",\n      config: {\n\n      },\n      totalBatches: 1,\n      taskPerBatch: 1,\n      createReader: (dbs: ECDb | IModelDb) => {\n        const quries = [\n          {\n            sql: `\n            WITH sequence(n) AS (\n              SELECT  1\n              UNION ALL\n              SELECT n + 1 FROM sequence WHERE n < 10000\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 < 1000000\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() * quries.length);\n        return dbs.createQueryReader(quries[idx].sql);\n      }\n    };\n\n    const verySmallFile = IModelTestUtils.resolveAssetFile(\"test.bim\");\n    const db = SnapshotDb.openFile(verySmallFile);\n    const simulator = new LoadSimulator(db, senario);\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"]}