lib/kafkr/consumer.rb in kafkr-0.10.0 vs lib/kafkr/consumer.rb in kafkr-0.11.0

@@ -16,10 +16,13 @@
       attr_reader :handlers
 
       def configuration
         FileUtils.mkdir_p "./.kafkr"
         @configuration ||= OpenStruct.new
+        @configuration.host = ENV.fetch("KAFKR_HOST", "localhost")
+        @configuration.port = ENV.fetch("KAFKR_PORT", 2000)
+        @configuration.timeout = ENV.fetch("KAFKR_CONSUMER_TIMEOUT", 120) 
         @configuration.suggest_handlers = false
         @configuration
       end
 
       def configure
@@ -148,10 +151,10 @@
       attempt = 0
       begin
         socket = TCPSocket.new(@host, @port)
         attempt = 0
 
-        Timeout.timeout(20) do
+        Timeout.timeout(Kafkr::Consumer.configuration.timeout) do
           sync_uid = send_message.call(message)
 
           loop do
             received_message = socket.gets
             raise LostConnection if received_message.nil?