lambda_wrap 0.25.0 → 0.26.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0c5e074725712f275d266a7e4a903eed3f9fc680
-  data.tar.gz: c0591eb2e37e873d7e270830728c6e0eb8ec60d3
+  metadata.gz: fb6c46e052f6c1185e9810f52f51992a1973c2aa
+  data.tar.gz: c4e14dda2fddb4fff94d757e877c6ce6ce0944ad
 SHA512:
-  metadata.gz: 90d7756da7775e043f6c8ad6fac06db8f0f7091798b83dbde2701f6e232a1084eae9a1521776cea56af2a7a5a24a5c24a7516969063e968128ec9d6cbaec6666
-  data.tar.gz: d204fd54dea9cb4eda93c3e39dc23288b0cc74259775a656f641bf59337ad68d3a58b98853a4b3284b3d4f52e5fcb4f703257d0c07bbed762d52382ba143c781
+  metadata.gz: c5e2cc04e1c7960382274252074adc6f2e0639d9f32076d13bd523c6a4808a96f691ffd0524e739fe8b3da77e4da4e04a7c8ef68cfe529bf908002a60f4c7e7c
+  data.tar.gz: 342207c240ef3003382d97b6bbc1728e5894fd984f31efb87b452519c5c9121d8ee5f48a1a531997424b009bb6aaf6435dace12280078c96f2477c08609feff0

data/lib/lambda_wrap/dynamo_db_manager.rb

@@ -25,6 +25,47 @@ module LambdaWrap
       )
     end
 
+    ##
+    # Updates the provisioned throughput read and write capacity of the requested table.
+    # If the table does not exist an error message is displayed. If current read/write capacity
+    # is equals to requested read/write capacity or the requested read/write capacity is 0 or less than 0
+    # no table updation is performed.
+    #
+    # *Arguments*
+    # [table_name]        The table name of the dynamoDB to be updated.
+    # [read_capacity]     The read capacity the table should be updated with.
+    # [write_capacity]    The write capacity the table should be updated with.
+    def update_table_capacity(table_name, read_capacity, write_capacity)
+      # Check if table exists.
+      begin
+        table_details = @client.describe_table(table_name: table_name).table
+      rescue Aws::DynamoDB::Errors::ResourceNotFoundException
+        raise "Update cannot be performed. Table #{table_name} does not exists."
+      end
+
+      if (read_capacity <= 0 || write_capacity <= 0)
+        puts "Table: #{table_name} not updated. Read/Write capacity should be greater than or equal to 1."
+      elsif (read_capacity == table_details.provisioned_throughput.read_capacity_units ||
+            write_capacity == table_details.provisioned_throughput.write_capacity_units)
+        puts "Table: #{table_name} not updated. Current and requested reads/writes are same.
+        Current ReadCapacityUnits provisioned for the table: #{table_details.provisioned_throughput.read_capacity_units}.
+        Requested ReadCapacityUnits: #{read_capacity}.
+        Current WriteCapacityUnits provisioned for the table: #{table_details.provisioned_throughput.write_capacity_units}.
+        Requested WriteCapacityUnits: #{write_capacity}. "
+      else
+        response = @client.update_table(
+          table_name: table_name,
+          provisioned_throughput: { read_capacity_units: read_capacity, write_capacity_units: write_capacity })
+
+        raise "Read and writes capacities was not updated for table: #{table_name}." unless (response.table_description.provisioned_throughput.read_capacity_units!= read_capacity ||
+        response.table_description.provisioned_throughput.write_capacity_units!= write_capacity)
+
+        puts "Updated new read/write capacity for table #{table_name}.
+        Read capacity updated to: #{read_capacity}.
+        Write capacity updated to: #{write_capacity}."
+      end
+    end
+
     ##
     # Publishes the database and awaits until it is fully available. If the table already exists,
     # it only adjusts the read and write

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lambda_wrap
 version: !ruby/object:Gem::Version
-  version: 0.25.0
+  version: 0.26.0
 platform: ruby
 authors:
 - Markus Thurner
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-12-19 00:00:00.000000000 Z
+date: 2017-02-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk