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
|
# The MIT License (MIT)
# Copyright (c) Microsoft Corporation. All rights reserved.
import os
import sys
from workload_utils import create_logger
from workload_configs import COSMOS_URI, COSMOS_KEY, PREFERRED_LOCATIONS
sys.path.append(r"./")
from azure.cosmos.aio import CosmosClient as AsyncClient
import asyncio
import time
from datetime import datetime
async def run_workload(client_id: str):
async with AsyncClient(COSMOS_URI, COSMOS_KEY, preferred_locations=PREFERRED_LOCATIONS,
enable_diagnostics_logging=True, logger=logger,
user_agent=client_id + "-" + datetime.now().strftime(
"%Y%m%d-%H%M%S")) as client:
await asyncio.sleep(1)
while True:
try:
database_account = await client._get_database_account()
logger.info("%s - Database account - writable locations: %s",
datetime.now().strftime("%Y%m%d-%H%M%S"),
database_account.WritableLocations)
logger.info("%s - Database account - readable locations: %s",
datetime.now().strftime("%Y%m%d-%H%M%S"),
database_account.ReadableLocations)
time.sleep(1)
except Exception as e:
logger.error(e)
raise e
if __name__ == "__main__":
file_name = os.path.basename(__file__)
prefix, logger = create_logger(file_name)
asyncio.run(run_workload(prefix))
|
