From 2a5c9e0c4541e0c995f5b53a26e0a9ae9195c031 Mon Sep 17 00:00:00 2001
From: Gervase Markham <gerv@gerv.net>
Date: Thu, 27 Jan 2011 12:42:01 +0000
Subject: Allow jobqueue.pl to run once and then exit. r=mkanat.

https://bugzilla.mozilla.org/show_bug.cgi?id=621256
---
 Bugzilla/JobQueue/Runner.pm | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

(limited to 'Bugzilla/JobQueue')

diff --git a/Bugzilla/JobQueue/Runner.pm b/Bugzilla/JobQueue/Runner.pm
index 20cf6439f..26755e78f 100644
--- a/Bugzilla/JobQueue/Runner.pm
+++ b/Bugzilla/JobQueue/Runner.pm
@@ -196,16 +196,34 @@ sub gd_setup_signals {
     $SIG{TERM} = sub { $self->gd_quit_event(); }
 }
 
+sub gd_other_cmd {
+    my ($self) = shift;
+    if ($ARGV[0] eq "once") {
+        $self->_do_work("work_once");
+
+        exit(0);
+    }
+    
+    $self->SUPER::gd_other_cmd();
+}
+
 sub gd_run {
     my $self = shift;
 
+    $self->_do_work("work");
+}
+
+sub _do_work {
+    my ($self, $fn) = @_;
+
     my $jq = Bugzilla->job_queue();
     $jq->set_verbose($self->{debug});
     foreach my $module (values %{ Bugzilla::JobQueue->job_map() }) {
         eval "use $module";
         $jq->can_do($module);
     }
-    $jq->work;
+
+    $jq->$fn;
 }
 
 1;
-- 
cgit v1.2.3-24-g4f1b