[#170425434] Add commands to (de-)register single target (#4)

Author: TokRa13

bin/deregister_single_target

@@ -0,0 +1,55 @@
+#!/usr/bin/env ruby
+
+require "aws-sdk-elasticloadbalancingv2"
+
+host_id = ARGV[0]
+
+fail "Please specify HOST_ID" if host_id.nil?
+
+REGION = "us-west-2".freeze
+class AWSClient
+  class << self
+    def elbv2
+      @elbv2 ||= Aws::ElasticLoadBalancingV2::Client.new(region: REGION)
+    end
+  end
+end
+
+tgs = AWSClient.elbv2.describe_target_groups.target_groups
+
+tgs.each do |tg|
+  targets = AWSClient.elbv2.describe_target_health(target_group_arn: tg.target_group_arn)
+                  .target_health_descriptions.map(&:target)
+  target_ports = targets.map(&:port).uniq
+
+  AWSClient.elbv2.deregister_targets({
+    target_group_arn: tg.target_group_arn,
+    targets: [
+      {
+        id: host_id,
+        port: target_ports[0]
+      }
+    ]
+  })
+end
+
+tgs.each do |tg|
+  targets = AWSClient.elbv2.describe_target_health(target_group_arn: tg.target_group_arn)
+                  .target_health_descriptions.map(&:target)
+  target_ports = targets.map(&:port).uniq
+
+  AWSClient.elbv2.wait_until(
+    :target_deregistered,
+    target_group_arn: tg.target_group_arn,
+    targets: [
+      {
+        id: host_id,
+        port: target_ports[0]
+      }
+    ]
+  )
+
+  puts "#{tg.target_group_name}:#{target_ports[0]} de-registered"
+end
+
+puts "\nTarget host de-registered from all target groups"

bin/register_single_target

@@ -0,0 +1,59 @@
+#!/usr/bin/env ruby
+
+require "aws-sdk-elasticloadbalancingv2"
+
+host_id = ARGV[0]
+tg_skip = ARGV[1]
+
+fail "Please specify HOST_ID" if host_id.nil?
+
+REGION = "us-west-2".freeze
+class AWSClient
+  class << self
+    def elbv2
+      @elbv2 ||= Aws::ElasticLoadBalancingV2::Client.new(region: REGION)
+    end
+  end
+end
+
+tgs = AWSClient.elbv2.describe_target_groups.target_groups
+
+if tg_skip
+  tgs.reject! { |tg| tg_skip.include? tg.target_group_name }
+end
+
+tgs.each do |tg|
+  targets = AWSClient.elbv2.describe_target_health(target_group_arn: tg.target_group_arn)
+                  .target_health_descriptions.map(&:target)
+  target_ports = targets.map(&:port).uniq
+
+  AWSClient.elbv2.register_targets({
+    target_group_arn: tg.target_group_arn,
+    targets: [
+      {
+        id: host_id,
+        port: target_ports[0]
+      }
+    ]
+  })
+end
+
+tgs.each do |tg|
+  targets = AWSClient.elbv2.describe_target_health(target_group_arn: tg.target_group_arn)
+                  .target_health_descriptions.map(&:target)
+  target_ports = targets.map(&:port).uniq
+
+  AWSClient.elbv2.wait_until(
+    :target_in_service,
+    target_group_arn: tg.target_group_arn,
+    targets: [
+      {
+        id: host_id,
+        port: target_ports[0]
+      }
+    ]
+  )
+  puts "#{tg.target_group_name}:#{target_ports[0]} registered"
+end
+
+puts "\nTarget host registered back to all target groups"

deploy-tools.gemspec

@@ -1,13 +1,18 @@
 Gem::Specification.new do |s|
   s.name        = 'deploy-tools'
-  s.version     = '0.1.2'
-  s.date        = '2022-06-28'
+  s.version     = '0.1.3'
+  s.date        = '2022-09-02'
   s.summary     = "Deploy tools"
   s.description = "A set of script used for deployment"
   s.authors     = ["Tony Nyurkin", "Serhii Voronoi"]
   s.email       = '[email protected]'
   s.files       = `git ls-files `.split("\n")
   s.bindir      = "bin"
-  s.executables = ["blue_green_switch", "detect_inactive_color"]
+  s.executables = [
+    "blue_green_switch",
+    "detect_inactive_color",
+    "deregister_single_target",
+    "register_single_target"
+  ]
   s.add_runtime_dependency 'aws-sdk-elasticloadbalancingv2', '~>1.44'
 end