Envoy Workload Improvements (#42335)

* improve envoy workloads

* react to comments

* adding markdown file

* fix cspell

Author: tvaron3

sdk/cosmos/azure-cosmos/tests/workloads/custom_tcp_connector.py

@@ -0,0 +1,53 @@
+# The MIT License (MIT)
+# Copyright (c) Microsoft Corporation. All rights reserved.
+import asyncio
+from typing import Any
+
+from aiohttp import TCPConnector, ClientRequest, tracing, ClientTimeout, ClientConnectorError, client_proto
+
+
+class ProxiedTCPConnector(TCPConnector):
+    """
+    A TCP connector that proxies all connections through a specified host and port.
+
+    This class extends TrackedTCPConnector to override connection behavior,
+    routing all traffic through a specified proxy server using non-encrypted http.
+    """
+
+    def __init__(
+            self,
+            *,
+            proxy_host: str,
+            proxy_port: int,
+            **kwargs: Any,
+    ) -> None:
+        """
+        Initialize the ProxiedTCPConnector.
+
+        Args:
+            proxy_host (str): The hostname of the proxy server.
+            proxy_port (int): The port number of the proxy server.
+            **kwargs: Additional keyword arguments passed to the base TCPConnector.
+        """
+        super().__init__(**kwargs)
+        self.__proxy_host = proxy_host
+        self.__proxy_port = proxy_port
+
+    async def _create_direct_connection(
+            self,
+            req: ClientRequest,
+            traces: list[tracing.Trace],
+            timeout: ClientTimeout,
+            *,
+            client_error: type[Exception] = ClientConnectorError,
+    ) -> tuple[asyncio.Transport, client_proto.ResponseHandler]:
+        """Override host, port, and schema to use proxy"""
+        req.url = (
+            req.url.with_host(self.__proxy_host).with_port(self.__proxy_port).with_scheme("http")
+        )
+        return await super()._create_direct_connection(
+            req,
+            traces,
+            timeout,
+            client_error=client_error,
+        )

sdk/cosmos/azure-cosmos/tests/workloads/dev.md

@@ -30,7 +30,7 @@ file when run. These logs are named in this format `<file name>-<process id>-<da
 1. Install envoy proxy https://www.envoyproxy.io/docs/envoy/latest/start/install
 1. Go to envoy folder and generate envoy configuration file using template. Template files are in `envoy/templates` directory. `<account_name>` is your Cosmos DB account name.
     - `cd envoy`
-    - `./generate_envoy_config.sh <template_file_path> <output_envoy_config_file> <account_name>`
+    - `./generate_envoy_config.sh <template_file_path> <output_envoy_config_file> <account_name> <write_region> <read_region>`
 1. Start envoy using the generated configuration file
     - `mkdir logs`
     - `envoy -c <envoy_config_file>.yaml --log-level debug --log-path logs/debug.txt`
@@ -45,6 +45,6 @@ file when run. These logs are named in this format `<file name>-<process id>-<da
 
 ### Close Workloads
 - If you want to keep the logs and stop the scripts,  
-   `./shutdown_workloads.sh --no-remove-logs` 
+   `./shutdown_workloads.sh --do-not-remove-logs` 
 - If you want to remove the logs and stop the scripts,        
    `./shutdown_workloads.sh`