speakeasy_client_sdk_ruby 1.0.1

data/lib/sdk/utils/retries.rb

@@ -0,0 +1,130 @@
+# Code generated by Speakeasy (https://speakeasyapi.dev). DO NOT EDIT.
+
+# typed: true
+# frozen_string_literal: true
+
+require 'sorbet-runtime'
+
+module SpeakeasyClientSDK
+  class BackoffStrategy
+    extend T::Sig
+
+    attr_accessor :initial_interval, :max_interval, :exponent, :max_elapsed_time
+
+    sig { params(initial_interval: Integer, max_interval: Integer, exponent: Float, max_elapsed_time: Integer).void }
+    def initialize(initial_interval, max_interval, exponent, max_elapsed_time)
+      @initial_interval = initial_interval
+      @max_interval = max_interval
+      @exponent = exponent
+      @max_elapsed_time = max_elapsed_time
+    end
+  end
+
+  class RetryConfig
+    extend T::Sig
+
+    attr_accessor :strategy, :backoff, :retry_connection_errors
+
+    sig { params(strategy: String, retry_connection_errors: bool).void }
+    def initialize(strategy, retry_connection_errors)
+      @strategy = strategy
+      @retry_connection_errors = retry_connection_errors
+    end
+  end
+
+  class RetriesContainer
+    extend T::Sig
+
+    attr_accessor :config, :status_codes
+
+    sig { params(config: RetryConfig, status_codes: Array[String]).void }
+    def initialize(config, status_codes)
+      @config = config
+      @status_codes = status_codes
+    end
+  end
+
+  class TemporaryError < StandardError
+    extend T::Sig
+
+    attr_accessor :response
+
+    sig { params(response: requests.Response).void }
+    def initialize(response)
+      @response = response
+    end
+  end
+
+  class PermanentError < StandardError
+    extend T::Sig
+
+    attr_accessor :inner
+
+    sig { params(inner: StandardError).void }
+    def initialize(inner)
+      @inner = inner
+    end
+  end
+
+  def self.retry(fun, retries)
+    if retries.config.strategy == 'backoff'
+      do_request = lambda do
+        begin
+          res = fun.call
+          retries.status_codes.each do |code|
+            if code.upcase.include? 'X'
+              code_range = code[0].to_i
+              s = res.status_code / 100
+              raise TemporaryError, res if s >= code_range && s < code_range + 1
+            else
+              parsed_code = code.to_i
+              raise TemporaryError, res if res.status_code == parsed_code
+            end
+          end
+        rescue Faraday::ConnectionFailed => e
+          if !retries.config.config.retry_connection_errors
+            raise
+          else
+            raise PermanentError, e
+          end
+        rescue Faraday::TimeoutError => e
+          if !retries.config.config.retry_connection_errors
+            raise
+          else
+            raise PermanentError, e
+          end
+        rescue TemporaryError
+          raise
+        rescue StandardError => e
+          raise PermanentError, e
+        end
+        return res
+      end
+      retry_with_backoff(do_request, retries.config.backoff.initial_interval,
+                         retries.config.backoff.max_interval, retries.config.backoff.exponent,
+                         retries.config.backoff.max_elapsed_time)
+    else
+      fun.call
+    end
+  end
+
+  def self.retry_with_backoff(fun, initial_interval = 500, max_interval = 60_000,
+                              exponent = 1.5, max_elapsed_time = 3_600_000)
+    start = round(Time.now.to_i * 1000)
+    x = 0
+    loop do
+      return fun.call
+    rescue PermanentError => e
+      raise e.inner
+    rescue TemporaryError => e
+      now = round(Time.now.to_i * 1000)
+      return e.response if now - start > max_elapsed_time
+
+      sleep = ((initial_interval / 1000) * exponent**x + rand)
+      sleep = max_interval / 1000 if sleep > max_interval / 1000
+
+      sleep(sleep)
+      x += 1
+    end
+  end
+end