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
69
70
71
72
73
74
|
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
import asyncio
from datetime import date, datetime, timezone
from devtools_testutils.perfstress_tests import PerfStressTest
from azure.monitor.query import LogsQueryClient as SyncLogsQueryClient, LogsBatchQuery
from azure.monitor.query.aio import LogsQueryClient as AsyncLogsQueryClient
from azure.identity import DefaultAzureCredential as SyncDefaultAzureCredential
from azure.identity.aio import DefaultAzureCredential as AsyncDefaultAzureCredential
class LogsBatchPerfTest(PerfStressTest):
def __init__(self, arguments):
super().__init__(arguments)
# auth configuration
self.workspace_id = self.get_from_env("LOG_WORKSPACE_ID")
# Create clients
self.logs_client = SyncLogsQueryClient(credential=SyncDefaultAzureCredential())
self.async_logs_client = AsyncLogsQueryClient(credential=AsyncDefaultAzureCredential())
self.requests = [
LogsBatchQuery(
query="AzureActivity | summarize count()",
start_time=datetime(2021, 7, 25, 0, 0, 0, tzinfo=timezone.utc),
end_time=datetime(2021, 7, 26, 0, 0, 0, tzinfo=timezone.utc),
workspace_id=self.workspace_id,
timespan=None,
),
LogsBatchQuery(
query="""AppRequests | take 10 |
summarize avgRequestDuration=avg(DurationMs) by bin(TimeGenerated, 10m), _ResourceId""",
start_time=datetime(2021, 7, 25, 0, 0, 0, tzinfo=timezone.utc),
end_time=datetime(2021, 7, 26, 0, 0, 0, tzinfo=timezone.utc),
workspace_id=self.workspace_id,
timespan=None,
),
LogsBatchQuery(
query="AppRequests | take 20", workspace_id=self.workspace_id, include_statistics=True, timespan=None
),
]
async def close(self):
"""This is run after cleanup.
Use this to close any open handles or clients.
"""
await self.async_logs_client.close()
await super().close()
def run_sync(self):
"""The synchronous perf test.
Try to keep this minimal and focused. Using only a single client API.
Avoid putting any ancillary logic (e.g. generating UUIDs), and put this in the setup/init instead
so that we're only measuring the client API call.
"""
self.logs_client.query_batch(self.requests)
async def run_async(self):
"""The asynchronous perf test.
Try to keep this minimal and focused. Using only a single client API.
Avoid putting any ancillary logic (e.g. generating UUIDs), and put this in the setup/init instead
so that we're only measuring the client API call.
"""
await self.async_logs_client.query_batch(self.requests)
|
