diff --git a/libsast/common.py b/libsast/common.py index d8a3718..6f7771a 100644 --- a/libsast/common.py +++ b/libsast/common.py @@ -74,10 +74,12 @@ def read_yaml(file_obj, text=False): def get_worker_count(): """Get worker count for pool.""" - # Default to 1 worker - worker_count = 1 - - # Try to get the CPU count + libsast_workers = os.getenv('LIBSAST_WORKERS') + if libsast_workers: + try: + return int(libsast_workers) + except ValueError: + return 1 try: worker_count = os.cpu_count() except Exception: @@ -86,13 +88,4 @@ def get_worker_count(): # Adjust worker count for Windows if sys.platform == 'win32': worker_count = min(worker_count, 61) - - # Override with environment variable if set - libsast_workers = os.getenv('LIBSAST_WORKERS') - if libsast_workers is not None: - try: - worker_count = int(libsast_workers) - except ValueError: - worker_count = 1 - return worker_count