cfn_manage 0.7.0 → 0.8.3

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

data/lib/cfn_manage/transfer_start_stop_handler.rb

@@ -1,97 +0,0 @@
-require 'cfn_manage/aws_credentials'
-require 'aws-sdk-transfer'
-
-module CfnManage
-
-  class TransferStartStopHandler
-
-    def initialize(server_id, skip_wait)
-      sftpId = server_id.split("/")
-      @server_id = sftpId.last
-      @skip_wait = skip_wait
-      credentials = CfnManage::AWSCredentials.get_session_credentials("startstoptransfer_#{@server_id}")
-      @client = Aws::Transfer::Client.new(retry_limit: 20)
-      if credentials != nil
-        @client = Aws::Transfer::Client.new(credentials: credentials, retry_limit: 20)
-      end
-    end
-
-    def start(configuration)
-
-      state = get_state()
-
-      if state != "OFFLINE"
-        $log.warn("SFTP Server #{@server_id} is in a state of #{state} and can not be started.")
-        return
-      end
-
-      $log.info("Starting SFTP Server #{@server_id}")
-      @client.start_server({
-        server_id: @server_id,
-      })
-      unless @skip_wait
-        $log.info("Waiting for SFTP to start #{@cluster_id}")
-        wait('ONLINE')
-      end
-
-      return configuration
-    end
-
-    def stop()
-
-      state = get_state()
-
-      if state != "ONLINE"
-        $log.warn("SFTP Server #{@server_id} is in a state of #{state} and can not be stopped.")
-        return {}
-      end
-
-      @client.stop_server({
-        server_id: @server_id,
-      })
-      unless @skip_wait
-        $log.info("Waiting for SFTP to stop #{@cluster_id}")
-        wait('OFFLINE')
-      end
-      return {}
-    end
-
-    def get_state()
-      resp = @client.describe_server({
-        server_id: @server_id,
-      })
-      return resp.server.state
-    end
-
-    def wait(completed_state)
-
-      state_count = 0
-      steady_count = 2
-      attempts = 0
-      until attempts == (max_attempts = 60*6) do
-        state = get_state()
-        $log.info("SFTP Server #{@server_id} state: #{state}, waiting for #{completed_state}")
-
-        if state == "#{completed_state}"
-          state_count = state_count + 1
-          $log.info("#{state_count}/#{steady_count}")
-        elsif ["START_FAILED", "STOP_FAILED"].include?(state)
-          $log.error("SFTP Server #{@server_id} failed to reach state #{completed_state} with current state #{state}!")
-          break
-        else
-          state_count = 0
-        end
-        break if state_count == steady_count
-        attempts = attempts + 1
-        sleep(15)
-      end
-
-      if attempts == max_attempts
-        $log.error("SFTP Server #{@server_id} did not enter #{completed_state} state, however continuing operations...")
-      end
-
-    end
-
-  end
-
-end