Support ddb billing mode in additional_settings

logstash-plugins · Feb 7, 2023 · 141c24d · 141c24d
1 parent 99f4ff7
commit 141c24d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-2.2.1
+2.3.0
diff --git a/lib/logstash/inputs/kinesis.rb b/lib/logstash/inputs/kinesis.rb
@@ -25,6 +25,8 @@
 # The library can optionally also send worker statistics to CloudWatch.
 class LogStash::Inputs::Kinesis < LogStash::Inputs::Base
   KCL = com.amazonaws.services.kinesis.clientlibrary.lib.worker
+  DDB = com.amazonaws.services.dynamodbv2
+  DDB_BILLING_MODE = com.amazonaws.services.dynamodbv2.model.BillingMode
   KCL_PROCESSOR_FACTORY_CLASS = com.amazonaws.services.kinesis.clientlibrary.interfaces.v2.IRecordProcessorFactory
   require "logstash/inputs/kinesis/worker"
 
@@ -135,8 +137,13 @@ def register
       # Call arbitrary "withX()" functions
       # snake_case => withCamelCase happens automatically
       @additional_settings.each do |key, value|
-          fn = "with_#{key}"
-          @kcl_config.send(fn, value)
+        fn = "with_#{key}"
+        arg = if key == "billing_mode"
+          DDB_BILLING_MODE::fromValue(value)
+        else
+          value
+        end
+        @kcl_config.send(fn, arg)
       end
 
       unless @http_proxy.to_s.empty?

diff --git a/logstash-input-kinesis.gemspec b/logstash-input-kinesis.gemspec
@@ -20,7 +20,7 @@ Gem::Specification.new do |spec|
 
   spec.platform      = 'java'
 
-  spec.requirements << "jar 'com.amazonaws:amazon-kinesis-client', '1.13.3'"
+  spec.requirements << "jar 'com.amazonaws:amazon-kinesis-client', '1.14.3'"
   spec.requirements << "jar 'com.amazonaws:aws-java-sdk-core', '1.11.1034'"
   spec.requirements << "jar 'com.amazonaws:aws-java-sdk-sts', '1.11.1034'"