summaryrefslogtreecommitdiff
path: root/lib/gitlab/cluster/mixins/puma_cluster.rb
blob: e9157d9f1e47db5459e623bd6aa0f9cf8e44a334 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
# frozen_string_literal: true

module Gitlab
  module Cluster
    module Mixins
      module PumaCluster
        def self.prepended(base)
          raise 'missing method Puma::Cluster#stop_workers' unless base.method_defined?(:stop_workers)
        end

        def stop_workers
          Gitlab::Cluster::LifecycleEvents.do_before_phased_restart

          super
        end
      end
    end
  end
end