Merge pull request #643 from penguincoder/rabbitmq-readiness-check

Add readiness check for rabbitmq
cachix · Jun 6, 2023 · f5daada · f5daada
2 parents b43ae85 + 2ef8093
commit f5daada
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/src/modules/services/rabbitmq.nix b/src/modules/services/rabbitmq.nix
@@ -143,6 +143,22 @@ in
     env.RABBITMQ_ENABLED_PLUGINS_FILE = plugin_file;
     env.RABBITMQ_NODENAME = "rabbit@localhost";
 
-    processes.rabbitmq.exec = "${cfg.package}/bin/rabbitmq-server";
+    processes.rabbitmq = {
+      exec = "${cfg.package}/bin/rabbitmq-server";
+
+      process-compose = {
+        readiness_probe = {
+          exec.command = "${cfg.package}/bin/rabbitmq-diagnostics -q ping";
+          initial_delay_seconds = 10;
+          period_seconds = 3;
+          timeout_seconds = 3;
+          success_threshold = 1;
+          failure_threshold = 5;
+        };
+
+        # https://github.com/F1bonacc1/process-compose#-auto-restart-if-not-healthy
+        availability.restart = "on_failure";
+      };
+    };
   };
 }