cfn_manage 0.2.1 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ff6c138751bd45eaebf5b59d6c6e250f3e70fc1d
-  data.tar.gz: ebb03da626d1719724d05a90c8cc0d337cafbad1
+  metadata.gz: e3c4afd538389130731a726436500698694da3f0
+  data.tar.gz: 13cdd8ba876f30ffdc270041a54f84166bf53495
 SHA512:
-  metadata.gz: 9d67d0dfb55ebd1dc1c072cb0ec03c8a083268f3d23a3083d9397176b29f5fe7beb425a8ccb995e4e50c30e29f1f95956e828d4cee92f5f3e59e95304b9ec726
-  data.tar.gz: 429336ecb21e627ee0b5dec6e68de12593628a77633b8991b135f101ea449343503dcfb8c13b8fc6410a939152ee5fcfe9c8d8c3ed683441f8ced0c7d60e9056
+  metadata.gz: c3b3804202de03e659d18b0be7dcd0c87d9500fb7233f718ec0c29d076ead3b8291888147baaec89122edafa54a3443189199361309cdf0b20933f59bf56b19a
+  data.tar.gz: 0ce6bf7c84d5effe320377d02b2643e43f0a4a2c3b5629e4b42eca85bac3d5d50a92b114a848e6654a147aaf09dcbc398de6ad61efbdd0e335c832831a934f3d

data/lib/dynamodb_start_stop_handler.rb

@@ -0,0 +1,53 @@
+require_relative '../lib/aws_credentials'
+
+module Base2
+
+  class DynamoDbStartStopHandler
+
+    @table
+    @table_name
+    @gsi_names
+
+    def initialize(table_name)
+      credentials = Base2::AWSCredentials.get_session_credentials("stoprun_#{instance_id}")
+      ec2_client = Aws::DynamoDB::Client.new(credentials: credentials)
+      @table = Aws::DynamoDB::Resource.new(client: ec2_client).table(table_name)
+      @table_name = table_name
+      @table.global_secondary_indexes.each do |gsi|
+        @gsi_names << gsi.index_name
+      end
+    end
+
+    def start(configuration)
+
+    end
+
+    def stop
+      configuration = {
+          read_capacity: @table.provisioned_throughput.read_capacity_units,
+          write_capacity: @table.provisioned_throughput.write_capacity_units
+      }
+
+      $log.info("Current dynamo table #{@table.name} capacity: #{configuration}")
+
+      # check if environment vairable set for
+      if ENV.key? ('cfn_manage_dynamodb_stop_capacity')
+        stop_capacity = ENV['cfn_manage_dynamodb_stop_capacity'].to_i
+
+        $log.info("Downsizing read/write capacity to #{stop_capacity}")
+        @table.update({
+            provisioned_throughput: {
+                read_capacity_units: stop_capacity, # required
+                write_capacity_units: stop_capacity, # required
+            }
+        })
+      end
+
+
+
+      configuration
+    end
+
+
+  end
+end

data/lib/rds_start_stop_handler.rb

@@ -89,7 +89,7 @@ module Base2
       @rds_instance.modify({ multi_az: multi_az, apply_immediately: true })
       # allow half an hour for instance to be converted
       wait_states = %w(modifying available)
-      self.wait_rds_instance_states( wait_states)
+      wait_rds_instance_states( wait_states)
     end
 
     def wait_rds_instance_states(wait_states)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cfn_manage
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.2.2
 platform: ruby
 authors:
 - Base2Services
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-11-23 00:00:00.000000000 Z
+date: 2017-11-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk
@@ -47,6 +47,7 @@ files:
 - lib/cf_common.rb
 - lib/cf_progress_tracker.rb
 - lib/cf_start_stop_environment.rb
+- lib/dynamodb_start_stop_handler.rb
 - lib/ec2_start_stop_handler.rb
 - lib/rds_start_stop_handler.rb
 - lib/start_stop_handler_factory.rb