rookout 0.1.12 → 0.1.17

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9c698c16bfbf7520a5f3fef8f0cda0f3140ba6be1f875e04223053b63ca421dc
-  data.tar.gz: 27a0f9fa6cd4f47749b48c552a8dd6a42c1eff4e359458417df725c618fd7864
+  metadata.gz: 412df2821122abb540f4baeed9fd11432ebdc2a50373130ba148c43863c62b9b
+  data.tar.gz: afe3d60125a31190bc17dcf2b27a5fa18cf7da70acde921cc959438a8a285dfc
 SHA512:
-  metadata.gz: 0756d328dbdd0249b3cb24f109283f28687d7f414b8c81d219a974973df1e768fa7481940c0246dd044ebb11e0af44e314d199b48530f8928f6a718518cc165b
-  data.tar.gz: e9e2b8468f70300e029391fdbd19e3da9b75a25f2fcad02f6f41f1327ca781c37bd7e0d76bf26b9137f0cfb702c841c6bc29ee167e6824499b08a0736b4cddfe
+  metadata.gz: 4a576e0968b7fdc91b80e4ef7b6b465d046cb44ad5eac73e6817898c422796dedc0a33fc4749480966814029143c83984c26edfecad18b508aaf9da03455205e
+  data.tar.gz: 608e9960132a87e86fd1ab9fa6bc045d493b7f95dece96fa5dd8bcbceedc2ea538b4a30a28702ec4e7c8f7951005b9826755b48a4d3171f3a77345ac643b24ea

data/lib/rookout/com_ws/agent_com_ws.rb

@@ -51,11 +51,11 @@ module Rookout
       def add message
         buffer = wrap_in_envelope message
         if buffer.length > Config.agent_com_max_message_limit
-          exc = Exceptions::RookMessageSizeExceeded.new buffer.length, Coonfig.agent_com_max_message_limit
-          warning = RookError.new exc, message
+          exc = Exceptions::RookMessageSizeExceeded.new buffer.length, Config.agent_com_max_message_limit
+          warning = Processor::RookError.new exc
           UserWarnings.notify_warning warning
 
-          Logger.instance.warn "Dropping message, size was #{buffer.length} which is over the message size limit"
+          Logger.instance.warning "Dropping message, size was #{buffer.length} which is over the message size limit"
           return
         end
 
@@ -100,10 +100,11 @@ module Rookout
 
         while @running
           begin
-            backoff.before_connection_attempt
             client = open_new_connection
 
             Logger.instance.debug "WebSocket connected successfully"
+            Logger.instance.info "Finished initialization"
+
             @token_valid = true
             backoff.after_connect
 

data/lib/rookout/com_ws/backoff.rb

@@ -4,25 +4,20 @@ module Rookout
 
     class Backoff
       def initialize
-        @connected = false
-        @last_successful_connection = Time.mktime 1970
-        reset_backoff
-      end
-
-      def before_connection_attempt
-        return unless Time.new > @last_successful_connection + Config.backoff_reset_time
+        @connect_time = nil
         reset_backoff
       end
 
       def after_disconnect
-        @connected = false
+        reset_backoff if @connect_time && Time.new > @connect_time + Config.backoff_reset_time
+        @connect_time = nil
+
         sleep @next_backoff
         @next_backoff = [@next_backoff * 2, Config.backoff_max_time].min
       end
 
       def after_connect
-        @connected = true
-        @last_successful_connection = Time.now
+        @connect_time = Time.now
       end
 
       private

data/lib/rookout/com_ws/websocket_client.rb

@@ -2,6 +2,7 @@ module Rookout
   module ComWs
     require "openssl"
     require "websocket/driver"
+    require "uri"
 
     require_relative "../config"
     require_relative "../logger"
@@ -11,14 +12,13 @@ module Rookout
       def initialize url, proxy, token
         @token = token
         @connection = WebsocketConnection.new url, proxy
+        @proxy = proxy
         @driver = nil
-        @error = nil
         @last_ping = nil
       end
 
       def connect
-        # TODO: add proxy support
-        @connection.connect
+        connection_error = nil
         @driver = WebSocket::Driver.client @connection
 
         headers.each do |key, value|
@@ -26,18 +26,20 @@ module Rookout
         end
 
         @driver.on :error do |error|
-          @error = error
+          connection_error = error
         end
 
         # Connect to the remote server
-        @driver.start
         # TODO: ADD CONNECT TIMEOUT
+        @connection.connect @driver
+        @driver.start
+
         while @driver.state == :connecting
           recv_data = @connection.read_char
           @driver.parse recv_data
         end
 
-        raise Exceptions::RookWebsocketException, @error if @driver.state != :open
+        raise Exceptions::RookWebsocketException, connection_error if @driver.state != :open
       end
 
       def connection_pump message_handler
@@ -88,14 +90,36 @@ module Rookout
         @secure = %w[https wss].include? @uri.scheme
 
         @socket = nil
+
+        @proxy_connected = false
       end
 
-      def connect
-        if @secure
-          @socket = ssl_connect
-        else
-          @socket = tcp_connect
+      def connect driver
+        @socket = tcp_connect
+
+        proxy_connect driver if @proxy
+
+        @socket = ssl_connect @socket if @secure
+      end
+
+      def proxy_connect driver
+        connection_error = nil
+        @proxy_connected = false
+        proxy = driver.proxy @proxy
+
+        proxy.on :connect do
+          @proxy_connected = true
         end
+
+        proxy.on :error do |error|
+          connection_error = error
+        end
+
+        proxy.start
+
+        proxy.parse read_char until @proxy_connected == true || !connection_error.nil?
+
+        raise Exceptions::RookProxyException, connection_error unless connection_error.nil?
       end
 
       attr_reader :url
@@ -118,13 +142,16 @@ module Rookout
       private
 
       def tcp_connect
-        TCPSocket.new @uri.host,
-                      @uri.port || (@secure ? 443 : 80)
+        if @proxy
+          uri = URI(@proxy)
+          TCPSocket.new uri.host, uri.port
+        else
+          TCPSocket.new @uri.host,
+                        @uri.port || (@secure ? 443 : 80)
+        end
       end
 
-      def ssl_connect
-        tcp_socket = tcp_connect
-
+      def ssl_connect tcp_socket
         ctx = ::OpenSSL::SSL::SSLContext.new
         ctx.min_version = ::OpenSSL::SSL::TLS1_2_VERSION
         cert_store = ::OpenSSL::X509::Store.new

data/lib/rookout/commit.rb

@@ -1,3 +1,3 @@
 module Rookout
-  COMMIT = "0f0cf1533a7338ef3e5ca1416415ff3b54dffe98".freeze
+  COMMIT = "c966c378ce29011cd64cc0cd52ca6f4acbed2025".freeze
 end

data/lib/rookout/exceptions.rb

@@ -94,7 +94,7 @@ module Rookout
         super "Message size of #{message_size} exceeds max size limit of #{max_message_size}. " \
                   "Change the depth of collection or change the default by setting ROOKOUT_MAX_MESSAGE_SIZE " \
                   "as environment variable or system property",
-              { message_size: message_size, max_message_size: max_message_size }
+              { "message_size" => message_size, "max_message_size" => max_message_size }
       end
     end
 
@@ -177,5 +177,23 @@ module Rookout
         super "Error from Websocket #{error}", { "error" => error }
       end
     end
+
+    class RookProxyException < ToolException
+      def initialize error
+        super "Error from proxy #{error}", { "error" => error }
+      end
+    end
+
+    class RookBadProtobuf < ToolException
+      def initialize
+        super 'Bad protobuf version. Please execute "bundle config force_ruby_platform true" before "bundler install".'
+      end
+    end
+
+    class RookBadProtobufPlatform < ToolException
+      def initialize platform
+        super "Bad protobuf platform: #{platform}"
+      end
+    end
   end
 end

data/lib/rookout/interface.rb

@@ -17,24 +17,22 @@ module Rookout
       throw_errors = options[:throw_errors] == true
       Config.debug = evaluate_flag options[:debug], "ROOKOUT_DEBUG"
 
+      puts "[Rookout] Running in debug mode" if Config.debug
+
       begin
+        verify_env
+
         require_relative "rookout_singleton"
 
-        # If we are running post fork, use previous start_options
-        if options[:post_fork]
-          # Don't re-enable the fork handler
-          @start_options[:fork] = false
-        else
-          configure_logging options
-          configure_git options
-
-          @start_options = configure_start_options options
-          print_config @start_options
-        end
+        configure options
 
         rook = RookoutSingleton.instance
         rook.connect(**@start_options)
-      rescue RookMissingToken, RookInvalidToken, RookInvalidOptions, RookVersionNotSupported => e
+      rescue LoadError
+        raise if throw_errors
+        STDERR.puts "[Rookout] Failed to load Rookout. Please make sure to force build native extensions by setting" \
+          "'bundle config force_ruby_platform true'"
+      rescue RookMissingToken, RookInvalidToken, RookInvalidOptions, RookVersionNotSupported, RookBadProtobuf => e
         raise if throw_errors
         STDERR.puts "[Rookout] Failed to start Rookout: #{e.message}"
       rescue RookCommunicationException => e
@@ -61,7 +59,32 @@ module Rookout
 
     TRUE_VALUE = [true, "y", "Y", "yes", "Yes", "YES", "true", "True", "TRUE", "1"].freeze
 
-    def configure_logging options
+    def configure options
+      # If we are running post fork, use previous start_options
+      if options[:post_fork]
+        # Don't re-enable the fork handler
+        @start_options[:fork] = false
+      else
+        configure_globals options
+
+        @start_options = configure_start_options options
+        print_config @start_options
+      end
+    end
+
+    def verify_env
+      # Only test alpine
+      return unless File.exist? "/etc/alpine-release"
+
+      protobuf = Gem::Specification.find_by_path "google/protobuf"
+      STDERR.puts RookBadProtobuf.new.message if protobuf.nil?
+      return unless protobuf.platform != "ruby"
+
+      error = RookBadProtobufPlatform.new protobuf.platform
+      STDERR.puts error.message
+    end
+
+    def configure_globals options
       if Config.debug
         log_to_stderr = true
         log_level = :DEBUG
@@ -75,9 +98,7 @@ module Rookout
       Config.logger_log_to_stderr = log_to_stderr unless log_to_stderr.nil?
       Config.logger_filename = log_file unless log_file.nil?
       Config.logger_log_level = log_level unless log_level.nil?
-    end
 
-    def configure_git options
       Config.user_git_origin = options[:git_origin] if options[:git_origin]
       Config.user_git_commit = options[:git_commit] if options[:git_commit]
     end
@@ -88,7 +109,7 @@ module Rookout
       proxy = evaluate_config options[:proxy], "ROOKOUT_PROXY"
       token = evaluate_config options[:token], "ROOKOUT_TOKEN"
 
-      raise RookMissingToken if token.nil? && host == "wss://control.rookout.com"
+      raise RookMissingToken if token.nil? && !(host_specified options)
       verify_token token if token
 
       labels = stringify_labels(options[:labels]) || parse_labels(ENV["ROOKOUT_LABELS"])
@@ -100,6 +121,10 @@ module Rookout
       { host: host, port: port, proxy: proxy, token: token, labels: labels, async_start: async_start, fork: fork }
     end
 
+    def host_specified options
+      !options[:host].nil? || !ENV["ROOKOUT_CONTROLLER_HOST"].nil? || !ENV["ROOKOUT_AGENT_HOST"].nil?
+    end
+
     def evaluate_flag argument, env_var_name
       return true? argument unless argument.nil?
       true? ENV[env_var_name]

data/lib/rookout/logger.rb

@@ -12,7 +12,8 @@ module Rookout
 
     def initialize
       # Detect unit tests
-      if ($PROGRAM_NAME.end_with?("minitest_runner.rb") ||
+      if (Config.debug ||
+          $PROGRAM_NAME.end_with?("minitest_runner.rb") ||
           $PROGRAM_NAME.end_with?("tunit_or_minitest_in_folder_runner.rb")) &&
          Dir.pwd.end_with?("ruby-sdk")
         Config.logger_log_level = :DEBUG

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

@@ -12,15 +12,15 @@ module Rookout
         end
 
         def call_method _name, _args
-          RubyObjectNamespace nil
+          Rookout::Processor::Namespaces::RubyObjectNamespace.new nil
         end
 
         def read_attribute _name
-          RubyObjectNamespace nil
+          Rookout::Processor::Namespaces::RubyObjectNamespace.new nil
         end
 
         def read_key _key
-          RubyObjectNamespace nil
+          Rookout::Processor::Namespaces::RubyObjectNamespace.new nil
         end
 
         def dump _log_object_errors

data/lib/rookout/version.rb

@@ -1,3 +1,3 @@
 module Rookout
-  VERSION = "0.1.12".freeze
+  VERSION = "0.1.17".freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rookout
 version: !ruby/object:Gem::Version
-  version: 0.1.12
+  version: 0.1.17
 platform: ruby
 authors:
 - Liran Haimovitch
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-12-22 00:00:00.000000000 Z
+date: 2021-01-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: binding_of_caller