Merge branch 'rack-timeout-rename' into 'master'

Moved RackTimeout observer to a different location and ignore transition states

See merge request gitlab-org/gitlab-ce!29214

2 files changed, 51 insertions, 46 deletions

diff --git a/lib/gitlab/cluster/rack_timeout_observer.rb b/lib/gitlab/cluster/rack_timeout_observer.rb
new file mode 100644
index 00000000000..5182b2be148
--- /dev/null
+++ b/lib/gitlab/cluster/rack_timeout_observer.rb
@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Cluster
+    class RackTimeoutObserver
+      TRANSITION_STATES = %i(ready active).freeze
+
+      def initialize
+        @counter = Gitlab::Metrics.counter(:rack_requests_total, 'Number of requests in a given rack state')
+      end
+
+      # returns the Proc to be used as the observer callback block
+      def callback
+        method(:log_timeout_exception)
+      end
+
+      private
+
+      def log_timeout_exception(env)
+        info = env[::Rack::Timeout::ENV_INFO_KEY]
+        return unless info
+        return if TRANSITION_STATES.include?(info.state)
+
+        @counter.increment(labels(info, env))
+      end
+
+      def labels(info, env)
+        params = controller_params(env) || grape_params(env) || {}
+
+        {
+          controller: params['controller'],
+          action: params['action'],
+          route: params['route'],
+          state: info.state
+        }
+      end
+
+      def controller_params(env)
+        env['action_dispatch.request.parameters']
+      end
+
+      def grape_params(env)
+        endpoint = env[Grape::Env::API_ENDPOINT]
+        route = endpoint&.route&.pattern&.origin
+        return unless route
+
+        { 'route' => route }
+      end
+    end
+  end
+end
diff --git a/lib/gitlab/rack_timeout_observer.rb b/lib/gitlab/rack_timeout_observer.rb
deleted file mode 100644
index 80d3f7dea60..00000000000
--- a/lib/gitlab/rack_timeout_observer.rb
+++ /dev/null
@@ -1,46 +0,0 @@
-# frozen_string_literal: true
-
-module Gitlab
-  class RackTimeoutObserver
-    def initialize
-      @counter = Gitlab::Metrics.counter(:rack_state_total, 'Number of requests in a given rack state')
-    end
-
-    # returns the Proc to be used as the observer callback block
-    def callback
-      method(:log_timeout_exception)
-    end
-
-    private
-
-    def log_timeout_exception(env)
-      info = env[::Rack::Timeout::ENV_INFO_KEY]
-      return unless info
-
-      @counter.increment(labels(info, env))
-    end
-
-    def labels(info, env)
-      params = controller_params(env) || grape_params(env) || {}
-
-      {
-        controller: params['controller'],
-        action: params['action'],
-        route: params['route'],
-        state: info.state
-      }
-    end
-
-    def controller_params(env)
-      env['action_dispatch.request.parameters']
-    end
-
-    def grape_params(env)
-      endpoint = env[Grape::Env::API_ENDPOINT]
-      route = endpoint&.route&.pattern&.origin
-      return unless route
-
-      { 'route' => route }
-    end
-  end
-end