summaryrefslogtreecommitdiff
path: root/lib/gitlab/sidekiq_status/server_middleware.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/gitlab/sidekiq_status/server_middleware.rb')
-rw-r--r--lib/gitlab/sidekiq_status/server_middleware.rb13
1 files changed, 13 insertions, 0 deletions
diff --git a/lib/gitlab/sidekiq_status/server_middleware.rb b/lib/gitlab/sidekiq_status/server_middleware.rb
new file mode 100644
index 00000000000..31dfa46ff9d
--- /dev/null
+++ b/lib/gitlab/sidekiq_status/server_middleware.rb
@@ -0,0 +1,13 @@
+module Gitlab
+ module SidekiqStatus
+ class ServerMiddleware
+ def call(worker, job, queue)
+ ret = yield
+
+ SidekiqStatus.unset(job['jid'])
+
+ ret
+ end
+ end
+ end
+end