logstash-output-awslogs 0.1.10 → 0.1.15

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 19a24fc8ec53b6f1aa7c2a42d2a1caa9e283960b
-  data.tar.gz: 972c27ef7ca2dfd7f7d6f84c710a9fdd6fd20a75
+  metadata.gz: 29735655683e0bf0d17d8726372472e951cbfc3c
+  data.tar.gz: 81fe5f0b3430e382cd049984cde731332f676252
 SHA512:
-  metadata.gz: 44676bc3a7ae7491843cc0440ea52c72d9f97a8d1689a99b4410dec7adc99188e27eddb021290c3411af34394a60ebb545cdefe16455cb522757ef590a4f58bf
-  data.tar.gz: b79b2661c24cf73751aa3cae1034002f3a1f6815b2f9cd20a10c2f85ed4c767ada1c03199b126553dd16e173c5e3406846d5418885f9e7020c1742b7dc1a0ddf
+  metadata.gz: 5f35c56827166827987c39716260dd1435f2ac0d7071208e674e6cf27d1a6102c9ae24e4cbac3fb67fce978499f5ce15e9e66eec61b07ca0c5b522c2e7c23068
+  data.tar.gz: 22eadde7bcff23206e3c99d87c5b0545de942e7f348260f8831a490565b206fc75f5b6d092db15a0091d7f512561d79698532e2da525d6da67412fabed580b81

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 => 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 #{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].key?(event_log_stream_name) && !sequence_tokens[event_log_group_name][event_log_stream_name].to_s.empty?
+            @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(e)
       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.10'
+  s.version       = '0.1.15'
   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.10
+  version: 0.1.15
 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