From 938953aadadb04ad05e49d8162ff76b24b5c5069 Mon Sep 17 00:00:00 2001 From: jamessantos10 Date: Mon, 1 Oct 2018 17:10:34 -0300 Subject: [PATCH] Update in_sqs.rb --- lib/fluent/plugin/in_sqs.rb | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/lib/fluent/plugin/in_sqs.rb b/lib/fluent/plugin/in_sqs.rb index a79cb04..5d7baa5 100644 --- a/lib/fluent/plugin/in_sqs.rb +++ b/lib/fluent/plugin/in_sqs.rb @@ -1,5 +1,6 @@ require 'fluent/plugin/input' require 'aws-sdk-sqs' +require 'json' module Fluent::Plugin class SQSInput < Input @@ -18,6 +19,7 @@ class SQSInput < Input config_param :visibility_timeout, :integer, default: nil config_param :delete_message, :bool, default: false config_param :stub_responses, :bool, default: false + config_param :raw_message, :bool, default: false def configure(conf) super @@ -53,7 +55,7 @@ def run wait_time_seconds: @wait_time_seconds, visibility_timeout: @visibility_timeout ).each do |message| - record = parse_message(message) + record = @raw_message ? parse_raw_message(message) : parse_message(message) message.delete if @delete_message @@ -66,6 +68,10 @@ def run private + def parse_raw_message(message) + JSON.parse(message.body) rescue message.body.to_s + end + def parse_message(message) { 'body' => message.body.to_s,