diff options
Diffstat (limited to 'config/initializers/cluster_events_before_phased_restart.rb')
| -rw-r--r-- | config/initializers/cluster_events_before_phased_restart.rb | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/config/initializers/cluster_events_before_phased_restart.rb b/config/initializers/cluster_events_before_phased_restart.rb index aae5470d6ae..cbb1dd1a53a 100644 --- a/config/initializers/cluster_events_before_phased_restart.rb +++ b/config/initializers/cluster_events_before_phased_restart.rb @@ -5,8 +5,10 @@ # # Follow-up the issue: https://gitlab.com/gitlab-org/gitlab/issues/34107 -if Gitlab::Runtime.puma? +if defined?(::Puma) Puma::Cluster.prepend(::Gitlab::Cluster::Mixins::PumaCluster) -elsif Gitlab::Runtime.unicorn? +end + +if defined?(::Unicorn::HttpServer) Unicorn::HttpServer.prepend(::Gitlab::Cluster::Mixins::UnicornHttpServer) end |