rookout 0.1.23 → 0.1.27

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8b41a3be6ece6ef09531502db594ac06453723797d3c9c0f33e61ffe15a67587
-  data.tar.gz: a772b70e1fd6df14a51f1bf74e4938a576ef9b7989692eaf09d658c190789ce2
+  metadata.gz: 29081116b19d38b4ff177eb1355605b69c735aa29cfe4b883d05099fe3556a5b
+  data.tar.gz: e39b5d4e540c790d194aea231ddf33a4d2743fa67ce2b42c2043c561e569873c
 SHA512:
-  metadata.gz: 7b81e9fb7362fdf9abb57988d9dc2eb0acc099c5149243b1bf4d787c4e05d5aa4f3bc3602214eba23742ba152ac90fc18342a6abd5b872189b53626f007cea71
-  data.tar.gz: 012b9eb8d6a15ba2f267190bc1edd1adea15472bc288e78d936cb592a2726b9bac66ba06067ab111e3757fc3562b8df95d69b013e38a2567f0b4c47385900106
+  metadata.gz: 1cc4883a3543f2c6026e7c739662ab8a2ca418bad4b9335a285b7bc1b45e2c30ff63359bcd6ebd4b2a490a72dbacd9b8500b7656a343ac5ddf6d8217fbf63e2e
+  data.tar.gz: 0dffd5e418e7f81f467b1667df7370806458de425e9b2998991798226bbe7ee9367bf639b15c30b53244aa6c3dae0dc7616dd97cce70beef65ac2429cf877109

data/lib/rookout/augs/aug_factory.rb

@@ -10,6 +10,7 @@ module Rookout
     require_relative "locations/location_exception_handler"
     require_relative "aug_rate_limiter"
     require_relative "aug"
+    require_relative "../utils"
 
     class AugFactory
       def initialize output
@@ -34,7 +35,8 @@ module Rookout
 
         rate_limit = create_rate_limit configuration
 
-        aug = Aug.new aug_id, action, condition, rate_limit, max_aug_time
+        max_aug_time_ns = Utils.milliseconds_to_nanoseconds max_aug_time
+        aug = Aug.new aug_id, action, condition, rate_limit, max_aug_time_ns
 
         location_configuration = configuration["location"]
         raise Exceptions::RookAugInvalidKey.new("location", configuration) unless location_configuration.is_a? Hash
@@ -72,7 +74,10 @@ module Rookout
 
         rate_limit_modifier = configuration["rateLimitModifier"] || 5
 
-        AugRateLimiter.new window_quota, window_size, rate_limit_modifier
+        window_quota_ns = Utils.milliseconds_to_nanoseconds window_quota
+        window_size_ns = Utils.milliseconds_to_nanoseconds window_size
+
+        AugRateLimiter.new window_quota_ns, window_size_ns, rate_limit_modifier
       end
     end
   end

data/lib/rookout/augs/aug_rate_limiter.rb

@@ -1,10 +1,12 @@
 module Rookout
   module Augs
     require_relative "../exceptions"
+    require_relative "../utils"
 
     class AugRateLimiter
       def initialize quota, window_size, active_limit
         @quota = quota
+        @has_quota = !@quota.nil? && (@quota > 0)
         @window_size = window_size
         @active_weight = quota / active_limit
 
@@ -18,24 +20,22 @@ module Rookout
         start_time ||= Time.now
 
         # If quota, verify it
-        if @quota && @quota > 0
+        if @has_quota
           # Get current time
-          now_ms = (start_time.to_f * 1000).to_i
+          now_ns = Utils.time_to_nanoseconds start_time
 
           # Calculate window keys
-          current_window_key = (now_ms / @window_size) * @window_size
-          prev_window_key = current_window_key - @window_size
+          current_window_key, prev_window_key = timestamp_to_window_keys now_ns
 
           @mutex.synchronize do
             # Clean old windows
-            cleanup now_ms
-
+            cleanup now_ns
             # Increase active count
             @active_count += 1
             active = true
 
             # If exceeding quota
-            if current_usage(now_ms, current_window_key, prev_window_key) > @quota
+            if current_usage(now_ns, current_window_key, prev_window_key) > @quota
               warning = Processor::RookError.new Exceptions::RookRuleRateLimited.new
               UserWarnings.notify_warning warning
               return
@@ -46,7 +46,9 @@ module Rookout
         begin
           yield
         ensure
-          @mutex.synchronize { record_usage current_window_key, (Time.now - start_time) * 1000 } if @quota && @quota > 0
+          if @has_quota
+            @mutex.synchronize { record_usage current_window_key, Utils.time_to_nanoseconds(Time.now) - now_ns }
+          end
         end
       ensure
         # Reduce active count
@@ -55,7 +57,15 @@ module Rookout
 
       private
 
-      def current_usage now_ms, current_window_key, prev_window_key
+      attr_reader :windows # will be used only in tests using <OBJ>.send(:windows)
+
+      def timestamp_to_window_keys now_ns
+        current_window_key = (now_ns / @window_size) * @window_size
+        prev_window_key = current_window_key - @window_size
+        [current_window_key, prev_window_key]
+      end
+
+      def current_usage now_ns, current_window_key, prev_window_key
         # Get usage information
         current_window_usage = @windows[current_window_key]
         if current_window_usage.nil?
@@ -65,7 +75,7 @@ module Rookout
         prev_window_usage = @windows[prev_window_key] || 0
 
         # Previous window weight
-        prev_weight = 1 - (now_ms - current_window_key) / @window_size.to_f
+        prev_weight = 1 - (now_ns - current_window_key) / @window_size.to_f
 
         # Final weighted usage
         (prev_window_usage * prev_weight) + (@active_count * @active_weight) + current_window_usage
@@ -80,16 +90,8 @@ module Rookout
         @windows[current_window_key] += [duration, 5].max.to_f
       end
 
-      def cleanup now_ms
-        # Don't bother with contention
-        return if @mutex.locked?
-
-        @mutex.synchronize do
-          # every 5 windows-times, clear windows older than 10 window-times
-          if @windows.length > 10
-            @windows.reject! { |key, _| key < (now_ms - @window_size * 5) }
-          end
-        end
+      def cleanup now_ns
+        @windows.reject! { |key, _| key < (now_ns - @window_size * 5) } if @windows.length > 10
       end
     end
   end

data/lib/rookout/com_ws/agent_com_ws.rb

@@ -25,7 +25,7 @@ module Rookout
 
       attr_reader :pending_messages
 
-      def initialize output, agent_host, agent_port, proxy, token, labels
+      def initialize output, agent_host, agent_port, proxy, token, labels, print_on_connect
         agent_host_with_protocl = agent_host.include?("://") ? agent_host : "ws://#{agent_host}"
         @uri = "#{agent_host_with_protocl}:#{agent_port}/v1"
         if proxy.nil? || proxy.empty?
@@ -48,6 +48,8 @@ module Rookout
         @running = false
         @ready_event = Concurrent::Event.new
         once("Com::Rookout::InitialAugsCommand") { @ready_event.set }
+
+        @print_on_initial_connection = print_on_connect
       end
 
       def add message
@@ -101,6 +103,7 @@ module Rookout
 
       private
 
+      # rubocop:disable Style/StderrPuts
       def connection_thread
         backoff = Backoff.new
 
@@ -108,6 +111,10 @@ module Rookout
           begin
             client = open_new_connection
 
+            if @print_on_initial_connection
+              @print_on_initial_connection = false
+              STDERR.puts "[Rookout] Successfully connected to controller"
+            end
             Logger.instance.debug "WebSocket connected successfully"
             Logger.instance.info "Finished initialization"
 
@@ -128,6 +135,7 @@ module Rookout
           Logger.instance.debug "Reconnecting"
         end
       end
+      # rubocop:enable Style/StderrPuts
 
       def open_new_connection
         client = WebsocketClient.new @uri, @proxy, @token

data/lib/rookout/com_ws/information.rb

@@ -17,7 +17,7 @@ module Rookout
         @labels["rookout_debug"] = "on" if Config.debug
 
         k8_namespace = create_cluster_namespace k8s_file_path
-        unless k8_namespace == ""
+        unless k8_namespace == "" || @labels.key?("k8s_namespace")
           @labels["k8s_namespace"] = k8_namespace
         end
 

data/lib/rookout/commit.rb

@@ -1,3 +1,3 @@
 module Rookout
-  COMMIT = "5319eced619aa4b48a8cbc41e22e0687dc48897c".freeze
+  COMMIT = "25edb2f6d66577295d8ee34842e067d6ad046d12".freeze
 end

data/lib/rookout/config.rb

@@ -25,7 +25,7 @@ module Rookout
       Rookout::Config.agent_com_ping_interval = 10
       Rookout::Config.agent_com_ping_timeout = 30
       Rookout::Config.agent_com_flush_timeout = 3
-      Rookout::Config.agent_com_max_queued_messages = 500
+      Rookout::Config.agent_com_max_queued_messages = 100
 
       attr_accessor :backoff_factor, :backoff_reset_time, :backoff_max_time
       Rookout::Config.backoff_factor = 0.2

data/lib/rookout/interface.rb

@@ -32,6 +32,7 @@ module Rookout
         configure options
 
         rook = RookoutSingleton.instance
+        @start_options[throw_errors: throw_errors]
         rook.connect(**@start_options)
       rescue LoadError
         raise if throw_errors

data/lib/rookout/logger.rb

@@ -113,8 +113,11 @@ module Rookout
     end
 
     def build_handlers
-      @handlers.push new_file_handler unless Config.logger_filename.nil? || Config.logger_filename.empty?
-      @handlers.push new_stderr_handler if Config.logger_log_to_stderr
+      if Config.logger_log_to_stderr
+        @handlers.push new_file_handler unless Config.logger_filename.nil? || Config.logger_filename.empty?
+        @handlers.push new_stderr_handler
+      end
+
       @handlers.push new_remote_handler
     end
 

data/lib/rookout/processor/namespaces/ruby_object_namespace.rb

@@ -49,6 +49,8 @@ module Rookout
             @dump_config = OBJECT_DUMP_CONFIG_STRING
           elsif @obj.is_a?(Hash) || @obj.is_a?(Array)
             @dump_config = OBJECT_DUMP_CONFIG_COLLECTION
+          else
+            @dump_config = OBJECT_DUMP_CONFIG_TOLERANT
           end
           self
         end

data/lib/rookout/rookout_singleton.rb

@@ -40,7 +40,8 @@ module Rookout
       @services_started = false
     end
 
-    def connect token: nil, host: nil, port: nil, proxy: nil, labels: [], async_start: false, fork: false
+    def connect token: nil, host: nil, port: nil, proxy: nil, labels: [], async_start: false, fork: false,
+                throw_errors: false
       raise Exceptions::RookInterfaceException, "Multiple connection attempts not supported!" unless @agent_com.nil?
 
       if fork
@@ -52,7 +53,7 @@ module Rookout
 
       Logger.instance.debug "Initiating AgentCom-\t#{host}:#{port}"
 
-      @agent_com = ComWs::AgentComWs.new @output, host, port, proxy, token, labels
+      @agent_com = ComWs::AgentComWs.new @output, host, port, proxy, token, labels, !throw_errors
       @output.agent_com = @agent_com
       @command_handler = ComWs::CommandHandler.new @agent_com, @aug_manager
 

data/lib/rookout/services/position.rb

@@ -18,6 +18,7 @@ module Rookout
       def initialize tracer
         @tracer = tracer
         @augs = {}
+        @augs_lock = Mutex.new
         @iseqs = []
         @trace_point = TracePoint.new :script_compiled do |tp|
           begin
@@ -41,7 +42,7 @@ module Rookout
       def add_aug location
         positions = evaluate_all_scripts_to_location location
         @tracer.add_breakpoint_aug positions, location unless positions.empty?
-        @augs[location.id] = location
+        @augs_lock.synchronize { @augs[location.id] = location }
       end
 
       def remove_aug aug_id
@@ -73,11 +74,13 @@ module Rookout
         filename = iseq.absolute_path
         return if filename.nil?
 
-        @augs.each_value do |location|
-          position = evaluate_script_to_location iseq, filename, location
-          next if position.nil?
+        @augs_lock.synchronize do
+          @augs.each_value do |location|
+            position = evaluate_script_to_location iseq, filename, location
+            next if position.nil?
 
-          @tracer.add_breakpoint_aug [position], location
+            @tracer.add_breakpoint_aug [position], location
+          end
         end
       end
 

data/lib/rookout/utils.rb

@@ -6,4 +6,15 @@ module Utils
   def uuid
     SecureRandom.uuid.gsub(/-/, "")
   end
+
+  def milliseconds_to_nanoseconds milliseconds
+    nano = milliseconds * (10**6)
+    nano.to_i
+  end
+
+  def time_to_nanoseconds time_obj
+    secs = time_obj.to_i
+    nsecs = time_obj.nsec
+    (10**9) * secs + nsecs
+  end
 end

data/lib/rookout/version.rb

@@ -1,3 +1,3 @@
 module Rookout
-  VERSION = "0.1.23".freeze
+  VERSION = "0.1.27".freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rookout
 version: !ruby/object:Gem::Version
-  version: 0.1.23
+  version: 0.1.27
 platform: ruby
 authors:
 - Liran Haimovitch
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-07-19 00:00:00.000000000 Z
+date: 2021-10-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: binding_of_caller
@@ -72,14 +72,14 @@ dependencies:
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 3.9.2
+        version: 3.17.3
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 3.9.2
+        version: 3.17.3
 - !ruby/object:Gem::Dependency
   name: google-style
   requirement: !ruby/object:Gem::Requirement