dblock/slack-ruby-client

View on GitHub
lib/slack/web/pagination/cursor.rb

Summary

Maintainability
A
2 hrs
Test Coverage
# frozen_string_literal: true
module Slack
  module Web
    module Api
      module Pagination
        class Cursor
          include Enumerable

          attr_reader :client, :verb, :sleep_interval, :max_retries, :params

          def initialize(client, verb, params = {})
            @client = client
            @verb = verb
            @params = params.dup
            @sleep_interval = @params.delete(:sleep_interval)
            @max_retries = @params.delete(:max_retries) || client.default_max_retries
          end

          def each
            next_cursor = nil
            retry_count = 0
            loop do
              query = { limit: client.default_page_size }.merge(params).merge(cursor: next_cursor)
              begin
                response = client.send(verb, query)
              rescue Slack::Web::Api::Errors::TooManyRequestsError => e
                raise e if retry_count >= max_retries

                client.logger.debug("#{self.class}##{__method__}") { e.to_s }
                retry_count += 1
                sleep(e.retry_after)
                next
              end
              yield response
              break unless response.response_metadata

              next_cursor = response.response_metadata.next_cursor
              break if next_cursor.nil? || next_cursor == ''

              retry_count = 0
              sleep(sleep_interval) if sleep_interval
            end
          end
        end
      end
    end
  end
end