cfn_manage 0.7.1 → 0.8.0

data/lib/cfn_manage/aurora_cluster_start_stop_handler.rb

@@ -1,97 +0,0 @@
-require 'cfn_manage/aws_credentials'
-
-module CfnManage
-
-  class AuroraClusterStartStopHandler
-
-    def initialize(cluster_id, skip_wait)
-      @cluster_id = cluster_id
-      @skip_wait = skip_wait
-      credentials = CfnManage::AWSCredentials.get_session_credentials("startstopcluster_#{cluster_id}")
-      @rds_client = Aws::RDS::Client.new(retry_limit: 20)
-      if credentials != nil
-        @rds_client = Aws::RDS::Client.new(credentials: credentials, retry_limit: 20)
-      end
-      rds = Aws::RDS::Resource.new(client: @rds_client)
-      @rds_cluster = rds.db_cluster(cluster_id)
-    end
-
-    def start(configuration)
-      if @rds_cluster.status == 'available'
-        $log.info("Aurora Cluster #{@cluster_id} is already in available state")
-        return
-      end
-
-      if @rds_cluster.engine_mode != 'provisioned'
-        $log.info("Aurora Cluster #{@cluster_id} is not a provisioned cluster and cannot be started using this method.")
-        return
-      end
-
-      # start rds cluster
-      if @rds_cluster.status == 'stopped'
-        $log.info("Starting Aurora cluster #{@cluster_id}")
-        @rds_client.start_db_cluster({ db_cluster_identifier: @cluster_id })
-        unless @skip_wait
-          # wait cluster to become available
-          $log.info("Waiting Aurora cluster to become available #{@cluster_id}")
-          wait('available')
-        end
-      else
-        $log.info("Aurora Cluster #{@cluster_id} is not in a stopped state. State: #{@rds_cluster.status}")
-      end
-    end
-
-    def stop
-      if @rds_cluster.status == 'stopped'
-        $log.info("Aurora Cluster #{@cluster_id} is already stopped")
-        return {}
-      end
-
-      if @rds_cluster.status != 'available'
-        $log.info("Aurora Cluster #{@cluster_id} is not in a available state. State: #{@rds_cluster.status}")
-        return {}
-      end
-
-      if @rds_cluster.engine_mode != 'provisioned'
-        $log.info("Aurora Cluster #{@cluster_id} is not a provisioned cluster and cannot be stopped using this method.")
-        return {}
-      end
-      # stop rds cluster and wait for it to be fully stopped
-      $log.info("Stopping aurora cluster #{@cluster_id}")
-      @rds_client.stop_db_cluster({ db_cluster_identifier: @cluster_id })
-      unless @skip_wait
-        $log.info("Waiting aurora cluster to be stopped #{@cluster_id}")
-        wait('stopped')
-      end
-      return {}
-    end
-
-    def wait(completed_state)
-      # reached state must be steady, at least a minute.
-      state_count = 0
-      steady_count = 4
-      attempts = 0
-      rds = Aws::RDS::Resource.new(client: @rds_client)
-      until attempts == (max_attempts = 60*6) do
-        cluster = rds.db_cluster(@cluster_id)
-        $log.info("Aurora Cluster #{cluster.db_cluster_identifier} state: #{cluster.status}, waiting for #{completed_state}")
-
-        if cluster.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 Aurora Cluster #{@cluster_id} did not enter #{state} state, however continuing operations...")
-      end
-    end
-
-  end
-
-end

data/lib/cfn_manage/documentdb_cluster_start_stop_handler.rb

@@ -1,89 +0,0 @@
-require 'cfn_manage/aws_credentials'
-
-module CfnManage
-
-  class DocumentDbClusterStartStopHandler
-
-    def initialize(cluster_id, skip_wait)
-      @cluster_id = cluster_id
-      @skip_wait = skip_wait
-      credentials = CfnManage::AWSCredentials.get_session_credentials("startstopcluster_#{cluster_id}")
-      @docdb_client = Aws::DocDB::Client.new(retry_limit: 20)
-      if credentials != nil
-        @docdb_client = Aws::DocDB::Client.new(credentials: credentials, retry_limit: 20)
-      end
-      cluster = @docdb_client.describe_db_clusters({ db_cluster_identifier: @cluster_id })
-      @docdb_cluster = cluster.db_clusters.first
-    end
-
-    def start(configuration)
-      if @docdb_cluster.status == 'available'
-        $log.info("DocDB Cluster #{@cluster_id} is already in available state")
-        return
-      end
-
-      # start docdb cluster
-      if @docdb_cluster.status == 'stopped'
-        $log.info("Starting DocDB cluster #{@cluster_id}")
-        @docdb_client.start_db_cluster({ db_cluster_identifier: @cluster_id })
-        unless @skip_wait
-          # wait cluster to become available
-          $log.info("Waiting DocDB cluster to become available #{@cluster_id}")
-          wait('available')
-        end
-      else
-        $log.info("DocDB Cluster #{@cluster_id} is not in a stopped state. State: #{@docdb_cluster.status}")
-      end
-    end
-
-    def stop
-      if @docdb_cluster.status == 'stopped'
-        $log.info("DocDB Cluster #{@cluster_id} is already stopped")
-        return {}
-      end
-
-      if @docdb_cluster.status != 'available'
-        $log.info("DocDB Cluster #{@cluster_id} is not in a available state. State: #{@docdb_cluster.status}")
-        return {}
-      end
-      # stop docdb cluster and wait for it to be fully stopped
-      $log.info("Stopping DocDB cluster #{@cluster_id}")
-      @docdb_client.stop_db_cluster({ db_cluster_identifier: @cluster_id })
-      unless @skip_wait
-        $log.info("Waiting DocDB cluster to be stopped #{@cluster_id}")
-        wait('stopped')
-      end
-      return {}
-    end
-
-    def wait(completed_state)
-      # reached state must be steady, at least a minute.
-      state_count = 0
-      steady_count = 4
-      attempts = 0
-
-      until attempts == (max_attempts = 60*6) do
-        # Declare client and cluster variable a second time inside the loop so it re-evaluates each time.
-        docdb = @docdb_client.describe_db_clusters({ db_cluster_identifier: @cluster_id })
-        cluster = docdb.db_clusters.first
-        $log.info("DocDB Cluster #{cluster.db_cluster_identifier} state: #{cluster.status}, waiting for #{completed_state}")
-
-        if cluster.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("DocDB Cluster #{@cluster_id} did not enter #{completed_state} state, however continuing operations...")
-      end
-    end
-
-  end
-
-end

data/lib/cfn_manage/ec2_start_stop_handler.rb

@@ -1,43 +0,0 @@
-require 'cfn_manage/aws_credentials'
-
-module CfnManage
-
-  class Ec2StartStopHandler
-
-    @instance
-
-    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)
-      if %w(running).include?(@instance.state.name)
-        $log.info("Instance #{@instance_id} already running")
-        return
-      end
-      $log.info("Starting instance #{@instance_id}")
-      @instance.start()
-    end
-
-    def stop
-      if %w(stopped stopping).include?(@instance.state.name)
-        $log.info("Instance #{@instance_id} already stopping or stopped")
-        return
-      end
-      $log.info("Stopping instance #{@instance_id}")
-      @instance.stop()
-
-      # empty configuration for ec2 instances
-      return {}
-    end
-
-    def wait(wait_states=[])
-      $log.debug("Not waiting for EC2 instance #{@instance_id}")
-    end
-
-  end
-end

data/lib/cfn_manage/ecs_cluster_start_stop_handler.rb

@@ -1,80 +0,0 @@
-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, scheduling_strategy: 'REPLICA', max_results: 100).each do |results|
-        @services.push(*results.service_arns)
-      end
-      $log.info("Found #{@services.count} services in ECS cluster #{cluster_id}")
-      @cluster = cluster_id
-      @skip_wait = skip_wait
-      @ignore_missing_ecs_config = (ENV.key? 'IGNORE_MISSING_ECS_CONFIG' and ENV['IGNORE_MISSING_ECS_CONFIG'] == '1')
-    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
-
-        if configuration.has_key?(service.service_name)
-          desired_count = configuration[service.service_name]['desired_count']
-        elsif @ignore_missing_ecs_config
-          $log.info("ECS service #{service.service_name} wasn't previosly stopped by cfn_manage. Option --ignore-missing-ecs-config set and setting desired count to 1")
-          desired_count = 1
-        else
-          $log.warn("ECS service #{service.service_name} wasn't previosly stopped by cfn_manage. Skipping ...")
-          next
-        end
-
-        $log.info("Starting ECS service #{service.service_name} with desired count of #{desired_count}")
-        @ecs_client.update_service({
-          desired_count: 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.empty? ? nil : configuration
-    end
-
-    def wait(completed_status)
-      $log.debug("Not waiting for ECS Services in cluster #{@cluster}")
-    end
-
-  end
-end

data/lib/cfn_manage/rds_start_stop_handler.rb

@@ -1,134 +0,0 @@
-require 'cfn_manage/aws_credentials'
-
-module CfnManage
-
-  class RdsStartStopHandler
-
-    def initialize(instance_id, skip_wait)
-      @instance_id = 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)
-      end
-      rds = Aws::RDS::Resource.new(client: @rds_client)
-      @rds_instance = rds.db_instance(instance_id)
-
-    end
-
-    def start(configuration)
-      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
-
-      if @rds_instance.db_instance_status == 'available'
-        $log.info("RDS Instance #{@instance_id} is already in available state")
-      end
-
-      # start rds instance
-      if @rds_instance.db_instance_status == 'stopped'
-        $log.info("Starting db instance #{@instance_id}")
-        @rds_client.start_db_instance({ db_instance_identifier: @instance_id })
-
-        # wait instance to become available
-        unless @skip_wait
-          $log.info("Waiting db instance to become available #{@instance_id}")
-          wait('available')
-        end
-      else
-        wait('available') unless @skip_wait
-      end
-
-      # convert rds instance to mutli-az if required
-      if configuration['is_multi_az']
-        $log.info("Converting to Multi-AZ instance after start (instance #{@instance_id})")
-        set_rds_instance_multi_az( true)
-      end unless configuration.nil?
-    end
-
-    def stop
-
-      configuration = {
-          is_multi_az: @rds_instance.multi_az
-      }
-      # RDS list of exluded engines that don't support RDS stop start
-      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
-
-      # check if available
-      if @rds_instance.db_instance_status != 'available'
-        $log.warn("RDS Instance #{@instance_id} not in available state, and thus can not be stopped")
-        $log.warn("RDS Instance #{@instance_id} state: #{@rds_instance.db_instance_status}")
-        return configuration
-      end
-
-      # check if already stopped
-      if @rds_instance.db_instance_status == 'stopped'
-        $log.info("RDS Instance #{@instance_id} is already stopped")
-        return configuration
-      end
-
-      #check if mutli-az RDS. if so, convert to single-az
-      if @rds_instance.multi_az
-        $log.info("Converting to Non-Multi-AZ instance before stop (instance #{@instance_id}")
-        set_rds_instance_multi_az(false)
-      end
-
-      # 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 })
-      unless @skip_wait
-        $log.info("Waiting db instance to be stopped #{@instance_id}")
-        wait('stopped')
-      end
-
-      return configuration
-    end
-
-    def set_rds_instance_multi_az(multi_az)
-      if @rds_instance.multi_az == multi_az
-        $log.info("Rds instance #{@rds_instance.db_instance_identifier} already multi-az=#{multi_az}")
-        return
-      end
-      @rds_instance.modify({ multi_az: multi_az, apply_immediately: true })
-      # allow half an hour for instance to be converted
-      wait('available')
-    end
-
-    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
-    end
-
-    private :set_rds_instance_multi_az
-
-  end
-
-end

data/lib/cfn_manage/spot_fleet_start_stop_handler.rb

@@ -1,57 +0,0 @@
-require 'cfn_manage/aws_credentials'
-
-module CfnManage
-
-  class SpotFleetStartStopHandler
-
-    def initialize(fleet_id, skip_wait)
-      @fleet_id = 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)
-      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
-
-    def wait(wait_states=[])
-      $log.debug("Not waiting for spot fleet #{@fleet_id}")
-    end
-
-  end
-
-end