heroku_queue_time_logger 0.0.1 → 0.0.2

data/heroku_queue_time_logger.gemspec

@@ -0,0 +1,17 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/heroku_queue_time_logger/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Roberts, Marc"]
+  gem.email         = ["marc@neutroncreations.com"]
+  gem.description   = %q{heroku queue time logger}
+  gem.summary       = %q{heroku queue time logger}
+  gem.homepage      = "http://github.com/neutroncreations/heroku_queue_time_logger"
+
+  gem.files         = `git ls-files`.split($\)
+  gem.executables   = gem.files.grep(%r{^bin/}).map{ |f| File.basename(f) }
+  gem.test_files    = gem.files.grep(%r{^(test|spec|features)/.*_spec.rb})
+  gem.name          = "heroku_queue_time_logger"
+  gem.require_paths = ["lib"]
+  gem.version       = HerokuQueueTimeLogger::VERSION
+end

data/lib/heroku_queue_time_logger/middleware.rb

@@ -0,0 +1,40 @@
+module HerokuQueueTimeLogger
+
+	# https://gist.github.com/a-warner/f5db30857ed3423cea79
+	# combination of https://gist.github.com/daveyeu/4960893
+	# and https://gist.github.com/jasonrclark/d82a1ea7695daac0b9ee
+	class Middleware
+	  DEFAULTS_OPTIONS = {
+	    :perf_headers => true
+	  }
+
+	  def initialize app, options = {}
+	    @options = DEFAULTS_OPTIONS.merge options
+	    @app = app
+	  end
+
+	  def call env
+	    now = Time.now.to_f
+
+	    # Delete Heroku's queue time header because it's incorrect and useless
+	    env.delete("HTTP_X_HEROKU_QUEUE_WAIT_TIME")
+
+	    microseconds = (now * 1_000_000).to_i
+	    env["HTTP_X_MIDDLEWARE_START"] = "t=#{microseconds}"
+
+	    perf_headers = {}
+	    if (request_start = env["HTTP_X_REQUEST_START"])
+	      request_start_microseconds = request_start.gsub("t=", "").to_i * 1_000
+	      queue_time_microseconds = [ microseconds - request_start_microseconds, 0 ].max
+	      env["HTTP_X_QUEUE_TIME"] = "t=#{queue_time_microseconds}"
+
+	      queue_time_milliseconds = (queue_time_microseconds / 1_000).to_i
+	      perf_headers["X-Queue-Time"] = queue_time_milliseconds.to_s if @options[:perf_headers]
+	    end
+
+	    status, headers, body = @app.call(env)
+
+	    [ status, headers.merge(perf_headers), body ]
+	  end
+	end
+end

data/lib/heroku_queue_time_logger/railtie.rb

@@ -0,0 +1,7 @@
+module HerokuQueueTimeLogger
+	class Railtie < Rails::Railtie
+	  initializer "heroku_queue_time_logger.configure_rails_initialization" do |app|
+	    app.middleware.insert 0, HerokuQueueTimeLogger::Middleware
+	  end
+	end
+end

data/lib/heroku_queue_time_logger/version.rb

@@ -0,0 +1,3 @@
+module HerokuQueueTimeLogger
+	VERSION = '0.0.2'
+end

data/lib/heroku_queue_time_logger.rb

@@ -0,0 +1,3 @@
+require 'heroku_queue_time_logger/version'
+require 'heroku_queue_time_logger/middleware'
+require 'heroku_queue_time_logger/railtie' if defined?(Rails)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: heroku_queue_time_logger
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -20,6 +20,11 @@ extra_rdoc_files: []
 files:
 - .gitignore
 - README.md
+- heroku_queue_time_logger.gemspec
+- lib/heroku_queue_time_logger.rb
+- lib/heroku_queue_time_logger/middleware.rb
+- lib/heroku_queue_time_logger/railtie.rb
+- lib/heroku_queue_time_logger/version.rb
 homepage: http://github.com/neutroncreations/heroku_queue_time_logger
 licenses: []
 post_install_message: