From eab1f126044d8408d8a919456714efb0431739ce Mon Sep 17 00:00:00 2001
From: dawn <dawn@tangled.org>
Date: Sat, 29 Aug 2026 02:23:31 +0900
Subject: [PATCH] nix/modules: drain local executors before mill

Signed-off-by: dawn <dawn@tangled.org>
---
 nix/modules/spindle.nix | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/nix/modules/spindle.nix b/nix/modules/spindle.nix
index 8b2e1f03..f9db8ee9 100644
--- a/nix/modules/spindle.nix
+++ b/nix/modules/spindle.nix
@@ -862,6 +862,7 @@ in
                 "docker.service"
                 "spindle-mill-executors.service"
               ];
+              partOf = ["spindle.service"];
               wantedBy = ["multi-user.target"];
               path = deps;
               serviceConfig = {
@@ -1035,6 +1036,7 @@ in
                   else "${pkgs.coreutils}/bin/env SPINDLE_ROLE=${cfg.role} SPINDLE_MILL_ARTIFACT_STORE=${cfg.mill.artifactStore} SPINDLE_MILL_DRAIN_TIMEOUT=${toString cfg.mill.drainTimeout}s ${cfg.package}/bin/spindle";
                 TimeoutStopSec = cfg.mill.drainTimeout + 510;
                 Restart = "always";
+                RestartMode = mkIf (localExecutors != {}) "direct";
               };
             };
           }
-- 
2.51.2