MovableInkAWS 0.1.12 → 0.1.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6068f47fd36870a62486fe861142c4f54d039d08699eb7067de12e2601345dce
-  data.tar.gz: 0ddae90ec17a4ed18309e8c6116a6b803cf93b8292611a6a74098772e461b2c0
+  metadata.gz: 9940e314b3dc8c1174f75e495f67a2879b057c2dff22e9cc10e5b85c848684c9
+  data.tar.gz: d65a2015d5c254bf1520a8bcffd5dbb3b73d9c2c587f3425204f69244be1b5b6
 SHA512:
-  metadata.gz: d6426c56c304a32b4943fddf5fa566aed036d06c1bf36c03c3432f1b4508b004894c15aca331e810d4d6bdde539587902212876f0f5839d24ba74a7c403e6f2f
-  data.tar.gz: 24ff69fdb3dfcf9e6f01a90c9fdc0e5a5bc957676554c63dbe179e4103e98a8e4620908bfb132d59c78c1704aa028b835077048bf86058ac21dc33486b3f0112
+  metadata.gz: 8d7ce1acf9e30f3f0e5b2015656ee19d6378f947f2f4ee026ab7f5f98e94dc74a4721633fe460defc7b3f66f14ed0c860beabd58d488d4e8bf910c8dd0952453
+  data.tar.gz: 5376c892106c465b0cb8c27fcfbe0164c19c4e4eaae0b633d61e13a9cbc2eb5979f993c61021362fa5b280a2134d346c2efba08bbdf8347caf2037c2b5522618

data/lib/movable_ink/aws/elasticache.rb

@@ -0,0 +1,48 @@
+module MovableInk
+  class AWS
+    module ElastiCache
+      def elasticache(region: my_region)
+        @elasticache_client ||= {}
+        @elasticache_client[region] ||= Aws::ElastiCache::Client.new(region: region)
+      end
+
+      def replication_group(role)
+        run_with_backoff do
+          @replication_group ||= elasticache.describe_replication_groups({replication_group_id: role}).replication_groups.first
+        end
+      end
+
+      def elasticache_primary(role)
+        replication_group(role).node_groups
+                               .first
+                               .primary_endpoint
+                               .address
+      end
+
+      def all_elasticache_replicas(role)
+        replication_group(role).node_groups
+                               .first
+                               .node_group_members
+                               .select{|ng| ng.current_role == 'replica'}
+                               .map{|member| member.read_endpoint.address}
+      end
+
+      def elasticache_replica_in_my_az(role)
+        members = replication_group(role).node_groups
+                                         .first
+                                         .node_group_members
+                                         .select{|ng| ng.preferred_availability_zone == availability_zone}
+        begin
+          members.select{|ng| ng.current_role == 'replica'}
+          .first
+          .read_endpoint
+          .address
+        rescue NoMethodError
+          members.first
+                 .read_endpoint
+                 .address
+        end
+      end
+    end
+  end
+end

data/lib/movable_ink/aws.rb

@@ -8,6 +8,8 @@ require_relative 'aws/route53'
 require_relative 'aws/ssm'
 require_relative 'aws/athena'
 require_relative 'aws/s3'
+require_relative 'aws/elasticache'
+
 
 module MovableInk
   class AWS
@@ -18,6 +20,7 @@ module MovableInk
     include SSM
     include Athena
     include S3
+    include ElastiCache
 
     class << self
       def regions

data/lib/movable_ink/version.rb

@@ -1,5 +1,5 @@
 module MovableInk
   class AWS
-    VERSION = '0.1.12'
+    VERSION = '0.1.13'
   end
 end

data/spec/elasticache_spec.rb

@@ -0,0 +1,51 @@
+require_relative '../lib/movable_ink/aws'
+
+describe MovableInk::AWS::ElastiCache do
+  let(:aws) { MovableInk::AWS.new }
+  let(:elasticache) { Aws::ElastiCache::Client.new(stub_responses: true) }
+
+  describe "elasticache_replica_in_my_az" do
+    let(:replication_group) { elasticache.stub_data(:describe_replication_groups, replication_groups: [
+          replication_group_id: 'foo',
+          node_groups: [{
+            primary_endpoint: {address: 'primary'},
+            node_group_members: [
+              {preferred_availability_zone: 'us-foo-1a',
+              read_endpoint: {address: 'address-1a'},
+              current_role: 'replica' },
+              {preferred_availability_zone: 'us-foo-1a',
+                read_endpoint: {address: 'address-1a-primary'},
+                current_role: 'primary' },
+              {preferred_availability_zone: 'us-foo-1b',
+                read_endpoint: {address: 'address-1b'},
+                current_role: 'primary' },
+              {preferred_availability_zone: 'us-foo-1c',
+                read_endpoint: {address: 'address-1c'},
+                current_role: 'replica' },
+            ]
+          }],
+        ]
+    )}
+
+    before(:each) do
+      elasticache.stub_responses(:describe_replication_groups, replication_group)
+      allow(aws).to receive(:mi_env).and_return('test')
+      allow(aws).to receive(:elasticache).and_return(elasticache)
+    end
+
+    it "should return the correct read address" do
+      allow(aws).to receive(:availability_zone).and_return('us-foo-1a')
+      expect(aws.elasticache_replica_in_my_az("foo")).to eq("address-1a")
+    end
+
+    it "should return the primary if there is no replica" do
+      allow(aws).to receive(:availability_zone).and_return('us-foo-1b')
+      expect(aws.elasticache_replica_in_my_az("foo")).to eq("address-1b")
+    end
+
+    it "should return an array of all replicas" do
+      allow(aws).to receive(:availability_zone).and_return('us-foo-1b')
+      expect(aws.all_elasticache_replicas("foo")).to eq(["address-1a", "address-1c"])
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: MovableInkAWS
 version: !ruby/object:Gem::Version
-  version: 0.1.12
+  version: 0.1.13
 platform: ruby
 authors:
 - Matt Chesler
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-07-23 00:00:00.000000000 Z
+date: 2018-08-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: aws-sdk
@@ -45,6 +45,7 @@ files:
 - lib/movable_ink/aws/athena.rb
 - lib/movable_ink/aws/autoscaling.rb
 - lib/movable_ink/aws/ec2.rb
+- lib/movable_ink/aws/elasticache.rb
 - lib/movable_ink/aws/errors.rb
 - lib/movable_ink/aws/route53.rb
 - lib/movable_ink/aws/s3.rb
@@ -54,6 +55,7 @@ files:
 - spec/autoscaling_spec.rb
 - spec/aws_spec.rb
 - spec/ec2_spec.rb
+- spec/elasticache_spec.rb
 - spec/route53_spec.rb
 - spec/s3_spec.rb
 - spec/sns_spec.rb