Add pool concurrency config

src/commands/create_keys.py

@@ -46,13 +46,21 @@
     type=str,
     callback=validate_eth_address,
 )
+@click.option(
+    '--pool-size',
+    help='Number of processes in a pool.',
+    # do not prompt
+    type=int,
+)
 @click.command(help='Creates the validator keys from the mnemonic.')
+# pylint: disable-next=too-many-arguments
 def create_keys(
     mnemonic: str,
     count: int,
     vault: HexAddress,
     data_dir: str,
     per_keystore_password: bool,
+    pool_size: int | None,
 ) -> None:
     config = VaultConfig(vault, Path(data_dir))
     config.load(mnemonic)
@@ -67,16 +75,18 @@ def create_keys(
         mnemonic=mnemonic,
         count=count,
         start_index=config.mnemonic_next_index,
+        pool_size=pool_size,
     )
     deposit_data = _export_deposit_data_json(
-        credentials=credentials, filename=str(deposit_data_file)
+        credentials=credentials, filename=str(deposit_data_file), pool_size=pool_size
     )
 
     _export_keystores(
         credentials=credentials,
         keystores_dir=str(keystores_dir),
         password_file=str(password_file),
         per_keystore_password=per_keystore_password,
+        pool_size=pool_size,
     )
 
     config.increment_mnemonic_index(count)
@@ -88,13 +98,15 @@ def create_keys(
     )
 
 
-def _export_deposit_data_json(credentials: list[Credential], filename: str) -> str:
+def _export_deposit_data_json(
+    credentials: list[Credential], filename: str, pool_size: int | None = None
+) -> str:
     with click.progressbar(
         length=len(credentials),
         label='Generating deposit data JSON\t\t',
         show_percent=False,
         show_pos=True,
-    ) as progress_bar, Pool() as pool:
+    ) as progress_bar, Pool(processes=pool_size) as pool:
         results = [
             pool.apply_async(
                 cred.deposit_datum_dict,
@@ -117,6 +129,7 @@ def _export_keystores(
     keystores_dir: str,
     password_file: str,
     per_keystore_password: bool,
+    pool_size: int | None = None,
 ) -> None:
     makedirs(path.abspath(keystores_dir), exist_ok=True)
     if not per_keystore_password:
@@ -126,7 +139,7 @@ def _export_keystores(
         label='Exporting validator keystores\t\t',
         show_percent=False,
         show_pos=True,
-    ) as progress_bar, Pool() as pool:
+    ) as progress_bar, Pool(processes=pool_size) as pool:
         results = [
             pool.apply_async(
                 cred.save_signing_keystore,

src/common/credentials.py

@@ -115,16 +115,22 @@ def deposit_datum_dict(self) -> dict[str, bytes]:
 
 class CredentialManager:
     @staticmethod
+    # pylint: disable-next=too-many-arguments
     def generate_credentials(
-        network: str, vault: HexAddress, mnemonic: str, count: int, start_index: int
+        network: str,
+        vault: HexAddress,
+        mnemonic: str,
+        count: int,
+        start_index: int,
+        pool_size: int | None = None,
     ) -> list[Credential]:
         credentials: list[Credential] = []
         with click.progressbar(
             length=count,
             label='Creating validator keys:\t\t',
             show_percent=False,
             show_pos=True,
-        ) as progress_bar, Pool() as pool:
+        ) as progress_bar, Pool(processes=pool_size) as pool:
 
             def bar_updated(result):
                 progress_bar.update(len(result))

src/config/settings.py

@@ -47,6 +47,7 @@ class Settings(metaclass=Singleton):
     ipfs_fetch_endpoints: list[str]
     validators_fetch_chunk_size: int
     sentry_dsn: str
+    pool_size: int | None
 
     # pylint: disable-next=too-many-arguments,too-many-locals
     def set(
@@ -121,6 +122,9 @@ def set(
             'VALIDATORS_FETCH_CHUNK_SIZE', default=100, cast=int
         )
         self.sentry_dsn = decouple_config('SENTRY_DSN', default='')
+        self.pool_size = decouple_config(
+            'POOL_SIZE', default=None, cast=lambda x: int(x) if x else None
+        )
 
     @property
     def network_config(self) -> NetworkConfig:

src/validators/utils.py

@@ -157,7 +157,7 @@ def load_keystores() -> Keystores | None:
 
     keystore_files = list_keystore_files()
     logger.info('Loading keystores from %s...', settings.keystores_dir)
-    with Pool() as pool:
+    with Pool(processes=settings.pool_size) as pool:
         # pylint: disable-next=unused-argument
         def _stop_pool(*args, **kwargs):
             pool.close()