vizsla 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 01dd09ce233d8fba9d914505bb601ccbab66eb39
-  data.tar.gz: 6578bab6b01119fe44cee971d50064eea2b32c6c
+  metadata.gz: 022a8c19393e8fec4151136366979fde7d45505d
+  data.tar.gz: dcb929f21d33a1dd4f0f0d50a42b770cb7fc6d5b
 SHA512:
-  metadata.gz: 853aa45a657bdc6848afebd9e3dd37386be461214241bba81a5d8cbcdbdc55c259d5788ad1daf8aafb63121d0b0679edc8aed3aaa77909d49953fba767ab159b
-  data.tar.gz: 136724d70d693bfa6c361e8336c497b8f8221d697d58bfcad0c9e143a37c94b1bd82f6750cc3bab1161e4633813e7d5a447beb2cdb0c7977df8fbd2919615ee5
+  metadata.gz: d2017c6757a5d87af0a5fd36b3cf43dda5948b21a267844f7602727fb973b5806a6c33095bd4388b00923532888522fa1061445b9eebda1990586ac6e0b6c37b
+  data.tar.gz: 8d0dbfc99b947ca32957c249a4bae2f8f470530e4c9810c24ab37bd0e9df310a488ec375d1945637b39555f929a64b620063cec475be5b68f866a4aa29475096

data/README.md

@@ -1,9 +1,5 @@
 # Vizsla
 
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/vizsla`. To experiment with that code, run `bin/console` for an interactive prompt.
-
-TODO: Delete this and the text above, and describe your gem
-
 ## Installation
 
 Add this line to your application's Gemfile:
@@ -32,7 +28,7 @@ To install this gem onto your local machine, run `bundle exec rake install`. To
 
 ## Contributing
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/vizsla. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.
+Bug reports and pull requests are welcome on GitHub at https://github.com/aastronautss/vizsla. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.
 
 
 ## License

data/lib/vizsla/helpers.rb

@@ -0,0 +1,7 @@
+module Vizsla
+  module Helpers
+    def to_milliseconds(time)
+      (time.to_f * 1000).round 1
+    end
+  end
+end

data/lib/vizsla/logger.rb

@@ -0,0 +1,23 @@
+module Vizsla
+  class RequestLogger
+    def request_response_time(query_time)
+      Rails.logger.debug "=" * 50
+      Rails.logger.debug "Total request/response time: #{query_time} seconds."
+      Rails.logger.debug "=" * 50
+    end
+
+    def log_events(events)
+      if events.empty?
+        Rails.logger.debug "=" * 50
+        Rails.logger.debug "No data collected. Are you calling ActiveRecord at all?"
+        Rails.logger.debug "=" * 50
+      else
+        events.keys.each do |event_name|
+          Rails.logger.debug "=" * 50
+          Rails.logger.debug events[event_name]
+          Rails.logger.debug "=" * 50
+        end
+      end
+    end
+  end
+end

data/lib/vizsla/middleware.rb

@@ -0,0 +1,27 @@
+require 'vizsla/timer'
+require 'vizsla/puppet_master'
+
+module Vizsla
+  class Middleware
+    def initialize(app)
+      @app = app
+    end
+
+    def call(env)
+      dup.__call(env)
+    end
+
+    def __call(env)
+      timer = Timer.new
+      timer.start!
+
+      status, headers, response = @app.call(env)
+
+      timer.stop!
+
+      PuppetMaster.new(timer).process
+
+      [status, headers, response]
+    end
+  end
+end

data/lib/vizsla/patches.rb

@@ -0,0 +1,45 @@
+require 'vizsla/helpers'
+
+module Vizsla
+  class Patches
+    include ::Vizsla::Helpers
+
+    class << self
+      def patch_postgres
+        if block_given?
+          @postgres_event_hanlder = Proc.new(&block)
+        end
+
+        ::PG::Connection.class_eval do
+          alias_method :exec_without_profiling, :exec
+
+          def exec(*args, &blk)
+            return exec_without_profiling(*args, &blk)
+
+            start_time   = Time.now
+            result       = exec_without_profiling(*args, &blk)
+            end_time     = Time.now
+
+            event_data = [
+              'sql.postgres_exec',
+              start_time,
+              end_time,
+              {
+                sql: args[0]
+              }
+            ]
+
+            ::Vizsla::Patches.handle_event :postgres, event_data
+
+            result
+          end
+        end
+      end
+
+      def handle_event(handler_name, event_data)
+        handler = self.get_instance_variable "@#{handler_name}_event_handler"
+        hanlder.call event_data unless hanlder.nil?
+      end
+    end
+  end
+end

data/lib/vizsla/puppet_master.rb

@@ -0,0 +1,15 @@
+require 'vizsla/logger'
+
+module Vizsla
+  class PuppetMaster
+    def initialize(timer)
+      @timer = timer
+      @logger = RequestLogger.new
+    end
+
+    def process
+      @logger.request_response_time @timer.elapsed
+      @logger.log_events @timer.events
+    end
+  end
+end

data/lib/vizsla/recorder.rb

@@ -0,0 +1,47 @@
+module Vizsla
+  class Recorder
+    THREAD_LOCAL_KEY = :_vizsla_current
+    LOCK = Mutex.new
+
+    class << self
+      def current
+        LOCK.synchronize do
+          Thread.current[THREAD_LOCAL_KEY]
+        end
+      end
+
+      def current=(val)
+        Thread.current[THREAD_LOCAL_KEY] = val
+      end
+
+      def start_recording
+        self.current = {}
+      end
+
+      def recording?
+        !self.current.nil?
+      end
+
+      def add_event(event)
+        return unless self.recording?
+        self.current[event.recorder_type] ||= []
+        self.current[event.recorder_type] << event.prettify_data
+      end
+      alias_method :<<, :add_event
+
+      def events
+        self.current
+      end
+
+      def stop_recording
+        LOCK.synchronize do
+          Thread.current[THREAD_LOCAL_KEY] = nil
+        end
+      end
+
+      at_exit do
+        self.stop_recording
+      end
+    end
+  end
+end

data/lib/vizsla/subscribers.rb

@@ -0,0 +1,124 @@
+require 'vizsla/recorder' unless defined?(::Vizsla::Recorder)
+require 'vizsla/patches' unless defined?(::Vizsla::Patches)
+
+module Vizsla
+  class Event
+    attr_reader :event
+
+    def initialize(event)
+      @event = event
+    end
+
+    def recorder_type
+      event[0]
+    end
+
+    def valid?
+      true
+    end
+
+    def prettify_data
+      {
+        event_started: event[1],
+        event_ended: event[2],
+        event_duration: event[2] - event[1],
+        event_payload: prettify_payload
+      }
+    end
+  end
+
+  class SQLEvent < Event
+    def valid?
+      event.last[:name] != "SCHEMA"
+    end
+
+    private
+
+    def prettify_payload
+      {
+        query: event.last[:sql]
+      }
+    end
+  end
+
+  class ControllerEvent < Event
+    private
+
+    def prettify_payload
+      payload = event.last
+      {
+        format: payload[:format],
+        controller: payload[:controller],
+        action: payload[:action],
+        path: payload[:path],
+        db_runtime: payload[:db_runtime]
+      }
+    end
+  end
+
+  class ViewEvent < Event
+    private
+
+    def prettify_payload
+      {
+        layout: event.last[:layout]
+      }
+    end
+  end
+
+  class Subscribers
+    def initialize
+      @events_data = Recorder
+      # @logger = RequestLogger.new
+      collect_events_data
+    end
+
+    def sql_hook
+      ActiveSupport::Notifications.subscribe "sql.active_record" do |*args|
+        event = SQLEvent.new(args)
+        @events_data << event if event.valid?
+      end
+    end
+
+    def process_action_hook
+      ActiveSupport::Notifications.subscribe "process_action.action_controller" do |*args|
+        event = ControllerEvent.new(args)
+        @events_data << event
+      end
+    end
+
+    def render_template_hook
+      ActiveSupport::Notifications.subscribe "render_template.action_view" do |*args|
+        event = ViewEvent.new(args)
+        @events_data << event
+      end
+    end
+
+    # ===---------------------------===
+    # Non-Rails Hooks
+    # ===---------------------------===
+
+    def postgres_hook
+      if !defined? Rails
+        ::Vizsla::Patches.patch_postgres do |event_data|
+          event = SQLEvent.new event_data
+          @events_data << event
+        end
+      end
+    end
+
+    # ===---------------------------===
+    # Aux
+    # ===---------------------------===
+
+    def collect_events_data
+      sql_hook
+      process_action_hook
+      render_template_hook
+    end
+
+    def report_events_data
+      @logger.log_events(@events_data)
+    end
+  end
+end

data/lib/vizsla/timer.rb

@@ -0,0 +1,31 @@
+require 'vizsla/recorder'
+
+module Vizsla
+  class Timer
+    attr_reader :events
+
+    def initialize
+      @start_time = nil
+      @stop_time = nil
+    end
+
+    def start!
+      @start_time = Time.now
+      Recorder.start_recording
+    end
+
+    def stop!
+      collect_events
+      Recorder.stop_recording
+      @stop_time = Time.now
+    end
+
+    def collect_events
+      @events = Recorder.events
+    end
+
+    def elapsed
+      "#{(@stop_time - @start_time).round 2}s"
+    end
+  end
+end

data/lib/vizsla/version.rb

@@ -1,3 +1,3 @@
 module Vizsla
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/lib/vizsla.rb

@@ -1,5 +1,9 @@
-require "vizsla/version"
+require 'vizsla/version'
+require 'vizsla/middleware'
+require 'vizsla/subscribers'
 
 module Vizsla
-  # Your code goes here...
+  class Core
+    @subscribers = Subscribers.new
+  end
 end

data/vizsla.gemspec

@@ -23,4 +23,5 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "bundler", "~> 1.14"
   spec.add_development_dependency "rake", "~> 10.0"
   spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_development_dependency 'pry', '~> 0.10'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: vizsla
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Tyler Guillen
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-04-04 00:00:00.000000000 Z
+date: 2017-04-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -52,6 +52,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.10'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.10'
 description: 
 email:
 - tyguillen@gmail.com
@@ -70,6 +84,14 @@ files:
 - bin/console
 - bin/setup
 - lib/vizsla.rb
+- lib/vizsla/helpers.rb
+- lib/vizsla/logger.rb
+- lib/vizsla/middleware.rb
+- lib/vizsla/patches.rb
+- lib/vizsla/puppet_master.rb
+- lib/vizsla/recorder.rb
+- lib/vizsla/subscribers.rb
+- lib/vizsla/timer.rb
 - lib/vizsla/version.rb
 - vizsla.gemspec
 homepage: https://google.com