introduce ROBOT_THREADPOOL_POOL_MAX_WORKERS and ROBOT_PROCESS_POOL_MAX_WORKERS environment vars

Author: d-biehl

.vscode/launch.json

@@ -92,7 +92,7 @@
 				"log.ini",
 				//"-w",
 				"--",
-				//"C:\\tmp\\robottest\\tests\\bdd_style.robot"				
+				//"C:\\tmp\\robottest\\tests\\bdd_style.robot"
 			]
 		},
 		{
@@ -184,6 +184,10 @@
 			"name": "Run Extension",
 			"type": "extensionHost",
 			"request": "launch",
+			"env": {
+				"ROBOT_THREADPOOL_POOL_MAX_WORKERS": "6",
+				"ROBOT_PROCESS_POOL_MAX_WORKERS": "4"
+			},
 			"args": [
 				"--extensionDevelopmentPath=${workspaceFolder}",
 			],

robotcode/language_server/robotframework/utils/process_pool.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import atexit
+import os
 import signal
 from concurrent.futures.process import ProcessPoolExecutor
 from typing import Any, Optional
@@ -30,7 +31,14 @@ def _terminate(*args: Any, **kwargs: Any) -> None:
 def get_process_pool() -> ProcessPoolExecutor:
     global _process_pool
     if _process_pool is None:
-        _process_pool = ProcessPoolExecutor(max_workers=PROCESS_POOL_MAX_WORKERS, initializer=_init_pool)
+        _process_pool = ProcessPoolExecutor(
+            max_workers=(
+                int(s)
+                if (s := os.environ.get("ROBOT_PROCESS_POOL_MAX_WORKERS", None)) and s.isnumeric()
+                else PROCESS_POOL_MAX_WORKERS
+            ),
+            initializer=_init_pool,
+        )
         atexit.register(_terminate)
 
         try:

robotcode/utils/async_tools.py

@@ -5,6 +5,7 @@
 import contextvars
 import functools
 import inspect
+import os
 import threading
 import weakref
 from collections import deque
@@ -443,6 +444,7 @@ async def check_canceled() -> bool:
 
     return True
 
+
 THREADPOOL_POOL_MAX_WORKERS = None
 
 __tread_pool_executor: Optional[ThreadPoolExecutor] = None
@@ -455,7 +457,13 @@ def run_in_thread(func: Callable[..., _T], /, *args: Any, **kwargs: Any) -> asyn
     func_call = functools.partial(ctx.run, func, *args, **kwargs)
 
     if __tread_pool_executor is None:
-        __tread_pool_executor = ThreadPoolExecutor(THREADPOOL_POOL_MAX_WORKERS)
+        __tread_pool_executor = ThreadPoolExecutor(
+            max_workers=(
+                int(s)
+                if (s := os.environ.get("ROBOT_THREADPOOL_POOL_MAX_WORKERS", None)) and s.isnumeric()
+                else THREADPOOL_POOL_MAX_WORKERS
+            )
+        )
 
     return cast("asyncio.Future[_T]", loop.run_in_executor(__tread_pool_executor, cast(Callable[..., _T], func_call)))
 

tests/robotcode/language_server/robotframework/parts/data/.vscode/settings.json

@@ -4,5 +4,8 @@
     },
     "robotcode.robot.env": {
         "ENV_VAR": "1"
-    }
+    },
+    "robotcode.languageServer.args": [
+        "--debugpy"
+    ]
 }

tests/robotcode/language_server/robotframework/parts/data/lib/alibrary.py

@@ -1,4 +1,5 @@
 from typing import Any
+
 from robot.api import logger