Add Retriable

lib/cool.rb

@@ -2,4 +2,5 @@
 
 # The list of classes in Cool module
 module Cool
+  autoload :Retriable, 'cool/retriable'
 end

lib/cool/retriable.rb

@@ -0,0 +1,22 @@
+module Cool
+  # Retriable
+  module Retriable
+    def retry_(new_method, on: StandardError, times: 3, &condition) # rubocop:disable Metrics/MethodLength, Metrics/LineLength
+      condition ||= ->(*) { false }
+      orig_method = "__retry_orig_#{new_method}".to_sym
+      alias_method orig_method, new_method
+      define_method(new_method) do |*args, &block|
+        (times - 1).times do
+          retval =
+            begin
+              method(orig_method).call(*args, &block)
+            rescue on
+              next
+            end
+          return retval unless condition.call(retval)
+        end
+        return method(orig_method).call(*args, &block)
+      end
+    end
+  end
+end

spec/lib/cool/retriable_spec.rb

@@ -0,0 +1,138 @@
+RSpec.describe Cool::Retriable do
+  describe '#retry_' do
+    it 'works' do
+      retriable = described_class
+      klass = Class.new do
+        extend retriable
+
+        ENUM = [:no, :no, :yes].to_enum
+        def do_it
+          ENUM.next
+        end
+        retry_(:do_it) { |retval| retval == :no }
+      end
+
+      expect(klass.new.do_it).to eq(:yes)
+    end
+
+    it 'works' do
+      retriable = described_class
+      klass = Class.new do
+        extend retriable
+
+        ENUM = [:no, :no, :no].to_enum
+        def do_it
+          ENUM.next
+        end
+        retry_(:do_it) { |retval| retval == :no }
+      end
+
+      expect(klass.new.do_it).to eq(:no)
+    end
+
+    it 'works' do
+      retriable = described_class
+      klass = Class.new do
+        extend retriable
+
+        ENUM = [nil, nil, :yes].to_enum
+        def do_it
+          ENUM.next
+        end
+        retry_(:do_it, &:nil?)
+      end
+
+      expect(klass.new.do_it).to eq(:yes)
+    end
+
+    it 'works' do
+      retriable = described_class
+      klass = Class.new do
+        extend retriable
+
+        ENUM = [[], {}, :yes].to_enum
+        def do_it
+          ENUM.next
+        end
+        retry_(:do_it, &:empty?)
+      end
+
+      expect(klass.new.do_it).to eq(:yes)
+    end
+
+    it 'works' do
+      retriable = described_class
+      klass = Class.new do
+        extend retriable
+
+        ENUM = [:no, :no, :yes].to_enum
+        def do_it
+          ENUM.next.tap { |value| raise if value == :no }
+        end
+        retry_(:do_it)
+      end
+
+      expect(klass.new.do_it).to eq(:yes)
+    end
+
+    it 'works' do
+      retriable = described_class
+      klass = Class.new do
+        extend retriable
+
+        ENUM = [:no, :no, :no].to_enum
+        def do_it
+          ENUM.next.tap { |value| raise if value == :no }
+        end
+        retry_(:do_it)
+      end
+
+      expect { klass.new.do_it }.to raise_error(StandardError)
+    end
+
+    it 'works' do
+      retriable = described_class
+      klass = Class.new do
+        extend retriable
+
+        ENUM = [:no].to_enum
+        def do_it
+          ENUM.next.tap { |value| raise if value == :no }
+        end
+        retry_(:do_it, times: 1)
+      end
+
+      expect { klass.new.do_it }.to raise_error(StandardError)
+    end
+
+    it 'works' do
+      retriable = described_class
+      klass = Class.new do
+        extend retriable
+
+        ENUM = [:error, nil, :yes].to_enum
+        def do_it
+          ENUM.next.tap { |value| raise if value == :error }
+        end
+        retry_(:do_it, &:nil?)
+      end
+
+      expect(klass.new.do_it).to eq(:yes)
+    end
+
+    it 'works' do
+      retriable = described_class
+      klass = Class.new do
+        extend retriable
+
+        ENUM = [:yes, :error].to_enum
+        def do_it
+          ENUM.next.tap { |value| raise if value == :error }
+        end
+        retry_(:do_it)
+      end
+
+      expect(klass.new.do_it).to eq(:yes)
+    end
+  end
+end

spec/spec_helper.rb

@@ -88,3 +88,5 @@
   # as the one that triggered the failure.
   Kernel.srand config.seed
 end
+
+require 'cool'