legionio 1.7.8 → 1.7.13

data/lib/legion/extensions/transport.rb

@@ -48,7 +48,7 @@ module Legion
         end
       end
 
-      def auto_create_exchange(exchange, default_exchange = false) # rubocop:disable Style/OptionalBooleanParameter
+      def auto_create_exchange(exchange, default_exchange: false)
         if Object.const_defined? exchange
           log.warn "#{exchange} is already defined"
           return
@@ -107,7 +107,7 @@ module Legion
       end
 
       def auto_generate_runner_messages(runner_info, messages_mod, ext_amqp)
-        runner_name   = runner_info[:runner_name]
+        runner_name = runner_info[:runner_name]
         runner_module = runner_info[:runner_module]
         return if runner_module.nil?
         return unless runner_module.respond_to?(:definition_for)
@@ -137,10 +137,17 @@ module Legion
           binding[:to] = nil unless binding.key?(:to)
           binding[:from] = default_exchange if !binding.key?(:from) || binding[:from].nil?
           bind_e_to_q(**binding)
+        rescue StandardError => e
+          log.warn '[transport] failed to build exchange-to-queue binding ' \
+                   "from=#{binding[:from].inspect} to=#{binding[:to].inspect} " \
+                   "routing_key=#{binding[:routing_key].inspect} binding=#{binding.inspect}"
+          handle_exception(e, handled: false, level: :warn)
+          raise e
         end
       end
 
       def bind_e_to_q(to:, from: default_exchange, routing_key: nil, **)
+        log.debug "[transport] building auto binding exchange: #{from}, routing_key: #{routing_key}, to: #{to}"
         if from.is_a? String
           from = "#{transport_class}::Exchanges::#{from.tr('.', '_').split('_').collect(&:capitalize).join}" unless from.include?('::')
           auto_create_exchange(from) unless Object.const_defined? from
@@ -153,6 +160,9 @@ module Legion
 
         routing_key = to.to_s.split('::').last.downcase if routing_key.nil?
         bind(from, to, routing_key: routing_key)
+      rescue StandardError => e
+        handle_exception(e, handled: false, level: :warn, from: from, to: to, routing_key: routing_key)
+        raise e
       end
 
       def build_e_to_e
@@ -168,6 +178,9 @@ module Legion
           end
 
           bind(binding[:from], binding[:to], binding)
+        rescue StandardError => e
+          handle_exception(e, handled: false, level: :warn)
+          raise e
         end
       end
 

data/lib/legion/region.rb

@@ -4,16 +4,42 @@ require 'net/http'
 
 module Legion
   module Region
+    include Legion::Logging::Helper if defined?(Legion::Logging::Helper)
+
     module_function
 
+    UNSET = Object.new.freeze
+    EXPECTED_METADATA_ERRORS = [
+      Net::OpenTimeout,
+      Net::ReadTimeout,
+      Errno::EHOSTUNREACH,
+      Errno::ECONNREFUSED,
+      Errno::ENETUNREACH,
+      IOError,
+      SocketError
+    ].freeze
+
     def current
       setting = defined?(Legion::Settings) ? Legion::Settings.dig(:region, :current) : nil
-      setting || detect_from_metadata
+      return setting unless blank_region?(setting)
+
+      @detected_region = UNSET unless instance_variable_defined?(:@detected_region)
+      return nil if @detected_region.equal?(UNSET) && @metadata_detection_complete == true
+      return @detected_region unless @detected_region.equal?(UNSET)
+
+      @detected_region = detect_from_metadata
+      @metadata_detection_complete = true
+      @detected_region
     rescue StandardError => e
       Legion::Logging.debug "Region#current failed: #{e.message}" if defined?(Legion::Logging)
       nil
     end
 
+    def reset!
+      remove_instance_variable(:@detected_region) if instance_variable_defined?(:@detected_region)
+      remove_instance_variable(:@metadata_detection_complete) if instance_variable_defined?(:@metadata_detection_complete)
+    end
+
     def local?(target_region)
       target_region.nil? || target_region == current
     end
@@ -76,6 +102,8 @@ module Legion
         response = http.request(req)
         response.is_a?(Net::HTTPSuccess) ? response.body.strip : nil
       end
+    rescue *EXPECTED_METADATA_ERRORS
+      nil
     rescue StandardError => e
       Legion::Logging.debug "Region#detect_aws_region failed: #{e.message}" if defined?(Legion::Logging)
       nil
@@ -90,9 +118,16 @@ module Legion
         response = http.request(req)
         response.is_a?(Net::HTTPSuccess) ? response.body.strip : nil
       end
+    rescue *EXPECTED_METADATA_ERRORS
+      nil
     rescue StandardError => e
       Legion::Logging.debug "Region#detect_azure_region failed: #{e.message}" if defined?(Legion::Logging)
       nil
     end
+
+    def blank_region?(value)
+      value.nil? || (value.respond_to?(:empty?) && value.empty?)
+    end
+    private_class_method :blank_region?
   end
 end