MovableInkAWS 2.7.5 → 2.7.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 96e412e659e0f48c3a469c1151da0e2e610aaafe2b0817ad22e7dae4eae39765
-  data.tar.gz: 5ae7c22cfedb9ddbd89c902f9531e01f88293e6843e1d057b9c6ca1f818b49b9
+  metadata.gz: 6f6dce584d3446fcb01d1b4ae891bd1d639f4319662c0ffebe0ab80509a04974
+  data.tar.gz: b6535994401231a30f9500393deb8d108271ae37e7224e9a47e608c92bfbe5a1
 SHA512:
-  metadata.gz: a877074eb719b30bdba76b22176e3cfc7965760a93de3d98cb9197a1e362c190d8a559f7cebdc5ebb0ad46138fdb8992622fd40c753cf593ced446aa5c68ed8f
-  data.tar.gz: 399aae9bbe6a4f14fa2fae2cb44c9bc2b378b898af8961bf2a8f01f3555f7db613c350cfedced06ffc357e863f7415a8e4a57f95c6eaabd76655ded6e080e2c9
+  metadata.gz: d679722fe0abdb344440a2e1dcabb913be3ba9ff279be3d8523c4e98e0d067e6d360362bc8dce7f71d0afd510eead66ab329702792426d5df9336bb3ca8dca04
+  data.tar.gz: '0238e618656e228fb021ce475303791d99c09c814e5b55f7948478dc78c3657b2a0d355acee87b8dbd5e0ed61997c82c00888970daae64e13c9927bc87a2b795'

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    MovableInkAWS (2.7.5)
+    MovableInkAWS (2.7.6)
       aws-sdk-athena (~> 1)
       aws-sdk-autoscaling (~> 1)
       aws-sdk-cloudwatch (~> 1)

data/lib/movable_ink/aws/ec2.rb

@@ -211,12 +211,26 @@ module MovableInk
         private_ip_addresses(ordered_instances)
       end
 
-      def redis_by_role(role, port)
-        instance_ip_addresses_by_role(role: role)
-          .shuffle
-          .inject([]) { |redii, instance|
-            redii.push({"host" => instance, "port" => port})
-          }
+      def redis_by_role(role, port, availability_zones = [])
+        if availability_zones.class != Array
+          raise MovableInk::AWS::Errors::AvailabilityZonesListInvalidError
+        end
+
+        redis_instances = []
+
+        def redis(host, port)
+          {"host" => host, "port" => port}
+        end
+
+        if availability_zones.length > 0
+          availability_zones.each do |az|
+            redis_instances << instance_ip_addresses_by_role(role: role, availability_zone: az).inject([]) { |redii, instance| redii.push(redis(instance, port)) }
+          end
+        else
+          redis_instances << instance_ip_addresses_by_role(role: role).inject([]) { |redii, instance| redii.push(redis(instance, port)) }
+        end
+
+        redis_instances.flatten.shuffle
       end
 
       def elastic_ips

data/lib/movable_ink/aws/errors.rb

@@ -8,6 +8,7 @@ module MovableInk
       class InvalidDiscoveryTypeError < StandardError; end
       class RoleNameRequiredError < StandardError; end
       class RoleNameInvalidError < StandardError; end
+      class AvailabilityZonesListInvalidError < StandardError; end
 
       class ExpectedError
         def initialize(error_class, message_patterns = [])

data/lib/movable_ink/version.rb

@@ -1,5 +1,5 @@
 module MovableInk
   class AWS
-    VERSION = '2.7.5'
+    VERSION = '2.7.6'
   end
 end

data/spec/ec2_spec.rb

@@ -589,6 +589,34 @@ describe MovableInk::AWS::EC2 do
           }
         ]])
       }
+      let(:redis_data_cross_azs) { ec2.stub_data(:describe_instances, reservations: [
+        instances: [
+          {
+            tags: [
+              {
+                key: 'mi:roles',
+                value: 'visitor_redis'
+              }
+            ],
+            private_ip_address: '10.0.0.1',
+            placement: {
+              availability_zone: availability_zone
+            }
+          },
+          {
+            tags: [
+              {
+                key: 'mi:roles',
+                value: 'visitor_redis'
+              }
+            ],
+            private_ip_address: '10.0.0.2',
+            placement: {
+              availability_zone: 'us-east-1z'
+            }
+          }
+        ]])
+      }
       let(:redii) { [{"host" => '10.0.0.1', "port" => 6379},{"host" => '10.0.0.2', "port" => 6379}] }
 
       it "should return redis IPs and ports" do
@@ -600,6 +628,20 @@ describe MovableInk::AWS::EC2 do
 
         expect(aws.redis_by_role('visitor_redis', port)).to match_array(redii)
       end
+
+      it "should return redis IPs and ports only from specified AZs" do
+        ec2.stub_responses(:describe_instances, redis_data_cross_azs)
+        allow(aws).to receive(:mi_env).and_return('test')
+        allow(aws).to receive(:availability_zone).and_return(availability_zone)
+        allow(aws).to receive(:my_region).and_return('us-east-1')
+        allow(aws).to receive(:ec2).and_return(ec2)
+
+        expect(aws.redis_by_role('visitor_redis', port, ['us-east-1z'])).to match_array([{"host" => '10.0.0.2', "port" => 6379}])
+      end
+
+      it "should throw exception if specified AZs list isnt array" do
+        expect { aws.redis_by_role('visitor_redis', port, 'us-east-1z') }.to raise_error(MovableInk::AWS::Errors::AvailabilityZonesListInvalidError)
+      end
     end
 
     context "elastic IPs" do

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: MovableInkAWS
 version: !ruby/object:Gem::Version
-  version: 2.7.5
+  version: 2.7.6
 platform: ruby
 authors:
 - MI SRE
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-09-22 00:00:00.000000000 Z
+date: 2022-09-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk-core