awspec 0.86.0 → 0.87.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 1d9ac6e4891eae5bfd514aee8f72a6a323b0b5c0
-  data.tar.gz: de536803219c6b32193384ad01b82ea0f23a6bfb
+  metadata.gz: f0bcfc84f264a0b85f2f18f22519ed7aab5ffde2
+  data.tar.gz: 23da8dcad88edde7095a351e0c438dbde7f81f84
 SHA512:
-  metadata.gz: 385162854f73f06848cca0c82d8530600b95d10b32d5caa81cf3ef2da487f9f164fdbbec5d7aae4f1c47e9bf9ff2db741d4866c978bd38ef692cc7a83f19106e
-  data.tar.gz: 1be74de00693e74ce843ee2ff3f83f150d201363c4a3e2cf80018eff794b53e52593a728033eea8586aa7e4d67fccd51d6cc2bb4fdacc69062a9a7e1e57f282c
+  metadata.gz: 4c9ccb900e383f560887452ec04f430eb8ee5a2c8b5532bce01d7b080b8c886e781d98e9a2c4c3d9108ed18af398b462f4291100256129341d9fb86d790cace9
+  data.tar.gz: 721e77e16de76fc711c7196330804ad48492ccc4dae6b1853a54bada63f04057cdbcdbef1be41fd393c9ed77514ef9464449276d4d03838dfb858d140f86d506

data/.gitignore

@@ -8,3 +8,5 @@
 /tmp/
 /spec/secrets.yml
 /.ruby-version
+*.sw*
+*.gem

data/Rakefile

@@ -23,7 +23,8 @@ if defined?(RSpec)
                'spec:core',
                'spec:generator_spec',
                'spec:generator_doc',
-               'spec:rubocop']
+               'spec:rubocop',
+               'spec:helper']
 
     task type: types
 
@@ -49,6 +50,10 @@ if defined?(RSpec)
       t.pattern = 'spec/generator/doc/*_spec.rb'
     end
 
+    RSpec::Core::RakeTask.new(:helper) do |t|
+      t.pattern = 'spec/lib/awspec/helper/*_spec.rb'
+    end
+
     RuboCop::RakeTask.new
   end
 end

data/lib/awspec/helper/client_wrap.rb

@@ -0,0 +1,46 @@
+module Awspec::Helper
+  class ClientWrap
+    attr_reader :client, :backoff, :iteration, :backoff_limit, :symbol
+    def initialize(real_client = nil, args = {})
+      raise ArgumentError, 'Client can not be nil' if real_client.nil?
+      @client        = real_client
+      @backoff       = args.key?(:backoff) ? args[:backoff] : 0.0
+      @orig_backoff  = @backoff
+      @iteration     = args.key?(:iteration) ? args[:iteration] : 1
+      @orig_iter     = @iteration
+      @backoff_limit = args.key?(:backoff_limit) ? args[:backoff_limit] : 30.0
+      # build the symbol we'll use to compare to any errors caught in method_missing
+      # below.
+      @symbol = real_client.class.to_s.split('::').shift(2).push('Errors', 'RequestLimitExceeded').join('::').to_sym
+    end
+
+    protected
+
+    # used to capture only the "RequestLimitExceeded" error from an aws
+    # client api call.  In the case of matching it we want to try again,
+    # backing off successively each time, until the backoff_limit is reached or
+    # exceeded, in which case, the error will be re-raised and it should fail
+    # as expected.
+    def method_missing(m, *args, &block)
+      begin
+        results = client.send(m, *args, &block)
+      rescue Exception => e # rubocop:disable Lint/RescueException
+        raise unless e.class.to_s == symbol.to_s && backoff < backoff_limit
+
+        @backoff = backoff + (iteration * iteration * 0.5)
+        @iteration += 1
+        sleep backoff
+        results = self.send(m, *args, &block)
+      end
+
+      reset_backoff
+
+      results
+    end
+
+    def reset_backoff
+      @backoff   = @orig_backoff
+      @iteration = @orig_iter
+    end
+  end
+end

data/lib/awspec/helper/finder.rb

@@ -35,6 +35,8 @@ require 'awspec/helper/finder/cloudformation'
 
 require 'awspec/helper/finder/account_attributes'
 
+require 'awspec/helper/client_wrap'
+
 module Awspec::Helper
   module Finder
     include Awspec::Helper::Finder::Alb
@@ -111,7 +113,10 @@ module Awspec::Helper
     CLIENTS.each do |method_name, client|
       define_method method_name do
         unless self.methods.include? "@#{method_name}"
-          instance_variable_set("@#{method_name}", client.new(CLIENT_OPTIONS))
+          instance_variable_set(
+            "@#{method_name}",
+            Awspec::Helper::ClientWrap.new(client.new(CLIENT_OPTIONS))
+          )
         end
       end
     end

data/lib/awspec/helper/finder/cloudwatch_logs.rb

@@ -50,7 +50,7 @@ module Awspec::Helper
           method_name = 'describe_' + type + '_filters'
           resources = []
           loop do
-            res = cloudwatch_logs_client.method(method_name).call(req)
+            res = cloudwatch_logs_client.send(method_name, req)
             case type
             when 'metric' then
               resources.push(*res.metric_filters)

data/lib/awspec/helper/finder/ec2.rb

@@ -41,14 +41,16 @@ module Awspec::Helper
         define_method 'find_' + type + '_gateway' do |*args|
           gateway_id = args.first
           method_name = 'describe_' + type + '_gateways'
-          res = ec2_client.method(method_name).call({
-                                                      filters: [{ name: type + '-gateway-id', values: [gateway_id] }]
-                                                    })
+          res = ec2_client.send(
+            method_name,
+            { filters: [{ name: type + '-gateway-id', values: [gateway_id] }] }
+          )
           resource = res[type + '_gateways'].single_resource(gateway_id)
           return resource if resource
-          res = ec2_client.method(method_name).call({
-                                                      filters: [{ name: 'tag:Name', values: [gateway_id] }]
-                                                    })
+          res = ec2_client.send(
+            method_name,
+            { filters: [{ name: 'tag:Name', values: [gateway_id] }] }
+          )
           res[type + '_gateways'].single_resource(gateway_id)
         end
       end

data/lib/awspec/helper/finder/iam.rb

@@ -7,7 +7,7 @@ module Awspec::Helper
         define_method 'find_iam_' + type do |*args|
           id = args.first
           selected = []
-          res = iam_client.method('list_' + type.pluralize).call
+          res = iam_client.send('list_' + type.pluralize)
           loop do
             selected += res[type.pluralize].select do |u|
               u[type + '_name'] == id || u[type + '_id'] == id || u.arn == id
@@ -40,16 +40,18 @@ module Awspec::Helper
 
       %w(user group role).each do |type|
         define_method 'select_iam_policy_by_' + type + '_name' do |name|
-          res = iam_client.method('list_attached_' + type + '_policies').call({
-                                                                                (type + '_name').to_sym => name
-                                                                              })
+          res = iam_client.send(
+            'list_attached_' + type + '_policies',
+            { (type + '_name').to_sym => name }
+          )
           res.attached_policies
         end
 
         define_method 'select_inline_policy_by_' + type + '_name' do |name|
-          res = iam_client.method('list_' + type + '_policies').call({
-                                                                       (type + '_name').to_sym => name
-                                                                     })
+          res = iam_client.send(
+            'list_' + type + '_policies',
+            { (type + '_name').to_sym => name }
+          )
           res.policy_names
         end
       end

data/lib/awspec/version.rb

@@ -1,3 +1,3 @@
 module Awspec
-  VERSION = '0.86.0'
+  VERSION = '0.87.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: awspec
 version: !ruby/object:Gem::Version
-  version: 0.86.0
+  version: 0.87.0
 platform: ruby
 authors:
 - k1LoW
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-10-04 00:00:00.000000000 Z
+date: 2017-10-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -368,6 +368,7 @@ files:
 - lib/awspec/generator/spec/vpc.rb
 - lib/awspec/generator/template.rb
 - lib/awspec/helper.rb
+- lib/awspec/helper/client_wrap.rb
 - lib/awspec/helper/color.rb
 - lib/awspec/helper/finder.rb
 - lib/awspec/helper/finder/account_attributes.rb