Merge pull request #112 from movableink/nbogetic/sc-89400/toggle-auto…

…scaling-lambda-failover [sc-89400] Allow toggling the autoscaling lambda
movableink · Sep 7, 2023 · 45b73d4 · 45b73d4
2 parents f882b2a + d94504d
commit 45b73d4
Show file tree

Hide file tree

Showing 5 changed files with 44 additions and 2 deletions.
diff --git a/Gemfile.lock b/Gemfile.lock
@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    MovableInkAWS (2.8.5)
+    MovableInkAWS (2.8.6)
       aws-sdk-athena (~> 1)
       aws-sdk-autoscaling (~> 1)
       aws-sdk-cloudwatch (~> 1)
@@ -10,6 +10,7 @@ PATH
       aws-sdk-eks (~> 1)
       aws-sdk-elasticache (~> 1)
       aws-sdk-iam (~> 1)
+      aws-sdk-lambda (~> 1)
       aws-sdk-rds (~> 1)
       aws-sdk-route53 (~> 1)
       aws-sdk-s3 (~> 1)
@@ -55,6 +56,9 @@ GEM
     aws-sdk-kms (1.62.0)
       aws-sdk-core (~> 3, >= 3.165.0)
       aws-sigv4 (~> 1.1)
+    aws-sdk-lambda (1.96.0)
+      aws-sdk-core (~> 3, >= 3.165.0)
+      aws-sigv4 (~> 1.1)
     aws-sdk-rds (1.171.0)
       aws-sdk-core (~> 3, >= 3.165.0)
       aws-sigv4 (~> 1.1)

diff --git a/MovableInkAWS.gemspec b/MovableInkAWS.gemspec
@@ -17,6 +17,7 @@ Gem::Specification.new do |s|
   s.add_runtime_dependency 'aws-sdk-eks', '~> 1'
   s.add_runtime_dependency 'aws-sdk-elasticache', '~> 1'
   s.add_runtime_dependency 'aws-sdk-iam', '~> 1'
+  s.add_runtime_dependency 'aws-sdk-lambda', '~> 1'
   s.add_runtime_dependency 'aws-sdk-rds', '~> 1'
   s.add_runtime_dependency 'aws-sdk-route53', '~> 1'
   s.add_runtime_dependency 'aws-sdk-s3', '~> 1'

diff --git a/lib/movable_ink/aws.rb b/lib/movable_ink/aws.rb
@@ -11,6 +11,7 @@
 require_relative 'aws/eks'
 require_relative 'aws/elasticache'
 require_relative 'aws/api_gateway'
+require_relative 'aws/lambda'
 require_relative 'consul/consul'
 require 'aws-sdk-cloudwatch'
 
@@ -28,6 +29,7 @@ class AWS
     include ApiGateway
     include EKS
     include IAM
+    include Lambda
 
     class << self
       def regions

diff --git a/lib/movable_ink/aws/lambda.rb b/lib/movable_ink/aws/lambda.rb
@@ -0,0 +1,35 @@
+require 'aws-sdk-lambda'
+
+module MovableInk
+  class AWS
+    module Lambda
+      def lambda(region: 'us-east-1')
+        @lambda_client ||= {}
+        @lambda_client[region] ||= Aws::Lambda::Client.new(region: region)
+      end
+
+      def disable_autoscaling_lambda(function_name)
+        lambda.update_function_configuration({
+          function_name: function_name,
+          environment: {
+            variables: {
+              "DRY_RUN" => "true",
+            }
+          }
+        })
+      end
+
+      def enable_autoscaling_lambda(function_name)
+        lambda.update_function_configuration({
+          function_name: function_name,
+          environment: {
+            variables: {
+              "DRY_RUN" => "false",
+            }
+          }
+        })
+      end
+
+    end
+  end
+end
diff --git a/lib/movable_ink/version.rb b/lib/movable_ink/version.rb
@@ -1,5 +1,5 @@
 module MovableInk
   class AWS
-    VERSION = '2.8.5'
+    VERSION = '2.8.6'
   end
 end