thirtysix 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 80a87d7c92d6d215816ac100818dac9e0ac9ee1e
-  data.tar.gz: c416ffc956da0afd1eb3105b2509975ce2729335
+  metadata.gz: 0ebddafbe1045a12501e5df96fa6d2b15e6c5c72
+  data.tar.gz: 711d3d9cea916f2c5ae650d3378d88767b176755
 SHA512:
-  metadata.gz: 354aaaa56a7eab627e3ffd95235b0b5611f6f54cfa5404abf95f4cfbea6415dd501dfbae24a5c590c74d65101347b46baa4ec61173f155c33b932977ea18700e
-  data.tar.gz: 20fe48621762137449d0af99e720da8d8d1b023247ee08246daf2a24bec5ae889bac8d9d4332ab5d8b310673bba381ded39a815b5c0de960793755622390aa35
+  metadata.gz: d5573dab0362dbed7e830a8738ca0853cc3c41690a719e03df7890d883ab323aeb959ca3c2308a8b65d08c0ce08a83aeebdde12e8fd372a9784995a5c5bffed3
+  data.tar.gz: 1fbdedda992ad6855922ea3cccec2bead0745d41175eaf98fbb6b4982c8ba04ba68bbbb6ebf92fbc8a38c5b3f6b55a717a6d3fd1b5ed69cb9e159e60261d6fb9

data/lib/thirtysix/rack/middleware.rb

@@ -0,0 +1,80 @@
+require 'objspace'
+require 'thirtysix/app_request'
+require 'thirtysix/rack/railtie' if defined?(Rails)
+
+module Thirtysix
+  module Rack
+    class Middleware
+      def initialize(app)
+        puts "Thirtysix: Initializing Middleware"
+        @app = app
+      end
+
+      def call(env)
+        start_time    = Time.now
+        start_memory  = get_memory
+        start_objects = ObjectSpace.count_objects_size
+        
+        status, headers, body = @app.call(env)
+
+        if env["ORIGINAL_FULLPATH"].include? "assets"
+          puts "Thirtysix: Skipping instrumentation for assets requests"
+        else
+          end_time    = Time.now
+          end_memory  = get_memory
+          end_objects = ObjectSpace.count_objects_size
+
+          app_request = Thirtysix::AppRequest.new
+          app_request.request_time_in_milliseconds = ((end_time - start_time) * 1000).to_i
+          app_request.memory_in_mb = end_memory - start_memory
+          app_request.build(start_objects, end_objects, env)
+
+          push app_request
+        end
+        
+        [status, headers, body]
+      end
+
+      #
+      # Returns current process' memory usage in MB
+      # Returns nil if any error occurred
+      # Only works on Darwin10
+      # Shamelessly stolen from: 
+      #   https://github.com/newrelic/rpm/blob/master/lib/new_relic/agent/samplers/memory_sampler.rb
+      #
+      def get_memory
+        (`ps -o rss #{$$}`.split("\n")[1].to_f / 1024.0) rescue nil
+      end
+
+      def thirtysix_api_key
+        ENV['THIRTYSIX_API_KEY']
+      end
+
+      def thirtysix_url
+        ENV['THIRTYSIX_URL']
+      end
+
+      def thirtysix_path
+        "/api/app_requests"
+      end
+
+      def endpoint_uri
+        @endpoint_uri ||= URI(thirtysix_url + thirtysix_path)
+      end
+
+      def push(app_request)
+        begin
+          http = Net::HTTP.new(endpoint_uri.host, endpoint_uri.port)
+          http.open_timeout = 0.2
+          http.read_timeout = 0.2
+          headers = {'Content-Type' => 'text/json', 'X-THIRTYSIX-API-KEY' => thirtysix_api_key}
+          request = Net::HTTP::Post.new(endpoint_uri.request_uri, headers)
+          request.body = app_request.to_json
+          response = http.request(request)
+        rescue Exception => e
+          puts "Thirtysix: We could not contact the Thirtysix client on #{thirtysix_url} [#{e.message}]"
+        end        
+      end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: thirtysix
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Gary Moore
@@ -18,6 +18,7 @@ extra_rdoc_files: []
 files:
 - Gemfile
 - README.md
+- lib/thirtysix/rack/middleware.rb
 homepage: http://github.com/gmoore/thirtysix
 licenses:
 - MIT