logstash-output-awslogs 0.1.9 → 0.1.14

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 976089b21d8679e1a4a82a82e496257ad6085a26
-  data.tar.gz: e05ab955c846840ca2e91d89520db15dc9cff0d3
+  metadata.gz: 66fdfa1f573c2124d47a0fa6fe7520283b55d64c
+  data.tar.gz: f31ebd44b0a4b584fb594651d08aa71c2814e9b2
 SHA512:
-  metadata.gz: 3fb50d9e90d952643b4002be3f252a3fa6aa1fd690384a160ece3f469223a0def2e34ead061f883df113bb882f7de10c2c0ad71cb33f35d64383c21359c4a1a7
-  data.tar.gz: '04870b92f6126ef8a297afa9a0bfef6f55b7ac630d5499d15fa67a9a06d62037dab5df10f32e4951690bba00190557ad6f7e45a545ca75eee3fbe0e212091964'
+  metadata.gz: 5604d859864682080ebecae322c1d7287efa1c5594e9886eab9959b669fe4b876b78e09268feff4c0a3a1fda1b1a5a694de4597d099114f2c84066c503235952
+  data.tar.gz: ec2e7a4ca3e63ed2ad5f19533cb4fb7cc1861ccaab375a5df7b5c3a5f8fa3738415b8a0b4e9cd0c12a430b544d933644cf532f81f3826d31e42d83d563ccd100

data/lib/logstash/outputs/awslogs.rb

@@ -1,8 +1,9 @@
-# encoding: utf-8
-require "logstash/outputs/base"
-require "logstash/namespace"
-require "logstash/plugin_mixins/aws_config"
-require "aws-sdk"
+# frozen_string_literal: true
+
+require 'logstash/outputs/base'
+require 'logstash/namespace'
+require 'logstash/plugin_mixins/aws_config'
+require 'aws-sdk'
 
 Aws.eager_autoload!
 
@@ -10,73 +11,74 @@ Aws.eager_autoload!
 class LogStash::Outputs::Awslogs < LogStash::Outputs::Base
   include LogStash::PluginMixins::AwsConfig::V2
 
-  config_name "awslogs"
-  default :codec, "line"
+  config_name 'awslogs'
+  default :codec, 'line'
 
-  config :log_group_name, :validate => :string, :required => true
-  config :log_stream_name, :validate => :string, :required => true
+  config :log_group_name, validate: :string, required: true
+  config :log_stream_name, validate: :string, required: true
 
   public
+
   def register
     @client = Aws::CloudWatchLogs::Client.new(aws_options_hash)
     @next_sequence_tokens = {}
   end # def register
 
   public
+
   def multi_receive_encoded(events_and_encoded)
     to_send = {}
     sequence_tokens = {}
 
-    events_and_encoded.each do |event, encoded|
+    events_and_encoded.each do |event, _encoded|
       event_log_stream_name = event.sprintf(log_stream_name)
       event_log_group_name = event.sprintf(log_group_name)
 
       next_sequence_token_key = [event_log_group_name, event_log_stream_name]
-      if ! to_send.keys.include? next_sequence_token_key
+      unless to_send.keys.include? next_sequence_token_key
         to_send.store(next_sequence_token_key, [])
       end
-      to_send[next_sequence_token_key].push({
+      to_send[next_sequence_token_key].push(
         timestamp: (event.timestamp.time.to_f * 1000).to_int,
-        message: event.get("message"),
-      })
+        message: event.get('message')
+      )
     end
 
     to_send.each do |event_log_names, _events|
       event_log_group_name = event_log_names[0]
       event_log_stream_name = event_log_names[1]
-      unless sequence_tokens.keys.include? event_log_group_name
-        sequence_tokens.store(event_log_group_name, {})
-        begin
-          @client.describe_log_streams({log_group_name: event_log_group_name}).each do |response|
-            response.log_streams.each do |log_stream_data|
-              sequence_tokens[event_log_group_name]["#{log_stream_data.log_stream_name}"] = "#{log_stream_data.upload_sequence_token}"
-            end
-          end
-        rescue Aws::CloudWatchLogs::Errors::ResourceNotFoundException => e
-          @logger.info("Will create log group/stream and retry")
-          begin
-            @client.create_log_group(:log_group_name => send_opts[:log_group_name])
-          rescue Aws::CloudWatchLogs::Errors::ResourceAlreadyExistsException => e
-            @logger.info("Log group #{send_opts[:log_group_name]} already exists")
-          rescue Exception => e
-            @logger.error(e)
-          end
-          begin
-            @client.create_log_stream(:log_group_name => send_opts[:log_group_name], :log_stream_name => send_opts[:log_stream_name])
-          rescue Aws::CloudWatchLogs::Errors::ResourceAlreadyExistsException => e
-            @logger.info("Log stream #{send_opts[:log_stream_name]} already exists")
-          rescue Exception => e
-            @logger.error(e)
+      next if sequence_tokens.keys.include? event_log_group_name
+
+      sequence_tokens.store(event_log_group_name, {})
+      begin
+        @client.describe_log_streams(log_group_name: event_log_group_name).each do |response|
+          response.log_streams.each do |log_stream_data|
+            sequence_tokens[event_log_group_name][log_stream_data.log_stream_name.to_s] = log_stream_data.upload_sequence_token.to_s
           end
-          retry
-        rescue Aws::CloudWatchLogs::Errors::ThrottlingException => e
-          @logger.info("Logs throttling, retry")
-          retry
         end
+      rescue Aws::CloudWatchLogs::Errors::ResourceNotFoundException => e
+        @logger.info('Will create log group/stream and retry')
+        begin
+          @client.create_log_group(log_group_name: event_log_group_name)
+        rescue Aws::CloudWatchLogs::Errors::ResourceAlreadyExistsException => e
+          @logger.info("Log group #{event_log_group_name} already exists")
+        rescue Exception => e
+          @logger.error(e)
+        end
+        begin
+          @client.create_log_stream(log_group_name: event_log_group_name, log_stream_name: event_log_stream_name)
+        rescue Aws::CloudWatchLogs::Errors::ResourceAlreadyExistsException => e
+          @logger.info("Log stream #{event_log_stream_name} already exists")
+        rescue Exception => e
+          @logger.error(e)
+        end
+        retry
+      rescue Aws::CloudWatchLogs::Errors::ThrottlingException => e
+        @logger.info('Logs throttling, retry')
+        retry
       end
     end
 
-
     to_send.each do |event_log_names, log_events|
       event_log_group_name = event_log_names[0]
       event_log_stream_name = event_log_names[1]
@@ -84,11 +86,11 @@ class LogStash::Outputs::Awslogs < LogStash::Outputs::Base
 
       ident_opts = {
         log_group_name: event_log_group_name,
-        log_stream_name: event_log_stream_name,
+        log_stream_name: event_log_stream_name
       }
-      send_opts = ident_opts.merge({
-        log_events: log_events,
-      })
+      send_opts = ident_opts.merge(
+        log_events: log_events
+      )
 
       if @next_sequence_tokens.keys.include? next_sequence_token_key
         send_opts[:sequence_token] = @next_sequence_tokens[next_sequence_token_key]
@@ -96,18 +98,26 @@ class LogStash::Outputs::Awslogs < LogStash::Outputs::Base
         send_opts[:sequence_token] = sequence_tokens[event_log_group_name][event_log_stream_name]
       else
         begin
-            @client.create_log_stream(ident_opts)
+          @client.create_log_stream(ident_opts)
+          until sequence_tokens[event_log_group_name][event_log_stream_name] do
+            @client.describe_log_streams(log_group_name: event_log_group_name).each do |response|
+              response.log_streams.each do |log_stream_data|
+                sequence_tokens[event_log_group_name][log_stream_data.log_stream_name.to_s] = log_stream_data.upload_sequence_token.to_s
+              end
+            end
+          end
+          send_opts[:sequence_token] = sequence_tokens[event_log_group_name][event_log_stream_name]
         rescue Aws::CloudWatchLogs::Errors::ResourceNotFoundException => e
-          @logger.info("Will create log group/stream and retry")
+          @logger.info('Will create log group/stream and retry')
           begin
-            @client.create_log_group(:log_group_name => send_opts[:log_group_name])
+            @client.create_log_group(log_group_name: send_opts[:log_group_name])
           rescue Aws::CloudWatchLogs::Errors::ResourceAlreadyExistsException => e
             @logger.info("Log group #{send_opts[:log_group_name]} already exists")
           rescue Exception => e
             @logger.error(e)
           end
           begin
-            @client.create_log_stream(:log_group_name => send_opts[:log_group_name], :log_stream_name => send_opts[:log_stream_name])
+            @client.create_log_stream(log_group_name: send_opts[:log_group_name], log_stream_name: send_opts[:log_stream_name])
           rescue Aws::CloudWatchLogs::Errors::ResourceAlreadyExistsException => e
             @logger.info("Log stream #{send_opts[:log_stream_name]} already exists")
           rescue Exception => e
@@ -115,23 +125,24 @@ class LogStash::Outputs::Awslogs < LogStash::Outputs::Base
           end
           retry
         rescue Aws::CloudWatchLogs::Errors::ThrottlingException => e
-          @logger.info("Logs throttling, retry")
+          @logger.info('Logs throttling, retry')
           retry
         end
       end
       begin
         resp = @client.put_log_events(send_opts)
+        @next_sequence_tokens.store(next_sequence_token_key, resp.next_sequence_token)
       rescue Aws::CloudWatchLogs::Errors::ResourceNotFoundException => e
-        @logger.info("Will create log group/stream and retry")
+        @logger.info('Will create log group/stream and retry')
         begin
-          @client.create_log_group(:log_group_name => send_opts[:log_group_name])
+          @client.create_log_group(log_group_name: send_opts[:log_group_name])
         rescue Aws::CloudWatchLogs::Errors::ResourceAlreadyExistsException => e
           @logger.info("Log group #{send_opts[:log_group_name]} already exists")
         rescue Exception => e
           @logger.error(e)
         end
         begin
-          @client.create_log_stream(:log_group_name => send_opts[:log_group_name], :log_stream_name => send_opts[:log_stream_name])
+          @client.create_log_stream(log_group_name: send_opts[:log_group_name], log_stream_name: send_opts[:log_stream_name])
         rescue Aws::CloudWatchLogs::Errors::ResourceAlreadyExistsException => e
           @logger.info("Log stream #{send_opts[:log_stream_name]} already exists")
         rescue Exception => e
@@ -139,10 +150,11 @@ class LogStash::Outputs::Awslogs < LogStash::Outputs::Base
         end
         retry
         # TODO: handle rejected events with debug message
-      @next_sequence_tokens.store(next_sequence_token_key, resp.next_sequence_token)
       rescue Aws::CloudWatchLogs::Errors::ThrottlingException => e
-        @logger.info("Logs throttling, retry")
+        @logger.info('Logs throttling, retry')
         retry
+      rescue Exception => e
+        @logger.error("Err: #{e}; Opts: #{send_opts.each { |k,v| "#{k.to_s} : #{v.to_s}" }}")
       end
     end
   end # def multi_receive_encoded

data/logstash-output-awslogs.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name          = 'logstash-output-awslogs'
-  s.version       = '0.1.9'
+  s.version       = '0.1.14'
   s.licenses      = ['Apache-2.0']
   s.summary       = 'Writes events to AWS CloudWatch logs.'
   s.homepage      = 'https://github.com/rickycook/logstash-output-awslogs'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: logstash-output-awslogs
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.1.14
 platform: ruby
 authors:
 - Anton Klyba
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-06-16 00:00:00.000000000 Z
+date: 2020-06-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: logstash-core-plugin-api