cfn_manage 0.7.0 → 0.8.3

data/lib/cfn_manage/alarm_start_stop_handler.rb

@@ -1,44 +0,0 @@
-require 'cfn_manage/aws_credentials'
-
-module CfnManage
-
-  class AlarmStartStopHandler
-
-    def initialize(alarm_name)
-      @alarm_id = alarm_name
-      credentials = CfnManage::AWSCredentials.get_session_credentials("startstopalarm_#{@asg_name}")
-      @cwclient = Aws::CloudWatch::Client.new(retry_limit: 20)
-      if credentials != nil
-        @cwclient = Aws::CloudWatch::Client.new(credentials: credentials, retry_limit: 20)
-      end
-
-      @cwresource = Aws::CloudWatch::Resource.new(client: @cwclient)
-      @alarm = @cwresource.alarm(alarm_name)
-    end
-
-    def start(configuration)
-      if @alarm.actions_enabled
-        $log.info("Alarm #{@alarm.alarm_arn} actions already enabled")
-        return
-      end
-      $log.info("Enabling alarm #{@alarm.alarm_arn}")
-      @alarm.enable_actions({})
-    end
-
-    def stop
-      if not @alarm.actions_enabled
-        $log.info("Alarm #{@alarm.alarm_arn} actions already disabled")
-        return {}
-      end
-      $log.info("Disabling actions on alarm #{@alarm.alarm_arn}")
-      @alarm.disable_actions({})
-      return {}
-    end
-
-    def wait(wait_states=[])
-      $log.debug("Not waiting for alarm #{@alarm_id}")
-    end
-
-  end
-
-end

data/lib/cfn_manage/asg_start_stop_handler.rb

@@ -1,181 +0,0 @@
-require 'cfn_manage/aws_credentials'
-
-module CfnManage
-
-  class AsgStartStopHandler
-
-    def initialize(asg_id, skip_wait)
-      @asg_name = asg_id
-      @skip_wait = skip_wait
-      @asg_suspend_termination = (ENV.key?('ASG_SUSPEND_TERMINATION') and ENV['ASG_SUSPEND_TERMINATION'] == '1')
-      credentials = CfnManage::AWSCredentials.get_session_credentials("stopasg_#{@asg_name}")
-      @asg_client = Aws::AutoScaling::Client.new(retry_limit: 20)
-      @ec2_client = Aws::EC2::Client.new(retry_limit: 20)
-      if credentials != nil
-        @asg_client = Aws::AutoScaling::Client.new(credentials: credentials, retry_limit: 20)
-        @ec2_client = Aws::EC2::Client.new(credentials: credentials, retry_limit: 20)
-      end
-
-      asg_details = @asg_client.describe_auto_scaling_groups(
-          auto_scaling_group_names: [@asg_name]
-      )
-      if asg_details.auto_scaling_groups.size() == 0
-        raise "Couldn't find ASG #{@asg_name}"
-      end
-      @asg = asg_details.auto_scaling_groups[0]
-    end
-
-    def stop
-      # check if already stopped
-      if @asg.min_size == @asg.max_size and @asg.max_size == @asg.desired_capacity and @asg.min_size == 0
-        $log.info("ASG #{@asg_name} already stopped")
-        # nil and false configurations are not saved
-        return nil
-      else
-
-        puts @asg.auto_scaling_group_name
-
-        unless @asg_suspend_termination 
-          # store asg configuration to S3
-          configuration = {
-              desired_capacity: @asg.desired_capacity,
-              min_size: @asg.min_size,
-              max_size: @asg.max_size
-          }
-
-          $log.info("Setting desired capacity to 0/0/0 for ASG #{@asg.auto_scaling_group_name}A")
-
-          puts @asg.auto_scaling_group_name
-          @asg_client.update_auto_scaling_group({
-              auto_scaling_group_name: "#{@asg.auto_scaling_group_name}",
-              min_size: 0,
-              max_size: 0,
-              desired_capacity: 0
-          })
-          return configuration
-        else
-
-          configuration = {
-            desired_capacity: @asg.desired_capacity,
-            min_size: @asg.min_size,
-            max_size: @asg.max_size,
-            suspended_processes: @asg.suspended_processes
-          }
-
-          $log.info("Suspending processes for ASG #{@asg.auto_scaling_group_name}A")
-
-          @asg_client.suspend_processes({
-            auto_scaling_group_name: "#{@asg.auto_scaling_group_name}",
-          })
-
-          $log.info("Stopping all instances in ASG #{@asg.auto_scaling_group_name}A")
-
-          @asg.instances.each do |instance|
-            @instance_id = instance.instance_id
-            @instance = Aws::EC2::Resource.new(client: @ec2_client, retry_limit: 20).instance(@instance_id)
-
-            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()
-          end
-
-          return configuration
-
-        end
-
-      end
-
-    end
-
-    def start(configuration)
-      if configuration.nil?
-        $log.warn("No configuration found for #{@asg_name}, skipping..")
-        return
-      end
-      $log.info("Starting ASG #{@asg_name} with following configuration\n#{configuration}")
-
-      unless @asg_suspend_termination 
-        # restore asg sizes
-        @asg_client.update_auto_scaling_group({
-          auto_scaling_group_name: @asg_name,
-          min_size: configuration['min_size'],
-          max_size: configuration['max_size'],
-          desired_capacity: configuration['desired_capacity']
-        })
-      else
-
-        $log.info("Starting instances for ASG #{@asg_name}...")
-
-        @asg.instances.each do |instance|
-          @instance_id = instance.instance_id
-          @instance = Aws::EC2::Resource.new(client: @ec2_client, retry_limit: 20).instance(@instance_id)
-
-          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
-
-        unhealthy = true
-
-        $log.info("Checking health status for instances for ASG #{@asg_name}")
-
-        while unhealthy do
-          
-          asg_curr_details = @asg_client.describe_auto_scaling_groups(
-            auto_scaling_group_names: [@asg_name]
-          )
-          @asg_status = asg_curr_details.auto_scaling_groups[0]
-
-          allHealthy = 0
-
-          @asg_status.instances.each do |instance|
-            @instance_health = instance.health_status
-            if @instance_health == "Healthy"
-              allHealthy += 1
-            else
-              $log.info("Instance #{instance.instance_id} not currently healthy...")
-              sleep(15)
-            end
-          end
-
-          if allHealthy == @asg_status.instances.length
-            $log.info("All instances healthy in ASG #{@asg_name}")
-            unhealthy = false
-            break
-          end
-
-        end
-
-        $log.info("Resuming all processes for ASG #{@asg_name}")
-
-        @asg_client.resume_processes({
-          auto_scaling_group_name: "#{@asg.auto_scaling_group_name}",
-        })
-
-        if configuration['suspended_processes'].any?
-
-          $log.info("Suspending processes stored in configuration for ASG #{@asg_name}")
-
-          @asg_client.suspend_processes({
-            auto_scaling_group_name: "#{@asg.auto_scaling_group_name}",
-            scaling_processes: configuration['suspended_processes'],
-          })
-        end
-
-      end
-
-    end
-
-    def wait(wait_states=[])
-      $log.debug("Not waiting for ASG #{@asg_name}")
-    end
-
-  end
-end

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