lib/pheme/queue_poller.rb in pheme-1.0.0 vs lib/pheme/queue_poller.rb in pheme-1.1.0
- old
+ new
@@ -1,7 +1,11 @@
+require_relative 'compression'
+
module Pheme
class QueuePoller
+ include Compression
+
attr_accessor :queue_url, :queue_poller, :connection_pool_block, :format, :max_messages, :poller_configuration
def initialize(queue_url:, connection_pool_block: false, max_messages: nil, format: :json, poller_configuration: {}, sqs_client: nil)
raise ArgumentError, "must specify non-nil queue_url" unless queue_url.present?
@queue_url = queue_url
@@ -82,10 +86,10 @@
def get_metadata(message_body)
message_body.except('Message', 'Records')
end
def get_content(body)
- body['Message']
+ decompress(body['Message'])
end
def parse_csv(message_contents)
parsed_body = SmarterCSV.process(StringIO.new(message_contents))
parsed_body.map{ |item| RecursiveOpenStruct.new(item, recurse_over_arrays: true) }