rookout 0.1.22 → 0.1.23

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 1189352fe3bb2817944ec553c3a658bcd0d677781ffee90190521e1ed55189c3
-  data.tar.gz: 63bf916c9f3140c4bf2be89a3469ffd3f451f37edc4741c39c6336ef1373269c
+  metadata.gz: 8b41a3be6ece6ef09531502db594ac06453723797d3c9c0f33e61ffe15a67587
+  data.tar.gz: a772b70e1fd6df14a51f1bf74e4938a576ef9b7989692eaf09d658c190789ce2
 SHA512:
-  metadata.gz: cace5dbc2f61f8e1987397345061080fecc85cbcaeb065976f8a8ae071621c01b717e46dd98df55a48fd399f7b2f02b467b786b335d9dcf4088e09a3094f1961
-  data.tar.gz: d33370845b6bb7166c93626e74b73a743d5a799f8b69722bae627561fbd747a36542c2a38e7f3135f099e90415d5b5219775407e942a8f3e25c60939a1be9012
+  metadata.gz: 7b81e9fb7362fdf9abb57988d9dc2eb0acc099c5149243b1bf4d787c4e05d5aa4f3bc3602214eba23742ba152ac90fc18342a6abd5b872189b53626f007cea71
+  data.tar.gz: 012b9eb8d6a15ba2f267190bc1edd1adea15472bc288e78d936cb592a2726b9bac66ba06067ab111e3757fc3562b8df95d69b013e38a2567f0b4c47385900106

data/lib/rookout/augs/aug.rb

@@ -28,6 +28,11 @@ module Rookout
       def execute frame, extracted, output
         return unless @enabled
 
+        if output.user_messages_queue_full?
+          Logger.instance.warning "Skipping aug-\t#{id} execution because the queue is full"
+          return
+        end
+
         namespace = create_namespaces frame, extracted
         return if @condition && !@condition.evaluate(namespace)
 

data/lib/rookout/com_ws/agent_com_ws.rb

@@ -23,6 +23,8 @@ module Rookout
     class AgentComWs
       include EventEmitter
 
+      attr_reader :pending_messages
+
       def initialize output, agent_host, agent_port, proxy, token, labels
         agent_host_with_protocl = agent_host.include?("://") ? agent_host : "ws://#{agent_host}"
         @uri = "#{agent_host_with_protocl}:#{agent_port}/v1"
@@ -62,6 +64,10 @@ module Rookout
         @pending_messages.push buffer if @pending_messages.length < Config.agent_com_max_queued_messages
       end
 
+      def queue_full?
+        @pending_messages.length >= Config.agent_com_max_queued_messages
+      end
+
       def connect
         @running = true
 

data/lib/rookout/com_ws/information.rb

@@ -11,11 +11,16 @@ module Rookout
       require_relative "git"
       include Git
 
-      def initialize labels
+      def initialize labels, k8s_file_path = "/var/run/secrets/kubernetes.io/serviceaccount/namespace"
         @agent_id = nil
         @labels = labels.clone
         @labels["rookout_debug"] = "on" if Config.debug
 
+        k8_namespace = create_cluster_namespace k8s_file_path
+        unless k8_namespace == ""
+          @labels["k8s_namespace"] = k8_namespace
+        end
+
         @ip_addr = local_ip
 
         @scm_info = create_scm_information
@@ -49,7 +54,7 @@ module Rookout
                                            scm: @scm_info
       end
 
-      attr_accessor :agent_id
+      attr_accessor :agent_id, :labels
 
       private
 
@@ -65,6 +70,18 @@ module Rookout
       end
       # rubocop:enable Style/ParallelAssignment
 
+      def create_cluster_namespace k8s_file_path
+        return_value = ""
+
+        # No Kubernetes is valid
+        if File.file? k8s_file_path
+          # Read the file contents and return it as result
+          return_value = File.read k8s_file_path
+        end
+
+        return_value
+      end
+
       def create_scm_information
         user_git_origin = Config.user_git_origin || ENV["ROOKOUT_REMOTE_ORIGIN"]
         user_git_commit = Config.user_git_commit || ENV["ROOKOUT_COMMIT"]

data/lib/rookout/com_ws/output.rb

@@ -49,6 +49,10 @@ module Rookout
         Logger.instance.remove_output self
       end
 
+      def user_messages_queue_full?
+        @user_message_bucket.exhausted? || (!@agent_com.nil? && @agent_com.queue_full?)
+      end
+
       def send_warning rule_id, error
         send_rule_status rule_id, :Warning, error
       end

data/lib/rookout/commit.rb

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

data/lib/rookout/services/tracer.rb

@@ -20,7 +20,7 @@ module Rookout
 
           begin
             trace_point.enable target: position.method, target_line: position.lineno
-          rescue ArgumentError => e
+          rescue RuntimeError, ArgumentError => e
             raise Exceptions::RookSetTracepointFailed.new(position.lineno, e)
           end
 

data/lib/rookout/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rookout
 version: !ruby/object:Gem::Version
-  version: 0.1.22
+  version: 0.1.23
 platform: ruby
 authors:
 - Liran Haimovitch
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-05-11 00:00:00.000000000 Z
+date: 2021-07-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: binding_of_caller