cfn_manage 0.2.8 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3d8a7ed727a31e5b2a377abd2cedf2e1e9dedd1ec5225d3a6e5c2c85f6f402b6
-  data.tar.gz: fb0c0023eb68cad7f4c11659100a9088574ba36f6ec108d8c93b01c83960fcf2
+  metadata.gz: 76bc39237a58017a9ed11fabb8f926b2732f91f25af127807b2b6403d8f6c21c
+  data.tar.gz: 35a655a89c72079b96323ba9e41d1afa36838ddadb45c31552a4c2b206717cbb
 SHA512:
-  metadata.gz: 4960e7f832c1463219ab949edcb2642f3079ff8eda2cfbc5f0b2c301c45e3c6607a2f02cfb999396aad4c3b4ab1e492faa2ceb7458421cff5042484ea1f81f45
-  data.tar.gz: bd52ce9e9d80cce6c50e841911861f10ee841701dd50f97651bab2d0e8427da499d532c5c74decbf5f0f53d0c51b81caeec1fe33e64d5616a33b6ed7e6724d24
+  metadata.gz: fb97e8f646522f3eec33e85ff24f4b3b2c713e827a2c136e63e95769a3e5f7421efbf71ca37cf7227c224c369fa3439621158031d5dfefbfb5e9253010f1f5d0
+  data.tar.gz: b80c71ab75a3384cded073a8564362b7d12a97260f1661b1a11913983403ebdf738128010a55124dd5b4e9332bf2bb702aae8f8b8a076b24c1c188b4df5a1199

data/lib/cf_start_stop_environment.rb

@@ -27,6 +27,7 @@ module Base2
           'AWS::RDS::DBInstance' => '100',
           'AWS::AutoScaling::AutoScalingGroup' => '200',
           'AWS::EC2::Instance' => '200',
+          'AWS::EC2::SpotFleet' => '200',
           'AWS::CloudWatch::Alarm' => '300'
       }
 

data/lib/spot_fleet_start_stop_handler.rb

@@ -0,0 +1,53 @@
+require_relative '../lib/aws_credentials'
+
+module Base2
+
+  class SpotFleetStartStopHandler
+
+    def initialize(fleet_id)
+      @fleet_id = fleet_id
+
+      credentials = Base2::AWSCredentials.get_session_credentials("startstopfleet_#{fleet_id}")
+      @ec2_client = Aws::EC2::Client.new(retry_limit: 20)
+      if credentials != nil
+        @ec2_client = Aws::EC2::Client.new(credentials: credentials, retry_limit: 20)
+      end
+
+      @fleet = @ec2_client.describe_spot_fleet_requests({spot_fleet_request_ids:[fleet_id]})
+      @fleet = @fleet.spot_fleet_request_configs[0].spot_fleet_request_config
+    end
+
+    def start(configuration)
+
+      $log.info("Setting fleet #{@fleet_id} capacity to #{configuration['target_capacity']}")
+      @ec2_client.modify_spot_fleet_request({
+          spot_fleet_request_id: @fleet_id,
+          target_capacity: configuration['target_capacity'],
+      })
+
+      return configuration
+    end
+
+    def stop
+
+      if @fleet.target_capacity == 0
+        $log.info("Spot fleet #{@fleet_id} already stopped")
+        return nil
+      end
+
+      configuration = {
+          target_capacity: @fleet.target_capacity
+      }
+
+      $log.info("Setting fleet #{@fleet_id} capacity to 0")
+      @ec2_client.modify_spot_fleet_request({
+          spot_fleet_request_id: @fleet_id,
+          target_capacity: 0,
+      })
+
+      return configuration
+    end
+
+  end
+
+end

data/lib/start_stop_handler_factory.rb

@@ -2,6 +2,7 @@ require_relative '../lib/asg_start_stop_handler'
 require_relative '../lib/ec2_start_stop_handler'
 require_relative '../lib/rds_start_stop_handler'
 require_relative '../lib/alarm_start_stop_handler'
+require_relative '../lib/spot_fleet_start_stop_handler'
 
 module Base2
 
@@ -24,6 +25,8 @@ module Base2
         when 'AWS::CloudWatch::Alarm'
           return Base2::AlarmStartStopHandler.new(resource_id)
 
+        when 'AWS::EC2::SpotFleet'
+          return Base2::SpotFleetStartStopHandler.new(resource_id)
         else
           return nil
       end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cfn_manage
 version: !ruby/object:Gem::Version
-  version: 0.2.8
+  version: 0.3.0
 platform: ruby
 authors:
 - Base2Services
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-04-12 00:00:00.000000000 Z
+date: 2018-04-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core
@@ -190,6 +190,7 @@ files:
 - lib/dynamodb_start_stop_handler.rb
 - lib/ec2_start_stop_handler.rb
 - lib/rds_start_stop_handler.rb
+- lib/spot_fleet_start_stop_handler.rb
 - lib/start_stop_handler_factory.rb
 homepage: https://github.com/base2Services/cfn-library/blob/master/README.md
 licenses: