-
Notifications
You must be signed in to change notification settings - Fork 1.2k
/
QueryThroughput.ts
68 lines (58 loc) · 2.01 KB
/
QueryThroughput.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
// Copyright (c) Microsoft Corporation.
// Licensed under the MIT license.
/**
* @summary Demonstrates query throughput scenarios.
*/
import * as dotenv from "dotenv";
dotenv.config();
import { Container, FeedOptions, SqlQuerySpec, CosmosClient } from "@azure/cosmos";
const key = process.env.COSMOS_KEY || "<cosmos key>";
const endpoint = process.env.COSMOS_ENDPOINT || "<cosmos endpoint>";
const containerId = process.env.COSMOS_CONTAINER || "<cosmos container>";
const databaseId = process.env.COSMOS_DATABASE || "<cosmos database>";
async function run() {
const client = new CosmosClient({
endpoint,
key,
});
const query1 = "Select * from c order by c._ts";
const query2 = "Select * from c";
const query3 = "Select value count(c.id) from c";
const container = client.database(databaseId).container(containerId);
const options = {
maxItemCount: 10000,
maxDegreeOfParallelism: 1000,
bufferItems: true,
};
const scenarios = [];
scenarios.push({ container, query: query1, options });
scenarios.push({ container, query: query2, options });
scenarios.push({ container, query: query3, options });
for (const scenario of scenarios) {
try {
console.log("Scenario starting: " + scenario.query);
const start = Date.now();
await runScenario(scenario.container, scenario.query, scenario.options);
console.log(
'Scenario complete: "' + scenario.query + '" - took ' + (Date.now() - start) / 1000 + "s",
);
} catch (e: any) {
console.log("Scenario failed: " + scenario.query + " - " + JSON.stringify(scenario.options));
}
}
}
async function runScenario(
container: Container,
query: string | SqlQuerySpec,
options: FeedOptions,
) {
const queryIterator = container.items.query(query, options);
let count = 0;
while (queryIterator.hasMoreResults() && count <= 100000) {
const { resources: results } = await queryIterator.fetchNext();
if (results !== undefined) {
count = count + results.length;
}
}
}
run().catch(console.error);