From f26bc4d36f92ba4e40143944c9f0af6a69dafd8e Mon Sep 17 00:00:00 2001
From: Evgeny Gusarov <evgeny@stakewise.io>
Date: Fri, 7 Jun 2024 19:55:06 +0300
Subject: [PATCH] Add pool-size param to validators-exit command

---
 src/commands/validators_exit.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/commands/validators_exit.py b/src/commands/validators_exit.py
index e290637b..64a70343 100644
--- a/src/commands/validators_exit.py
+++ b/src/commands/validators_exit.py
@@ -102,6 +102,12 @@ class ValidatorExit:
     envvar='LOG_LEVEL',
     help='The log level.',
 )
+@click.option(
+    '--pool-size',
+    help='Number of processes in a pool.',
+    envvar='POOL_SIZE',
+    type=int,
+)
 @click.command(help='Performs a voluntary exit for active vault validators.')
 # pylint: disable-next=too-many-arguments
 def validators_exit(
@@ -116,6 +122,7 @@ def validators_exit(
     data_dir: str,
     verbose: bool,
     log_level: str,
+    pool_size: int | None,
 ) -> None:
     # pylint: disable=duplicate-code
     vault_config = VaultConfig(vault, Path(data_dir))
@@ -134,6 +141,7 @@ def validators_exit(
         hashi_vault_url=hashi_vault_url,
         verbose=verbose,
         log_level=log_level,
+        pool_size=pool_size,
     )
     try:
         # Try-catch to enable async calls in test - an event loop