awful 0.0.95 → 0.0.96

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 34dbb29593ea88811949beec598cfc225de269bc
-  data.tar.gz: 7e123a61645aed0bedbe3675755f368950b572e2
+  metadata.gz: 20cf14fecdfbb426f917378fdb87a71c3d03b06d
+  data.tar.gz: 08cc9b0dc46451f3d57156272996521bcd5044ba
 SHA512:
-  metadata.gz: 1a9ef4276d0c04d62f9c0dbf66311bdc3e1bcbf0c522766e406bbeddd014b6cfdf422506ea2d59482bab5535acba107271163a716e95efe8ed55457c4e6fd993
-  data.tar.gz: 423f56071f4be30f7da82d9975d740a61ee9e12fc11f941688c4ad3989c9efb6a17d95cee8ecc7736ba6602356f168848c5dacf284ab97decb61e8b07a194b9d
+  metadata.gz: 6b38dbe10ec76867dcdcad38513053d6394aabef26d46b79f15ad2a34c2513e129823a3e0096c223a247c413cae5ddeaaa2555c2c445036a85ced8be5915cffc
+  data.tar.gz: d0aea0c0acbe2ad27c93e55eeadd1fabe2e5fec205ce5e45294a57a651032e6c86ebac64aa881ad8082ea813a56bc9f8f67a901ebcd744b5c520589b0f5027c0

data/lib/awful/dynamodb.rb

@@ -82,18 +82,53 @@ module Awful
     end
 
     desc 'throughput NAME', 'get or update provisioned throughput for table NAME'
-    method_option :read_capacity_units,  aliases: '-r', type: :numeric, default: nil, desc: 'Read capacity units'
-    method_option :write_capacity_units, aliases: '-w', type: :numeric, default: nil, desc: 'Write capacity units'
+    method_option :read_capacity_units,  aliases: '-r', type: :numeric, default: nil,   desc: 'Read capacity units'
+    method_option :write_capacity_units, aliases: '-w', type: :numeric, default: nil,   desc: 'Write capacity units'
+    method_option :gsi,                  aliases: '-g', type: :array,   default: [],    desc: 'GSIs to update'
+    method_option :all,                                 type: :boolean, default: false, desc: 'Update all GSIs for the table'
+    method_option :table,                               type: :boolean, default: true,  desc: 'Update througput on table'
     def throughput(name)
-      provisioned = dynamodb.describe_table(table_name: name).table.provisioned_throughput.to_h
-      provisioned.merge!(symbolize_keys(options))
-      if options.empty?         # just print current throughput
-        provisioned.tap do |p|
-          puts YAML.dump(stringify_keys(p))
-        end
-      else                      # update from options
-        dynamodb.update_table(table_name: name, provisioned_throughput: only_keys_matching(provisioned, %i[read_capacity_units write_capacity_units]))
+      table = dynamodb.describe_table(table_name: name).table
+
+      ## current is hash of current provisioned throughput
+      current = table.provisioned_throughput.to_h
+      table.global_secondary_indexes.each do |gsi|
+        current[gsi.index_name] = gsi.provisioned_throughput.to_h
+      end
+
+      ## if no updates requested, just print throughput and return table details
+      unless options[:read_capacity_units] or options[:write_capacity_units]
+        puts YAML.dump(stringify_keys(current))
+        return table
       end
+
+      ## parameters for update request
+      params = { table_name: name }
+
+      ## add table throughput unless told not to
+      params[:provisioned_throughput] = {
+        read_capacity_units:  options[:read_capacity_units]  || current[:read_capacity_units],
+        write_capacity_units: options[:write_capacity_units] || current[:write_capacity_units]
+      } if options[:table]
+
+      ## list of requested GSIs, or all for this table
+      gsis = options[:gsi]
+      gsis = table.global_secondary_indexes.map(&:index_name) if options[:all]
+      params[:global_secondary_index_updates] = gsis.map do |gsi|
+        {
+          update: {
+            index_name: gsi,
+            provisioned_throughput: {
+              read_capacity_units:  options[:read_capacity_units]  || current[gsi][:read_capacity_units],
+              write_capacity_units: options[:write_capacity_units] || current[gsi][:write_capacity_units]
+            }
+          }
+        }
+      end
+
+      ## make the update request
+      params.reject! { |_,v| v.empty? } # sdk hates empty global_secondary_index_updates
+      dynamodb.update_table(params)
     end
 
     desc 'enable_streams NAME', 'enable/disable streams on the table'

data/lib/awful/version.rb

@@ -1,3 +1,3 @@
 module Awful
-  VERSION = '0.0.95'
+  VERSION = '0.0.96'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: awful
 version: !ruby/object:Gem::Version
-  version: 0.0.95
+  version: 0.0.96
 platform: ruby
 authors:
 - Ric Lister
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-03-12 00:00:00.000000000 Z
+date: 2016-03-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler