Alter exit code if NoValidatorKeysStateException

courtneyeh · Nov 28, 2023 · a92a630 · a92a630
1 parent f4c8281
commit a92a630
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/teku/src/main/java/tech/pegasys/teku/cli/BeaconNodeCommand.java b/teku/src/main/java/tech/pegasys/teku/cli/BeaconNodeCommand.java
@@ -348,6 +348,9 @@ public Integer call() {
           .ifPresentOrElse(this::reportUserError, () -> reportUnexpectedError(e));
     } catch (Throwable t) {
       reportUnexpectedError(t);
+      if (t instanceof NoValidatorKeysStateException) {
+        return 2;
+      }
     }
     return 1;
   }

diff --git a/teku/src/main/java/tech/pegasys/teku/cli/subcommand/ValidatorClientCommand.java b/teku/src/main/java/tech/pegasys/teku/cli/subcommand/ValidatorClientCommand.java
@@ -38,6 +38,7 @@
 import tech.pegasys.teku.infrastructure.logging.LoggingConfigurator;
 import tech.pegasys.teku.networks.Eth2NetworkConfiguration;
 import tech.pegasys.teku.storage.server.DatabaseStorageException;
+import tech.pegasys.teku.validator.client.NoValidatorKeysStateException;
 
 @Command(
     name = "validator-client",
@@ -110,6 +111,9 @@ public Integer call() {
               parentCommand::reportUserError, () -> parentCommand.reportUnexpectedError(e));
     } catch (Throwable t) {
       parentCommand.reportUnexpectedError(t);
+      if (t instanceof NoValidatorKeysStateException) {
+        return 2;
+      }
     }
     return 1;
   }