From ce9e859a9c0125334fdce19e96eee78d268f8baf Mon Sep 17 00:00:00 2001
From: Eelco Dolstra <eelco.dolstra@logicblox.com>
Date: Wed, 17 Jun 2015 21:35:20 +0200
Subject: [PATCH] hydra-queue-runner: Implement --unlock

---
 src/hydra-queue-runner/hydra-queue-runner.cc | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/src/hydra-queue-runner/hydra-queue-runner.cc b/src/hydra-queue-runner/hydra-queue-runner.cc
index c5f657f4..56feaa54 100644
--- a/src/hydra-queue-runner/hydra-queue-runner.cc
+++ b/src/hydra-queue-runner/hydra-queue-runner.cc
@@ -1266,14 +1266,20 @@ int main(int argc, char * * argv)
     return handleExceptions(argv[0], [&]() {
         initNix();
 
-        parseCmdLine(argc, argv, [&](Strings::iterator & arg, const Strings::iterator & end) {
-            return false;
-        });
-
         signal(SIGINT, SIG_DFL);
         signal(SIGTERM, SIG_DFL);
         signal(SIGHUP, SIG_DFL);
 
+        bool unlock = false;
+
+        parseCmdLine(argc, argv, [&](Strings::iterator & arg, const Strings::iterator & end) {
+            if (*arg == "--unlock")
+                unlock = true;
+            else
+                return false;
+            return true;
+        });
+
         settings.buildVerbosity = lvlVomit;
         settings.useSubstitutes = false;
         settings.lockCPU = false;
@@ -1281,6 +1287,9 @@ int main(int argc, char * * argv)
         /* FIXME: need some locking to prevent multiple instances of
            hydra-queue-runner. */
         State state;
-        state.run();
+        if (unlock)
+            state.clearBusy(0);
+        else
+            state.run();
     });
 }