From 714d13a6de258abef040a2dd46040f74a7d29722 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C3=B6rg=20Herbel?= <joerg.herbel@checkmk.com>
Date: Fri, 22 Nov 2024 10:33:23 +0100
Subject: [PATCH] Scheduler main: Check for cancellation between base setup and
 setup steps

Barely makes a difference but is the correct thing to do.
---
 src/bin/scheduler/main.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/bin/scheduler/main.rs b/src/bin/scheduler/main.rs
index 77d17eb2..3b15648b 100644
--- a/src/bin/scheduler/main.rs
+++ b/src/bin/scheduler/main.rs
@@ -63,6 +63,10 @@ fn run() -> Result<(), Terminate> {
     setup::base_directories::setup(&global_config, &plans)?;
     info!("Base setup completed");
 
+    if global_config.cancellation_token.is_cancelled() {
+        return Err(Terminate::Cancelled);
+    }
+
     write_phase(&SchedulerPhase::Setup, &global_config)?;
     let (plans, setup_failures) = setup::steps::run::run(&global_config, plans)?;
     write_setup_failures(setup_failures.into_iter(), &global_config)?;