Measure proxy timing: needs influxdb code

2 files changed, 23 insertions, 0 deletions

diff --git a/config/initializers/proxy_timing.rb b/config/initializers/proxy_timing.rb
new file mode 100644
index 00000000000..9161bd9b6c9
--- /dev/null
+++ b/config/initializers/proxy_timing.rb
@@ -0,0 +1 @@
+Rails.application.config.middleware.use(Gitlab::Middleware::ProxyTiming)
diff --git a/lib/gitlab/middleware/proxy_timing.rb b/lib/gitlab/middleware/proxy_timing.rb
new file mode 100644
index 00000000000..9094d6b1e7d
--- /dev/null
+++ b/lib/gitlab/middleware/proxy_timing.rb
@@ -0,0 +1,22 @@
+# This Rack middleware is intended to measure the latency between
+# gitlab-workhorse forwarding a request to the Rails application and the
+# time this middleware is reached.
+
+module Gitlab
+  module Middleware
+    class ProxyTiming
+      def initialize(app)
+        @app = app
+      end
+      
+      def call(env)
+        proxy_start = env['HTTP_GITLAB_WORHORSE_PROXY_START'].to_f / 1_000_000_000
+        if proxy_start > 0
+          # send measurement
+          puts "\n\n\n#{(Time.now - proxy_start).to_f}\n\n\n"
+        end
+        @app.call(env)
+      end
+    end
+  end
+end