From b752ee8aa93ead6d9e39219444a20761d9f01de5 Mon Sep 17 00:00:00 2001
From: Jacob Vosmaer <contact@jacobvosmaer.nl>
Date: Thu, 21 Aug 2014 15:51:31 +0200
Subject: [PATCH] Add rake task to drop a project's PostReceive jobs

---
 lib/tasks/gitlab/sidekiq.rake | 47 +++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 lib/tasks/gitlab/sidekiq.rake

diff --git a/lib/tasks/gitlab/sidekiq.rake b/lib/tasks/gitlab/sidekiq.rake
new file mode 100644
index 0000000000..7e2a6668e5
--- /dev/null
+++ b/lib/tasks/gitlab/sidekiq.rake
@@ -0,0 +1,47 @@
+namespace :gitlab do
+  namespace :sidekiq do
+    QUEUE = 'queue:post_receive'
+
+    desc 'Drop all Sidekiq PostReceive jobs for a given project'
+    task :drop_post_receive , [:project] => :environment do |t, args|
+      unless args.project.present?
+        abort "Please specify the project you want to drop PostReceive jobs for:\n  rake gitlab:sidekiq:drop_post_receive[group/project]"
+      end
+      project_path = Project.find_with_namespace(args.project).repository.path_to_repo
+
+      Sidekiq.redis do |redis|
+        unless redis.exists(QUEUE)
+          abort "Queue #{QUEUE} is empty"
+        end
+
+        temp_queue = "#{QUEUE}_#{Time.now.to_i}"
+        redis.rename(QUEUE, temp_queue)
+
+        # At this point, then post_receive queue is empty. It may be receiving
+        # new jobs already. We will repopulate it with the old jobs, skipping the
+        # ones we want to drop.
+        dropped = 0
+        while (job = redis.lpop(temp_queue)) do
+          if repo_path(job) == project_path
+            dropped += 1
+          else
+            redis.rpush(QUEUE, job)
+          end
+        end
+        # The temp_queue will delete itself after we have popped all elements
+        # from it
+
+        puts "Dropped #{dropped} jobs containing #{project_path} from #{QUEUE}"
+      end
+    end
+
+    def repo_path(job)
+      job_args = JSON.parse(job)['args']
+      if job_args
+        job_args.first
+      else
+        nil
+      end
+    end
+  end
+end