sleuth 0.2.0

data/lib/sleuth.rb

@@ -0,0 +1,54 @@
+require 'rack/client'
+
+require 'active_support/core_ext/module/delegation'
+require 'active_support/core_ext/class/attribute_accessors'
+require 'active_support/notifications'
+require 'active_support/buffered_logger'
+
+module Sleuth
+  TRANSACTION_HEADER = "X_SLEUTH_TRANSACTION"
+
+  class << self
+    delegate :head, :get, :post, :put, :delete, :to => :http
+
+    def http
+      @http ||= Rack::Client.new {
+        use OutboundHeader
+      }
+    end
+
+    def instrument(payload)
+      ActiveSupport::Notifications.instrument(:sleuth, payload) do
+        yield
+      end
+    end
+
+    def transaction(current_name, parent = nil)
+      ActiveSupport::Notifications.transaction do
+        Transaction.create(current_name, current_id, parent)
+
+        yield
+      end
+    end
+
+    def current_id
+      ActiveSupport::Notifications.transaction_id
+    end
+
+    def current_transaction
+      Transaction.running[current_id]
+    end
+
+    def watch(log_path)
+      logger = ActiveSupport::BufferedLogger.new(log_path)
+      ActiveSupport::Notifications.subscribe('sleuth') do |*args|
+        logger.debug(Transaction.message_for(*args))
+      end
+    end
+  end
+end
+
+current_dir = File.expand_path(File.dirname(__FILE__) + '/sleuth')
+require current_dir + '/middleware'
+require current_dir + '/outbound_handler'
+require current_dir + '/transaction'

data/lib/sleuth/middleware.rb

@@ -0,0 +1,20 @@
+module Sleuth
+  class Middleware
+    def initialize(app, name)
+      @app, @name = app, name
+    end
+
+    def call(env)
+      parent = env["HTTP_#{TRANSACTION_HEADER}"]
+
+      Sleuth.transaction(@name, parent) do
+        request = Rack::Request.new(env)
+        code, headers, body = Sleuth.instrument("Received #{request.request_method} #{request.url}") do
+          @app.call(env)
+        end
+        headers[TRANSACTION_HEADER] = Sleuth.current_transaction.full_name
+        [code, headers, body]
+      end
+    end
+  end
+end

data/lib/sleuth/outbound_handler.rb

@@ -0,0 +1,15 @@
+module Sleuth
+  class OutboundHeader
+    def initialize(app)
+      @app = app
+    end
+
+    def call(env)
+      request = Rack::Request.new(env)
+      Sleuth.instrument("Sending #{request.request_method} #{request.url}") do
+        env["HTTP_#{TRANSACTION_HEADER}"] = Sleuth.current_transaction.full_name
+        @app.call(env)
+      end
+    end
+  end
+end

data/lib/sleuth/transaction.rb

@@ -0,0 +1,36 @@
+module Sleuth
+  class Event < ActiveSupport::Notifications::Event
+    def formatted_time
+      "%s%06d" % [time.utc.iso8601, time.usec]
+    end
+
+    def formatted_end
+      "%s%06d" % [@end.utc.iso8601, @end.usec]
+    end
+  end
+
+  class Transaction < Struct.new(:name, :id, :stamp, :pid, :parent)
+    mattr_reader :running
+    @@running = {}
+
+    def self.create(name, id, parent)
+      running[id] = new(name, id, Time.now.to_i, Process.pid, parent)
+    end
+
+    def self.message_for(*args)
+      event = Event.new(*args)
+      transaction = running[event.transaction_id]
+      transaction.message_for(event)
+    end
+
+    def message_for(event)
+      parts = [event.formatted_time, event.formatted_end,
+               full_name, parent || '-', event.duration, event.payload.inspect]
+      "%s %s %s %s %0.4f -- %s" % parts
+    end
+
+    def full_name
+      "#{name}-#{id}-#{stamp}-#{pid}"
+    end
+  end
+end

metadata

@@ -0,0 +1,77 @@
+--- !ruby/object:Gem::Specification 
+name: sleuth
+version: !ruby/object:Gem::Version 
+  version: 0.2.0
+platform: ruby
+authors: 
+- Tim Carey-Smith
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-11-05 00:00:00 -08:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: halorgium-activesupport
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        version: 3.0.pre
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: rack-client
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - "="
+      - !ruby/object:Gem::Version 
+        version: 0.2.0
+    version: 
+description: 
+email: dev@spork.in
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/sleuth/middleware.rb
+- lib/sleuth/outbound_handler.rb
+- lib/sleuth/transaction.rb
+- lib/sleuth.rb
+has_rdoc: true
+homepage: 
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Transaction logging middleware
+test_files: []
+