More resilient no provision on shutdown (#610)

jenkinsci · Dec 20, 2024 · 9a63017 · 9a63017
1 parent 7288787
commit 9a63017
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/main/java/com/microsoft/azure/vmagent/AzureVMCloud.java b/src/main/java/com/microsoft/azure/vmagent/AzureVMCloud.java
@@ -634,6 +634,12 @@ public Collection<PlannedNode> provision(CloudState cloudState, int workLoad) {
         LOGGER.log(Level.FINE, "Starting for label {0} workLoad {1}",
                 new Object[]{cloudState.getLabel(), workLoad}
         );
+
+        if (Jenkins.get().isQuietingDown()) {
+            LOGGER.log(Level.FINE, "Skipping provision as Jenkins is in quiet-down");
+            return Collections.emptyList();
+        }
+
         final AzureVMAgentTemplate template = AzureVMCloud.this.getAzureAgentTemplate(cloudState.getLabel());
 
         // round up the number of required machine

diff --git a/src/main/java/com/microsoft/azure/vmagent/AzureVMNoDelayProvisionerStrategy.java b/src/main/java/com/microsoft/azure/vmagent/AzureVMNoDelayProvisionerStrategy.java
@@ -48,6 +48,11 @@ public NodeProvisioner.StrategyDecision apply(@NonNull NodeProvisioner.StrategyS
             return NodeProvisioner.StrategyDecision.CONSULT_REMAINING_STRATEGIES;
         }
 
+        if (Jenkins.get().isQuietingDown()) {
+            LOGGER.log(Level.FINE, "Skipping provision as Jenkins is quieting down");
+            return NodeProvisioner.StrategyDecision.CONSULT_REMAINING_STRATEGIES;
+        }
+
         final Label label = strategyState.getLabel();
 
         LoadStatistics.LoadStatisticsSnapshot snapshot = strategyState.getSnapshot();