instruments 0.0.1 → 0.0.2

data/lib/instruments.rb

@@ -1,2 +1,91 @@
 module Instruments
+  def self.defaults=(args)
+    @logger = args[:logger] || Kernel
+    @method = args[:method] || :puts
+    @default_data = args[:data] || {}
+  end
+
+  def self.write(data)
+    @logger.send(@method, data.merge(@default_data))
+  end
+
+  if defined?(::Sinatra)
+    module ::Sinatra
+      module Instrumentation
+        def route(verb, action, *)
+          condition {@instrumented_route = action}
+          super
+        end
+
+        def instrument_routes
+          before do
+            @start_request = Time.now
+          end
+          after do
+            t = Integer((Time.now - @start_request)*1000)
+            Instruments.write({
+              :lib => "sinatra",
+              :action => "http-request",
+              :route => @instrumented_route,
+              :elapsed => t,
+              :method => env["REQUEST_METHOD"].downcase,
+              :status => response.status
+            }.merge(params))
+          end
+        end
+      end
+      register Instrumentation
+    end
+  end
+
+  if defined?(::Sequel)
+    module ::Sequel
+      class Database
+        def log_yield(sql, args=nil)
+          sql = "#{sql}; #{args.inspect}" if args
+          t0 = Time.now
+          begin
+            yield
+          rescue => e
+            log_exception(e, sql)
+            raise
+          ensure
+            t1 = Time.now
+            log_duration(Integer((t1-t0)*1000), sql) unless e
+          end
+        end
+
+        def log_duration(t, sql)
+          Instruments.write(:info => true, :action => action(sql), :elapsed => t, :sql => sql)
+        end
+
+        def log_exception(e, sql)
+          Instruments.write(:error => true, :exception => e.class, :sql => sql)
+        end
+
+        def action(sql)
+          sql[/(\w+){1}/].downcase
+        end
+      end
+    end
+  end
+
+  if defined?(::Excon)
+    module ::Excon
+      module Instrumentation
+        def self.instrument(name, params={}, &blk)
+          t0 = Time.now
+          res = yield if block_given?
+          t1 = Time.now
+          Instruments.write(
+            :lib => "excon",
+            :action => "http-request",
+            :elapsed => Integer((t1-t0)*1000)
+          )
+        end
+      end
+    end
+    Excon.defaults[:instrumentor] = ::Excon::Instrumentation
+  end
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: instruments
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors: