cfn_manage 0.4.3 → 0.5.0

data/lib/{ec2_start_stop_handler.rb → cfn_manage/ec2_start_stop_handler.rb}

@@ -1,16 +1,17 @@
-require_relative '../lib/aws_credentials'
+require 'cfn_manage/aws_credentials'
 
-module Base2
+module CfnManage
 
   class Ec2StartStopHandler
 
     @instance
 
-    def initialize(instance_id)
-      credentials = Base2::AWSCredentials.get_session_credentials("stoprun_#{instance_id}")
+    def initialize(instance_id, skip_wait)
+      credentials = CfnManage::AWSCredentials.get_session_credentials("stoprun_#{instance_id}")
       ec2_client = Aws::EC2::Client.new(credentials: credentials, retry_limit: 20)
       @instance = Aws::EC2::Resource.new(client: ec2_client, retry_limit: 20).instance(instance_id)
       @instance_id = instance_id
+      @skip_wait = skip_wait
     end
 
     def start(configuration)
@@ -34,8 +35,9 @@ module Base2
       return {}
     end
 
-
-
+    def wait(wait_states=[])
+      $log.debug("Not waiting for EC2 instance #{@instance_id}")
+    end
 
   end
-end
+end

data/lib/cfn_manage/ecs_cluster_start_stop_handler.rb

@@ -0,0 +1,66 @@
+require 'cfn_manage/aws_credentials'
+
+module CfnManage
+  class EcsClusterStartStopHandler
+
+    def initialize(cluster_id, skip_wait)
+      credentials = CfnManage::AWSCredentials.get_session_credentials("stoprun_#{cluster_id}")
+      @ecs_client = Aws::ECS::Client.new(credentials: credentials, retry_limit: 20)
+      @services = @ecs_client.list_services(cluster: cluster_id).service_arns
+      $log.info("Found #{@services.count} services in ECS cluster #{cluster_id}")
+      @cluster = cluster_id
+      @skip_wait = skip_wait
+    end
+
+    def start(configuration)
+      @services.each do |service_arn|
+
+        $log.info("Searching for ECS service #{service_arn} in cluster #{@cluster}")
+        service = @ecs_client.describe_services(services:[service_arn], cluster: @cluster).services[0]
+
+        if service.desired_count != 0
+          $log.info("ECS service #{service.service_name} is already running")
+          next
+        end
+
+        $log.info("Starting ECS service #{service.service_name} with desired count of #{configuration[service.service_name]['desired_count']}")
+        @ecs_client.update_service({
+          desired_count: configuration[service.service_name]['desired_count'],
+          service: service_arn,
+          cluster: @cluster
+        })
+
+      end
+    end
+
+    def stop
+      configuration = {}
+      @services.each do |service_arn|
+
+        $log.info("Searching for ECS service #{service_arn} in cluster #{@cluster}")
+        service = @ecs_client.describe_services(services:[service_arn], cluster: @cluster).services[0]
+
+        if service.desired_count == 0
+          $log.info("ECS service #{service.service_name} is already stopped")
+          next
+        end
+
+        configuration[service.service_name] = { desired_count: service.desired_count }
+        $log.info("Stopping ECS service #{service.service_name}")
+        @ecs_client.update_service({
+          desired_count: 0,
+          service: service_arn,
+          cluster: @cluster
+        })
+
+      end
+
+      return configuration
+    end
+
+    def wait(completed_status)
+      $log.debug("Not waiting for ECS Services in cluster #{@cluster}")
+    end
+
+  end
+end

data/lib/{rds_start_stop_handler.rb → cfn_manage/rds_start_stop_handler.rb}

@@ -1,13 +1,14 @@
-require_relative '../lib/aws_credentials'
+require 'cfn_manage/aws_credentials'
 
-module Base2
+module CfnManage
 
   class RdsStartStopHandler
 
-    def initialize(instance_id)
+    def initialize(instance_id, skip_wait)
       @instance_id = instance_id
-
-      credentials = Base2::AWSCredentials.get_session_credentials("startstoprds_#{instance_id}")
+      @skip_wait = skip_wait
+      @excluded_engines = %w(aurora aurora-mysql aurora-postgresql) # RDS list of exluded engines that don't support RDS stop start
+      credentials = CfnManage::AWSCredentials.get_session_credentials("startstoprds_#{instance_id}")
       @rds_client = Aws::RDS::Client.new(retry_limit: 20)
       if credentials != nil
         @rds_client = Aws::RDS::Client.new(credentials: credentials, retry_limit: 20)
@@ -18,9 +19,7 @@ module Base2
     end
 
     def start(configuration)
-      # RDS list of exluded engines that don't support RDS stop start
-      excluded_engines = %w(aurora aurora-mysql aurora-postgresql)
-      if excluded_engines.include? @rds_instance.engine
+      if @excluded_engines.include? @rds_instance.engine
          $log.info("RDS Instance #{@instance_id} engine is #{@rds_instance.engine} and cannot be started by instance.")
          return
       end
@@ -35,10 +34,12 @@ module Base2
         @rds_client.start_db_instance({ db_instance_identifier: @instance_id })
 
         # wait instance to become available
-        $log.info("Waiting db instance to become available #{@instance_id}")
-        wait_rds_instance_states( %w(starting available))
+        unless @skip_wait
+          $log.info("Waiting db instance to become available #{@instance_id}")
+          wait('available')
+        end
       else
-        wait_rds_instance_states( %w(available))
+        wait('available') unless @skip_wait
       end
 
       # convert rds instance to mutli-az if required
@@ -54,8 +55,7 @@ module Base2
           is_multi_az: @rds_instance.multi_az
       }
       # RDS list of exluded engines that don't support RDS stop start
-      excluded_engines = %w(aurora aurora-mysql aurora-postgresql)
-      if excluded_engines.include? @rds_instance.engine
+      if @excluded_engines.include? @rds_instance.engine
          $log.info("RDS Instance #{@instance_id} engine is #{@rds_instance.engine} and cannot be stoped by instance.")
          return configuration
       end
@@ -82,8 +82,10 @@ module Base2
       # stop rds instance and wait for it to be fully stopped
       $log.info("Stopping instance #{@instance_id}")
       @rds_client.stop_db_instance({ db_instance_identifier: @instance_id })
-      $log.info("Waiting db instance to be stopped #{@instance_id}")
-      wait_rds_instance_states(%w(stopping stopped))
+      unless @skip_wait
+        $log.info("Waiting db instance to be stopped #{@instance_id}")
+        wait('stopped')
+      end
 
       return configuration
     end
@@ -95,40 +97,37 @@ module Base2
       end
       @rds_instance.modify({ multi_az: multi_az, apply_immediately: true })
       # allow half an hour for instance to be converted
-      wait_states = %w(modifying available)
-      wait_rds_instance_states( wait_states)
+      wait('available')
     end
 
-    def wait_rds_instance_states(wait_states)
-      wait_states.each do |state|
-        # reached state must be steady, at least a minute. Modifying an instance to/from MultiAZ can't be shorter
-        # than 40 seconds, hence steady count is 4
-        state_count = 0
-        steady_count = 4
-        attempts = 0
-        rds = Aws::RDS::Resource.new(client: @rds_client)
-        until attempts == (max_attempts = 60*6) do
-          instance = rds.db_instance(@instance_id)
-          $log.info("Instance #{instance.db_instance_identifier} state: #{instance.db_instance_status}, waiting for #{state}")
-
-          if instance.db_instance_status == "#{state}"
-            state_count = state_count + 1
-            $log.info("#{state_count}/#{steady_count}")
-          else
-            state_count = 0
-          end
-          break if state_count == steady_count
-          attempts = attempts + 1
-          sleep(15)
+    def wait(completed_state)
+      # reached state must be steady, at least a minute. Modifying an instance to/from MultiAZ can't be shorter
+      # than 40 seconds, hence steady count is 4
+      state_count = 0
+      steady_count = 4
+      attempts = 0
+      rds = Aws::RDS::Resource.new(client: @rds_client)
+      until attempts == (max_attempts = 60*6) do
+        instance = rds.db_instance(@instance_id)
+        $log.info("Instance #{instance.db_instance_identifier} state: #{instance.db_instance_status}, waiting for #{completed_state}")
+
+        if instance.db_instance_status == "#{completed_state}"
+          state_count = state_count + 1
+          $log.info("#{state_count}/#{steady_count}")
+        else
+          state_count = 0
         end
+        break if state_count == steady_count
+        attempts = attempts + 1
+        sleep(15)
+      end
 
-        if attempts == max_attempts
-          $log.error("RDS Database Instance #{@instance_id} did not enter #{state} state, however continuing operations...")
-        end
+      if attempts == max_attempts
+        $log.error("RDS Database Instance #{@instance_id} did not enter #{state} state, however continuing operations...")
       end
     end
 
-    private :set_rds_instance_multi_az, :wait_rds_instance_states
+    private :set_rds_instance_multi_az
 
   end
 

data/lib/{spot_fleet_start_stop_handler.rb → cfn_manage/spot_fleet_start_stop_handler.rb}

@@ -1,13 +1,13 @@
-require_relative '../lib/aws_credentials'
+require 'cfn_manage/aws_credentials'
 
-module Base2
+module CfnManage
 
   class SpotFleetStartStopHandler
 
-    def initialize(fleet_id)
+    def initialize(fleet_id, skip_wait)
       @fleet_id = fleet_id
-
-      credentials = Base2::AWSCredentials.get_session_credentials("startstopfleet_#{fleet_id}")
+      @skip_wait = skip_wait
+      credentials = CfnManage::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)
@@ -48,6 +48,10 @@ module Base2
       return configuration
     end
 
+    def wait(wait_states=[])
+      $log.debug("Not waiting for spot fleet #{@fleet_id}")
+    end
+
   end
 
-end
+end

data/lib/cfn_manage/start_stop_handler_factory.rb

@@ -0,0 +1,44 @@
+require 'cfn_manage/asg_start_stop_handler'
+require 'cfn_manage/ec2_start_stop_handler'
+require 'cfn_manage/rds_start_stop_handler'
+require 'cfn_manage/aurora_cluster_start_stop_handler'
+require 'cfn_manage/alarm_start_stop_handler'
+require 'cfn_manage/spot_fleet_start_stop_handler'
+require 'cfn_manage/ecs_cluster_start_stop_handler'
+
+module CfnManage
+
+  class StartStopHandlerFactory
+
+    #   Factory method to get start/stop handler based on CloudFormation
+    # resource type. If resource_id passed in does not exist, it is
+    # very likely that exception will be raised
+    def self.get_start_stop_handler(resource_type, resource_id, skip_wait)
+      case resource_type
+        when 'AWS::AutoScaling::AutoScalingGroup'
+          return CfnManage::AsgStartStopHandler.new(resource_id, skip_wait)
+
+        when 'AWS::EC2::Instance'
+          return CfnManage::Ec2StartStopHandler.new(resource_id, skip_wait)
+
+        when 'AWS::RDS::DBInstance'
+          return CfnManage::RdsStartStopHandler.new(resource_id, skip_wait)
+
+        when 'AWS::RDS::DBCluster'
+          return CfnManage::AuroraClusterStartStopHandler.new(resource_id, skip_wait)
+
+        when 'AWS::CloudWatch::Alarm'
+          return CfnManage::AlarmStartStopHandler.new(resource_id)
+
+        when 'AWS::EC2::SpotFleet'
+          return CfnManage::SpotFleetStartStopHandler.new(resource_id, skip_wait)
+
+        when 'AWS::ECS::Cluster'
+          return CfnManage::EcsClusterStartStopHandler.new(resource_id, skip_wait)
+
+        else
+          return nil
+      end
+    end
+  end
+end

data/lib/cfn_manage/version.rb

@@ -0,0 +1,3 @@
+module CfnManage
+  VERSION="0.5.0".freeze
+end

metadata

@@ -1,15 +1,16 @@
 --- !ruby/object:Gem::Specification
 name: cfn_manage
 version: !ruby/object:Gem::Version
-  version: 0.4.3
+  version: 0.5.0
 platform: ruby
 authors:
 - Base2Services
 - Nikola Tosic
+- Angus Vine
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-10-18 00:00:00.000000000 Z
+date: 2018-10-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core
@@ -171,6 +172,124 @@ dependencies:
     - - "<"
       - !ruby/object:Gem::Version
         version: '2'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-ecs
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '2'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.9'
+- !ruby/object:Gem::Dependency
+  name: rspec-core
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.8'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.8'
+- !ruby/object:Gem::Dependency
+  name: rspec-expectations
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.8'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.8'
+- !ruby/object:Gem::Dependency
+  name: rspec-mocks
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.8'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.8'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.16'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.16'
 description: ''
 email: itsupport@base2services.com
 executables:
@@ -179,20 +298,22 @@ extensions: []
 extra_rdoc_files: []
 files:
 - bin/cfn_manage
-- bin/cfn_manage.rb
 - bin/usage.txt
-- lib/alarm_start_stop_handler.rb
-- lib/asg_start_stop_handler.rb
-- lib/aurora_cluster_start_stop_handler.rb
-- lib/aws_credentials.rb
-- lib/cf_common.rb
-- lib/cf_progress_tracker.rb
-- lib/cf_start_stop_environment.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
+- lib/cfn_manage.rb
+- lib/cfn_manage/alarm_start_stop_handler.rb
+- lib/cfn_manage/asg_start_stop_handler.rb
+- lib/cfn_manage/aurora_cluster_start_stop_handler.rb
+- lib/cfn_manage/aws_credentials.rb
+- lib/cfn_manage/cf_common.rb
+- lib/cfn_manage/cf_progress_tracker.rb
+- lib/cfn_manage/cf_start_stop_environment.rb
+- lib/cfn_manage/ec2_start_stop_handler.rb
+- lib/cfn_manage/ecs_cluster_start_stop_handler.rb
+- lib/cfn_manage/rds_start_stop_handler.rb
+- lib/cfn_manage/spot_fleet_start_stop_handler.rb
+- lib/cfn_manage/start_stop_handler_factory.rb
+- lib/cfn_manage/version.rb
+homepage: https://github.com/base2Services/cfn-start-stop-stack/blob/master/README.md
 licenses:
 - MIT
 metadata: {}
@@ -212,7 +333,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.6.13
+rubygems_version: 2.7.7
 signing_key: 
 specification_version: 4
 summary: Manage AWS Cloud Formation stacks