callstacking-rails 0.1.30 → 0.1.32

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7423b24771b1ee8ad73efbd79f287546e2b92876743424250064541c0a91eb0e
-  data.tar.gz: 79ef13978d8252f26950486c3589b7d8ad2d5414a35683ba95ea0768b6187e52
+  metadata.gz: 002a4d9f1b5077534ba5ea5dfc3b418fa77325a4028cd4ed9322803cb0200fb5
+  data.tar.gz: e9eb4f9d680d978e66f38f754b8c8e251e243628b8317dfa004df95545661048
 SHA512:
-  metadata.gz: 52d1c88ad8385cf8f9f228672927fb8bcc3397fd4799e8dd3987885fe68982bbb6a45e32b7bad7ebae8aa02f9362e727636dc3c23f2b98a04427a1662023b8d5
-  data.tar.gz: 6f4ef1b35a429613b39fc898edb8e390711895134a0fddf295a988e9e738a182423fdfb923ae17008af36190086e7ef56189d597dae54def16ff913edcee7dd6
+  metadata.gz: 70c5b03becd0e1e054a5d7142130dfce3ae8e7310a9a249ea6742a95f45b5f3f52a8f59832820c1cd981fbd556ede7f2b672de1ad56cc61e16a978a216fd0b55
+  data.tar.gz: 54719e4784c2c945e05eafdbc202de21c4bf425a5d0a4ce1c9403e9154851ba18a63883e2bf32712d23342cb5f9c0257d65bfd02630b9cb6bd4b58f878479baf

data/lib/callstacking/rails/client/base.rb

@@ -21,8 +21,6 @@ module Callstacking
           # https://github.com/lostisland/awesome-faraday
           @connection ||= Faraday.new(url) do |c|
             c.response :json
-            c.use Faraday::Response::Logger, Logger.new('/tmp/callstacking-rails.log')
-            # c.use Faraday::Response::Logger, nil, { headers: false, bodies: false }
             c.response :follow_redirects
             c.use Faraday::Response::RaiseError # raise exceptions on 40x, 50x responses
             c.request :json # This will set the "Content-Type" header to application/json and call .to_json on the body
@@ -35,7 +33,7 @@ module Callstacking
           end
         end
 
-        def get(url, params = {})
+        def get(url, params = {}, headers = {})
           if async
             threads << Thread.new do
               connection.get(url, params, headers)

data/lib/callstacking/rails/client/trace.rb

@@ -6,6 +6,7 @@ module Callstacking
       class Trace < Base
         CREATE_URL = "/api/v1/traces.json"
         UPDATE_URL = "/api/v1/traces/:id.json"
+        SHOW_URL = "/api/v1/traces/:id.json"
 
         def initialize(url, auth_token)
           super
@@ -37,6 +38,10 @@ module Callstacking
         def upsert(trace_id, traces)
           patch(UPDATE_URL.gsub(':id', trace_id), {}, traces)
         end
+
+        def show(trace_id, params = {})
+          get(SHOW_URL.gsub(':id', trace_id), params)
+        end
       end
     end
   end

data/lib/callstacking/rails/engine.rb

@@ -13,46 +13,81 @@ require "callstacking/rails/client/trace"
 require "callstacking/rails/cli"
 require "callstacking/rails/time_based_uuid"
 require "callstacking/rails/helpers/instrument_helper"
+require "callstacking/rails/logger"
 
 module Callstacking
   module Rails
     class Engine < ::Rails::Engine
-      EXCLUDED_TEST_CLASSES = %w[test/dummy/app/models/salutation.rb test/dummy/app/controllers/application_controller.rb].freeze
+      EXCLUDED_TEST_CLASSES = %w[test/dummy/app/models/salutation.rb 
+                                 test/dummy/app/controllers/application_controller.rb].freeze
       
-      cattr_accessor :spans, :trace, :settings, :instrumenter, :loader
+      cattr_accessor :spans, :traces, :settings, :instrumenter, :loader, :lock
 
       isolate_namespace Callstacking::Rails
 
+      @@spans||={}
+      @@traces||={}
+      @@lock||=Mutex.new
+      @@instrumenter||=Instrument.new
       @@settings||=Callstacking::Rails::Settings.new
-      @@spans||=Spans.new
-      @@trace||=Trace.new(@@spans)
-      @@instrumenter||=Instrument.new(@@spans)
 
       initializer "engine_name.assets.precompile" do |app|
         app.config.assets.precompile << "checkpoint_rails_manifest.js"
       end
 
       config.after_initialize do
-        puts "Call Stacking loading (#{Callstacking::Rails::Env.environment})"
-          
-        @@loader = Callstacking::Rails::Loader.new(@@instrumenter, excluded: @@settings.excluded + EXCLUDED_TEST_CLASSES)
-        @@loader.on_load
+        Logger.log "Call Stacking loading (#{Callstacking::Rails::Env.environment})"
+
+        spans[Thread.current.object_id]||=Spans.new
+        instrumenter.add_span(spans[Thread.current.object_id])
+
+        @@loader = Callstacking::Rails::Loader.new(instrumenter,
+                                                   excluded: settings.excluded + EXCLUDED_TEST_CLASSES)
+        loader.on_load
       end
 
+      # Serialize all tracing requests for now.
+      #  Can enable parallel tracing later.
       def self.start_tracing(controller)
-        @@settings.enable!
-        @@loader.reset!
-        @@instrumenter.enable!(@@loader.klasses.to_a)
-        @@trace.begin_trace(controller)
+        Logger.log("Callstacking::Rails::Engine.start_tracing")
+        
+        settings.enable!
+
+        lock.synchronize do
+          spans[Thread.current.object_id]||=Spans.new
+          span = spans[Thread.current.object_id]
+
+          instrumenter.add_span(span)
+
+          if instrumenter.instrumentation_required?
+            loader.reset!
+            instrumenter.enable!(loader.klasses.to_a)
+          end
+
+          traces[Thread.current.object_id] = Trace.new(span)
+          trace = traces[Thread.current.object_id]
+
+          trace.begin_trace(controller)
+        end
 
         true
       end
 
       def self.stop_tracing(controller)
-        @@settings.disable!
-        @@instrumenter.disable!
-        @@trace.end_trace(controller)
+        Logger.log("Callstacking::Rails::Engine.stop_tracing")
+        
+        settings.disable!
+
+        trace = nil
+        lock.synchronize do
+          trace = traces.delete(Thread.current.object_id)
+          if traces.empty?
+            instrumenter.disable!
+          end
+        end
 
+        trace&.end_trace(controller)
+        
         true
       end
     end

data/lib/callstacking/rails/instrument.rb

@@ -7,8 +7,8 @@ module Callstacking
       attr_accessor :spans
       attr_reader :settings, :span_modules
 
-      def initialize(spans)
-        @spans = spans
+      def initialize
+        @spans = {}
         @span_modules = Set.new
         @settings = Callstacking::Rails::Settings.new
       end
@@ -34,6 +34,9 @@ module Callstacking
         new_method = nil
         if RUBY_VERSION < "2.7.8"
           new_method = tmp_module.define_method(method_name) do |*args, &block|
+            settings = tmp_module.instance_variable_get(:@settings)
+            return super(*args, &block) if settings.disabled?
+            
             method_name = __method__
 
             path = method(__method__).super_method.source_location.first
@@ -42,19 +45,23 @@ module Callstacking
             p, l = caller.find { |c| c.to_s =~ /#{::Rails.root.to_s}/}&.split(':')
 
             spans = tmp_module.instance_variable_get(:@spans)
+            span  = spans[Thread.current.object_id]
             klass = tmp_module.instance_variable_get(:@klass)
 
             arguments = Callstacking::Rails::Instrument.arguments_for(method(__method__).super_method, args)
 
-            spans.call_entry(klass, method_name, arguments, p || path, l || line_no)
+            span.call_entry(klass, method_name, arguments, p || path, l || line_no)
             return_val = super(*args, &block)
-            spans.call_return(klass, method_name, p || path, l || line_no, return_val)
+            span.call_return(klass, method_name, p || path, l || line_no, return_val)
 
             return_val
           end
           new_method.ruby2_keywords if new_method.respond_to?(:ruby2_keywords)
         else
           new_method = tmp_module.define_method(method_name) do |*args, **kwargs, &block|
+            settings = tmp_module.instance_variable_get(:@settings)
+            return super(*args, **kwargs, &block) if settings.disabled?
+
             method_name = __method__
 
             path = method(__method__).super_method.source_location.first
@@ -63,13 +70,14 @@ module Callstacking
             p, l = caller.find { |c| c.to_s =~ /#{::Rails.root.to_s}/}&.split(':')
 
             spans = tmp_module.instance_variable_get(:@spans)
+            span  = spans[Thread.current.object_id]
             klass = tmp_module.instance_variable_get(:@klass)
 
             arguments = Callstacking::Rails::Instrument.arguments_for(method(__method__).super_method, args)
 
-            spans.call_entry(klass, method_name, arguments, p || path, l || line_no)
+            span.call_entry(klass, method_name, arguments, p || path, l || line_no)
             return_val = super(*args, **kwargs, &block)
-            spans.call_return(klass, method_name, p || path, l || line_no, return_val)
+            span.call_return(klass, method_name, p || path, l || line_no, return_val)
 
             return_val
           end
@@ -80,8 +88,6 @@ module Callstacking
       end
 
       def enable!(klasses)
-        reset!
-
         Array.wrap(klasses).each do |klass|
           instrument_klass(klass, application_level: true)
         end
@@ -97,6 +103,10 @@ module Callstacking
         reset!
       end
 
+      def instrumentation_required?
+        span_modules.empty?
+      end
+
       def reset!
         span_modules.clear
       end
@@ -120,6 +130,10 @@ module Callstacking
         f.filter h
       end
 
+      def add_span(span)
+        spans[Thread.current.object_id] ||= span
+      end
+
       private
       def find_or_initialize_module(klass)
         name = klass&.name rescue nil
@@ -137,6 +151,7 @@ module Callstacking
 
           new_module.instance_variable_set("@klass", klass)
           new_module.instance_variable_set("@spans", spans)
+          new_module.instance_variable_set("@settings", settings)
 
           klass.prepend new_module
           klass.singleton_class.prepend new_module if klass.class == Module

data/lib/callstacking/rails/loader.rb

@@ -1,13 +1,21 @@
 require "rails"
+require "callstacking/rails/logger"
 
 module Callstacking
   module Rails
     class Loader
-      attr_accessor :instrumenter, :klasses, :excluded
+      attr_accessor :instrumenter, :klasses, :excluded, :settings
       def initialize(instrumenter, excluded: [])
         @excluded = excluded
         @instrumenter = instrumenter
         @klasses = Set.new
+        @settings = Callstacking::Rails::Settings.new
+
+        preloaded_klasses
+      end
+
+      def preloaded_klasses
+        ObjectSpace.each_object(Module){|ob| filter_klass(ob, (Object.const_source_location(ob.to_s)&.first rescue nil))}
       end
 
       def on_load
@@ -15,13 +23,7 @@ module Callstacking
           klass = tp.self
           path  = tp.path
 
-          excluded_klass = excluded.any? { |ex| path =~ /#{ex}/ }
-
-          if path =~ /#{::Rails.root.to_s}/ &&
-            !klasses.include?(klass) &&
-            !excluded_klass
-              klasses << klass
-          end
+          filter_klass(klass, path)
         end
 
         trace.enable
@@ -31,6 +33,21 @@ module Callstacking
         instrumenter.instrument_method(ActionView::PartialRenderer, :render, application_level: false)
         instrumenter.instrument_method(ActionView::TemplateRenderer, :render, application_level: false)
       end
+
+      private
+      def filter_klass(klass, path)
+        return if klass.nil? || path.nil?
+        return if path == false
+
+        excluded_klass = excluded.any? { |ex| path =~ /#{ex}/ }
+
+        if path =~ /#{::Rails.root.to_s}/ &&
+          !klasses.include?(klass) &&
+          !excluded_klass
+          instrumenter.instrument_klass(klass) if settings.enabled?
+          klasses << klass
+        end
+      end
     end
   end
 end

data/lib/callstacking/rails/logger.rb

@@ -0,0 +1,18 @@
+module Callstacking
+  module Rails
+    class Logger
+      def self.log(message)
+        puts message
+        
+        if ENV['GITHUB_OUTPUT'].present?
+          File.open(ENV['GITHUB_OUTPUT'], 'a') do |file|
+            # Write your progress output to the file
+            # This could be inside a loop or condition, depending on your needs
+            file.puts "::set-output name=progress_output::#{message}"
+          end
+        end
+      end
+    end
+  end
+end
+

data/lib/callstacking/rails/settings.rb

@@ -35,14 +35,14 @@ module Callstacking
       end
 
       def self.enable!
-        ::Rails.cache.write(CACHE_KEY, true)
+        Thread.current[CACHE_KEY] = true
       end
       def enable!
         self.class.enable!
       end
 
       def self.disable!
-        ::Rails.cache.write(CACHE_KEY, false)
+        Thread.current[CACHE_KEY] = false
       end
 
       def disable!
@@ -50,11 +50,9 @@ module Callstacking
       end
 
       def enabled?
-        return ::Rails.cache.read(CACHE_KEY) unless ::Rails.cache.read(CACHE_KEY).nil?
-        return false if ENV[ENV_KEY] == 'false'
-        return false if settings.nil?
-        
-        settings[:enabled]
+        return ActiveRecord::Type::Boolean.new.cast(ENV[ENV_KEY]) if ENV[ENV_KEY].present?
+        return Thread.current[CACHE_KEY] if Thread.current[CACHE_KEY].present?
+        false
       end
 
       def excluded

data/lib/callstacking/rails/trace.rb

@@ -7,18 +7,22 @@ module Callstacking
     class Trace
       include Callstacking::Rails::Helpers::HeadsUpDisplayHelper
       
-      attr_accessor :spans, :client, :lock
+      attr_accessor :spans, :client, :lock, :traces
       attr_reader :settings
       cattr_accessor :current_trace_id
       cattr_accessor :current_tuid
+      cattr_accessor :trace_log
 
       ICON = '💥'
       MAX_TRACE_ENTRIES = 3000
+      
+      @@trace_log||={}
 
       def initialize(spans)
-        @traces   = []
-        @spans    = spans
-        @settings = Callstacking::Rails::Settings.new
+
+        @traces    = []
+        @spans     = spans
+        @settings  = Callstacking::Rails::Settings.new
 
         @lock     = Mutex.new
         @client   = Callstacking::Rails::Client::Trace.new(settings.url, settings.auth_token)
@@ -29,6 +33,8 @@ module Callstacking
 
       def begin_trace(controller)
         @trace_id, @tuid = init_uuids(controller.request&.request_id || SecureRandom.uuid, TimeBasedUUID.generate)
+        trace_log[@trace_id] = controller.request&.original_url
+        
         init_callbacks(@tuid)
 
         start_request(@trace_id, @tuid,
@@ -50,6 +56,10 @@ module Callstacking
         inject_hud(@settings, controller.request, controller.response)
       end
 
+      def self.trace_log_clear
+        trace_log.clear
+      end
+
       private
 
       def init_callbacks(tuid)

data/lib/callstacking/rails/version.rb

@@ -1,5 +1,5 @@
 module Callstacking
   module Rails
-    VERSION = "0.1.30"
+    VERSION = "0.1.32"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: callstacking-rails
 version: !ruby/object:Gem::Version
-  version: 0.1.30
+  version: 0.1.32
 platform: ruby
 authors:
 - Jim Jones
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-05-08 00:00:00.000000000 Z
+date: 2023-05-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -112,6 +112,7 @@ files:
 - lib/callstacking/rails/helpers/instrument_helper.rb
 - lib/callstacking/rails/instrument.rb
 - lib/callstacking/rails/loader.rb
+- lib/callstacking/rails/logger.rb
 - lib/callstacking/rails/settings.rb
 - lib/callstacking/rails/setup.rb
 - lib/callstacking/rails/spans.rb